squared 0.7.6 → 0.8.0

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

@@ -2,10 +2,12 @@
 
 module Squared
   module Workspace
+    GIT_PROTO = %r{\A(https?|ssh|git|file)://}i.freeze
+    private_constant :GIT_PROTO
+
     module Git
       GIT_REPO = Support.hashobj
-      GIT_PROTO = %r{\A(https?|ssh|git|file)://}i.freeze
-      private_constant :GIT_REPO, :GIT_PROTO
+      private_constant :GIT_REPO
 
       attr_reader :revfile
 
@@ -187,6 +189,7 @@ module Squared
                      A|after-context=i B|before-context=i color=b C|context=i m|max-count=n max-depth=n
                      open-files-in-pager=b threads=n].freeze,
             mv: %w[k f|force n|dry-run v|verbose].freeze,
+            remote: %w[f m=b t=b no-tags tags mirror=b].freeze,
             revert: %w[e S=bm? n|no-commit reference cleanup=b gpg-sign=b? m|mainline=i s|signoff strategy=b
                        X|strategy-option=b].freeze,
             rm: %w[r cached f|force n|dry-run ignore-unmatch pathspec-file-nul q|quiet sparse
@@ -233,6 +236,16 @@ module Squared
                      f|force-rebase ignore-date ignore-whitespace i|interactive keep-base m|merge no-ff q|quiet quit
                      reset-author-date root show-current-patch signoff v|verbose empty=b x|exec=q gpg-sign=b? onto=b
                      r|rebase-merges=b s|strategy=b X|strategy-option=b trailer=q whitespace=b].freeze,
+          remote: {
+            rename: %w[no-progress progress].freeze,
+            remove: [].freeze,
+            'set-head': %w[a|auto d|delete].freeze,
+            'set-branches': %w[add].freeze,
+            'get-url': %w[all push].freeze,
+            'set-url': %w[add delete push].freeze,
+            show: %w[n].freeze,
+            prune: %w[n dry-run].freeze
+          }.freeze,
           reset: %w[N pathspec-file-nul q|quiet pathspec-from-file=p].freeze,
           restore: %w[ignore-skip-worktree-bits ignore-unmerged m|merge ours p|patch pathspec-file-nul q|quiet S|staged
                       theirs W|worktree conflict=b pathspec-from-file=p s|source=b].freeze,
@@ -332,7 +345,7 @@ module Squared
         end
 
         subtasks({
-          'branch' => %i[create track delete move copy list current].freeze,
+          'branch' => %i[create track delete move copy list all current].freeze,
           'checkout' => %i[commit branch track detach path].freeze,
           'commit' => %i[add all amend amend-orig fixup].freeze,
           'diff' => %i[head branch files view between contain].freeze,
@@ -345,12 +358,13 @@ module Squared
           'pull' => %i[origin remote all].freeze,
           'rebase' => %i[branch onto send].freeze,
           'refs' => %i[heads tags remote].freeze,
+          'remote' => %i[add select].freeze,
           'reset' => %i[commit index patch mode undo].freeze,
           'restore' => %i[source staged worktree].freeze,
           'rev' => %i[commit branch build output].freeze,
           'show' => %i[format oneline textconv].freeze,
           'sparse-checkout' => %i[add reapply list clean disable].freeze,
-          'stash' => %i[push pop apply branch drop clear list all staged worktree].freeze,
+          'stash' => %i[push pop apply drop command].freeze,
           'submodule' => %i[status update branch url sync].freeze,
           'switch' => %i[branch create detach].freeze,
           'tag' => %i[add sign delete list].freeze
@@ -481,31 +495,37 @@ module Squared
                     when :add, :sign
                       format_desc action, flag, 'name,message?,commit?,remote?'
                       task flag, [:name, :message, :commit, :remote] do |_, args|
-                        remote = if (name = args.name)
-                                   message = args.message
-                                   commit = commithead args.commit
-                                   args.remote
-                                 else
-                                   commit, name, message = choice_commit(reflog: false, series: true,
-                                                                         values: [
-                                                                           ['Enter tag name', true],
-                                                                           'Enter message'
-                                                                         ])
-                                   choice_remote
-                                 end
+                        if (name = args.name)
+                          message = args.message
+                          commit = commithead args.commit
+                          remote = args.remote
+                          if !remote && commit && !commit.start_with?('@') && !git_grep('remote', commit).empty?
+                            remote = commit
+                            commit = nil
+                          end
+                        else
+                          commit, name, message = choice_commit(reflog: false, series: true,
+                                                                values: [['Enter tag name', true], 'Enter message'])
+                          remote = choice_remote
+                        end
                         ret = tag(flag, refs: [name], message: message, commit: commit, remote: remote)
                         success?(ret, !remote)
                       end
                     end
                   when 'stash'
-                    format_desc(action, flag, 'opts*', after: case flag
+                    format_desc(action, flag, 'opts*', before: ('clear|list|all|staged|worktree' if flag == :command),
+                                                       after: case flag
                                                               when :push then 'pathspec*,:'
-                                                              when :branch then 'name,stash/:'
-                                                              when :clear, :list, :all then nil
+                                                              when :command then nil
                                                               else 'stash?|:'
                                                               end)
                     task flag do |_, args|
-                      stash flag, args.to_a
+                      args = args.to_a
+                      if flag == :command
+                        param_guard(action, flag, args: args)
+                        flag = args.shift.to_sym
+                      end
+                      stash flag, args
                     end
                   when 'log', 'diff', 'range-diff'
                     case flag
@@ -531,7 +551,7 @@ module Squared
                             end
                             args = args.drop(index.size)
                           end
-                          log!(flag, args, index: index)
+                          log_(flag, args, index: index)
                         end
                         next
                       end
@@ -549,7 +569,7 @@ module Squared
                           opts.concat(refs.shellsplit) if refs
                         end
                         if action == 'log'
-                          log!(flag, opts, range: range)
+                          log_(flag, opts, range: range)
                         else
                           diff(flag, opts, range: range, from: action.to_sym)
                         end
@@ -613,7 +633,7 @@ module Squared
                           end
                         end
                         param_guard(action, flag, args: grep)
-                        log!(flag, opts, grep: grep)
+                        log_(flag, opts, grep: grep)
                       end
                     end
                   when 'checkout'
@@ -730,12 +750,7 @@ module Squared
                       task flag do |_, args|
                         branch flag, args.to_a
                       end
-                    when :current
-                      format_desc action, flag
-                      task flag do
-                        branch flag
-                      end
-                    else
+                    when :move, :copy
                       format_desc action, flag, 'branch,oldbranch?'
                       task flag, [:branch, :oldbranch] do |_, args|
                         if (branch = args.branch)
@@ -746,6 +761,11 @@ module Squared
                         end
                         branch(flag, refs: [oldbranch, branch])
                       end
+                    else
+                      format_desc action, flag
+                      task flag do
+                        branch flag
+                      end
                     end
                   when 'switch'
                     case flag
@@ -794,10 +814,10 @@ module Squared
                                end
                         success?(reset(flag, args, commit: commit))
                       end
-                    when :index, :undo
-                      format_desc(action, flag, ('opts*,pathspec*' if flag == :index))
+                    when :index
+                      format_desc action, flag, 'opts*,pathspec*'
                       task flag do |_, args|
-                        reset(flag, flag == :index ? args.to_a : [])
+                        reset(flag, args.to_a, refs: [])
                       end
                     when :mode
                       format_desc action, flag, 'mode,ref/:'
@@ -812,6 +832,11 @@ module Squared
                         ref = commithead args.ref
                         reset(flag, refs: args.extras, ref: !ref || ref == ':' ? choice_commit(reflog: false) : ref)
                       end
+                    when :undo
+                      format_desc action, flag
+                      task flag do
+                        reset flag
+                      end
                     end
                   when 'show'
                     case flag
@@ -976,6 +1001,51 @@ module Squared
                     task flag do |_, args|
                       sparse_checkout flag, args.to_a
                     end
+                  when 'remote'
+                    if flag == :add
+                      format_desc action, flag, 'name,url,opts*'
+                      task flag, [:name, :url] do |_, args|
+                        name = param_guard(action, flag, args: args, key: :name)
+                        url = param_guard(action, flag, args: args, key: :url)
+                        raise ArgumentError, message('invalid url', hint: url) unless url.match?(GIT_PROTO)
+
+                        git(:remote, args.extras, extras: [name, url])
+                      end
+                    else
+                      format_desc action, flag, "#{OPT_GIT[:remote].keys.join('|')},opts*" if TASK_METADATA
+                      task flag, [:command] do |_, args|
+                        command = param_guard(action, flag, args: args, key: :command)
+                        cmd, opts = git_session('remote', opts: args.extras)
+                        param = case command
+                                when 'rename'
+                                  'New name'
+                                when 'set-branches'
+                                  'Branches'
+                                when 'set-head'
+                                  'Branch' if opts.empty?
+                                when 'set-url'
+                                  'URL'
+                                end
+                        remote, target = choice_remote(force: true, values: param ? [[param, true]] : [])
+                        cmd << '-v' if has_value!(opts, 'v', 'verbose') && command == 'show'
+                        cmd << command << remote
+                        op = OptionPartition.new(opts, OPT_GIT[:remote].fetch(command.to_sym, []), cmd,
+                                                 project: self, strict: strict?)
+                        case command
+                        when 'remove'
+                          exit 1 unless confirm_basic('Remove?', remote, 'N')
+                        when 'get-url', 'show', 'prune'
+                          status = false
+                        else
+                          raise ArgumentError, message('unrecognized command', command, hint: action) unless param
+                          raise ArgumentError, message('missing target', command, hint: action) unless target
+
+                          op << target
+                        end
+                        op.clear
+                        success?(source(banner: !quiet?), status)
+                      end
+                    end
                   when 'git'
                     before = case flag
                              when :blame
@@ -1056,7 +1126,8 @@ module Squared
                 cur ||= line.delete_prefix!('* ')
                 heads << line if matchany?(reg, line)
               end
-              raise_error 'head not found', hint: 'for-each-ref' unless cur
+              raise message('head not found', hint: 'for-each-ref') unless cur
+
               opts << 'ff-only' if opts.empty? && option('ff-only', notequals: '0')
               (heads.dup << cur).each_with_index do |branch, i|
                 next unless (i < heads.size && cur != branch) || i == heads.size
@@ -1169,7 +1240,7 @@ module Squared
           option('no-tags') { opts[:'no-tags'] = true }
           opts.delete(:'recurse-submodules') || opts.delete(:'no-recurse-submodules') if append_submodules(from: :clone)
           append_hash opts
-          cmd << '--quiet' if option('quiet') || !verbose
+          cmd << '--quiet' if quiet? || option('quiet')
           append_value(data[0], path, delim: true)
           source(sync: sync, banner: sync && !quiet?, multiple: !sync || quiet?)
         end
@@ -1219,18 +1290,17 @@ module Squared
                 else
                   op << out
                 end
-              elsif !op.empty?
-                op.add_first(prefix: ':')
               elsif flag == :branch
-                raise ArgumentError, 'no branch name'
+                op.add_first(prefix: ':', expect: 'no branch name')
               end
-              op.clear
+              op.add_first
+                .clear
             when :clear
               n = sub_style file.read.lines.size, theme[:inline]
               s = sub_style name, theme[:active]
               source(stdout: true) if confirm("Remove #{n} stash entries from #{s}?", 'N')
               return
-            when :list
+            else
               op.clear
               out, banner, from = source(io: true)
               print_item banner
@@ -1321,6 +1391,7 @@ module Squared
 
           sync = kwargs.fetch(:sync) { invoked_sync?('revbuild', flag) }
           kwargs = kwargs.key?(:include) || kwargs.key?(:exclude) ? kw.call : @revbuild || { pass: true }
+          from = symjoin 'git', 'revbuild'
           case flag
           when :build
             op = OptionPartition.new(opts, VAL_GIT[:revbuild].map { |key| "#{key}=b?" }, project: self, strict: strict?)
@@ -1333,7 +1404,7 @@ module Squared
                    .compact
             OptionPartition.uniq!(args)
           end
-          run_p(*Array(kwargs[:before]), sync: sync, from: :revbuild) if kwargs[:before]
+          run_p(*Array(kwargs[:before]), sync: sync, from: symjoin(from, 'before')) if kwargs[:before]
           force = if (force = env('REVBUILD_FORCE', strict: true))
                     force != '0' && force != 'false'
                   elsif (force = ENV['GIT_FORCE'] || ENV['REVBUILD_FORCE'])
@@ -1348,7 +1419,7 @@ module Squared
             end
           end
           start = time_epoch
-          ret = build(*@output, sync: sync, from: symjoin('git', 'revbuild'))
+          ret = build(*@output, sync: sync, from: from)
         rescue => e
           print_error(e, pass: true)
         else
@@ -1356,7 +1427,7 @@ module Squared
             print_status('revbuild', subject: name, start: start, loglevel: Logger::WARN, from: :failed)
             workspace.rev_timeutc name, 'build'
           else
-            run_p(*Array(kwargs[:after]), sync: sync, from: :revbuild) if kwargs[:after] && ret != false
+            run_p(*Array(kwargs[:after]), sync: sync, from: symjoin(from, 'after')) if kwargs[:after] && ret != false
             print_status('revbuild', subject: name, start: start)
             workspace.rev_write(name, { 'revision' => sha, 'files' => status_digest(*args, **kwargs) },
                                 sync: sync, utc: 'build')
@@ -1463,7 +1534,7 @@ module Squared
           ret
         end
 
-        def log!(flag, opts = [], range: [], index: [], grep: [])
+        def log_(flag, opts = [], range: [], index: [], grep: [])
           cmd, opts = git_session('log', opts: opts)
           op = OptionPartition.new(opts, collect_hash(OPT_GIT[:log]), cmd, project: self, strict: strict?,
                                                                            no: collect_hash(OPT_GIT[:no][:log]),
@@ -1512,7 +1583,8 @@ module Squared
               op.add_quote(branch) if branch
               if !index.empty?
                 if op.arg?('cached')
-                  raise_error "single commit: #{index.join(', ')}", hint: 'cached' unless index.size == 1
+                  raise message('single commit only', index.join(', '), hint: flag) unless index.size == 1
+
                   op << index.first
                 else
                   op.merge(index)
@@ -1575,9 +1647,9 @@ module Squared
             next if line.empty?
 
             branch, origin, hint = line.split('...')
-            if hint && !hint.match?(/^\[(\D+0,\D+0)\]$/)
-              raise_error 'work tree is not usable', hint: hint[1..-2]
-            elsif (!origin || origin.empty?) && !dryrun?
+            raise message('work tree is not usable', hint: hint[1..-2]) if hint && !hint.match?(/^\[(\D+0,\D+0)\]$/)
+
+            if (!origin || origin.empty?) && !dryrun?
               return nil if pass
 
               unless (origin = option('upstream', prefix: 'git', ignore: false))
@@ -1610,7 +1682,7 @@ module Squared
           elsif flag == :'amend-orig' || option('edit', equals: '0')
             co << '--no-edit'
           end
-          pu << '--force-with-lease' if amend
+          pu << (option('f', 'force') ? '--force' : '--force-with-lease') if amend
           pu.merge(repotrack(origin, branch))
           adding = git_spawn 'diff --name-only --no-color'
           source op
@@ -1680,7 +1752,8 @@ module Squared
             cmd << shell_quote(target)
             cmd << shell_quote(ref) if ref
           when :track
-            raise_error 'invalid upstream', hint: ref unless ref.include?('/')
+            raise message('invalid upstream', hint: ref) unless ref.include?('/')
+
             if ref.delete_prefix!('^')
               cmd << '--unset-upstream' << shell_quote(ref)
               remote = false
@@ -1691,7 +1764,7 @@ module Squared
             end
           when :delete
             remote&.each { |val| source git_output('push --delete', *val.split('/', 2).quote!) }
-            force, list = refs.partition { |val| val.start_with?('~', '^') }
+            force, list = refs.partition { |val| val.start_with?(/[~^]/) }
             force.each do |val|
               r = '-r' if val.delete!('~')
               source git_output('branch', val.delete!('^') ? '-D' : '-d', r, shell_quote(val))
@@ -1738,15 +1811,15 @@ module Squared
                 data = line.sub(/^\*?\s+/, '').split(/\s+/, 3)
                 a = sub_style(data[0], theme[:inline], styles: (:underline if !first && line.start_with?('*')))
                 b = commitstyle data[1]
-                r = /\A(?:\[((?~\]\s))\]\s)?(.+)\z/m.match(data[2])
-                if (r1 = r[1]) && r1 =~ /^(.+):(?: ([a-z]+) (\d+),)? ([a-z]+) (\d+)$/
+                md = /\A(?:\[((?~\]\s))\]\s)?(.+)\z/m.match(data[2])
+                if (r1 = md[1]) && r1 =~ /^(.+):(?: ([a-z]+) (\d+),)? ([a-z]+) (\d+)$/
                   write = ->(s1, s2) { "#{s1.capitalize.rjust(7)}: #{sub_style(s2, theme[:warn])}" }
                   r1 = $1
                   r2 = $2 && write.call($2, $3)
                   r3 = write.call($4, $5)
                 end
                 r1 = nil if r1 == "origin/#{data[0]}"
-                ["#{"\n" unless index == 0} Branch: #{a.subhint(r1)}", r2, r3, " Commit: #{b}", "Message: #{r[2]}"]
+                ["#{"\n" unless index == 0} Branch: #{a.subhint(r1)}", r2, r3, " Commit: #{b}", "Message: #{md[2]}"]
                   .compact
                   .join("\n")
               end
@@ -1902,9 +1975,9 @@ module Squared
           list_result(ret, 'files', grep: op.extras, from: from)
         end
 
-        def git(flag, opts = [])
+        def git(flag, opts = [], extras: [])
           cmd, opts = git_session(flag, opts: opts)
-          list = OPT_GIT[:git].fetch(flag, []) + OPT_GIT.fetch(flag, [])
+          list = OPT_GIT[:git].fetch(flag, []) + OPT_GIT.fetch(flag, []).yield_self { |a| a.is_a?(Array) ? a : [] }
           case flag
           when :add
             list.concat(OPT_GIT[:log][:diff_context])
@@ -1953,6 +2026,12 @@ module Squared
               end
             end
             return source(git_session('status -s'), banner: false) unless append_pathspec(op.extras)
+          when :remote
+            op.unshift(*extras)
+            op.adjoin('add')
+              .add_first(expect: 'missing name')
+              .add_first(expect: 'missing URL', quote: true)
+              .clear
           when :mv
             refs = projectmap op.extras
             raise 'no source/destination' unless refs.size > 1
@@ -1981,7 +2060,7 @@ module Squared
           when :revert, :mv, :rm
             source(sync: false, stderr: true)
           else
-            success?(source, flag == :'sparse-checkout' || (flag == :add && !op.arg?('verbose')))
+            success?(source, %i[remote sparse-checkout].include?(flag) || (flag == :add && !op.arg?('verbose')))
           end
         end
 
@@ -1993,7 +2072,7 @@ module Squared
           build? && !!workspace.revfile
         end
 
-        def enabled?(*, **kwargs)
+        def enabled?(**kwargs)
           super || (kwargs[:base] == false && !!clone?)
         end
 
@@ -2001,6 +2080,10 @@ module Squared
 
         def source(cmd = @session, exception: true, io: false, sync: true, stdout: false, stderr: false, banner: true,
                    multiple: false, hint: nil, from: nil, timeout: nil, send: :system, **kwargs)
+          unless cmd
+            print_error('no git session started', subject: project, hint: from, pass: true)
+            return
+          end
           cmd = cmd.target if cmd.is_a?(OptionPartition)
           if io && banner == false
             from = nil
@@ -2070,7 +2153,7 @@ module Squared
           elsif stdout?
             styles = theme.fetch(:banner, []).reject { |val| val.to_s.end_with?('!') }
             styles << :bold if styles.size <= 1
-            puts print_footer("#{size} #{size == 1 ? type.sub(/(?:(?<!l)e)?s\z/, '') : type}",
+            puts print_footer("#{size} #{size == 1 ? type.sub(/((?<!l)e)?s\z/, '') : type}",
                               sub: opt_style(styles, /^(\d+)(.+)$/))
           end
           on :last, from
@@ -2159,7 +2242,7 @@ module Squared
           end
           op << '--verbose' if (flag || from == :fetch) && stdout? && !op.arg?('quiet')
           if remote
-            op.append(remote, delim: true)
+            op.append(remote)
             if (val = option('refspec', target: target, strict: true))
               op.append(split_escape(val))
             else
@@ -2277,14 +2360,24 @@ module Squared
                                                                       stdout: stdout, banner: banner, **kwargs)
         end
 
-        def dryrun?(*, target: @session, prefix: target&.first)
+        def git_grep(cmd, *grep, equals: true, **kwargs)
+          ret = []
+          git_spawn(cmd, **kwargs).lines(chomp: true).each do |line|
+            if grep.any? { |val| val.is_a?(Regexp) ? val.match?(line) : equals ? line == val : line.include?(val) }
+              ret << line
+            end
+          end
+          ret
+        end
+
+        def dryrun?(target: @session, prefix: target&.first, **)
           Array(target).include?('--dry-run') || !option('dry-run', target: target, prefix: prefix).nil?
         end
 
-        def quiet?(*, target: @session, **)
-          return false unless target
+        def quiet?(target: @session, **)
+          return silent? unless target
 
-          target.include?('--quiet') || (target.include?('-q') && target.first.stripext == 'git')
+          silent? || target.include?('--quiet') || (target.include?('-q') && target.first.stripext == 'git')
         end
 
         def gitpath(*args)
@@ -2293,7 +2386,8 @@ module Squared
 
         def repotrack(origin, branch, quote: true)
           n = origin&.index('/')
-          raise_error "missing #{origin ? 'branch' : 'remote'} name", hint: origin unless n && branch
+          raise message("missing #{origin ? 'branch' : 'remote'} name", hint: origin) unless n && branch
+
           branch = "#{branch}:#{origin[n.succ..-1]}" unless origin.end_with?("/#{branch}")
           ret = [origin[0..n.pred], branch]
           ret.quote! if quote