summaryrefslogtreecommitdiff
path: root/lib/pry/commands/shell_command.rb
blob: 6069a88275fc549cfdf6bd9ba7f2f80f40886f20 (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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
# frozen_string_literal: true

class Pry
  class Command
    class ShellCommand < Pry::ClassCommand
      match(/\.(.*)/)
      group 'Input and Output'
      description "All text following a '.' is forwarded to the shell."
      command_options listing: '.<shell command>', use_prefix: false,
                      takes_block: true

      banner <<-'BANNER'
        Usage: .COMMAND_NAME

        All text following a "." is forwarded to the shell.

        .ls -aF
        .uname
      BANNER

      def process(cmd)
        if cmd =~ /^cd\s*(.*)/i
          process_cd parse_destination(Regexp.last_match(1))
        else
          pass_block(cmd)
          if command_block
            command_block.call `#{cmd}`
          else
            pry_instance.config.system.call(output, cmd, pry_instance)
          end
        end
      end

      private

      def parse_destination(dest)
        return "~" if dest.empty?
        return dest unless dest == "-"

        state.old_pwd || raise(CommandError, "No prior directory available")
      end

      def process_cd(dest)
        state.old_pwd = Dir.pwd
        Dir.chdir(File.expand_path(path_from_cd_path(dest) || dest))
      rescue Errno::ENOENT
        raise CommandError, "No such directory: #{dest}"
      end

      def cd_path_env
        Pry::Env['CDPATH']
      end

      def cd_path_exists?
        cd_path_env && cd_path_env.length.nonzero?
      end

      def path_from_cd_path(dest)
        return if !(dest && cd_path_exists?) || special_case_path?(dest)

        cd_path_env.split(File::PATH_SEPARATOR).each do |path|
          return path if File.directory?(path) && path.split(File::SEPARATOR).last == dest
        end

        nil
      end

      def special_case_path?(dest)
        ['.', '..', '-'].include?(dest) || dest =~ /\A[#{File::PATH_SEPARATOR}~]/
      end
    end

    Pry::Commands.add_command(Pry::Command::ShellCommand)
  end
end