squared 0.7.3 → 0.7.5

data/lib/squared/workspace/project/ruby.rb

@@ -16,11 +16,13 @@ module Squared
                    D|describe=q? e|execute=q E|execute-continue=q p|execute-print=q f|rakefile=p job-stats=b? j|jobs=i?
                    I|libdir=p R|rakelib=p rakelibdir=p r|require=b suppress-backtrace=q T|tasks=q? t|trace=b?
                    W|where=q?].freeze,
-          irb: %w[d f U w E=b I=p r=b W=im? autocomplete colorize echo echo-on-assignment extra-doc-dir inf-ruby-mode
+          irb: %w[d f U w E=b I=p r=bm W=im? autocomplete colorize echo echo-on-assignment extra-doc-dir inf-ruby-mode
                   inspect multiline no-pager noautocomplete nocolorize noecho noecho-on-assignment noinspect
                   nomultiline noprompt noscript nosingleline noverbose regexp-completor sample-book-mode script
                   simple-prompt single-irb singleline tracer truncate-echo-on-assignment type-completor verbose
                   back-trace-limit=i context-mode=i prompt=b prompt-mode=b].freeze,
+          rdbg: %w[no-color no-rc no-sigint-hook c|command n|nonstop stop-at-load cookie=q e=q host=q x|init-script=p
+                   O|open=q? port=i port-range=b session-name=q sock-path=q util=q].freeze,
           rbs: %w[I=pm r=bm no-stdlib no-collection collection=p log-level=b log-output=p repo=p].freeze,
           rubocop: %w[D P r=bm auto-gen-config a|autocorrect A|autocorrect-all d|debug disable-pending-cops
                       display-only-correctable display-only-fail-level-offenses display-only-failed
@@ -32,6 +34,8 @@ module Squared
                       start-server stderr stop-server C|cache=b cache-root=p config=p exclude-limit=i fail-level=b
                       f|format=b except=q only=q o|out=p plugin=p require=p show-cops=q show-docs-url=q
                       s|stdin=p].freeze,
+          pry: %w[f I=pm no-color no-history no-multiline no-pager no-plugins simple-prompt c|context=q
+                  d|disable-plugin=q e=q gem=b r|require=bm s|select-plugin=q].freeze,
           no: {
             rubocop: %w[auto-gen-enforced-style auto-gen-only-exclude auto-gen-timestamp color display-cop-names
                         offense-counts parallel server].freeze
@@ -158,7 +162,9 @@ module Squared
           rake: %w[I libdir r require].freeze,
           irb: %w[I r].freeze,
           rbs: %w[I r repo].freeze,
+          rdbg: %w[cookie].freeze,
           rubocop: %w[format plugin r require].freeze,
+          pry: %w[I d disable-plugin r require].freeze,
           gem: {
             contents: %w[s spec-dir].freeze,
             dependency: %w[s source].freeze,
@@ -194,11 +200,13 @@ module Squared
           'outdated' => %i[major minor patch].freeze,
           'ruby' => %i[file script version reshim].freeze,
           'gem' => %i[install uninstall outdated update pristine build push exec command].freeze,
-          'bundle' => %i[config install update cache exec reinstall command].freeze,
+          'bundle' => %i[config install update cache exec check reinstall command].freeze,
           'rake' => nil,
           'irb' => nil,
+          'rdbg' => nil,
           'rbs' => nil,
-          'rubocop' => nil
+          'rubocop' => nil,
+          'pry' => nil
         })
 
         attr_reader :gemdir
@@ -303,13 +311,36 @@ module Squared
                       queue.call
                     end
                   end
-                when 'irb'
+                when 'irb', 'pry'
+                  next unless action == 'irb' || gemspec?(action)
+
                   format_desc action, nil, 'opts*,args*|:'
                   task action do |_, args|
                     opts = args.to_a
                     args = Array(opts.delete(':') && readline('Enter file [arguments]', force: false))
                     name = gemname if gemlib.any? { |file| exist?(file, "#{gemname}.rb") }
-                    irb(*args, opts: opts, name: name, verbose: false)
+                    __send__(action, *args, opts: opts, name: name, verbose: false)
+                  end
+                when 'rdbg'
+                  format_desc action, nil, 'file?/:,opts*,debuggee?/::'
+                  task action do |_, args|
+                    opts = args.to_a
+                    cmd = opts.delete('::')
+                    if (n = opts.index('c'))
+                      args = opts.slice!(n.succ..-1)
+                    else
+                      file = opts.shift
+                      args = []
+                      if (file == ':' && (d = path)) || (file && (d = basepath(file)).directory?)
+                        args << choice_index('Select file', d.glob('*.rb').map { |f| f.relative_path_from(path) })
+                      elsif exist?(file, type: 'f')
+                        args << file
+                      else
+                        opts.unshift(file)
+                      end
+                    end
+                    args << readline('Enter debuggee [arguments]', force: false) if cmd
+                    rdbg(*args.compact, opts: opts, banner: false, verbose: false)
                   end
                 when 'rbs'
                   next unless @steepfile
@@ -322,9 +353,9 @@ module Squared
                       data[$1 || $3.gsub(/[: ]/, '-')] = target
                       next
                     end
-                    next unless target && line =~ /^\s*(check|signature)\s+(["'])(.+)\2/
+                    next unless target && line =~ /^\s*(?:(check)|signature)\s+(["'])(.+)\2/
 
-                    target[$1 == 'check' ? 1 : 0] << $3
+                    target[$1 ? 1 : 0] << $3
                   end
                   next if data.empty?
 
@@ -459,6 +490,16 @@ module Squared
                         task flag do |_, args|
                           bundle(flag, opts: args.to_a, banner: flag == :exec ? verbose? : true)
                         end
+                      when :check
+                        format_desc action, flag, 'gemfile?,d/ry-run?'
+                        task flag do |_, args|
+                          args = args.to_a
+                          opts = []
+                          opts << 'dry-run' if has_value!(args, 'd', 'dry-run')
+                          opts << "gemfile=#{shell_quote(args.pop, option: false, force: false)}" if exist?(args.first)
+                          opts.concat(args)
+                          bundle(:check, opts: opts, banner: true)
+                        end
                       when :reinstall
                         format_desc action, flag, 'f/orce?,l/ocal?,opts*'
                         task flag do |_, args|
@@ -537,9 +578,6 @@ module Squared
                           if s && (%w[system latest].include?(s) || SEM_VER.match?(s) || s.start_with?(/(path|ref):/))
                             rbvm = exist?(ver.last) && !exist?(ver.first)
                             case vmname
-                            when nil
-                              print_error('no version manager detected', subject: name)
-                              next
                             when 'asdf'
                               unless rbvm
                                 asdf(:set, **{ name: args.name, version: s, banner: verbose? }.compact)
@@ -555,6 +593,9 @@ module Squared
                                 run(banner: verbose?)
                                 next
                               end
+                            when nil
+                              print_error('no version manager detected', subject: name)
+                              next
                             end
                             File.write(basepath(ver.last), "#{s}\n") if SEM_VER.match?(s)
                             next
@@ -563,16 +604,14 @@ module Squared
                             out = []
                             tool = args.name || (args.local && !SEM_VER.match?(args.local) ? args.local : 'ruby')
                             trim = ->(s) { s[/^\D+\d+\.\d+(?:\.\S+)?/, 0].sub(/^([a-z]+)-/i, '\1 ') }
-                            out << trim.call(case (vm, bin = vmname(bin: true))
+                            vm, bin = vmname(bin: true)
+                            out << trim.call(case vm
                                              when 'rvm'
-                                               ver.shift
                                                `rvm current`[/^\S+/, 0]
                                              when 'rbenv'
-                                               ver.shift
                                                name = `rbenv version-name`
                                                (name =~ SEM_VER) == 0 ? "ruby #{name}" : name
                                              when 'chruby'
-                                               ver.shift
                                                chruby = session_output 'source', bin
                                                `#{chruby.with('ruby --version')}`
                                              when 'mise'
@@ -592,7 +631,7 @@ module Squared
                                                end
                                                opt = [@asdf.first]
                                                opt.unshift('--no-header') unless @@asdf.version == 15
-                                               cur = `asdf current #{opt.join(' ')}`
+                                               exit 1 unless (cur = `asdf current #{opt.join(' ')}`)
                                                if cur.match?(/\sfalse\b/)
                                                  `ruby --version`
                                                else
@@ -626,32 +665,53 @@ module Squared
                                                      when 'mise'
                                                        `mise which #{tool}`
                                                      else
-                                                       `which #{tool}`
+                                                       `which ruby`
                                                      end)
                               rescue => e
                                 log.debug e
                               end
                               if ver
-                                ver = mise + ver if vm == 'mise'
+                                if vm == 'mise'
+                                  ver = mise + ver
+                                  pat = /^\s*#{tool}\s*=\s*\[?(.+?)\]?\s*$/
+                                elsif vm != 'asdf'
+                                  ver.shift
+                                end
+                                append = ->(file, hint) { out << message("found #{file}", hint: hint) }
                                 catch :found do
                                   path.ascend do |dir|
-                                    ver.filter { |val| dir.join(val).exist? }.each do |val|
-                                      dir += val
-                                      file = File.read(dir)
+                                    ver.select { |val| dir.join(val).exist? }.each do |val|
+                                      file = dir + val
                                       hint = if val.include?('mise')
-                                               line = file[/^\s*#{tool}\s*=\s*\[?(.+?)\]?\s*$/, 1]
-                                               line&.gsub(/["']/, '')
+                                               file.read[pat, 1]&.gsub(/["']/, '')
                                              else
-                                               file.lines
+                                               file.read
+                                                   .lines
                                                    .map { |line| line.sub(/#.*$/, '').strip }
                                                    .reject(&:empty?)
                                                    .join(', ')
                                              end
-                                      out << message("found #{dir}", hint: hint)
+                                      append.call(file, hint)
                                       throw :found if hint&.include?(out.first[/^(?:j|truffle)?ruby ([\d.]+)/, 1])
                                     rescue
                                       nil
                                     end
+                                    file = case vm
+                                           when 'rbenv'
+                                             '.rbenv/version'
+                                           when 'rvm'
+                                             '.rvm/config/default'
+                                           when 'mise'
+                                             '.config/mise/config.toml'
+                                           else
+                                             next
+                                           end
+                                    if File.exist?(file = dir.join(file))
+                                      hint = out.first.split(' ', 2).last
+                                      pat ||= SEM_VER
+                                      append.call(file, file.read[/^(.*(?:#{Regexp.escape(hint)}|#{pat}).*)$/, 1])
+                                      throw :found
+                                    end
                                   end
                                 end
                               end
@@ -673,7 +733,7 @@ module Squared
                           when 'mise'
                             cmd = session('mise', 'reshim', args.name || 'ruby')
                           else
-                            print_error('not supported by version manager', subject: args.name)
+                            print_error('reshim not supported by version manager', subject: args.name)
                           end
                           success?(run(banner: verbose?), verbose?) if cmd
                         end
@@ -780,7 +840,7 @@ module Squared
             elsif se || ia || up
               items = []
             end
-            OptionPartition.new(opts, bundleopts(:outdated), cmd << "--#{flag}", project: self)
+            OptionPartition.new(opts, bundleopts(:outdated), cmd << "--#{flag}", project: self, strict: strict?)
                            .clear
           elsif (up = option('u', 'update', prefix: 'bundle'))
             flag = case up
@@ -851,10 +911,8 @@ module Squared
                       when '~>'
                         if c < v && cur[0] == val[0] && !semmajor?(cur, val)
                           styles[0] = :yellow
-                        elsif semmajor?(val, lat)
-                          styles[1] = :underline
                         else
-                          styles[1] = :bold
+                          styles[1] = semmajor?(val, lat) ? :underline : :bold
                         end
                       end
                     end
@@ -943,8 +1001,8 @@ module Squared
             from = symjoin 'ruby', flag
           end
           opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
-          op = OptionPartition.new(opts, OPT_RUBY[:ruby], ruby_session, project: self, multiple: [/^-e/], args: true,
-                                                                        stdin: true)
+          op = OptionPartition.new(opts, OPT_RUBY[:ruby], ruby_session, project: self, strict: strict?,
+                                                                        multiple: [/^-e/], args: true, stdin: true)
           if kwargs[:command]
             op << quote_option('e', kwargs[:command])
           elsif kwargs[:file]
@@ -987,7 +1045,7 @@ module Squared
             opts.concat(args)
           end
           op = OptionPartition.new(opts, gemopts(flag), gem_session(flag),
-                                   project: self, no: OPT_GEM[:no][flag == :update ? :install : flag])
+                                   project: self, strict: strict?, no: OPT_GEM[:no][flag == :update ? :install : flag])
           from = symjoin 'gem', flag
           if flag == :outdated
             op.adjoin(gempwd, start: 0) if gempwd
@@ -1341,7 +1399,7 @@ module Squared
           end
           opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
           invalid = ->(a) { raise_error ArgumentError, "unrecognized args: #{a.join(', ')}", hint: flag }
-          cmd = bundle_session(flag)
+          cmd = bundle_session flag
           case flag
           when :cache, :check, :clean, :init, :install, :lock, :pack, :package, :platform
             pre = true
@@ -1387,7 +1445,8 @@ module Squared
                                                       flag
                                                     end),
                                    cmd,
-                                   project: self, no: OPT_BUNDLE[:no][flag], args: flag == :exec || flag == :config)
+                                   project: self, strict: strict?, no: OPT_BUNDLE[:no][flag],
+                                   args: flag == :exec || flag == :config)
           op.concat(args) unless pre
           output = false
           case flag
@@ -1398,7 +1457,7 @@ module Squared
               a = op.dup
               b, c = op.slice!(0, 2)
               d = op.arg?('global', 'local')
-              getname = -> { op << (b || readline('Enter name', force: true)) }
+              getname = -> { b ? op << b : op.readline('Enter name') }
               case pre
               when 'get'
                 getname.call
@@ -1409,7 +1468,7 @@ module Squared
                   c = op.shift
                 end
                 getname.call
-                op << (c || readline('Enter value', force: true))
+                c ? op << c : op.readline('Enter value', quote: true)
                 output = true
               when 'unset'
                 if d
@@ -1433,9 +1492,8 @@ module Squared
               end
               op.clear
             else
-              val = readline('Enter arguments', force: false)
-              op << (val.empty? ? 'list' : val)
-              output = val.start_with?('set', 'unset')
+              output = op.readline('Enter arguments', 'list')
+                         .start_with?('set', 'unset')
             end
           when :plugin
             case plu
@@ -1446,7 +1504,7 @@ module Squared
             end
           when :exec
             if op.empty? || (op.remove(':') && op.append(quote: false))
-              op << readline('Enter arguments', force: true)
+              op.readline('Enter arguments')
             else
               op.append(quote: false)
             end
@@ -1470,7 +1528,7 @@ module Squared
 
         def rake(*args, sync: true, banner: verbose?, with: nil, pass: PASS_RUBY[:rake], **kwargs)
           opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
-          op = OptionPartition.new(opts, OPT_RUBY[:rake], rake_session, project: self)
+          op = OptionPartition.new(opts, OPT_RUBY[:rake], rake_session, project: self, strict: strict?)
           op.adjoin(quote_option('f', rakefile)) if rakefile && !op.arg?('f', 'rakefile')
           op.concat(args)
           op.append(escape: true)
@@ -1481,16 +1539,30 @@ module Squared
 
         def irb(*args, banner: verbose?, with: nil, pass: PASS_RUBY[:irb], **kwargs)
           opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
-          op = OptionPartition.new(opts, OPT_RUBY[:irb], session('irb'), project: self, first: [/\.rb$/])
-          r = []
-          r << 'bundler/setup' if !op.arg?('r') && bundle_load
-          r.concat(Array(kwargs[:name])) if kwargs[:name]
-          r.each { |val| op.add_option('r', val, merge: true) }
-          op.merge((gemlib + Array(kwargs[:path])).map { |val| quote_option('I', val, merge: true) })
+          run_repl(*args, opts: opts, banner: banner, from: :irb, **kwargs)
+        end
+
+        def pry(*args, banner: verbose?, with: nil, pass: PASS_RUBY[:pry], **kwargs)
+          opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
+          run_repl(*args, opts: opts, banner: banner, from: :pry, delim: false, **kwargs)
+        end
+
+        def rdbg(*args, banner: verbose?, with: nil, pass: PASS_RUBY[:rdbg], **kwargs)
+          opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
+          cmd = session 'rdbg'
+          args.each_with_index do |val, index|
+            break if val == '--'
+            next unless exist?(val, type: 'f')
+
+            args[index] = shell_quote basepath(val)
+            cmd.merge(args.slice!(0, index + 1))
+            break
+          end
+          op = OptionPartition.new(opts, OPT_RUBY[:rdbg], cmd, project: self, strict: strict?, first: [/\.rb$/])
           op.concat(args)
-          op.append(delim: true)
+          op.append(quote: false, delim: true)
           print_run(op, banner, **kwargs)
-          run(banner: false, exception: kwargs.fetch(:exception, exception?), from: :irb)
+          run(banner: false, exception: kwargs.fetch(:exception, exception?), from: :rdbg)
         end
 
         def rbs(flag, *args, banner: verbose?, with: nil, pass: nil, **kwargs)
@@ -1502,7 +1574,7 @@ module Squared
           end
           opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
           cmd, opts = rbs_session(opts: opts)
-          op = OptionPartition.new(opts, [], cmd << flag, project: self)
+          op = OptionPartition.new(opts, [], cmd << flag, project: self, strict: strict?)
           case flag
           when :prototype
             sig = args.shift
@@ -1545,7 +1617,7 @@ module Squared
 
         def rubocop(*args, sync: true, banner: verbose?, with: nil, pass: PASS_RUBY[:rubocop], **kwargs)
           opts = session_opts(with, args: args, kwargs: kwargs, pass: pass)
-          op = OptionPartition.new(opts, OPT_RUBY[:rubocop], session('rubocop'), project: self,
+          op = OptionPartition.new(opts, OPT_RUBY[:rubocop], session('rubocop'), project: self, strict: strict?,
                                                                                  no: OPT_RUBY[:no][:rubocop])
           if @rubocopfile && !op.arg?('c', 'config') && !rootpath('.rubocop.yml', ascend: true).exist?
             op.add_path(@rubocopfile, option: 'c')
@@ -1720,13 +1792,13 @@ module Squared
               parse.call(target.resolve_feature_path(gemname)&.last)
             end
             unless gemdir || pwd_set { parse.call(`#{bundle_output('show', gemname)}`) }
-              raise_error Errno::ENOENT, 'gems home'
+              raise Errno::ENOENT, 'gems home'
             end
           end
         rescue => e
           log.error e
-          self.version = nil
           @gemdir = nil
+          self.version = nil
           self.autodetect = false
         else
           gemdir?
@@ -1738,6 +1810,20 @@ module Squared
 
         private
 
+        def run_repl(*args, opts:, banner:, from:, delim: true, **kwargs)
+          op = OptionPartition.new(opts, OPT_RUBY[from], session(from), project: self, strict: strict?,
+                                                                        first: [/\.rb$/])
+          r = []
+          r << 'bundler/setup' if !op.arg?('r') && bundle_load
+          r.concat(Array(kwargs[:name])) if kwargs[:name]
+          r.each { |val| op.add_option('r', val, merge: true) }
+          op.merge((gemlib + Array(kwargs[:path])).map { |val| quote_option('I', val, merge: true) })
+          op.concat(args)
+          op.append(delim: delim)
+          print_run(op, banner, **kwargs)
+          run(banner: false, exception: kwargs.fetch(:exception, exception?), from: from)
+        end
+
         def ruby_session(*cmd, **kwargs)
           session('ruby', *preopts, *cmd, **kwargs)
         end
@@ -1759,7 +1845,7 @@ module Squared
         def rbs_session(*cmd, opts: nil)
           return session('rbs', *cmd) unless opts
 
-          op = OptionPartition.new(opts, OPT_RUBY[:rbs], project: self)
+          op = OptionPartition.new(opts, OPT_RUBY[:rbs], project: self, strict: strict?)
           [session('rbs', *op.to_a, *cmd), op.extras]
         end
 
@@ -1943,13 +2029,11 @@ module Squared
         end
 
         def gemdir?
-          return false unless gemdir
-
-          gemdir.exist? && !gemdir.empty? && gemdir.writable?
+          !gemdir.nil? && gemdir.exist? && !gemdir.empty? && gemdir.writable?
         end
 
         def compat?(min, max = Float::INFINITY, gem: nil)
-          return false unless !gem || (bundle = gem.match?(/bundler?/))
+          return false if gem && !gem.match?(/bundler?/)
 
           n = @rubygems
           if n.is_a?(Numeric) && n > 0
@@ -1960,14 +2044,20 @@ module Squared
                   when String
                     n
                   when Array
-                    n[bundle ? 1 : 0]
+                    n[gem ? 1 : 0]
                   else
-                    min.tap { min = (bundle && bundle_load) || Gem::VERSION }
+                    min.tap { min = (gem && bundle_load) || Gem::VERSION }
                   end
             semgte?(ver.to_s, min.to_s) && (max == Float::INFINITY || !semgte?(ver.to_s, max.to_s))
           end
         end
 
+        def gemspec?(name)
+          !Gem::Specification.find_by_name(name).nil?
+        rescue Gem::MissingSpecError
+          false
+        end
+
         def serve?
           !Gem::Specification.find_by_name('webrick').nil?
         rescue Gem::MissingSpecError => e