squared 0.4.18 → 0.5.0

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

@@ -21,8 +21,7 @@ module Squared
           compose: {
             common: %w[all-resources compatibility dry-run ansi|b env-file=p f|file=p parallel=b profile=b progress=b
                        project-directory=p p|project-name=e].freeze,
-            build: %w[check no-cache pull push with-dependencies q|quiet build-arg=qq builder=b m|memory=b
-                      ssh=qq].freeze,
+            build: %w[no-cache pull push with-dependencies q|quiet build-arg=qq builder=b m|memory=b ssh=qq].freeze,
             exec: %w[dry-run privileged d|detach e|env=qq index=i T|no-TTY=b? user=e w|workdir=q].freeze,
             run: %w[build dry-run no-deps quiet-pull remove-orphans rm P|service-ports use-aliases cap-add=b cap-drop=b
                     d|detach entrypoint=q e|env=qq i|interactive=b? l|label=q name=b T|no-TTY=b? p|publish=e pull=b
@@ -55,15 +54,15 @@ module Squared
             commit: %w[a|author=q c|change=q m|message=q pause=b?].freeze,
             inspect: %w[s|size f|format=q].freeze,
             start: %w[a|attach i|interactive detach-keys=q].freeze,
-            stop: %w[s|signal=b t|time=i t|timeout=i].freeze,
-            restart: %w[s|signal=b t|time=i t|timeout=i].freeze,
+            stop: %w[s|signal=b t|time=i].freeze,
+            restart: %w[s|signal=b t|time=i].freeze,
             kill: %w[s|signal=b].freeze,
             stats: %w[no-trunc format|q].freeze
           }.freeze,
           image: {
             list: %w[a|all digests no-trunc f|filter=q format=q].freeze,
             push: %w[a|all-tags disable-content-trust=b? platform=b q|quiet].freeze,
-            rm: %w[f|force no-prune platform=b].freeze,
+            rm: %w[f|force no-prune].freeze,
             save: %w[o|output=p platform=b].freeze
           }.freeze,
           network: {
@@ -73,11 +72,8 @@ module Squared
         }.freeze
         VAL_DOCKER = {
           run: {
-            common: %w[source src destination dst target readonly ro].freeze,
-            bind: %w[bind-propagation].freeze,
-            volume: %w[volume-subpath volume-nocopy volume-opt].freeze,
-            tmpfs: %w[tmpfs-size tmpfs-mode].freeze,
-            image: %w[image-path].freeze
+            bind: %w[type source src destination dst target readonly ro bind-propagation].freeze,
+            tmpfs: %w[type destination dst target tmpfs-size tmpfs-mode].freeze
           }.freeze
         }.freeze
         private_constant :COMPOSEFILE, :BAKEFILE, :OPT_DOCKER, :VAL_DOCKER
@@ -95,9 +91,8 @@ module Squared
         end
 
         subtasks({
-          'build' => %i[tag context].freeze,
+          'build' => %i[tag context bake].freeze,
           'compose' => %i[build run exec up].freeze,
-          'bake' => %i[build check].freeze,
           'image' => %i[list rm push tag save].freeze,
           'container' => %i[run create exec update commit inspect diff start stop restart pause unpause top stats kill
                             rm].freeze,
@@ -116,6 +111,7 @@ module Squared
           @mounts = mounts
           @secrets = secrets
           @registry = tagjoin registry, kwargs[:username]
+          @file = nil
           initialize_ref Docker.ref
           initialize_logger(**kwargs)
           initialize_env(**kwargs)
@@ -128,11 +124,11 @@ module Squared
 
         def populate(*, **)
           super
-          return unless ref?(Docker.ref) || @only
+          return unless ref?(Docker.ref)
 
           namespace name do
             Docker.subtasks do |action, flags|
-              next if task_pass?(action)
+              next if @pass.include?(action)
 
               namespace action do
                 flags.each do |flag|
@@ -145,31 +141,18 @@ module Squared
                         param = param_guard(action, flag, args: args, key: flag)
                         buildx(:build, args.extras, "#{flag}": param)
                       end
-                    end
-                  when 'bake'
-                    break unless bake?
-
-                    case flag
-                    when :build
+                    when :bake
                       format_desc action, flag, ':?,opts*,target*,context?'
                       task flag do |_, args|
                         args = args.to_a
                         if args.first == ':'
                           choice_command :bake
                         else
-                          buildx :bake, args
+                          buildx flag, args
                         end
                       end
-                    when :check
-                      format_desc action, flag, 'target'
-                      task flag, [:target] do |_, args|
-                        target = param_guard(action, flag, args: args, key: :target)
-                        buildx :bake, ['allow=fs.read=*', 'call=check', target]
-                      end
                     end
                   when 'compose'
-                    break unless compose?
-
                     case flag
                     when :build, :up
                       format_desc action, flag, 'opts*,service*'
@@ -177,7 +160,7 @@ module Squared
                         compose! flag, args.to_a
                       end
                     when :exec, :run
-                      format_desc action, flag, "service,command#{flag == :exec ? '' : '?'}|:,args*,opts*"
+                      format_desc action, flag, "service,command#{flag == :exec ? '' : '?'},args*,opts*"
                       task flag, [:service] do |_, args|
                         service = param_guard(action, flag, args: args, key: :service)
                         compose!(flag, args.extras, service: service)
@@ -215,8 +198,8 @@ module Squared
                     when :push
                       format_desc action, flag, 'tag,registry/username?,opts*'
                       task flag, [:tag] do |_, args|
-                        tag = param_guard(action, flag, args: args, key: :tag)
-                        image(flag, args.extras, id: tag)
+                        id = param_guard(action, flag, args: args, key: :tag)
+                        image(flag, args.extras, id: id)
                       end
                     else
                       format_desc(action, flag, case flag
@@ -282,11 +265,10 @@ module Squared
           when Enumerable
             ret.merge(opts.to_a)
           end
+
           [args, flags].each_with_index do |target, index|
-            if target.is_a?(String)
-              ret << target
-            elsif (target = append_any(target, target: []))
-              ret.merge(target.map { |arg| index == 0 ? fill_option(arg) : quote_option('build-arg', arg) })
+            if (data = append_any(target, target: []))
+              ret.merge(data.map { |arg| index == 0 ? fill_option(arg) : quote_option('build-arg', arg) })
             end
           end
           case from
@@ -296,12 +278,12 @@ module Squared
               ret << quote_option('secret', @secrets, double: true)
             when Hash
               append = lambda do |type|
-                Array(@secrets[type]).each { |arg| ret << quote_option('secret', "type=#{type},#{arg}", double: true) }
+                as_a(@secrets[type]).each { |arg| ret << quote_option('secret', "type=#{type},#{arg}", double: true) }
               end
               append.call(:file)
               append.call(:env)
             else
-              Array(@secrets).each { |arg| ret << quote_option('secret', arg) }
+              as_a(@secrets).each { |arg| ret << quote_option('secret', arg) }
             end
             if (val = option('tag', ignore: false))
               append_tag val
@@ -310,8 +292,8 @@ module Squared
             end
             append_context
           when :bake, :compose
-            if (val = option(from == :bake ? 'target' : 'service', ignore: false))
-              ret.merge(split_escape(val).map! { |s| shell_escape(s) })
+            option(from == :bake ? 'target' : 'service', ignore: false) do |a|
+              ret.merge(split_escape(a).map! { |b| shell_escape(b) })
             end
           end
           ret
@@ -324,17 +306,17 @@ module Squared
           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) || self.tag)
+            append_tag(tag || option('tag', ignore: false) || @tag)
             append_context context
           when :bake
             unless op.empty?
               args = op.dup
-              op.reset
+              op.extras.clear
               if Dir.exist?(args.last)
                 if projectpath?(val = args.pop)
                   context = val
                 else
-                  op.push(val)
+                  op.extras << val
                 end
               end
               op.append(args, escape: true)
@@ -356,7 +338,7 @@ module Squared
           when :build, :up
             op.append(escape: true)
           when :exec, :run
-            append_command flag, service, op.extras
+            append_command(flag, service, op.extras, from: from)
           end
           run(from: from)
         end
@@ -373,54 +355,44 @@ module Squared
           when :run, :create, :exec
             if rc && !op.arg?('mount')
               run = VAL_DOCKER[:run]
-              all = collect_hash VAL_DOCKER[:run]
-              delim = Regexp.new(",\\s*(?=#{all.join('|')})")
-              Array(@mounts).each do |val|
+              both = run[:bind] + run[:tmpfs]
+              diff = run[:bind].reject { |val| run[:tmpfs].include?(val) }
+              delim = Regexp.new(",\\s*(?=#{both.join('|')})")
+              as_a(@mounts).each do |val|
                 args = []
-                type = nil
+                tmpfs = true
                 val.split(delim).each do |opt|
                   k, v, q = split_option opt
+                  next unless both.include?(k)
+
                   if k == 'type'
-                    case v
-                    when 'bind', 'volume', 'image', 'tmpfs'
-                      type = v
-                    else
-                      raise_error("unknown type: #{v}", hint: flag)
-                    end
-                  elsif all.include?(k)
-                    unless type
-                      run.each_pair do |key, val|
-                        if val.include?(k)
-                          type = key.to_s unless key == :common
-                          break
-                        end
-                      end
-                    end
-                    case k
-                    when 'readonly', 'ro'
-                      args << k
-                      next
-                    when 'source', 'src', 'destination', 'dst', 'target', 'volume-subpath', 'image-path'
-                      v = path + v
-                      v = shell_quote(v, option: false, force: false) if q == ''
-                    end
-                    args << "#{k}=#{q + v + q}"
-                  elsif verbose
-                    log_message(Logger::INFO, 'unrecognized option', subject: from, hint: k)
+                    tmpfs = false if v == 'bind'
+                    next
+                  elsif diff.include?(k)
+                    tmpfs = false
+                  end
+                  case k
+                  when 'readonly', 'ro'
+                    args << k
+                    next
+                  when 'source', 'src', 'destination', 'dst', 'target'
+                    v = path + v
+                    v = shell_quote(v, option: false, force: false) if q == ''
+                    tmpfs = false if k[0] == 's'
                   end
+                  args << "#{k}=#{q + v + q}"
                 end
-                raise_error('missing type', hint: flag) unless type
-                cmd << "--mount type=#{type},#{args.join(',')}"
+                cmd << "--mount type=#{tmpfs ? 'tmpfs' : 'bind'},#{args.join(',')}"
               end
             end
-            append_command(flag, id || tagmain, op.extras)
+            append_command(flag, id || tagmain, op.extras, from: from)
           when :update
-            raise_error('missing container', hint: flag) if op.empty?
+            raise_error('missing container', hint: from) if op.empty?
             op.append(escape: true)
           when :commit
             latest = op.shift || tagmain
             cmd << id << latest
-            raise_error("unknown args: #{op.join(', ')}", hint: flag) unless op.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
@@ -463,7 +435,7 @@ module Squared
               when :rm
                 status = %w[created exited dead]
               end
-              ps = docker_output('ps -a', *status.map { |s| quote_option('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
@@ -527,10 +499,10 @@ module Squared
           when :push
             id ||= option('tag', ignore: false) || tagmain
             registry ||= op.shift || option('registry') || @registry
-            raise_error(id ? "unknown args: #{op.join(', ')}" : 'no id/tag given', hint: flag) unless id && op.empty?
-            raise_error('username/registry not provided', hint: flag) unless 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}")
+            uri = shell_quote "#{registry}/#{id}"
             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)
@@ -562,14 +534,6 @@ module Squared
           super || dockerfile.exist?
         end
 
-        def compose?(file = dockerfile)
-          COMPOSEFILE.include?(File.basename(file))
-        end
-
-        def bake?(file = dockerfile)
-          BAKEFILE.include?(File.basename(file))
-        end
-
         def dockerfile(val = nil)
           if val == 'Dockerfile'
             @file = false
@@ -599,19 +563,22 @@ module Squared
           session('docker', *cmd, main: false, options: false, **kwargs)
         end
 
-        def append_command(flag, val, list, target: @session)
-          if list.delete(':')
-            list << readline('Enter command [args]', force: true)
-          elsif (args = env('DOCKER_ARGS'))
+        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)
-              target << basic_option('name', dnsname("#{name}_%s" % rand_s(6)))
+              target << basic_option('name', dnsname("#{name}_%s" % if RUBY_VERSION >= '3.1'
+                                                                      require 'random/formatter'
+                                                                      Random.new.alphanumeric(6)
+                                                                    else
+                                                                      (0...6).map { rand(97..122).chr }.join
+                                                                    end))
             end
           when :exec
-            raise_error('no command args', hint: flag) if list.empty?
+            raise_error('no command args', hint: from) if list.empty?
           end
           target << val << list.shift
           target << list.join(' && ') unless list.empty?
@@ -620,7 +587,7 @@ module Squared
         def append_file(type, target: @session)
           return unless type == 2 || type == 4 || @file.is_a?(Array)
 
-          files = Array(@file).map { |val| quote_option('file', path + val) }
+          files = as_a(@file).map { |val| quote_option('file', path + val) }
           if target.is_a?(Set)
             target.merge(files)
           else
@@ -636,18 +603,20 @@ module Squared
         end
 
         def append_tag(val, target: @session)
-          ver = option('version', target: target, ignore: false)
-          list = case val
-                 when String
-                   val.split(',')
-                 when Array
-                   val
-                 else
-                   []
-                 end
-          list.each do |s|
-            s = "#{s.sub(/:latest$/, '')}:#{ver}" if ver && (!s.include?(':') || s.end_with?(':latest'))
-            target << basic_option('tag', tagname(s))
+          case val
+          when String
+            val.split(',')
+          when Array
+            val
+          else
+            []
+          end.yield_self do |list|
+            ver = option('version', target: target, ignore: false)
+            list.each do |s|
+              s = "#{s}:#{ver}" if ver && (!s.include?(':') || s.delete_suffix!(':latest'))
+              target << basic_option('tag', tagname(s))
+            end
+            target
           end
         end
 
@@ -657,7 +626,7 @@ module Squared
             index = 0
             all = option('all', prefix: 'docker')
             y = from == :'image:rm' && option('y', prefix: 'docker')
-            pat = /\b(?:#{dnsname(name)}|#{tagname(project)}|#{tagmain.split(':', 2).first})\b/
+            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']
@@ -694,7 +663,7 @@ module Squared
                 cols.each do |key|
                   next if (key == 'Tag' && !dd) || (key == 'Size' && data[key] == '0B')
 
-                  puts "#{g + f} #{key}: #{Array(data[key]).join(', ')}" unless data[key].to_s.empty?
+                  puts "#{g + f} #{key}: #{as_a(data[key]).join(', ')}" unless data[key].to_s.empty?
                 end
                 w = 9 + flag.to_s.size + 4 + ee.size
                 puts g + sub_style(ARG[:BORDER][6] + (ARG[:BORDER][1] * w), styles: theme[:inline])
@@ -706,10 +675,14 @@ module Squared
               end
               yield id
             end
-            puts log_message(Logger::INFO, 'none detected', subject: name, hint: hint || from) if !found && !y
+            puts log_message(Logger::INFO, 'none detected', subject: "#{name}:#{from}", hint: hint) if found || y
           end
         rescue StandardError => e
-          on_error e, from
+          log.error e
+          ret = on(:error, from, e)
+          raise if exception && ret != true
+
+          warn log_message(Logger::WARN, e, pass: true) if warning?
         end
 
         def confirm_command(*args, title: nil, target: nil, as: nil)
@@ -766,7 +739,7 @@ module Squared
             cmd = docker_output ctx
             case flag
             when :tag
-              args = tagjoin @registry, tag
+              args = tagjoin @registry, @tag
             when :save
               opts = "#{opts}.tar" unless opts.end_with?('.tar')
               cmd << quote_option('output', File.expand_path(opts))
@@ -777,9 +750,13 @@ module Squared
             else
               cmd << opts << '--'
             end
-            cmd.merge(Array(out).map! { |val| parse.call(val) })
+            cmd.merge(if out.is_a?(Array)
+                        out.map! { |val| parse.call(val) }
+                      else
+                        [parse.call(out)]
+                      end)
             cmd << args
-            print_success if success?(run(cmd)) && ctx.match?(/\A(?:network|tag|save)/)
+            print_success if success?(run(cmd)) && ctx.start_with?(/(?:network|tag|save)/)
           end
         end
 
@@ -809,9 +786,10 @@ module Squared
 
         def tagname(val)
           val = val.split(':').map! { |s| charname(s.sub(/^\W+/, '')) }
-          ret = val.join(':')
-          ret = val.first if val.size > 1 && ret.size > 128
-          ret[0..127]
+          val.join(':').yield_self do |s|
+            s = val.first if val.size > 1 && s.size > 128
+            s[0..127]
+          end
         end
 
         def dnsname(val)
@@ -825,6 +803,14 @@ module Squared
         def tagmain
           tag.is_a?(Array) ? tag.first : tag
         end
+
+        def compose?(file = dockerfile)
+          COMPOSEFILE.include?(File.basename(file))
+        end
+
+        def bake?(file = dockerfile)
+          BAKEFILE.include?(File.basename(file))
+        end
       end
 
       Application.implement Docker