squared 0.8.4 → 0.9.0.pre1

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

@@ -16,19 +16,19 @@ module Squared
                       cgroup-parent=b iidfile=p label=q network=b no-cache-filter=b o|output=q platform=q q|quiet
                       resource=q secret=qq shm-size=b ssh=qq t|tag=b target=b ulimit=q].freeze,
             bake: %w[print list=q set=qq var=qq].freeze,
-            shared: %w[check load no-cache pull push allow=q call=b? f|file=p metadata-file=p policy=q progress=b
-                       provenance=q sbom=q].freeze
+            shared: %w[check load no-cache provenance pull push sbom allow=qq call=b? f|file=p metadata-file=p policy=q
+                       progress=b].freeze
           }.freeze,
           compose: {
-            common: %w[all-resources ansi=b compatibility dry-run env-file=p f|file=p parallel=n profile=b progress=b
+            common: %w[1 all-resources compatibility dry-run ansi=b env-file=p f|file=p parallel=n profile=b progress=b
                        project-directory=p p|project-name=b].freeze,
             build: %w[check no-cache print pull push with-dependencies q|quiet build-arg=qq builder=b m|memory=b
                       provenance=q sbom=q ssh=qq].freeze,
             create: %w[build force-recreate no-build no-recreate quiet-pull remove-orphans y|yes pull=b scale=i].freeze,
             exec: %w[d|detach privileged e|env=qq index=i T|no-tty user=b w|workdir=q].freeze,
             run: %w[build d|detach no-deps q|quiet quiet-build quiet-pull remove-orphans rm P|service-ports use-aliases
-                    cap-add=b cap-drop=b entrypoint=q e|env=qq env-from-file=p i|interactive=b? l|label=q name=b
-                    T|no-tty p|publish=q pull=b u|user=b v|volume=qq w|workdir=q].freeze,
+                    cap-add=b cap-drop=b entrypoint=q e|env=qq env-from-file=p i|interactive l|label=q name=b T|no-tty
+                    p|publish=q pull=b u|user=b v|volume=qq w|workdir=q].freeze,
             up: %w[abort-on-container-exit abort-on-container-failure always-recreate-deps attach-dependencies build
                    d|detach force-recreate menu no-build no-color no-deps no-log-prefix no-recreate no-start quiet-build
                    quiet-pull remove-orphans V|renew-anon-volumes timestamps wait w|watch y|yes attach=b
@@ -50,7 +50,7 @@ module Squared
                        network-alias=b oom-score-adj=b pid=b pids-limit=n platform=q p|publish=q pull=b restart=b
                        runtime=b security-opt=q shm-size=b stop-signal=b stop-timeout=i storage-opt=q sysctl=q tmpfs=q
                        ulimit=q u|user=b userns=b uts=b v|volume=qq volume-driver=b volumes-from=b w|workdir=q].freeze,
-            run: %w[d|detach detach-keys=q sig-proxy=b?].freeze,
+            run: %w[d|detach detach-keys=q sig-proxy].freeze,
             update: %w[blkio-weight=i cpu-period=i cpu-quota=i cpu-rt-period=i cpu-rt-runtime=i c|cpu-shares=i cpus=f
                        cpuset-cpus=b cpuset-mems=b m|memory=b memory-reservation=b memory-swap=n pids-limit=n
                        restart=q].freeze,
@@ -81,16 +81,17 @@ module Squared
             connect: %w[alias=b driver-opt=q gw-priority=n ip=b ip6=q link=b link-local-ip=q].freeze,
             create: %w[attachable config-only gateway=q ingress internal ipv4 ipv6 aux-address=q config-from=b
                        d|driver=b ip-range=q ipam-driver=b ipam-opt=q label=q o|opt=q scope=b subnet=q].freeze,
-            disconnect: %w[f|force].freeze
+            disconnect: %w[f|force].freeze,
+            rm: %w[f|force].freeze
           }.freeze
         }.freeze
         VAL_DOCKER = {
           run: {
             common: %w[source src destination dst target readonly ro].freeze,
-            bind: %w[bind-propagation].freeze,
+            bind: %w[bind-create-src bind-propagation].freeze,
             volume: %w[volume-subpath volume-nocopy volume-opt].freeze,
             tmpfs: %w[tmpfs-size tmpfs-mode].freeze,
-            image: %w[image-path].freeze
+            image: %w[image-subpath].freeze
           }.freeze,
           ls: {
             compose: %w[Name Image Command Service RunningFor Status Ports CreatedAt ExitCode Health ID Labels
@@ -127,7 +128,8 @@ module Squared
           'image' => %i[ls rm pull push tag save inspect history].freeze,
           'container' => %i[attach commit create diff exec inspect kill logs pause restart rm run start stats stop top
                             unpause update].freeze,
-          'network' => %i[connect disconnect create].freeze,
+          'run' => %i[rmi rmd mount volume entrypoint].freeze,
+          'network' => %i[connect disconnect create rm].freeze,
           'ls' => nil
         })
 
@@ -139,8 +141,7 @@ module Squared
           self.global = nil
           return unless dockerfile(kwargs[:file]).exist?
 
-          @tag = env('TAG', strict: true,
-                            ignore: false) || kwargs[:tag] || tagname(@project, @version || 'latest')
+          @tag = env('TAG', strict: true, ignore: false) || kwargs[:tag] || tagname(@project, @version || 'latest')
           @context = kwargs[:context]
           @client = kwargs[:client]&.each_with_object([]) do |(key, val), out|
             val = case (key = key.to_s)
@@ -149,7 +150,7 @@ module Squared
                   when 'tlscacert', 'tlscert', 'tlskey'
                     basepath!(val, type: 'f')
                   when 'context', 'host', 'log-level'
-                    quote_option(key, val)
+                    quote_option key, val
                   when 'debug', 'tls', 'tlsverify'
                     out << key
                     next
@@ -158,7 +159,7 @@ module Squared
           end
           @mounts = Array(kwargs[:mounts])
           @secrets = kwargs[:secrets]
-          self.registry = tagjoin kwargs[:registry], kwargs[:username]
+          @registry = tagjoin kwargs[:registry], kwargs[:username]
           @oci = tagjoin kwargs[:oci], kwargs[:username]
           initialize_ref Docker.ref
           initialize_logger kwargs[:log]
@@ -303,13 +304,22 @@ module Squared
                           end
                         end
                       else
-                        format_desc action, flag, "opts*,id/name#{case flag
-                                                                  when :stats then '*'
-                                                                  when :logs, :diff, :attach, :top then nil
-                                                                  else '+'
-                                                                  end}"
+                        single = %i[logs diff attach top].freeze
+                        format_desc action, flag, "opts*,id/name#{if flag == :stats
+                                                                    '*'
+                                                                  elsif !single.include?(flag)
+                                                                    '+'
+                                                                  end}|:"
                         task flag do |_, args|
