squared 0.5.4 → 0.5.6

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

@@ -4,7 +4,7 @@ module Squared
   module Workspace
     module Project
       class Docker < Base
-        COMPOSEFILE = %w[compose.yaml compose.yml docker-compose.yaml compose.yml docker-compose.yml].freeze
+        COMPOSEFILE = %w[compose.yaml compose.yml docker-compose.yaml 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
         OPT_DOCKER = {
@@ -12,56 +12,57 @@ module Squared
                      tlskey=p].freeze,
           buildx: {
             common: %w[builder=b D|debug],
-            build: %w[load pull push add-host=q annotation=q attest=q build-arg=qq ent=q iidfile=p label=q a-file=p
-                      network=b no-cache-filter=b o|output=q platform=b q|quiet secret=qq shm-size=b ssh=qq t|tag=b
-                      target=b ulimit=q].freeze,
-            bake: %w[load print pull push list=q metadata-file=p set=q].freeze,
-            shared: %w[check no-cache allow=q call=b? f|file=p progress=b provenance=q sbom=q].freeze
+            build: %w[add-host=q annotation=q attest=q build-arg=qq build-context=qq cache-from=q cache-to=q
+                      cgroup-parent=b ent=q iidfile=p label=q a-file=p network=b no-cache-filter=b o|output=q platform=b
+                      q|quiet secret=qq shm-size=b ssh=qq t|tag=b target=b ulimit=q].freeze,
+            bake: %w[print list=q set=q].freeze,
+            shared: %w[check load no-cache pull push allow=q call=b? f|file=p metadata-file=p progress=b provenance=q
+                       sbom=q].freeze
           }.freeze,
           compose: {
-            common: %w[all-resources compatibility dry-run ansi|b env-file=p f|file=p parallel=b profile=b progress=b
+            common: %w[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=e].freeze,
-            build: %w[check 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
-                    u|user=e v|volume=q w|workdir=q].freeze,
-            up: %w[y abort-on-container-exit abort-on-container-failure always-recreate-deps attach-dependencies build
-                   d|detach dry-run force-recreate menu no-build no-color no-deps no-log-prefix no-recreate no-start
-                   quiet-pull remove-orphans V|renew-anon-volumes timestamps wait w|watch attach=b exit-code-from=b
-                   no-attach=b pull=b scale=i t|timeout=i wait-timeout=i].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,
+            exec: %w[d|detach privileged e|env=qq index=i T|no-TTY=b? user=e 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 q e|env=qq env-from-file=p i|interactive=b? l|label=q name=b T|no-TTY=b?
+                    p|publish=e pull=b u|user=e v|volume=q 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
+                   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
           }.freeze,
           container: {
             create: %w[init i|interactive no-healthcheck oom-kill-disable privileged P|publish-all q|quiet read-only
-                       rm runtime t|tty use-api-socket io-maxbandwidth=b io-maxiops=b add-host=q annotation=q
-                       a|attach=b blkio-weight=i blkio-weight-device=i cap-add=b cap-drop=b cgroup-parent=b cgroupns=b
-                       cidfile=p device=q device-cgroup-rule=q device-read-bps=q device-read-iops=q device-write-bps=q
+                       rm runtime t|tty use-api-socket io-maxbandwidth=b io-maxiops=b add-host=q annotation=q a|attach=b
+                       blkio-weight=i blkio-weight-device=i cap-add=b cap-drop=b cgroup-parent=b cgroupns=b cidfile=p
+                       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 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 m|memory=b memory-reservation=b memory-swap=n
-                       memory-swappiness=n mount=qq name=b network=b network-alias=b oom-score-adj=b pid=b pids-limit=n
-                       platform=b p|publish=e 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=q
-                       volume-driver=b volumes-from=b w|workdir=q].freeze,
+                       entrypoint=q e|env=qq env-file=p expose=e gpus=q group-add=b health-cmd=q health-interval=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 m|memory=b memory-reservation=b memory-swap=n memory-swappiness=n
+                       mount=qq name=b network=b network-alias=b oom-score-adj=b pid=b pids-limit=n platform=b
+                       p|publish=e 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=q volume-driver=b
+                       volumes-from=b w|workdir=q].freeze,
             run: %w[d|detach detach-keys=q sig-proxy=b?].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
                        cpuset-cpus=b cpuset-mems=b m|memory=b memory-reservation=b memory-swap=b pids-limit=n
                        restart=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,
             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,
             kill: %w[s|signal=b].freeze,
-            stats: %w[no-trunc format|q].freeze
+            stats: %w[a|all no-stream no-trunc format|q].freeze
           }.freeze,
           image: {
-            list: %w[a|all digests no-trunc f|filter=q format=q].freeze,
+            list: %w[a|all q|quiet digests no-trunc tree 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,
             save: %w[o|output=p platform=b].freeze
@@ -96,7 +97,7 @@ module Squared
 
         subtasks({
           'build' => %i[tag context].freeze,
-          'compose' => %i[build run exec up].freeze,
+          'compose' => %i[build run exec up down].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
@@ -147,11 +148,11 @@ module Squared
                       end
                     end
                   when 'bake'
-                    next unless bake?
+                    break unless bake?
 
                     case flag
                     when :build
-                      format_desc action, flag, ':?,opts*,target*,context?'
+                      format_desc action, flag, 'opts*,target*,context?|:'
                       task flag do |_, args|
                         args = args.to_a
                         if args.first == ':'
@@ -168,16 +169,16 @@ module Squared
                       end
                     end
                   when 'compose'
-                    next unless compose?
+                    break unless compose?
 
                     case flag
-                    when :build, :up
-                      format_desc action, flag, 'opts*,service*'
+                    when :build, :up, :down
+                      format_desc action, flag, 'opts*,service*|:'
                       task flag do |_, args|
                         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)
@@ -186,7 +187,7 @@ module Squared
                   when 'container'
                     case flag
                     when :exec, :commit
-                      format_desc(action, flag, flag == :exec ? 'id/name,opts*,args+' : 'id/name,tag?,opts*')
+                      format_desc(action, flag, flag == :exec ? 'id/name,opts*,args+|:' : 'id/name,tag?,opts*')
                       task flag, [:id] do |_, args|
                         if flag == :exec && !args.id
                           choice_command flag
@@ -196,7 +197,7 @@ module Squared
                         end
                       end
                     when :run, :create
-                      format_desc action, flag, 'image,opts*,args*'
+                      format_desc action, flag, 'image,opts*,args*|:'
                       task flag, [:image] do |_, args|
                         if args.image
                           container(flag, args.extras, id: args.image)
@@ -222,7 +223,8 @@ module Squared
                       format_desc(action, flag, case flag
                                                 when :rm, :save then 'id*,opts*'
                                                 when :tag then 'version?'
-                                                else 'opts*,args*' end)
+                                                else 'opts*,args*'
+                                                end)
                       task flag do |_, args|
                         args = args.to_a
                         if args.empty? && flag != :list
@@ -259,12 +261,11 @@ module Squared
 
           ret = docker_session
           if from == :run
-            case (n = filetype)
-            when 1, 2
+            if bake?(n = filetype)
               ret << 'buildx' << 'bake'
               append_file n
               from = :bake
-            when 3, 4
+            elsif compose?(n)
               ret << 'compose' << 'build'
               append_file n
               from = :compose
@@ -335,7 +336,7 @@ module Squared
                   op.push(val)
                 end
               end
-              op.append(args, escape: true)
+              op.append(args, escape: true, strip: /^:/)
               contextdir context if context
             end
           end
@@ -348,15 +349,27 @@ module Squared
           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
-            op.append(escape: true)
-          when :exec, :run
-            append_command flag, service, op.extras
+          op.parse(OPT_DOCKER[:compose].fetch(flag, []))
+          multiple = case flag
+                     when :build, :up, :down then true
+                     else false
+                     end
+          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 : 5)
+                      end
           end
-          run(from: from)
+          if multiple
+            op.concat(service) if service
+            op.append(delim: true, escape: true, strip: /^:/)
+          else
+            raise_error('no services were found', hint: flag) unless service
+            append_command(flag, service, op.extras, prompt: '::')
+          end
+          run(from: :"compose:#{flag}")
         end
 
         def container(flag, opts = [], id: nil)
@@ -388,10 +401,10 @@ module Squared
                   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
+                        next unless val.include?(k)
+
+                        type = key.to_s unless key == :common
+                        break
                       end
                     end
                     case k
@@ -414,7 +427,7 @@ module Squared
             append_command(flag, id || tagmain, op.extras)
           when :update
             raise_error('missing container', hint: flag) if op.empty?
-            op.append(escape: true)
+            op.append(escape: true, strip: /^:/)
           when :commit
             latest = op.shift || tagmain
             cmd << id << latest
@@ -467,7 +480,7 @@ module Squared
               end
               return
             else
-              op.append(escape: true)
+              op.append(escape: true, strip: /^:/)
             end
           end
           run(from: from)
@@ -544,7 +557,7 @@ module Squared
 
         def network(flag, opts = [], target: nil)
           cmd, opts = docker_session('network', flag, opts: opts)
-          op = OptionPartition.new(opts, OPT_DOCKER[:network][flag], cmd, project: self)
+          op = OptionPartition.new(opts, OPT_DOCKER[:network].fetch(flag, []), cmd, project: self)
           op.clear
           from = :"network:#{flag}"
           list_image(flag, docker_output('ps -a'), from: from) do |img|
@@ -561,10 +574,14 @@ module Squared
         end
 
         def compose?(file = dockerfile)
+          return file == 3 || file == 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)
+
           BAKEFILE.include?(File.basename(file))
         end
 
@@ -585,6 +602,21 @@ module Squared
 
         private
 
+        def read_composefile(*keys, target: nil)
+          require 'yaml'
+          target = ENV['COMPOSE_FILE']&.split(workspace.windows? ? ';' : ':') if !target || target.empty?
+          Array(target || dockerfile).each do |val|
+            doc = YAML.load_file(path + val)
+            if keys.empty?
+              yield doc
+            elsif (data = doc.dig(*keys))
+              yield data
+            end
+          rescue StandardError => e
+            log.debug e
+          end
+        end
+
         def docker_session(*cmd, opts: nil)
           return session('docker', *cmd) unless opts
 
@@ -597,9 +629,9 @@ 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)
+        def append_command(flag, val, list, target: @session, prompt: ':')
+          if list.delete(prompt)
+            list << readline('Enter command [args]', force: flag == :exec)
           elsif (args = env('DOCKER_ARGS'))
             list << args
           end
@@ -616,13 +648,22 @@ module Squared
         end
 
         def append_file(type, target: @session)
-          return unless type == 2 || type == 4 || @file.is_a?(Array)
+          return if ENV['COMPOSE_FILE'] && compose?(type)
 
+          unless @file.is_a?(Array)
+            case type
+            when 2, 4
+              return
+            when 3
+              return unless COMPOSEFILE.map { |val| path + val }.select(&:exist?).size > 1
+            end
+          end
           files = Array(@file).map { |val| quote_option('file', path + val) }
           if target.is_a?(Set)
-            target.merge(files)
+            opts = target.to_a.insert(2, *files)
+            target.clear.merge(opts)
           else
-            target.concat(files)
+            target.insert(2, *files)
           end
         end
 
@@ -788,7 +829,7 @@ module Squared
             bake?(val) ? 1 : 2
           when '.yml', '.yaml'
             if compose?(val)
-              path.children.any? { |file| bake?(file) } ? 1 : 3
+              @only&.include?('compose') || path.children.none? { |file| bake?(file) } ? 3 : 1
             else
               4
             end