squared 0.5.0 → 0.5.2

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

@@ -8,7 +8,7 @@ module Squared
         DIR_PYTHON = (DEP_PYTHON + %w[README.rst]).freeze
         OPT_PYTHON = {
           common: %w[b B d E h i I O OO 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 v|verbose w|wheel x|skip-dependency-check C|config-setting=q installer=b
+          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,
           venv: %w[clear copies symlinks system-site-packages upgrade upgrade-deps without-scm-ignore-files without-pip
                    prompt=q].freeze
@@ -28,11 +28,18 @@ module Squared
           freeze: %w[all exclude-editable l|local user exclude=b path=p r|requirement=p].freeze
         }.freeze
         OPT_POETRY = {
-          common: %w[ansi no-ansi no-cache n|no-interaction no-plugins P|project=p q|quiet v|verbose].freeze,
+          common: %w[ansi no-ansi no-cache n|no-interaction no-plugins q|quiet v|verbose P|project=p].freeze,
           build: %w[clean config-settings=qq f|format=b o|output=p].freeze,
-          publish: %w[build dry-run client-cert=p cert=p dist-dir=p p|password=b r|repository=b skip-existing
+          publish: %w[build dry-run skip-existing cert=p client-cert=p dist-dir=p p|password=b r|repository=b
                       u|username=b].freeze
         }.freeze
+        OPT_PDM = {
+          common: %w[I|ignore-python no-cache n|non-interactive].freeze,
+          build: %w[C=bm no-clean no-isolation no-sdist no-wheel quiet verbose config-setting=q d|dest=p p|project=p
+                    k|skip=b].freeze,
+          publish: %w[no-build no-very-ssl quiet S|sign skip-existing verbose ca-certs=p c|comment=q d|dest=p identity=b
+                      p|password=q p|project=p r|repository=q k|skip=b u|username=b].freeze
+        }.freeze
         OPT_HATCH = {
           common: %w[color interactive no-color no-interactive cache-dir=p config=p data-dir=p e|env=b p|project=b
                      q|quiet v|verbose].freeze,
@@ -41,11 +48,11 @@ module Squared
                       p|publisher=b r|repo=b u|user=q].freeze
         }.freeze
         OPT_TWINE = {
-          publish: %w[attestations disable-progress-bar non-interactive skip-existing verbose s|sign c|comment=q
-                      config-file=p cert=p client-cert=p i|identity=b p|password=q r|repository=b repository-url=q
+          publish: %w[attestations disable-progress-bar non-interactive s|sign skip-existing verbose cert=p
+                      client-cert=p c|comment=q config-file=p i|identity=b p|password=q r|repository=b repository-url=q
                       sign-with=b u|username=q].freeze
         }.freeze
-        private_constant :DEP_PYTHON, :DIR_PYTHON, :OPT_PYTHON, :OPT_PIP, :OPT_POETRY, :OPT_HATCH, :OPT_TWINE
+        private_constant :DEP_PYTHON, :DIR_PYTHON, :OPT_PYTHON, :OPT_PIP, :OPT_POETRY, :OPT_PDM, :OPT_HATCH, :OPT_TWINE
 
         class << self
           def populate(*); end
@@ -71,7 +78,7 @@ module Squared
 
         attr_reader :venv, :editable
 
-        def initialize(*, venv: nil, editable: '.', verbose: nil, **kwargs)
+        def initialize(*, editable: '.', verbose: nil, **kwargs)
           super
           if @pass.include?(Python.ref)
             initialize_ref Python.ref
@@ -80,18 +87,20 @@ module Squared
             initialize_build(Python.ref, **kwargs)
             initialize_env(**kwargs)
           end
-          dependfile_set DEP_PYTHON
           @verbose = verbose.size if verbose.is_a?(String) && verbose.match?(/\Av+\z/)
+          dependfile_set DEP_PYTHON
           editable_set editable
-          venv_set venv if venv
+          venv_set kwargs[:venv]
         end
 
         subtasks({
           'venv' => %i[exec create remove show].freeze,
           'pip' => %i[uninstall freeze].freeze,
           'install' => %i[user force upgrade target editable].freeze,
-          'build' => %i[python poetry hatch].freeze,
-          'publish' => %i[poetry twine hatch].freeze,
+          'outdated' => %i[major minor patch].freeze,
+          'build' => %i[python poetry pdm hatch].freeze,
+          'publish' => %i[poetry pdm hatch twine].freeze,
+          'run' => nil,
           'exec' => nil
         })
 
@@ -109,6 +118,62 @@ module Squared
 
               if flags.nil?
                 case action
+                when 'run'
+                  next unless pyprojectfile
+
+                  format_desc action, nil, "script+|#{indexchar}index+|#,pattern*"
+                  task action, [:command] do |_, args|
+                    found = 0
+                    ['tool.poetry.scripts', 'tool.pdm.scripts', 'project.scripts'].each_with_index do |table, index|
+                      next if (list = read_pyproject(table)).empty?
+
+                      if args.command == '#'
+                        format_list(list, "run[#{indexchar}N]", 'scripts', grep: args.extras, from: pyprojectfile)
+                        found |= 1
+                      else
+                        args.to_a.each do |val|
+                          if (n, = indexitem(val))
+                            if (script, = list[n - 1])
+                              case index
+                              when 0
+                                script = session_output 'poetry', 'run', script
+                              when 1
+                                script = pdm_session 'run', script
+                              else
+                                venv_init
+                              end
+                              found |= 1
+                              run(script, from: :run)
+                            elsif exception
+                              indexerror n, list
+                            else
+                              found |= 2
+                              log.warn "run script #{n} of #{list.size} (out of range)"
+                            end
+                          else
+                            case index
+                            when 0
+                              found |= 1
+                              run(session_output('poetry', 'run', val), from: :run)
+                            when 1
+                              found |= 1
+                              run(pdm_session('run', val), from: :run)
+                            else
+                              raise_error("script: #{val}", hint: 'unknown') if exception
+                              found |= 2
+                              log.warn "run script \"#{val}\" (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
+                  end
                 when 'exec'
                   format_desc action, nil, 'command|:,args*'
                   task action do |_, args|
@@ -123,7 +188,7 @@ module Squared
                             end
                             args.join(' ')
                           end
-                    Kernel.exec(cmd, chdir: path)
+                    shell(cmd, name: :exec, chdir: path)
                   end
                 end
               else
@@ -214,14 +279,30 @@ module Squared
                           depend flag, args.to_a
                         end
                       end
+                    when 'outdated'
+                      format_desc action, flag, 'eager?,user?'
+                      task flag do |_, args|
+                        outdated flag, args.to_a
+                      end
                     when 'build'
+                      case flag
+                      when :poetry
+                        next unless build_backend == 'poetry.core.masonry.api'
+                      when :pdm
+                        next unless build_backend == 'pdm.backend'
+                      when :hatch
+                        next unless build_backend == 'hatchling.build'
+                      end
                       format_desc(action, flag, 'opts*', after: case flag
                                                                 when :python then 'srcdir?'
+                                                                when :poetry then 'output?'
+                                                                when :pdm then 'dest?'
                                                                 when :hatch then 'location?'
                                                                 end)
                       task flag do |_, args|
                         build! flag, args.to_a
                       end
+                      break unless flag == :python
                     when 'publish'
                       format_desc(action, flag, 'opts*', after: case flag
                                                                 when :hatch then 'artifacts?'
@@ -269,7 +350,7 @@ module Squared
           end
         end
 
-        def outdated(*, sync: invoked_sync?('outdated'))
+        def outdated(flag = nil, opts = [], sync: invoked_sync?('outdated'))
           cmd = pip_session 'list', '--outdated'
           append_global
           cmd = session_done cmd
@@ -279,28 +360,35 @@ module Squared
           print_item banner if sync
           start = 0
           found = 0
-          major = 0
+          major = []
+          minor = []
+          patch = []
           pwd_set(from: :outdated) do
             buffer = []
             out = ->(val) { sync ? puts(val) : buffer << val }
             IO.popen(runenv || {}, cmd).each do |line|
-              next if line.match?(/^[\s-]+$/)
+              next if line.match?(/^[ -]+$/)
 
               if start > 0
                 unless stdin?
-                  data = line.scan(SEM_VER)
-                  next unless (cur = data.shift) && (lat = data.shift)
+                  cur, lat = line.scan(SEM_VER)
+                  next unless cur && lat
 
                   latest = lat.join
                   current = cur.join
                   semver cur
                   semver lat
-                  if semmajor?(cur, lat)
-                    type = 2
-                    major += 1
-                  else
-                    type = cur[2] == lat[2] ? 0 : 1
-                  end
+                  name = line.split(' ', 2).first
+                  type = if semmajor?(cur, lat)
+                           major << name
+                           2
+                         elsif cur[2] == lat[2]
+                           patch << name
+                           0
+                         else
+                           minor << name
+                           1
+                         end
                   if type == 0
                     styles = color(:yellow)
                   else
@@ -334,7 +422,18 @@ module Squared
               puts buffer
             end
             if found > 0
-              puts print_footer empty_status('Updates are available', 'major', major)
+              print_status(major.size, minor.size, patch.size, from: :outdated)
+              pkg = case flag
+                    when :major
+                      major + minor + patch
+                    when :minor
+                      minor + patch
+                    when :patch
+                      patch
+                    end
+              unless !pkg || pkg.empty?
+                install(:upgrade, pkg, strategy: opts.include?('eager') ? 'eager' : nil, user: opts.include?('user'))
+              end
             elsif start == 0
               puts 'No updates were found'
             end
@@ -342,7 +441,7 @@ module Squared
           on :last, :outdated
         end
 
-        def install(flag, opts = [], strategy: nil)
+        def install(flag, opts = [], strategy: nil, user: nil)
           cmd = pip_session 'install'
           out = append_pip(flag, opts, from: :install)
           case flag
@@ -352,6 +451,7 @@ module Squared
           when :upgrade
             raise_error('no packages listed', hint: flag) if out.empty?
             cmd << '--upgrade'
+            cmd << '--user' if user
             cmd << basic_option('upgrade-strategy', strategy) if strategy
             append_value out
           end
@@ -366,6 +466,9 @@ module Squared
           when :poetry
             cmd = poetry_session 'build'
             list = OPT_POETRY[:build] + OPT_POETRY[:common]
+          when :pdm
+            cmd, opts = pdm_session('build', opts: opts)
+            list = OPT_PDM[:build]
           when :hatch
             cmd, opts = hatch_session('build', opts: opts)
             list = OPT_HATCH[:build]
@@ -373,7 +476,7 @@ module Squared
           srcdir = nil
           op = OptionPartition.new(opts, list, cmd, project: self, single: singleopt(flag))
           op.each do |opt|
-            if !srcdir && basepath(opt).exist? && projectpath?(opt)
+            if !srcdir && basepath(opt.chomp('*')).exist? && projectpath?(opt.chomp('*'))
               srcdir = opt
             else
               op.found << opt
@@ -381,12 +484,13 @@ module Squared
           end
           op.swap
           case flag
-          when :poetry
+          when :poetry, :pdm
             if srcdir
-              if op.arg?('o', 'output')
+              args = flag == :pdm ? ['d', 'dest'] : ['o', 'output']
+              if op.arg?(*args)
                 op.extras << srcdir
               else
-                op << quote_option('output', path + srcdir)
+                op << quote_option(args.last, path + srcdir)
               end
               srcdir = nil
             end
@@ -408,21 +512,35 @@ module Squared
           when :poetry
             poetry_session 'publish'
             list = OPT_POETRY[:publish] + OPT_POETRY[:common]
-          when :twine
-            session 'twine', 'upload'
-            list = OPT_TWINE[:publish]
+          when :pdm
+            opts = pdm_session('publish', opts: opts).last
+            list = OPT_PDM[:publish]
           when :hatch
             opts = hatch_session('publish', opts: opts).last
             list = OPT_HATCH[:publish]
+          when :twine
+            session 'twine', 'upload'
+            list = OPT_TWINE[:publish]
           end
           op = OptionPartition.new(opts, list, @session, project: self, single: singleopt(flag))
-          if op.empty?
-            dist = path + 'dist'
-            raise_error('no source files found', hint: dist) unless dist.directory? && !dist.empty?
-            op.extras << "#{dist}/*" unless flag == :poetry
+          dist = lambda do
+            (path + 'dist').tap do |dir|
+              raise_error('no source files found', hint: dir) unless dir.directory? && !dir.empty?
+            end
           end
-          op.append
-          run(from: :"#{flag}:publish")
+          case flag
+          when :hatch, :twine
+            if op.empty?
+              op.extras << "#{dist.call}/*"
+            else
+              op.map! { |val| path + val }
+            end
+            op.append
+          else
+            dist.call unless op.arg?(*(flag == :poetry ? ['dist-dir'] : ['d', 'dest']))
+            op.clear(pass: false)
+          end
+          run(from: :"#{flag}:publish", interactive: "Publish #{sub_style(project, styles: theme[:active])}")
         end
 
         def pip(flag, opts = [])
@@ -495,11 +613,19 @@ module Squared
           ret
         end
 
+        def pdm_session(*cmd, opts: nil)
+          create_session(*cmd, name: 'pdm', common: OPT_PDM[:common], opts: opts)
+        end
+
         def hatch_session(*cmd, opts: nil)
-          return session('hatch', *preopts, *cmd, path: venv.nil?) unless opts
+          create_session(*cmd, name: 'hatch', common: OPT_HATCH[:common], opts: opts)
+        end
+
+        def create_session(*cmd, name:, common:, opts: nil)
+          return session(name, *preopts, *cmd, path: venv.nil?) unless opts
 
-          op = OptionPartition.new(opts, OPT_HATCH[:common], project: self, single: singleopt)
-          ret = session('hatch', *op.to_a, *cmd, path: venv.nil?)
+          op = OptionPartition.new(opts, common, project: self, single: singleopt)
+          ret = session(name, *op.to_a, *cmd, path: venv.nil?)
           [ret, op.extras]
         end
 
@@ -577,13 +703,77 @@ module Squared
           append_nocolor(target: target)
         end
 
-        def editable_set(val)
-          @editable = case val
-                      when '.', Pathname
-                        val
-                      when String
-                        Pathname.new(editable)
+        def build_backend
+          @build_backend ||= read_pyproject('build-system', 'build-backend') || ''
+        end
+
+        def read_pyproject(table, key = nil)
+          return [] unless (file = pyprojectfile)
+
+          unless (ret = (@pyproject ||= {})[table])
+            ret = []
+            start = /^\s*\[#{Regexp.escape(table)}\]\s*$/
+            ch = nil
+            found = false
+            File.foreach(file) do |line|
+              if found
+                break if line.match?(/^\s*\[[\w.-]+\]\s*$/)
+
+                if ch
+                  val = line.rstrip
+                  case ch
+                  when '}', ']'
+                    ch = nil if val.end_with?(ch)
+                    val = "\n#{val}"
+                  else
+                    if val.chomp!(ch)
+                      ch = nil
+                    else
+                      val = line
+                    end
+                  end
+                  ret.last[1] += val
+                elsif (data = line.match(/^\s*(\S+)\s*=\s*([+-]?[\d.]+|true|false|("""|'''|["'\[{])(.*?))\s*$/))
+                  if (val = data[4])
+                    case (ch = data[3])
+                    when '{', '['
+                      val = "#{ch}#{val}"
+                      ch = ch == '{' ? '}' : ']'
+                      ch = nil if val.end_with?(ch)
+                    else
+                      if val.chomp!(ch)
+                        ch = nil
+                      elsif ch.size == 1
+                        next
                       end
+                    end
+                  else
+                    val = case (val = data[2])
+                          when 'true'
+                            true
+                          when 'false'
+                            false
+                          else
+                            val.include?('.') ? val.to_f : val.to_i
+                          end
+                  end
+                  ret << [data[1], val]
+                end
+              else
+                found = line.match?(start)
+              end
+            end
+            @pyproject[table] = ret
+          end
+          return ret.find { |val| val[0] == key }&.last if key
+
+          ret
+        end
+
+        def pyprojectfile
+          return unless (ret = basepath(DEP_PYTHON[2])).exist?
+
+          ret
         end
 
         def singleopt(flag = nil)
@@ -626,7 +816,18 @@ module Squared
           @venv&.join(workspace.windows? ? 'Scripts' : 'bin')
         end
 
+        def editable_set(val)
+          @editable = case val
+                      when '.', Pathname
+                        val
+                      when String
+                        Pathname.new(editable)
+                      end
+        end
+
         def venv_set(val)
+          return unless val
+
           if val.is_a?(Array)
             val, *opts = val
             @venvopts = opts