-                          container flag, args.to_a
+                          args = args.to_a
+                          if choice!(args, first: false)
+                            args.concat(Array(choice_command(:container, multiple: !single.include?(flag))))
+                            if flag == :rm
+                              stop = filter_ps(:stop, column: 'ID').select { |id| args.include?(id) }
+                              container :stop, stop if !stop.empty? && confirm_basic('Stop?', stop.join(', '))
+                            end
+                          end
+                          container flag, args
                         end
                       end
                     when 'image'
@@ -358,14 +368,47 @@ module Squared
                           end
                         end
                       end
+                    when 'run'
+                      format_desc action, flag, 'id|:,opts*'
+                      task flag, [:id] do |_, args|
+                        id = if args.id == ':'
+                               choice_command(:image, multiple: false)
+                             else
+                               param_guard(action, flag, args: args, key: :id)
+                             end
+                        args = case flag
+                               when :mount
+                                 mount = []
+                                 loop do
+                                   s = readline("Enter mount ##{mount.size.succ}", force: false)
+                                   break if s.empty?
+
+                                   mount << (s.include?('type=') ? s : "type=bind,#{s}")
+                                 end
+                                 mount.map { |val| quote_option('mount', val, double: true) }
+                               when :volume
+                                 choice_command(:volume, multiple: true, series: true).map do |val|
+                                   basic_option 'volume', val
+                                 end
+                               when :entrypoint
+                                 ["entrypoint=#{shell_quote(readline('Entrypoint', force: true), double: true)}"]
+                               when :rmi
+                                 %w[rm i t]
+                               when :rmd
+                                 %w[rm d]
+                               end + args.extras
+                        container(:run, args.uniq, id: id)
+                      end
                     when 'network'
