squared 0.4.7 → 0.4.9

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

@@ -4,8 +4,6 @@ module Squared
   module Workspace
     module Project
       class Docker < Base
-        include Prompt
-
         COMPOSEFILE = %w[compose.yaml compose.yml docker-compose.yaml compose.yml docker-compose.yml].freeze
         BAKEFILE = %w[docker-bake.json docker-bake.hcl docker-bake.override.json docker-bake.override.hcl].freeze
         DIR_DOCKER = (COMPOSEFILE + BAKEFILE).freeze
@@ -37,15 +35,15 @@ module Squared
             run: %w[d|detach init i|interactive no-healthcheck oom-kill-disable privileged P|publish-all q|quiet
                     read-only rm runtime t|tty add-host=q annotation=q a|attach=b blkio-weight-device=i cap-add=b
                     cap-drop=b cgroup-parent=b cgroupns=b cidfile=p detach-keys=q device=q device-cgroup-rule=q
-                    device-read-bps=q device-read-iops=q device-write-bps=q device-write-iops=q
-                    disable-content-trust=b? dns=e dns-option=e dns-search=e domainname=b entrypoint=q e|env=qq
-                    env-file=p expose=e gpus=q group-add=b health-cmd=q health-interval=b health-retries=i
-                    health-start-interval=b health-start-period=b health-timeout=b h|hostname=e io-maxbandwidth=b
-                    io-maxiops=b ip=b ip6=e ipc=b isolation=b kernel-memory=b l|label=q label-file=p link=b
-                    link-local-ip=b log-driver=b log-opt=q mac-address=e memory-swappiness=b mount=qq name=b network=b
-                    network-alias=b oom-score-adj=b pid=b platform=b p|publish=e pull=b restart=b runtime=b
-                    security-opt=q shm-size=b sig-proxy=b? stop-signal=b stop-timeout=i storage-opt=q sysctl=q tmpfs=q
-                    ulimit=q user=e userns=b uts=b v|volume=q volume-driver=b volumes-from=b w|workdir=q].freeze,
+                    device-read-bps=q device-read-iops=q device-write-bps=q device-write-iops=q disable-content-trust=b?
+                    dns=e dns-option=e dns-search=e domainname=b entrypoint=q e|env=qq env-file=p expose=e gpus=q
+                    group-add=b health-cmd=q health-interval=b health-retries=i health-start-interval=b
+                    health-start-period=b health-timeout=b h|hostname=e io-maxbandwidth=b io-maxiops=b ip=b ip6=e ipc=b
+                    isolation=b kernel-memory=b l|label=q label-file=p link=b link-local-ip=b log-driver=b log-opt=q
+                    mac-address=e memory-swappiness=b mount=qq name=b network=b network-alias=b oom-score-adj=b pid=b
+                    platform=b p|publish=e pull=b restart=b runtime=b security-opt=q shm-size=b sig-proxy=b?
+                    stop-signal=b stop-timeout=i storage-opt=q sysctl=q tmpfs=q ulimit=q user=e userns=b uts=b
+                    v|volume=q volume-driver=b volumes-from=b w|workdir=q].freeze,
             exec: %w[d|detach i|interactive privileged t|tty detach-keys=q e|env=qq env-file=p user=e
                      w|workdir=q].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
@@ -125,7 +123,7 @@ module Squared
           return unless ref?(Docker.ref)
 
           namespace name do
-            @@tasks[Docker.ref].each do |action, flags|
+            Docker.subtasks do |action, flags|
               next if @pass.include?(action)
 
               namespace action do
@@ -151,31 +149,38 @@ module Squared
                     when :build, :up
                       format_desc action, flag, 'opts*,service*'
                       task flag do |_, args|
-                        composex flag, args.to_a
+                        compose! flag, args.to_a
                       end
                     when :exec, :run
                       format_desc action, flag, "service,command#{flag == :exec ? '' : '?'},args*,opts*"
                       task flag, [:service] do |_, args|
                         service = param_guard(action, flag, args: args, key: :service)
-                        composex(flag, args.extras, service: service)
+                        compose!(flag, args.extras, service: service)
                       end
                     end
                   when 'container'
                     case flag
                     when :exec, :commit
