squared 0.6.9 → 0.7.0

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

@@ -33,7 +33,8 @@ module Squared
                    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
                    exit-code-from=b no-attach=b pull=b scale=i t|timeout=i wait-timeout=i].freeze,
-            down: %w[remove-orphans v|volumes rmi=b t|timeout=i].freeze
+            down: %w[remove-orphans v|volumes rmi=b t|timeout=i].freeze,
+            publish: %w[app resolve-image-digests with-env y|yes oci-version=b].freeze
           }.freeze,
           container: {
             create: %w[init i|interactive no-healthcheck oom-kill-disable privileged P|publish-all q|quiet read-only
@@ -64,6 +65,7 @@ module Squared
           }.freeze,
           image: {
             ls: %w[a|all digests no-trunc q|quiet tree f|filter=q format=q].freeze,
+            pull: %w[a|all-tags platform=q q|quiet].freeze,
             push: %w[a|all-tags platform=q q|quiet].freeze,
             rm: %w[f|force no-prune platform=q].freeze,
             save: %w[o|output=p platform=q].freeze
@@ -107,9 +109,9 @@ module Squared
 
         subtasks({
           'build' => %i[tag context].freeze,
-          'compose' => %i[build create run exec up down service].freeze,
-          'bake' => %i[build check].freeze,
-          'image' => %i[ls rm push tag save].freeze,
+          'compose' => %i[build create publish run exec up down service].freeze,
+          'bake' => %i[build compose check].freeze,
+          'image' => %i[ls rm pull push tag save].freeze,
           'container' => %i[run create exec update commit inspect diff start stop restart pause unpause top stats kill
                             rm].freeze,
           'network' => %i[connect disconnect].freeze,
@@ -119,17 +121,19 @@ module Squared
         attr_reader :context
         attr_accessor :tag
 
-        def initialize(*, file: nil, context: nil, tag: nil, secrets: nil, mounts: [], registry: nil, **kwargs)
+        def initialize(*, mounts: [], **kwargs)
           super
-          return unless dockerfile(file).exist?
+          self.global = nil
+          return unless dockerfile(kwargs[:file]).exist?
 
-          @context = context
-          self.tag = tag || tagname("#{@project}:#{@version || 'latest'}")
+          self.tag = kwargs[:tag] || tagname("#{@project}:#{@version || 'latest'}")
+          @context = kwargs[:context]
           @mounts = mounts
-          @secrets = secrets
-          @registry = tagjoin registry, kwargs[:username]
+          @secrets = kwargs[:secrets]
+          @registry = tagjoin kwargs[:registry], kwargs[:username]
+          @oci = tagjoin kwargs[:oci], kwargs[:username]
           initialize_ref Docker.ref
-          initialize_logger(**kwargs)
+          initialize_logger kwargs[:log]
           initialize_env(**kwargs)
           @output[4] = merge_opts(kwargs[:args], @output[4]) if kwargs[:args]
         end
@@ -163,25 +167,21 @@ module Squared
                                                  end
                     cmd << '-a' if has_value!(args, 'a', 'all') && command != 'network'
                     data = VAL_DOCKER[:ls][command.to_sym]
-                    cols = if has_value!(args, 's', 'standard')
-                             data.first(data.index('CreatedAt'))
-                           else
-                             [].tap do |out|
-                               args.each do |val|
-                                 if val =~ /^(\d+)$/
-                                   out << data[$1.to_i.pred]
-                                 elsif val =~ /^(\d+)(-|\.{2,3})(\d+)$/
-                                   j = $1.to_i.pred
-                                   k = $3.to_i - ($2 == '..' ? 2 : 1)
-                                   out.concat(data[j..k]) if k > j
-                                 end
-                               end
-                               next unless out.empty?
-
-                               out.replace(choice_index('Select a column', data, multiple: true, attempts: 1))
-                             end
-                           end
-                    cmd << quote_option('format', "table #{cols.map! { |val| "{{.#{val}}}" }.join("\t")}")
+                    if has_value!(args, 's', 'standard')
+                      cols = data.first(data.index('CreatedAt'))
+                    else
+                      cols = args.each_with_object([]) do |val, out|
+                        if val =~ /^(\d+)$/
+                          out << data[$1.to_i.pred]
+                        elsif val =~ /^(\d+)(-|\.{2,3})(\d+)$/
+                          j = $1.to_i.pred
+                          k = $3.to_i - ($2 == '..' ? 2 : 1)
+                          out.concat(data[j..k]) if k > j
+                        end
+                      end
+                      cols = choice_index('Select a column', data, multiple: true, attempts: 1) if cols.empty?
+                    end
+                    cmd << quote_option('format', "table #{cols.map { |val| "{{.#{val}}}" }.join("\t")}")
                     run(cmd, banner: false, from: :ls)
                   end
                 end
@@ -199,14 +199,14 @@ module Squared
                       break unless bake?
 
                       case flag
-                      when :build
-                        format_desc action, flag, 'opts*,target*,context|:'
+                      when :build, :compose
+                        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(:bake, args, from: (:'buildx:bake' if flag == :compose))
                           end
                         end
                       when :check
@@ -239,6 +239,13 @@ module Squared
                             compose!(flag, [command], service: service.empty? || service)
                           end
                         end
+                      when :publish
+                        next unless @oci
+
+                        format_desc action, flag, 'tag?,repository?,opts*'
+                        task flag, [:tag] do |_, args|
+                          compose! flag, args.to_a
+                        end
                       else
                         format_desc action, flag, 'opts*,service*|:'
                         task flag do |_, args|
@@ -275,6 +282,13 @@ module Squared
                     when 'image'
                       case flag
                       when :push
+                        next unless @registry
+
+                        format_desc action, flag, 'tag?,registry/username?,opts*'
+                        task flag, [:tag] do |_, args|
+                          image flag, args.to_a
+                        end
+                      when :pull
                         format_desc action, flag, 'tag,registry/username?,opts*'
                         task flag, [:tag] do |_, args|
                           id = param_guard(action, flag, args: args, key: :tag)
@@ -349,7 +363,7 @@ module Squared
           [args, flags].each_with_index do |item, i|
             next unless item && (data = append_any(item, target: []))
 
-            ret.merge(data.map! { |arg| i == 0 ? fill_option(arg) : quote_option('build-arg', arg) })
+            ret.merge(data.map { |arg| i == 0 ? fill_option(arg) : quote_option('build-arg', arg) })
           end
           case from
           when :run
@@ -357,13 +371,13 @@ module Squared
             when String
               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) }
+              append = lambda do |key|
+                ret.merge(Array(@secrets[key]).map { |arg| quote_option('secret', "type=#{key},#{arg}", double: true) })
               end
               append.call(:file)
               append.call(:env)
             else
-              Array(@secrets).each { |arg| ret << quote_option('secret', arg) }
+              ret.merge(Array(@secrets).map { |arg| quote_option('secret', arg) })
             end
             if (val = option('tag', ignore: false))
               append_tag val
@@ -377,18 +391,18 @@ module Squared
           ret
         end
 
-        def buildx(flag, opts = [], tag: nil, context: nil)
+        def buildx(flag, opts = [], tag: nil, context: nil, from: nil)
           cmd, opts = docker_session('buildx', opts: opts)
-          op = OPT_DOCKER[:buildx].yield_self do |data|
-            OptionPartition.new(opts, data[:common], cmd, project: self)
-                           .append(flag, quote: false)
-                           .parse(data[flag == :bake ? :bake : :build] + data[:shared])
-          end
+          data = OPT_DOCKER[:buildx]
+          op = OptionPartition.new(opts, data[:common], cmd, project: self)
+          op.append(flag, quote: false)
+            .parse(data[flag == :bake ? :bake : :build] + data[:shared])
           case flag
           when :build, :context
             append_tag(tag || option('tag', ignore: false) || self.tag)
             append_context context
           when :bake
+            append_file(0, index: 3) unless from || op.arg?('f', 'file') || !anypath?(*COMPOSEFILE)
             unless op.empty?
               args = op.dup
               op.reset
@@ -404,11 +418,11 @@ module Squared
             end
           end
           op.clear(pass: false)
-          run(from: :"buildx:#{flag}")
+          run(from: from || symjoin('buildx', flag))
         end
 
-        def compose!(flag, opts = [], service: nil, multiple: false)
-          from = :"compose:#{flag}"
+        def compose!(flag, opts = [], id: nil, service: nil, multiple: false)
+          from = symjoin 'compose', flag
           if flag == :service
             command = opts.first
             if service == true
@@ -422,23 +436,31 @@ module Squared
           else
             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')
+            append_file(filetype, force: flag == :publish) unless op.arg?('f', 'file')
             op << flag
             op.parse(OPT_DOCKER[:compose].fetch(flag, []))
-            if op.remove(':') || service == ':'
-              keys = Set.new
-              read_composefile('services', target: op.values_of('f', 'file')) { |data| keys.merge(data.keys) }
-              service = unless keys.empty?
-                          choice_index('Add services', keys, multiple: multiple, force: !multiple,
-                                                             attempts: multiple ? 1 : 3)
-                        end
-            end
-            if multiple
-              op.concat(service) if service
-              op.append(delim: true, escape: true, strip: /^:/)
+            if flag == :publish
+              id ||= option('tag', ignore: false) || op.shift || tagmain
+              registry ||= option('registry') || op.shift || @oci
+              emptyargs op, flag
+              op << shell_quote(tagjoin(registry, id))
+              return unless confirm_command(op.to_s, title: from, target: id)
             else
-              raise_error ArgumentError, 'no service was selected', hint: flag unless service
-              append_command(flag, service, op.extras, prompt: '::')
+              if op.remove(':') || service == ':'
+                keys = Set.new
+                read_composefile('services', target: op.values_of('f', 'file')) { |data| keys.merge(data.keys) }
+                service = unless keys.empty?
+                            choice_index('Add services', keys, multiple: multiple, force: !multiple,
+                                                               attempts: multiple ? 1 : 3)
+                          end
+              end
+              if multiple
+                op.concat(service) if service
+                op.append(delim: true, escape: true, strip: /^:/)
+              else
+                raise_error ArgumentError, 'no service was selected', hint: flag unless service
+                append_command(flag, service, op.extras, prompt: '::')
+              end
             end
           end
           run(from: from)
@@ -446,35 +468,32 @@ module Squared
 
         def container(flag, opts = [], id: nil)
           cmd, opts = docker_session('container', flag, opts: opts)
-          rc = flag == :run || flag == :create
-          op = OPT_DOCKER[:container].yield_self do |data|
-            list = data.fetch(flag, [])
-            list += data[:create] if flag == :run
-            list += data[:update] if rc
-            OptionPartition.new(opts, list, cmd, project: self, args: rc || flag == :exec)
-          end
-          from = :"container:#{flag}"
+          data = OPT_DOCKER[:container]
+          list = data.fetch(flag, [])
+          list += data[:create] if (rc = flag == :run)
+          list += data[:update] if rc ||= flag == :create
+          op = OptionPartition.new(opts, list, cmd, project: self, args: rc || flag == :exec)
+          from = symjoin 'container', flag
           case flag
           when :run, :create, :exec
             if rc && !op.arg?('mount')
               all = collect_hash VAL_DOCKER[:run]
               delim = Regexp.new(",\\s*(?=#{all.join('|')})")
               Array(@mounts).each do |val|
-                args = []
                 type = nil
-                val.split(delim).each do |opt|
-                  k, v, q = split_option opt
+                args = val.split(delim).each_with_object([]) do |opt, out|
+                  k, v, q = OptionPartition.parse_option(opt)
                   if k == 'type'
                     case v
                     when 'bind', 'volume', 'image', 'tmpfs'
                       type = v
                     else
-                      raise_error TypeError, "unknown: #{v}", hint: flag
+                      raise_error TypeError, "unknown: #{v || "''"}", hint: flag
                     end
                   elsif all.include?(k)
                     unless type
-                      VAL_DOCKER[:run].each_pair do |key, a|
-                        next unless a.include?(k)
+                      VAL_DOCKER[:run].each_pair do |key, items|
+                        next unless items.include?(k)
 
                         type = key.to_s unless key == :common
                         break
@@ -482,13 +501,14 @@ module Squared
                     end
                     case k
                     when 'readonly', 'ro'
-                      args << k
+                      out << k
                       next
                     when 'source', 'src', 'destination', 'dst', 'target', 'volume-subpath', 'image-path'
+                      raise_error ArgumentError, "#{k}: no path value", hint: flag unless v
                       v = basepath v
                       v = shell_quote(v, option: false, force: false) if q == ''
                     end
-                    args << "#{k}=#{q + v + q}"
+                    out << "#{k}=#{q}#{v}#{q}"
                   elsif !silent?
                     log_message('unrecognized option', subject: from, hint: k)
                   end
@@ -514,7 +534,7 @@ module Squared
             opts = []
             append_option('platform', target: opts, equals: true)
             opts << case option('disable-content-trust', ignore: false)
-                    when 'false', '0'
+                    when '0', 'false'
                       '--disable-content-trust=false'
                     else
                       '--disable-content-trust'
@@ -536,9 +556,9 @@ module Squared
         def image(flag, opts = [], sync: true, id: nil, registry: nil)
           cmd, opts = docker_session('image', flag, opts: opts)
           op = OptionPartition.new(opts, OPT_DOCKER[:image].fetch(flag, []), cmd, project: self)
-          exception = self.exception
+          exception = exception?
           banner = true
-          from = :"image:#{flag}"
+          from = symjoin 'image', flag
           case flag
           when :ls
             if opts.size == op.size
@@ -565,9 +585,7 @@ module Squared
           when :rm
             unless id
               if op.empty?
-                list_image(:rm, from: from) do |val|
-                  image(:rm, opts, sync: sync, id: val)
-                end
+                list_image(:rm, from: from) { |val| image(:rm, opts, sync: sync, id: val) }
               else
                 op.each { |val| run(cmd.temp(val), sync: sync, from: from) }
               end
@@ -581,46 +599,46 @@ module Squared
           when :tag, :save
             list_image(flag, from: from) do |val|
               op << val
-              if flag == :tag
-                op << tagname("#{project}:#{op.first}")
-                break
-              end
+              next unless flag == :tag
+
+              op << tagname("#{project}:#{op.first}")
+              break
             end
-          when :push
-            id ||= option('tag', ignore: false) || tagmain
-            registry ||= op.shift || option('registry') || @registry
-            unless id && op.empty?
-              if id
-                raise_error ArgumentError, "unrecognized args: #{op.join(', ')}", hint: flag
-              else
-                raise_error 'no id/tag', hint: flag
-              end
+          when :pull
+            if !id
+              id = tagmain
+            elsif !op.arg?('a', 'all-tags') && !id.include?(':')
+              id = "#{project}:#{id}"
+            end
+            unless registry
+              registry = op.shift
+              registry ||= option('registry') || @registry unless id.include?('/')
             end
+            cmd << shell_quote(tagjoin(registry, id))
+          when :push
+            id ||= option('tag', ignore: false) || op.shift || tagmain
+            registry ||= option('registry') || op.shift || @registry
+            emptyargs op, flag
             raise_error ArgumentError, 'username/registry not specified', hint: flag unless registry
-            registry.chomp!('/')
-            uri = shell_quote "#{registry}/#{id}"
+            uri = shell_quote tagjoin(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)
 
-            cmd = img
+            @session = img
             sync = false
-            exception = true
+            exception ||= true
             banner = false
           end
-          run(cmd, sync: sync, exception: exception, banner: banner, from: from).tap do |ret|
-            success?(ret, flag == :tag || flag == :save)
-          end
+          success?(run(sync: sync, exception: exception, banner: banner, from: from), flag == :tag || flag == :save)
         end
 
         def network(flag, opts = [], target: nil)
           cmd, opts = docker_session('network', flag, opts: opts)
           OptionPartition.new(opts, OPT_DOCKER[:network].fetch(flag, []), cmd, project: self)
                          .clear
-          from = :"network:#{flag}"
-          list_image(flag, docker_output('ps -a'), from: from) do |img|
-            success?(run(cmd.temp(target, img), from: from))
-          end
+          from = symjoin 'network', flag
+          list_image(flag, docker_output('ps -a'), from: from) { |id| success?(run(cmd.temp(target, id), from: from)) }
         end
 
         def build?
@@ -669,7 +687,7 @@ module Squared
             elsif (data = doc.dig(*keys))
               yield data
             end
-          rescue StandardError => e
+          rescue => e
             log.debug e
           end
         end
@@ -686,11 +704,11 @@ module Squared
         end
 
         def append_command(flag, val, list, target: @session, prompt: ':')
-          if list.delete(prompt)
-            list << readline('Enter command [args]', force: flag == :exec)
-          else
-            env('DOCKER_ARGS') { |args| list << args }
-          end
+          list << if list.delete(prompt)
+                    readline('Enter command [args]', force: flag == :exec)
+                  else
+                    env('DOCKER_ARGS')
+                  end
           case flag
           when :run
             unless session_arg?('name', target: target)
@@ -703,10 +721,10 @@ module Squared
           target << list.join(' && ') unless list.empty?
         end
 
-        def append_file(type, target: @session, index: 2)
-          return if !@file || (ENV['COMPOSE_FILE'] && compose?(type))
+        def append_file(type, target: @session, index: 2, force: false)
+          return unless @file && !(ENV['COMPOSE_FILE'] && compose?(type))
 
-          unless @file.is_a?(Array)
+          unless @file.is_a?(Array) || force
             case type
             when 2, 4
               return
@@ -714,14 +732,7 @@ module Squared
               return unless COMPOSEFILE.select { |val| basepath!(val) }.size > 1
             end
           end
-          files = Array(@file).map { |val| quote_option('file', basepath(val)) }
-          if target.is_a?(Set)
-            opts = target.to_a.insert(index, *files)
-            target.clear
-                  .merge(opts)
-          else
-            target.insert(index, *files)
-          end
+          target.insert(index, *Array(@file).map { |val| quote_option('file', basepath(val)) })
         end
 
         def append_context(ctx = nil, target: @session)
@@ -735,7 +746,7 @@ module Squared
           ver = option('version', target: target, ignore: false)
           case val
           when String
-            split_escape val
+            val.split(',')
           else
             Array(val)
           end.each do |s|
@@ -828,7 +839,7 @@ module Squared
             end
             list_empty(hint: hint || from) if index == 1 && !y
           end
-        rescue StandardError => e
+        rescue => e
           on_error e, from
         end
 
@@ -900,18 +911,18 @@ module Squared
           when :tag
             args = tagjoin @registry, tag
           when :save
-            opts = "#{opts}.tar" unless opts.end_with?('.tar')
+            opts = opts.sub_ext('.tar')
             cmd << quote_option('output', File.expand_path(opts))
             if args
-              cmd << basic_option('platform', args)
+              cmd << quote_option('platform', args)
               args = nil
             end
           else
             cmd << opts << '--'
           end
-          cmd.merge(Array(out).map! { |val| val.split(/\s+/, 2).first })
+          cmd.merge(Array(out).map { |val| val.split(/\s+/, 2).first })
           cmd << args
-          success?(run(cmd), ctx.start_with?(/network|tag|save/))
+          success?(run(cmd), ctx.start_with?('network', 'tag', 'save'))
         end
 
         def filetype(val = dockerfile)
@@ -934,11 +945,12 @@ module Squared
         end
 
         def tagjoin(*args, char: '/')
-          args.join(char) unless (args = args.compact).empty?
+          args.compact!
+          args.map { |val| val.chomp(char) }.join(char) unless args.empty?
         end
 
         def tagname(val)
-          val = val.split(':').map! { |s| charname(s.sub(/^\W+/, '')) }
+          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]
@@ -955,6 +967,14 @@ module Squared
         def tagmain
           tag.is_a?(Array) ? tag.first : tag
         end
+
+        def emptyargs(list, hint = nil)
+          raise_error ArgumentError, "unrecognized args: #{list.join(', ')}", hint: hint unless list.empty?
+        end
+
+        def anypath?(*args)
+          args.any? { |val| basepath!(val) }
+        end
       end
 
       Application.implement Docker