squared 0.6.4 → 0.6.6

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

@@ -7,9 +7,9 @@ module Squared
         DEP_PYTHON = %w[poetry.lock setup.cfg pyproject.toml setup.py requirements.txt].freeze
         DIR_PYTHON = (DEP_PYTHON + %w[README.rst]).freeze
         OPT_PYTHON = {
-          common: %w[b=+ B d E h i I O P q=+ s S u v=+ x c=q m=b W=b X=q check-hash-based-pycs=b].freeze,
-          build: %w[n|no-isolation s|sdist x|skip-dependency-check v|verbose w|wheel C|config-setting=q installer=b
-                    o|outdir=p].freeze,
+          common: %w[b=+ B d E h i I O P q=+ s S u v=+ V=+ x c=q m=b W=b X=q check-hash-based-pycs=b].freeze,
+          build: %w[C=bm n|no-isolation s|sdist x|skip-dependency-check v|verbose w|wheel config-json=q config-setting=q
+                    installer=b o|outdir=p].freeze,
           venv: %w[clear copies symlinks system-site-packages upgrade upgrade-deps without-scm-ignore-files without-pip
                    prompt=q].freeze
         }.freeze
@@ -68,6 +68,7 @@ module Squared
                       sign-with=b u|username=q].freeze
         }.freeze
         PASS_PYTHON = {
+          python: %w[c v V].freeze,
           pip: {
             debug: %w[platform].freeze,
             install: %w[C config-settings c constraint extra-index-url no-binary only-binary platform
@@ -178,7 +179,7 @@ module Squared
                               indexerror n, list
                             else
                               found |= 2
-                              log.warn "run script #{n} of #{list.size} (out of range)"
+                              log.warn "run script #{n} of #{list.size}".subhint('out of range')
                             end
                           else
                             case i
@@ -191,18 +192,18 @@ module Squared
                             else
                               raise_error "script: #{val}" if exception
                               found |= 2
-                              log.warn "run script \"#{val}\" (not indexed)"
+                              log.warn "run script \"#{val}\"".subhint('not indexed')
                             end
                           end
                         end
                       end
                       break
                     end
-                    unless found.anybits?(1)
-                      puts log_message(found == 0 ? Logger::INFO : Logger.WARN,
-                                       "no scripts #{found == 0 ? 'found' : 'executed'}",
-                                       subject: name, hint: pyprojectfile)
-                    end
+                    next if found.anybits?(1)
+
+                    puts log_message(found == 0 ? Logger::INFO : Logger.WARN,
+                                     "no scripts #{found == 0 ? 'found' : 'executed'}",
+                                     subject: name, hint: pyprojectfile)
                   end
                 when 'exec'
                   format_desc action, nil, ':|command,args*'
@@ -234,11 +235,12 @@ module Squared
                         when :remove
                           next unless projectpath?(venv)
 
-                          format_desc action, flag, 'c/reate?,d/epend?'
+                          format_desc action, flag, 'c/reate?,d/epend?,opts*'
                           task flag do |_, args|
+                            args = args.to_a
                             rm_rf(venv, verbose: true)
-                            venv_init if has_value?(args = args.to_a, 'c', 'create')
-                            depend if has_value?(args, 'd', 'depend')
+                            venv_init if has_value!(args, 'c', 'create')
+                            depend :force, args if has_value!(args, 'd', 'depend')
                           end
                         when :exec
                           format_desc action, flag, 'command,args*'
@@ -286,11 +288,11 @@ module Squared
                           pip(flag, opts: args.to_a, banner: true)
                         end
                       when :reinstall
-                        if venv && projectpath?(venv)
-                          format_desc action, flag
-                          task flag do
-                            ns['venv:remove'].invoke('depend')
-                          end
+                        next unless venv && projectpath?(venv)
+
+                        format_desc action, flag
+                        task flag do
+                          ns['venv:remove'].invoke('depend')
                         end
                       end
                     when 'install'
@@ -309,7 +311,7 @@ module Squared
                           install flag, (case args.strategy
                                          when 'eager'
                                            'eager'
-                                         when /(?:\Aonly-if|needed\z)/
+                                         when /^only-if|needed$/
                                            'only-if-needed'
                                          end.yield_self do |val|
                                            if val
@@ -330,7 +332,7 @@ module Squared
                         end
                       end
                     when 'outdated'
-                      format_desc(action, flag, "eager?,no-deps?,#{shortname('h', 'i', 's', 'd')}",
+                      format_desc(action, flag, "eager?,no-deps?,#{shortname('h', 'i', 's', 'u', 'd')}",
                                   before: ('user?' unless venv))
                       task flag do |_, args|
                         outdated flag, args.to_a
@@ -408,18 +410,28 @@ module Squared
         def outdated(flag = nil, opts = [], sync: invoked_sync?('outdated', flag))
           cmd = pip_session 'list --outdated'
           cmd << if flag
-                   se = has_value!(opts, 's', 'select')
+                   se = has_value! opts, 's', 'select'
                    ia = has_value!(opts, 'i', 'interactive') && !se
+                   up = has_value! opts, 'u', 'update'
                    hide = has_value! opts, 'h', 'hide'
                    dryrun = has_value! opts, 'd', 'dry-run'
                    if !sync || stdin?
                      se = false
                      ia = false
                    elsif se || ia
+                     up = true
                      items = []
                    end
                    '--not-required' if opts.include?('no-deps')
                  else
+                   if (up = option('u', 'update'))
+                     flag = case up
+                            when 'major', 'minor'
+                              up.to_sym
+                            else
+                              :patch
+                            end
+                   end
                    '--not-required' unless option('not-required', equals: '0')
                  end
           cmd << '--local' if option('l', 'local')
@@ -441,11 +453,10 @@ module Squared
             buffer = []
             out = ->(val) { sync ? puts(val) : buffer << val }
             if workspace.windows?
-              (venv ? command(runenv, cmd) : `#{cmd}`).lines
+              (venv ? command(runenv, cmd) : `#{cmd}`).lines(chomp: true)
             else
-              IO.popen(runenv || {}, cmd)
+              IO.popen(runenv || {}, cmd).readlines(chomp: true)
             end.each do |line|
-              line.chomp!
               next if line.match?(/^[ -]+$/)
 
               if start > 0
@@ -506,34 +517,33 @@ module Squared
               puts buffer
             end
             if found > 0
-              if se
-                choice('Select a package', items.map(&:first),
-                       multiple: true, force: false, index: true, border: true).map! { |n| items[n.pred].last }
-              elsif ia
-                items.map(&:last)
-              else
-                case flag
-                when :major
-                  major + minor + patch
-                when :minor
-                  minor + patch
+              items = if se
+                        choice('Select a package', items.map(&:first),
+                               multiple: true, force: false, index: true, border: true).map! { |n| items[n.pred].last }
+                      elsif ia
+                        items.map(&:last)
+                      else
+                        case flag
+                        when :major
+                          major + minor + patch
+                        when :minor
+                          minor + patch
+                        else
+                          patch
+                        end
+                      end
+              if up && !items.empty?
+                base = %w[eager no-deps]
+                base << 'user' unless venv
+                opts = (base & opts).map! { |val| val == 'eager' ? "upgrade-strategy=#{val}" : val }
+                if dryrun
+                  opts.map! { |val| fill_option(val) }
+                  print_run pip_output('install --upgrade', *opts, *items.quote!), false
                 else
-                  patch
+                  install(:upgrade, opts, packages: items, banner: false)
                 end
-              end.tap do |packages|
-                unless packages.empty?
-                  base = %w[eager no-deps]
-                  base << 'user' unless venv
-                  opts = (base & opts).map! { |val| val == 'eager' ? "upgrade-strategy=#{val}" : val }
-                  if dryrun
-                    opts.map! { |val| fill_option(val) }
-                    print_run pip_output('install --upgrade', *opts, *packages.quote!), false
-                  else
-                    install(:upgrade, opts, packages: packages, banner: false)
-                  end
-                end
-                print_status(major.size, minor.size, patch.size, from: :outdated)
               end
+              print_status(major.size, minor.size, patch.size, from: :outdated)
             elsif start == 0 || hide
               puts 'No updates were found'
             end
@@ -653,6 +663,27 @@ module Squared
           run(from: :"#{flag}:publish", interactive: ['Publish', 'N', project])
         end
 
+        def python(*args, sync: true, banner: verbose?, with: nil, pass: PASS_PYTHON[:python], **kwargs)
+          op = OptionPartition.new(session_opts(with, args: args, kwargs: kwargs, pass: pass), OPT_PYTHON[:common],
+                                   session('python', path: venv.nil?),
+                                   project: self, multiple: [/^-c/], single: singleopt(:python), args: true,
+                                   stdin: true)
+          op.concat(args)
+          if op.include?('-')
+            op.exist?(add: true)
+          else
+            op.append_any { |val| OptionPartition.parse_arg!('c', val) }
+            if op.arg?('c')
+              op.clear
+            else
+              op.exist?(add: true, first: true) unless op.arg?('m')
+              op.append(escape: kwargs.fetch(:escape, false), quote: kwargs.fetch(:quote, false))
+            end
+          end
+          print_run(op, banner, **kwargs)
+          run(sync: sync, banner: banner, exception: kwargs.fetch(:exception, exception), from: :python)
+        end
+
         def pip(flag, *args, sync: true, banner: verbose?, with: nil, pass: nil, **kwargs)
           flag = flag.to_sym
           pass = PASS_PYTHON[:pip].fetch(pip_install?(flag) ? :install : flag, []) + %w[v verbose] if pass.nil?
@@ -726,29 +757,30 @@ module Squared
             op.clear
           end
           print_run(op, banner, **kwargs)
-          run(sync: sync, banner: banner, from: :"pip:#{flag}").yield_self { |val| ret || val }
+          run(sync: sync, banner: banner, exception: kwargs.fetch(:exception, exception), from: :"pip:#{flag}")
+            .yield_self { |val| ret || val }
         end
 
-        def variable_set(key, *val, **, &blk)
+        def variable_set(key, *args, **, &blk)
           if block_given?
             case key
             when :dependfile, :venv, :editable
-              val = block_args val, &blk
+              args = block_args args, &blk
             end
           end
           case key
           when :dependfile
-            req = basepath(*val)
-            if (index = DEP_PYTHON.index(req.basename.to_s))
+            val = basepath(*args)
+            if (index = DEP_PYTHON.index(val.basename.to_s))
               @dependindex = index
-              @dependfile = req
+              @dependfile = val
             else
-              log.warn "variable_set: @#{key}=#{req} (not supported)"
+              log.warn "variable_set: @#{key}=#{val} (not supported)"
             end
           when :editable
-            editable_set val.first
+            editable_set args.first
           when :venv
-            instance_variable_set(:"@#{key}", (basepath(*val) unless val.empty?))
+            instance_variable_set(:"@#{key}", (basepath(*args) unless args.empty?))
           else
             super
           end
@@ -775,14 +807,14 @@ module Squared
         def python_session(*cmd, opts: nil)
           return session('python', *preopts(quiet: false), *cmd, path: venv.nil?) unless opts
 
-          op = OptionPartition.new(opts, OPT_PYTHON[:common], project: self, single: /\A(?:v+|OO)\z/)
+          op = OptionPartition.new(opts, OPT_PYTHON[:common], project: self, single: singleopt(:python))
           [session('python', *op.to_a, *cmd, path: venv.nil?), op.extras]
         end
 
         def poetry_session(*cmd)
-          session('poetry', *cmd, *preopts).tap do |ret|
-            option('project', ignore: false) { |val| ret << quote_option('project', basepath(val)) }
-          end
+          ret = session('poetry', *cmd, *preopts)
+          option('project', ignore: false) { |val| ret << quote_option('project', basepath(val)) }
+          ret
         end
 
         def pdm_session(*cmd, opts: nil)
@@ -849,7 +881,7 @@ module Squared
         def append_editable(target: @session)
           return if requirements? && editable == '.'
 
-          if (val = option('editable', 'e', target: target, ignore: false))
+          if (val = option('e', 'editable', target: target, ignore: false))
             OptionPartition.delete_key(target, 'e', 'editable')
             case val
             when '0', 'false'
@@ -873,7 +905,7 @@ module Squared
                 quote_option 'cache-dir', basepath(val)
               end
             end,
-            option('proxy', target: target) { |val| shell_option('proxy', val) },
+            option('proxy', target: target) { |val| quote_option('proxy', val) },
             option('python', target: target) { |val| quote_option('python', basepath(val)) }
           ])
           append_nocolor(target: target)
@@ -947,15 +979,14 @@ module Squared
         end
 
         def pyprojectfile
-          return @pyprojectfile || nil unless @pyprojectfile.nil?
-
-          @pyprojectfile = (file = basepath(DEP_PYTHON[2])).exist? ? file : false
+          @pyprojectfile = basepath!(DEP_PYTHON[2]) || false if @pyprojectfile.nil?
+          @pyprojectfile || nil
         end
 
         def singleopt(flag = nil)
           case flag
           when :python
-            /\A(?:v+|q+|b+)\z/
+            /\A(?:v+|q+|b+|V+|O+)\z/
           when :pdm
             /\Av+\z/
           when :twine
@@ -966,14 +997,14 @@ module Squared
         end
 
         def preopts(quiet: true)
-          [].tap do |ret|
-            case verbose
-            when FalseClass
-              ret << '--quiet' if quiet
-            when Numeric
-              ret << "-#{'v' * verbose}" if verbose > 0
-            end
+          ret = []
+          case verbose
+          when FalseClass
+            ret << '--quiet' if quiet
+          when Numeric
+            ret << "-#{'v' * verbose}" if verbose > 0
           end
+          ret
         end
 
         def variables
@@ -1006,20 +1037,21 @@ module Squared
         def venv_set(val)
           return unless val
 
+          write = ->(level, hint) { log.add(level, "venv: #{@venv}".subhint(hint)) }
           if val.is_a?(Array)
             val, *opts = val
             @venvopts = opts
           end
-          @venv = basepath(val)
+          @venv = basepath val
           if projectpath?(@venv)
             if @venv.exist?
-              log.debug "venv found: #{@venv}"
-            elsif @path.directory? && !@path.empty?
+              write.call(Logger::DEBUG, 'found')
+            elsif path.directory? && !path.empty?
               @venv.mkpath
-              log.info "venv mkdir: #{@venv}"
+              write.call(Logger::INFO, 'mkdir')
             end
           elsif !@venv.directory?
-            log.warn "venv invalid: #{@venv}"
+            write.call(Logger::WARN, 'invalid')
             @venv = nil
           end
         end
@@ -1027,20 +1059,21 @@ module Squared
         def venv_init
           return if !venv || (venvbin.directory? && !venvbin.empty?)
 
-          puts log_message(venv, subject: 'venv', hint: 'init')
+          puts log_message(venv, subject: 'venv', hint: 'init') unless silent?
           opts = @venvopts&.map { |val| OptionPartition.strip(val) }&.flatten
           venv_create(venv, opts || ["prompt=#{name}", 'upgrade-deps'], env: false, banner: false)
-          puts log_message(venv, subject: 'venv', hint: 'created')
+          puts log_message(venv, subject: 'venv', hint: 'created') unless silent?
         end
 
-        def venv_create(dir, opts = [], env: nil, banner: true)
+        def venv_create(dir, opts = [], env: nil, banner: banner?)
           cmd, opts = python_session('-m venv', opts: opts)
           op = OptionPartition.new(opts, OPT_PYTHON[:venv], cmd, project: self)
           status = op.append(dir, delim: true)
                      .clear(pass: false)
                      .arg?(/\A-v+\z/)
-          run(op, env, exception: true, banner: banner)
-          puts(dir.directory? ? "Success: #{dir}" : 'Failed') if banner && !status
+          success?(run(op, env, exception: true, banner: banner), banner, !status) do |ret|
+            puts(ret && dir.directory? ? "Success: #{dir}" : 'Failed')
+          end
         end
 
         def pip_install?(flag)
@@ -1061,7 +1094,7 @@ module Squared
         end
 
         def installable?
-          setuptools? || !pyprojectfile.nil?
+          setuptools? || !!pyprojectfile
         end
 
         def setuptools?