squared 0.4.6 → 0.4.8

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

@@ -125,7 +125,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 +151,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*'
+                      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 +195,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 +204,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 (target = args.target)
+                        network(flag, args.extras, target: target)
+                      else
+                        choice_command flag
+                      end
                     end
                   end
                 end
@@ -251,13 +261,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
@@ -278,45 +288,44 @@ module Squared
 
         def buildx(flag, opts = [], tag: nil, context: nil)
           cmd, opts = docker_session('buildx', opts: opts)
-          opts = option_sanitize(opts, OPT_DOCKER[:buildx][:common]).first
-          cmd << flag
-          list = OPT_DOCKER[:buildx][flag == :bake ? :bake : :build] + OPT_DOCKER[:buildx][:shared]
-          out = option_sanitize(opts, list).first
+          op = OptionPartition.new(opts, OPT_DOCKER[:buildx][:common], cmd, project: self)
+          op << flag
+          op.parse(OPT_DOCKER[:buildx][flag == :bake ? :bake : :build] + OPT_DOCKER[:buildx][:shared])
           case flag
           when :build, :context
             append_tag(tag || option('tag', ignore: false) || @tag)
             append_context context
           when :bake
-            unless out.empty?
-              args = out.dup
-              out.clear
+            unless op.empty?
+              args = op.dup
+              op.extras.clear
               if Dir.exist?(args.last)
                 if projectpath?(val = args.pop)
                   context = val
                 else
-                  out << val
+                  op.extras << val
                 end
               end
-              append_value(args, escape: true)
+              op.append(args, escape: true)
               contextdir(context) if context
             end
           end
-          option_clear(out, pass: false)
+          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)
-          opts = option_sanitize(opts, OPT_DOCKER[:compose][:common]).first
-          append_file filetype unless session_arg?('f', 'file')
-          cmd << flag
-          out = option_sanitize(opts, OPT_DOCKER[:compose][flag]).first
+          op = OptionPartition.new(opts, OPT_DOCKER[:compose][:common], cmd, project: self)
+          append_file filetype unless op.arg?('f', 'file')
+          op << flag
+          op.parse(OPT_DOCKER[:compose][flag])
           from = :"compose:#{flag}"
           case flag
           when :build, :up
-            append_value(out, escape: true)
+            op.append(escape: true)
           when :exec, :run
-            append_command(flag, service, out, from: from)
+            append_command(flag, service, op.extras, from: from)
           end
           run(from: from)
         end
@@ -324,12 +333,12 @@ module Squared
         def container(flag, opts = [], id: nil)
           cmd, opts = docker_session('container', flag, opts: opts)
           list = OPT_DOCKER[:container].fetch(flag, [])
-          list += OPT_DOCKER[:container][:update] if flag == :run
-          out = option_sanitize(opts, list, args: flag == :run || flag == :exec).first
+          list.concat(OPT_DOCKER[:container][:update]) if flag == :run
+          op = OptionPartition.new(opts, list, cmd, project: self, args: flag == :run || flag == :exec)
           from = :"container:#{flag}"
           case flag
           when :run, :exec
-            if flag == :run && !session_arg?('mount')
+            if flag == :run && !op.arg?('mount')
               run = VAL_DOCKER[:run]
               both = run[:bind] + run[:tmpfs]
               diff = run[:bind].reject { |val| run[:tmpfs].include?(val) }
@@ -361,14 +370,14 @@ module Squared
                 cmd << "--mount type=#{tmpfs ? 'tmpfs' : 'bind'},#{args.join(',')}"
               end
             end
-            append_command(flag, id.to_s.empty? ? tagmain : id, out, from: from)
+            append_command(flag, id.to_s.empty? ? tagmain : id, op.extras, from: from)
           when :update
-            raise_error('missing container', hint: from) if out.empty?
-            append_value(out, escape: true)
+            raise_error('missing container', hint: from) if op.empty?
+            op.append(escape: true)
           when :commit
-            latest = out.shift || tagmain
+            latest = op.shift || tagmain
             cmd << id << latest
-            raise_error("unknown args: #{out.join(', ')}", hint: from) unless out.empty?
+            raise_error("unknown args: #{op.join(', ')}", hint: from) unless op.empty?
             return unless confirm_command(cmd.to_s, title: from, target: id, as: latest)
 
             registry = option('registry') || @registry
@@ -386,7 +395,7 @@ module Squared
             opts << '--quiet' unless verbose
             return image(:push, opts, id: latest, registry: registry)
           else
-            if out.empty?
+            if op.empty?
               status = []
               no = true
               case flag
@@ -411,13 +420,13 @@ 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
               return
             else
-              append_value(out, escape: true)
+              op.append(escape: true)
             end
           end
           run(from: from)
@@ -425,11 +434,11 @@ module Squared
 
         def image(flag, opts = [], sync: true, id: nil, registry: nil)
           cmd, opts = docker_session('image', flag, opts: opts)
-          out = option_sanitize(opts, OPT_DOCKER[:image][flag]).first
+          op = OptionPartition.new(opts, OPT_DOCKER[:image][flag], cmd, project: self)
           from = :"image:#{flag}"
           case flag
           when :list