-                      format_desc action, flag, 'target,opts*'
-                      task flag, [:target] do |_, args|
-                        target = flag == :create ? param_guard(action, flag, args: args, key: :target) : args.target
-                        if target
-                          network(flag, args.extras, target: target)
+                      format_desc action, flag, 'name,opts*'
+                      task flag, [:name] do |_, args|
+                        name = flag == :create ? param_guard(action, flag, args: args, key: :name) : args.name
+                        if name
+                          network(flag, args.extras, name: name)
+                        elsif flag == :rm
+                          choice_command :'network:rm'
                         else
-                          choice_command flag
+                          choice_command(flag, id: choice_command(:container, force: false), series: true)
                         end
                       end
                     end
@@ -497,7 +540,7 @@ module Squared
             if flag == :publish
               id ||= option('tag', ignore: false) || op.shift || tagmain
               registry ||= option('registry') || op.shift || @oci
-              emptyargs op, flag
+              op.clear(strict: true)
               op << shell_quote(tagjoin(registry, id))
               return unless confirm_command(op.to_s, title: from, target: id)
             else
@@ -512,10 +555,10 @@ module Squared
               if multiple
                 op.concat(service) if service
                 op.append(delim: true, escape: true, strip: ':')
-              else
-                raise ArgumentError, message('no service was selected', hint: flag) unless service
-
+              elsif service
                 append_command(flag, service, op.extras, prompt: '::')
+              else
+                raise ArgumentError, message('no service was selected', hint: flag)
               end
             end
           end
@@ -530,6 +573,7 @@ module Squared
           list += ct[:create] if rc
           list += ct[:update] if rc ||= flag == :create
           op = OptionPartition.new(opts, list, cmd, args: rc || flag == :exec, **dockerinit)
+          option('entrypoint') { |val| op.add_option('entrypoint', val, escape: false, double: true) } if flag == :run
           from = symjoin 'container', flag
           case flag
           when :run, :create, :exec
@@ -557,23 +601,35 @@ module Squared
                       end
                     end
                     case k
-                    when 'readonly', 'ro'
-                      out << k
+                    when 'readonly', 'ro', 'volume-nocopy', 'bind-create-src'
+                      out << k if %w[bind volume].include?(type)
                       next
-                    when 'source', 'src', 'destination', 'dst', 'target', 'volume-subpath', 'image-path'
+                    when 'source', 'src', 'destination', 'dst', 'target', 'volume-subpath', 'image-subpath'
                       raise ArgumentError, message('no path value', k, hint: flag) unless v
 
-                      v = basepath v
+                      case k
+                      when 'source', 'src'
+                        if type == 'bind'
+                          v = basepath v
+                        elsif type == 'tmpfs'
+                          next
+                        end
+                      when 'destination', 'dst', 'target'
+                        unless v.start_with?('/')
+                          print_error(Logger::ERROR, "path is not absolute: #{v}", subject: from, hint: k)
+                          next
+                        end
+                      end
                       v = shell_quote(v, option: false, force: false) if q == ''
                     end
                     out << "#{k}=#{q}#{v}#{q}"
                   elsif !silent?
-                    log_message('unrecognized option', subject: from, hint: k)
+                    puts log_message('unrecognized option', subject: from, hint: k)
                   end
                 end
                 raise TypeError, message('none specified', hint: flag) unless type
 
-                cmd << "--mount type=#{type},#{args.join(',')}"
+                op << "--mount type=#{type},#{args.join(',')}"
               end
             end
             append_command(flag, id || tagmain, op.extras)
