ruby/lib/irb/cmd/edit.rb
Stan Lo 223d4448c8 [ruby/irb] show_doc command should take non-string argument too
(https://github.com/ruby/irb/pull/478)

Given that `show_doc` already supports syntax like `String#gsub`, it
should be able to take it in non-string form too, like `edit` and
`show_source` do. This ensures users can have a consistent syntax on
argument between different commands.
2022-12-12 17:35:48 +00:00

61 lines
1.6 KiB
Ruby

require 'shellwords'
require_relative "nop"
module IRB
# :stopdoc:
module ExtendCommand
class Edit < Nop
category "Misc"
description 'Open a file with the editor command defined with `ENV["EDITOR"]`.'
class << self
def transform_args(args)
# Return a string literal as is for backward compatibility
if args.nil? || args.empty? || string_literal?(args)
args
else # Otherwise, consider the input as a String for convenience
args.strip.dump
end
end
end
def execute(*args)
path = args.first
if path.nil? && (irb_path = @irb_context.irb_path)
path = irb_path
end
if !File.exist?(path)
require_relative "show_source"
source =
begin
ShowSource.find_source(path, @irb_context)
rescue NameError
# if user enters a path that doesn't exist, it'll cause NameError when passed here because find_source would try to evaluate it as well
# in this case, we should just ignore the error
end
if source && File.exist?(source.file)
path = source.file
else
puts "Can not find file: #{path}"
return
end
end
if editor = ENV['EDITOR']
puts "command: '#{editor}'"
puts " path: #{path}"
system(*Shellwords.split(editor), path)
else
puts "Can not find editor setting: ENV['EDITOR']"
end
end
end
end
# :startdoc:
end