-                      format_desc(action, flag, 'id/name,opts*', after: flag == :exec ? 'args+' : 'tag?')
+                      format_desc(action, flag, flag == :exec ? 'id/name,opts*,args+' : 'id/name,tag?,opts*')
                       task flag, [:id] do |_, args|
-                        id = param_guard(action, flag, args: args, key: :id)
-                        container(flag, args.extras, id: id)
+                        if flag == :exec && !args.id
+                          choice_command flag
+                        else
+                          id = param_guard(action, flag, args: args, key: :id)
+                          container(flag, args.extras, id: id)
+                        end
                       end
                     when :run
                       format_desc action, flag, 'image,opts*,args*'
                       task flag, [:image] do |_, args|
-                        image = param_guard(action, flag, args: args, key: :image)
-                        container(flag, args.extras, id: image)
+                        if args.image
+                          container(flag, args.extras, id: args.image)
+                        else
+                          choice_command flag
+                        end
                       end
                     else
-                      format_desc(action, flag, 'opts*', after: "id/name#{flag == :update ? '+' : '*'}")
+                      format_desc(action, flag, 'opts*,id/name', after: flag == :update ? '+' : '*')
                       task flag do |_, args|
                         container flag, args.to_a
                       end
@@ -188,7 +193,7 @@ module Squared
                         tag = param_guard(action, flag, args: args, key: :tag)
                         image(flag, args.extras, id: tag)
                       end
-                    else
+                    when :list, :rm
                       format_desc(action, flag, flag == :rm ? 'id*,opts*' : 'opts*,args*')
                       task flag do |_, args|
                         image flag, args.to_a
@@ -197,8 +202,11 @@ module Squared
                   when 'network'
                     format_desc action, flag, 'target,opts*'
                     task flag, [:target] do |_, args|
-                      target = param_guard(action, flag, args: args, key: :target)
-                      network(flag, args.extras, target: target)
+                      if args.target
+                        network(flag, args.extras, target: args.target)
+                      else
+                        choice_command flag
+                      end
                     end
                   end
                 end
@@ -251,13 +259,13 @@ module Squared
           when :build
             case @secrets
             when String
-              quote_option('secret', @secrets, double: true)
+              ret << quote_option('secret', @secrets, double: true)
             when Hash
               append = lambda do |type|
                 as_a(@secrets[type]).each { |arg| ret << quote_option('secret', "type=#{type},#{arg}", double: true) }
               end
-              append.(:file)
-              append.(:env)
+              append.call(:file)
+              append.call(:env)
             else
               as_a(@secrets).each { |arg| ret << quote_option('secret', arg) }
             end
@@ -297,14 +305,14 @@ module Squared
                 end
               end
               op.append(args, escape: true)
-              contextdir(context) if context
+              contextdir context if context
             end
           end
           op.clear(pass: false)
           run(from: :"buildx:#{flag}")
         end
 
-        def composex(flag, opts = [], service: nil)
+        def compose!(flag, opts = [], service: nil)
           cmd, opts = docker_session('compose', opts: opts)
           op = OptionPartition.new(opts, OPT_DOCKER[:compose][:common], cmd, project: self)
           append_file filetype unless op.arg?('f', 'file')
@@ -351,7 +359,7 @@ module Squared
                     args << k
                     next
                   when 'source', 'src', 'destination', 'dst', 'target'
-                    v = basepath(v)
+                    v = path + v
                     v = shell_quote(v, option: false, force: false) if q == ''
                     tmpfs = false if k[0] == 's'
                   end
@@ -410,7 +418,7 @@ module Squared
               when :rm
                 status = %w[created exited dead]
               end
-              ps = docker_output('ps -a', *status.map { |s| "--filter='status=#{s}'" })
+              ps = docker_output('ps -a', *status.map { |s| "--filter=\"status=#{s}\"" })
               list_image(flag, ps, no: no, hint: "status: #{status.join(', ')}", from: from) do |img|
                 run(cmd.temp(img), from: from)
               end
@@ -449,8 +457,7 @@ module Squared
               op << id
             else
               if op.empty?
-                list_image(flag, docker_output('image', 'ls', '-a'), from: from) do |val|
-                  puts val
+                list_image(flag, docker_output('image ls -a'), from: from) do |val|
                   image(:rm, opts, sync: sync, id: val)
                 end
               else