@@ -583,7 +639,7 @@ module Squared
             op.append(escape: true, strip: ':')
           when :commit
             latest = op.shift || tagmain
-            cmd << id << latest
+            op << id << latest
             raise ArgumentError, message('unrecognized args', op.join(', '), hint: flag) unless op.empty?
             return unless confirm_command(cmd.to_s, title: from, target: id, as: latest)
 
@@ -604,7 +660,7 @@ module Squared
           else
             if op.empty?
               ps, status, no = filter_ps(flag, from: from)
-              cmd << '--no-stream' if flag == :stats
+              op << '--no-stream' if flag == :stats
               list_image(flag, ps, no: no, hint: status, multiple: flag != :attach, from: from) do |img|
                 run(cmd.temp(img), from: from)
               end
@@ -633,11 +689,12 @@ module Squared
           op = OptionPartition.new(opts, list, cmd, **dockerinit)
           exception = exception?
           banner = true
+          status = false
           from = symjoin 'image', flag
           case flag
           when :ls
             if opts.size == op.size
-              index = 0
+              index = -1
               name = nil
               opts.reverse_each do |val|
                 next unless (arg = OptionPartition.parse_arg!('name', val))
@@ -648,8 +705,9 @@ module Squared
               end
               list_image(:run, filter: filter, from: from) do |val|
                 args = if name
+                         index += 1
                          opts.dup << "name=#{index == 0 ? name : "#{name}-#{index}"}"
-                       elsif choice!(opts, only: true)
+                       elsif choice!(opts, empty: true, only: true)
                          OptionPartition.strip(readline('Enter arguments', force: false))
                        else
                          opts
@@ -659,7 +717,6 @@ module Squared
                   break
                 end
                 container(:run, args, id: val)
-                index += 1
               end
               return
             end
@@ -679,16 +736,15 @@ module Squared
               banner = false
             end
           when :tag, :save
-            found = false
             list_image(flag, filter: filter, from: from) do |val|
               op << val
-              found = true
+              status = true
               if flag == :tag
                 op << tagname(project, op.first)
                 break
               end
             end
-            raise ArgumentError, message('target not specified', hint: flag) unless found
+            raise ArgumentError, message('target not specified', hint: flag) unless status
           when :pull
             if !id
               id = tagmain
@@ -699,13 +755,13 @@ module Squared
               registry = op.shift
               registry ||= option('registry') || self.registry unless id.include?('/')
             end
-            cmd << shell_quote(tagjoin(registry, id))
+            op << shell_quote(tagjoin(registry, id))
           when :push
             id ||= option('tag', ignore: false) || op.shift || tagmain
             registry ||= option('registry') || op.shift || self.registry
-            emptyargs op, flag
             raise ArgumentError, message('username/registry not specified', hint: flag) unless registry
 
+            op.clear(strict: true)
             uri = shell_quote tagjoin(registry, id)
             op << uri
             img = docker_output 'image tag', id, uri
@@ -725,16 +781,17 @@ module Squared
               op.append(escape: true, strip: ':')
             end
           end
-          success?(run(sync: sync, exception: exception, banner: banner, from: from), flag == :tag || flag == :save)
+          success?(run(sync: sync, exception: exception, banner: banner, from: from), status)
         end
 
-        def network(flag, opts = [], target: nil)
+        def network(flag, opts = [], name: nil)
           cmd, opts = docker_session('network', flag, opts: opts)
           op = OptionPartition.new(opts, OPT_DOCKER[:network].fetch(flag, []), cmd, **dockerinit)
                               .clear
           from = symjoin 'network', flag
-          if flag == :create
-            op.add_quote(target)
+          case flag
+          when :create, :rm
+            op.add_quote(name)
             run(from: from)
           else
             list_image(flag, docker_output('ps -a'), from: from) do |id|
@@ -752,13 +809,13 @@ module Squared
         end
 
         def compose?(file = dockerfile)
-          return file == 3 || file == 4 if file.is_a?(Numeric)
+          return file.between?(3, 4) if file.is_a?(Numeric)
 
           COMPOSEFILE.include?(File.basename(file))
         end
 
         def bake?(file = dockerfile)
