summaryrefslogtreecommitdiff
path: root/lib/irb/cmd/edit.rb
blob: 0103891cf48334241e9eaa8966e4587fcf7e1aef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
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