@@ -480,7 +487,7 @@ module Squared
           op.clear
           from = :"network:#{flag}"
           list_image(flag, docker_output('ps -a'), from: from) do |img|
-            puts 'Success' if run(cmd.temp(target, img), from: from) == true && stdout? && banner?
+            print_success if success?(run(cmd.temp(target, img), from: from))
           end
         end
 
@@ -522,6 +529,9 @@ module Squared
         end
 
         def append_command(flag, val, list, target: @session, from: nil)
+          if (args = env('DOCKER_ARGS'))
+            list << args
+          end
           case flag
           when :run
             unless session_arg?('name', target: target)
@@ -532,13 +542,13 @@ module Squared
             raise_error('no command args', hint: from) if list.empty?
           end
           target << val << list.shift
-          target << shell_quote(list.join(' && '), double: true, option: false) unless list.empty?
+          target << list.join(' && ') unless list.empty?
         end
 
         def append_file(type, target: @session)
           return unless type == 2 || type == 4 || @file.is_a?(Array)
 
-          files = as_a(@file).map { |val| quote_option('file', basepath(val)) }
+          files = as_a(@file).map { |val| quote_option('file', path + val) }
           if target.is_a?(Set)
             target.merge(files)
           else
@@ -569,11 +579,11 @@ module Squared
           end
         end
 
-        def list_image(flag, cmd, hint: nil, from: nil, no: true, &blk)
+        def list_image(flag, cmd, hint: nil, from: nil, no: true)
           pwd_set do
             found = false
             index = 0
-            pat = /^(?:#{dnsname(name)}|#{tagname(project)}|#{tagmain.split(':').first})(?:[_.,:-]|$)/
+            pat = /^(?:#{dnsname(name)}|#{tagname(project)}|#{tagmain.split(':', 2).first})(?:[_.,:-]|$)/
             IO.popen(session_done(cmd << '--format=json')).each do |line|
               data = JSON.parse(line)
               id = data['ID']
@@ -586,7 +596,7 @@ module Squared
                                        id
                                      end)
               ee = data['Image'] || rt || aa
-              next unless ee.match?(pat) || aa.match?(pat)
+              next unless option('all') || ee.match?(pat) || aa.match?(pat)
 
               bb = index.succ.to_s
               cc = bb.size + 1
@@ -618,7 +628,7 @@ module Squared
               next unless confirm("#{h + b}? [#{c}] ", d, timeout: 60)
 
               puts if @@print_order == 0
-              blk.call id
+              yield id
             end
             puts log_message(Logger::INFO, 'none detected', subject: "#{name}:#{from}", hint: hint) unless found
           end
@@ -646,6 +656,35 @@ module Squared
           confirm("#{a} #{b}#{c ? " as #{c}" : ''}? [y/N] ", 'N', timeout: 60)
         end
 
+        def choice_command(flag)
+          msg, cmd, index = case flag
+                            when :run
+                              ['Choose an image', 'images -a', 2]
+                            when :exec
+                              ['Choose a container', 'ps -a', 0]
+                            else
+                              ['Choose a network', 'network ls', 0]
+                            end
+          lines = `#{docker_output(cmd)}`.lines
+          header = lines.shift
+          if lines.empty?
+            puts log_message(Logger::INFO, 'none found', subject: name, hint: "docker #{cmd}")
+          else
+            puts " #  #{header}"
+            case flag
+            when :run, :exec
+              values = [['Options', flag == :run], ['Arguments', flag == :exec]]
+              cmd = flag.to_s
+            else
+              values = [['Options', false], ['Container', true]]
+              cmd = "network #{flag}"
+            end
+            out, opts, args = choice_index(msg, lines, values: values)
+            ret = run docker_output(cmd, opts, '--', out.split(/\s+/)[index], args)
+            print_success if success?(ret && cmd.start_with?('network'))
+          end
+        end
+
         def filetype(val = dockerfile)
           case File.extname(val)
           when '.hcl', '.json'
@@ -662,7 +701,7 @@ module Squared
         end
 
         def contextdir(val = nil)
-          val && projectpath?(val) ? shell_quote(basepath(val)) : '.'
+          val && projectpath?(val) ? shell_quote(path + val) : '.'
         end
 
         def tagname(val)