-          return file == 1 || file == 2 if file.is_a?(Numeric)
+          return file.between?(1, 2) if file.is_a?(Numeric)
 
           BAKEFILE.include?(File.basename(file))
         end
@@ -770,11 +827,11 @@ module Squared
                       val.size > 1 ? val : val.first
                     elsif val == true
                       DIR_DOCKER.find { |file| exist?(file) }
-                    elsif val != 'Dockerfile'
+                    elsif val != DIR_DOCKER.last
                       val
                     end
           end
-          basepath((@file.is_a?(Array) ? @file.first : @file) || 'Dockerfile')
+          basepath((@file.is_a?(Array) ? @file.first : @file) || DIR_DOCKER.last)
         end
 
         private
@@ -790,7 +847,7 @@ module Squared
               yield data
             end
           rescue => e
-            log.debug e
+            print_error(e, subject: name, hint: 'compose')
           end
         end
 
@@ -873,7 +930,7 @@ module Squared
           target
         end
 
-        def filter_ps(flag, meth = nil, from: meth || :'container:ps')
+        def filter_ps(flag, column: nil, from: :'container:ps')
           no = false
           status = case flag.to_sym
                    when :start
@@ -899,6 +956,8 @@ module Squared
                    end
           cmd = docker_output("#{from.to_s.split(':').first} ps -a",
                               *status.map { |s| quote_option('filter', "status=#{s}") })
+          return IO.popen(cmd.temp("--format=\"{{.#{column}}}\"")).map(&:chomp) if column
+
           [cmd, status, no]
         end
 
@@ -907,15 +966,15 @@ module Squared
           pwd_set(from: from) do
             index = 1
             all = option('all', prefix: 'docker')
-            y = from.to_s.end_with?(':rm') && option('y', prefix: 'docker')
+            y = from == :'image:rm' && option('y', prefix: 'docker')
             filter = env('DOCKER_FILTER', filter).to_s
             pat = if OptionPartition.pattern?(filter)
                     Regexp.new(filter)
                   elsif filter.match?(/[:_-]$/)
                     /\b#{Regexp.escape(filter)}/
                   else