-            if opts.size == out.size
+            if opts.size == op.size
               index = 0
               name = nil
               opts.reverse_each { |opt| break opts.delete(opt) if (name = opt[/^name=["']?(.+?)["']?$/, 1]) }
@@ -443,28 +452,29 @@ module Squared
               end
               return
             else
-              option_clear out
+              op.clear
             end
           when :rm
             if id
-              cmd << id
-            elsif !out.empty?
-              out.each { |val| run(cmd.temp(val), sync: sync, from: from) }
-              return
+              op << id
             else
-              list_image(flag, docker_output('image ls -a'), from: from) do |val|
-                image(:rm, opts, sync: sync, id: val)
+              if op.empty?
+                list_image(flag, docker_output('image ls -a'), from: from) do |val|
+                  image(:rm, opts, sync: sync, id: val)
+                end
+              else
+                op.each { |val| run(cmd.temp(val), sync: sync, from: from) }
               end
               return
             end
           when :push
             id ||= option('tag', ignore: false) || tagmain
-            registry ||= out.shift || option('registry') || @registry
-            raise_error(id ? "unknown args: #{out.join(', ')}" : 'no id/tag given', hint: from) unless id && out.empty?
+            registry ||= op.shift || option('registry') || @registry
+            raise_error(id ? "unknown args: #{op.join(', ')}" : 'no id/tag given', hint: from) unless id && op.empty?
             raise_error('username/registry not provided', hint: from) unless registry
             registry.chomp!('/')
             uri = shell_quote("#{registry}/#{id}")
-            cmd << uri
+            op << uri
             img = docker_output 'image', 'tag', id, uri
             return unless confirm_command(img.to_s, cmd.to_s, target: id, as: registry, title: from)
 
@@ -475,7 +485,8 @@ module Squared
 
         def network(flag, opts = [], target: nil)
           cmd, opts = docker_session('network', flag, opts: opts)
-          option_sanitize(opts, OPT_DOCKER[:network][flag]).first
+          op = OptionPartition.new(opts, OPT_DOCKER[:network][flag], cmd, project: self)
+          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?
@@ -510,10 +521,9 @@ module Squared
         def docker_session(*cmd, opts: nil)
           return session('docker', *cmd) unless opts
 
-          out = []
-          opts = option_sanitize(opts, OPT_DOCKER[:common], target: out).first
-          ret = session('docker', *out, *cmd)
-          [ret, opts]
+          op = OptionPartition.new(opts, OPT_DOCKER[:common], project: self)
+          ret = session('docker', *op.to_a, *cmd)
+          [ret, op.extras]
         end
 
         def docker_output(*cmd, **kwargs)
@@ -521,9 +531,12 @@ 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')
+            unless session_arg?('name', target: target)
               require 'random/formatter'
               target << basic_option('name', dnsname("#{name}_#{Random.new.alphanumeric(6)}"))
             end
@@ -531,24 +544,29 @@ 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)
 
-          target.merge(as_a(@file).map { |val| quote_option('file', basepath(val)) })
+          files = as_a(@file).map { |val| quote_option('file', basepath(val)) }
+          if target.is_a?(Set)
+            target.merge(files)
+          else
+            target.concat(files)
+          end
         end
 
         def append_context(ctx = nil, target: @session)
-          if @file.is_a?(String) && !session_arg?('f', 'file') && !bake?(dockerfile) && !compose?(dockerfile)
+          if @file.is_a?(String) && !session_arg?('f', 'file', target: target) && !bake? && !compose?
             target << quote_option('file', dockerfile)
           end
           target << contextdir(ctx || context)
         end
 
         def append_tag(val, target: @session)
-          ver = option('version', ignore: false)
+          ver = option('version', target: target, ignore: false)
           list = case val
                  when String
                    val.split(',')
@@ -563,7 +581,7 @@ 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
@@ -580,14 +598,14 @@ 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
               a = sub_style(ee, styles: theme[:inline])
               b = "Execute #{sub_style(flag, styles: theme[:active])} on #{a}#{ee == id ? '' : " (#{id})"}"
               c, d = no ? ['y/N', 'N'] : ['Y/n', 'Y']
-              e = time_format(time_offset(data['CreatedAt']), pass: ['ms'])
+              e = time_format(time_since(data['CreatedAt']), pass: ['ms'])
               f = sub_style(ARG[:BORDER][0], styles: theme[:inline])
               g = ' ' * (cc + 1)
               h = "#{sub_style(bb.rjust(cc), styles: theme[:current])} #{f} "
@@ -612,7 +630,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
@@ -640,6 +658,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], ['Command 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)
+            puts 'Success' if ret && cmd.start_with?('network')
+          end
+        end
+
         def filetype(val = dockerfile)
           case File.extname(val)
           when '.hcl', '.json'
@@ -678,11 +725,11 @@ module Squared
           tag.is_a?(Array) ? tag.first : tag
         end
 
-        def compose?(file)
+        def compose?(file = dockerfile)
           COMPOSEFILE.include?(File.basename(file))
         end
 
-        def bake?(file)
+        def bake?(file = dockerfile)
           BAKEFILE.include?(File.basename(file))
         end
       end