-                    filter = filter.empty? ? '([:_-]|$)' : "[:_-]#{filter}"
-                    /\b(#{dnsname(name)}|#{tagname(project)}|#{tagmain.split(':', 2).first})#{filter}/
+                    names = [dnsname(name), subdomainname(name), tagname(project), tagmain.split(':').first]
+                    /\b(#{names.join('|')})#{filter.empty? ? '([:_-]|$)' : "[:_-]#{filter}"}/
                   end
             IO.popen(cmd.temp('--format=json')).each do |line|
               data = JSON.parse(line)
@@ -989,18 +1048,22 @@ module Squared
           confirm "#{s} #{sub_style(target, theme[:subject])}#{" as #{sub_style(as, theme[:inline])}" if as}?", 'N'
         end
 
-        def choice_command(flag, *action)
+        def choice_command(flag, *action, id: nil, multiple: nil, series: false, force: true)
           msg, cmd = case flag
-                     when :exec
+                     when :exec, :container
                        ['Choose a container', 'ps -a']
                      when :bake
                        ['Choose a target', 'buildx bake --list=type=targets']
-                     when :connect, :disconnect
+                     when :connect, :disconnect, :'network:rm'
                        ['Choose a network', 'network ls']
                      when :service
                        ['Choose a service',
                         'compose ps -a ' \
                         '--format="table {{.Service}}\t{{.Name}}\t{{.Image}}\t{{.Command}}\t{{.Status}}\t{{.Ports}}"']
+                     when :volume
+                       ['Choose a volume',
+                        'volume ls ' \
+                        '--format="table {{.Name}}\t{{.Driver}}\t{{.Mountpoint}}"']
                      else
                        ['Choose an image',
                         'images -a ' \
@@ -1011,28 +1074,40 @@ module Squared
             puts log_message('none found', subject: name, hint: "docker #{cmd.split(' ', 3)[0, 2].join(' ')}")
             return
           end
-          puts " #  #{lines.shift}"
-          multiple = false
+          parse = lambda do |list|
+            ret = Array(list).map { |val| val.split(/\s+/, 2).first }.reject(&:empty?)
+            return ret.first if multiple == false
+
+            ret unless ret.empty?
+          end
+          puts "#{"\n" if id && !printfirst?} #  #{lines.shift}"
           ctx = flag.to_s
           case flag
           when :run, :exec
             values = [['Options', flag == :run], ['Arguments', flag == :exec]]
           when :rm, :bake
             values = ['Options']
-            multiple = true
+            multiple = true if multiple.nil?
             ctx = flag == :rm ? 'image rm' : "buildx bake -f #{shell_quote(dockerfile)}"
           when :save
             values = [['Output', true], 'Platform']
-            multiple = true
+            multiple = true if multiple.nil?
           when :service
-            values = []
-            multiple = true
+            values = ['Options']
+            multiple = true if multiple.nil?
             ctx = 'compose'
           when :connect, :disconnect
-            values = ['Options', ['Container', true]]
+            values = id ? ['Options'] : ['Options', ['Container', true]]
             ctx = "network #{flag}"
+          when :'network:rm'
+            values = ['Options']
+            multiple = true if multiple.nil?
+            ctx = 'network rm'
+          when :container, :image, :volume
+            return parse.call(choice_index(msg, lines, multiple: multiple, series: series, force: force,
+                                                       attempts: force ? 3 : 1))
           end
-          out, opts, args = choice_index(msg, lines, multiple: multiple, values: values)
+          out, opts, args = choice_index(msg, lines, values: values, multiple: multiple, series: series)
           cmd = docker_output(ctx, *action)
           case flag
           when :tag
@@ -1047,8 +1122,10 @@ module Squared
           else
             cmd << opts << '--'
           end
-          cmd.merge(Array(out).map { |val| val.split(/\s+/, 2).first })
-          success?(run(cmd << args, { 'NO_COLOR' => 'true' }), ctx.start_with?('network', 'tag', 'save'))
+          cmd.merge(parse.call(out))
+          cmd << args
+          cmd.merge(Array(id)) if id
+          success?(run(cmd, { 'NO_COLOR' => 'true' }), ctx.start_with?('network', 'tag', 'save'))
         end
 
         def filetype(val = dockerfile)
@@ -1079,7 +1156,7 @@ module Squared
           val = val.compact
                    .join(':')
                    .split(':')
-                   .map { |s| charname(s.sub(/^\W+/, '')) }
+                   .map { |s| subdomainname(s) }
           ret = val.join(':')
           return ret unless ret.size > max
 
@@ -1088,15 +1165,19 @@ module Squared
             s = val.join(':')
             return s unless s.size > max
           end
-          ret[0...max]
+          ret[...max]
         end
 
-        def dnsname(val)
-          charname(val[/^[^a-z\d]*(.*?)[^a-z\d]*$/i, 1].gsub(/-{2,}/, '-'))[0..62].downcase
+        def dnsname(val, max: 63, char: nil)
+          ret = val.sub(/^[^a-z]+/i, '')
+                   .sub(/[^a-z\d]+$/i, '')
+                   .gsub(/[^a-z\d#{char}]+/i, '-')
+                   .gsub(/-{2,}/, '-')[...max]
+          char ? ret : ret.downcase
         end
 
-        def charname(val)
-          val.gsub(/[^\w.-]+/, '_')
+        def subdomainname(val)
+          dnsname(val, max: 128, char: '.').gsub(/\.{2,}/, '.')
         end
 
         def tagmain
@@ -1104,11 +1185,7 @@ module Squared
         end
 
         def dockerinit
-          { project: self, strict: strict?, explicit: '=', bool: %w[true false 1 0] }
-        end
-
-        def emptyargs(list, hint = nil)
-          raise ArgumentError, message('unrecognized args', list.join(', '), hint: hint) unless list.empty?
+          { project: self, strict: strict?, explicit: '=', bool: %w[1 0 true false] }
         end
 
         def anypath?(*args)