capistrano-nomad 0.6.4 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 707ded6b01d5903708a362ef6e98f3c56d5e52fd1557bfad47f1844cb7d6c96e
-  data.tar.gz: 68ea61e7025400e99c97683b8f3390357a0f04cf5b173846bba378d8bdd5fa3d
+  metadata.gz: d227d6f15f433cd6410884afc5572cf9c7e115a7365e00e244d3b95ca320f148
+  data.tar.gz: a0cdd0e131319cd0f1a17a926afe70619d24550018de9731c96a610b075300c2
 SHA512:
-  metadata.gz: 19a5ee3bcddf663ed3820245f10d0c35dee3cf2d1659b44f61947696c7ea24c263696c3c25d6dba032d14293706f0b9b68ee3e9cdbd41b66309e43eb92a0e297
-  data.tar.gz: b20019a85564cc4eee312bfaa17934becab4b73c2505c7a0b9511b1759b320c56fda3656b7f2b80b3cd090e347f05f5c0a2ef3e283a078850636c37ea58f20fb
+  metadata.gz: fd4aec6aecd6698d4ed23d4e1a2c3d3e8ded18467c1a4d8daee191e29304e2f91f85ca7d604681dba4ec6cb885308f0fb6985fb4e9d952e459fc7859a3b3917a
+  data.tar.gz: 6009d60f5bd99e35fdb2961f6b9d860ea9db40ac844cebacf87b21ae58381eeefd35022fb58c7a6b88fa74da7e3acff3ee26a0f9a71a92e1b61495102bf53a21

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    capistrano-nomad (0.6.4)
+    capistrano-nomad (0.7.0)
       activesupport (<= 7.0.8)
       byebug
       capistrano (~> 3.0)

data/README.md

@@ -37,8 +37,8 @@ Within `deploy.rb`
 set :nomad_jobs_path, "nomad/jobs"
 set :nomad_var_files_path, "nomad/vars"
 
-# Accessible in all .erb files
-set :nomad_erb_vars, (lambda do
+# Make variables available to all template .erb files
+set :nomad_template_vars, (lambda do
   {
     env_name: fetch(:stage).to_sym,
     domain: fetch(:domain),
@@ -46,19 +46,45 @@ set :nomad_erb_vars, (lambda do
   }
 end)
 
-# Docker image types
+# Make helpers available to all template .erb files
+nomad_template_helpers do
+  def restart_stanza(interval = "1m")
+    <<-EOF
+      restart {
+        interval = "#{interval}"
+        attempts = 3
+        mode = "delay"
+      }
+    EOF
+  end
+end
+
+# Use hosted Docker image
+nomad_docker_image_type :postgres,
+  alias_digest: "postgres:5.0.0"
+
+# Use Docker image that will be built locally relative to project and push
 nomad_docker_image_type :backend,
   path: "local/path/backend",
   alias: ->(image_type:) { "gcr.io/axsuul/#{image_type}" },
   target: "release",
   build_args: { foo: "bar" }
+
+# Use Docker image that will be built locally from an absolute path and push
 nomad_docker_image_type :redis,
   path: "/absolute/path/redis",
   alias: "gcr.io/axsuul/redis"
 
+# Use Docker image that will be built remotely on server
+nomad_docker_image_type :restic,
+  path: "containers/restic",
+  alias: "my-project/restic:local",
+  strategy: :remote_build
+
 # Jobs
-nomad_job :frontend
 nomad_job :backend, docker_image_types: [:backend], var_files: [:rails]
+nomad_job :frontend
+nomad_job :postgres, docker_image_types: [:postgres]
 nomad_job :redis, docker_image_types: [:redis]
 nomad_job :"traefik-default", template: :traefik, erb_vars: { role: :default }
 nomad_job :"traefik-secondary", template: :traefik, erb_vars: { role: :secondary }
@@ -68,21 +94,29 @@ nomad_namespace :analytics do
 end
 ```
 
-Deploy all with
+Deploy all jobs
 
 ```shell
 cap production nomad:all:deploy
 ```
 
-Deploy individually with
+Deploy individual jobs
 
 ```shell
 cap production nomad:app:deploy
-cap production nomad:redis:purge
 cap production nomad:analytics:grafana:deploy
 ```
 
-Open console with
+Manage jobs
+
+```shell
+cap production nomad:app:stop
+cap production nomad:redis:purge
+cap production nomad:analytics:grafana:restart
+cap production nomad:postgres:status
+```
+
+Open console
 
 ```shell
 cap production nomad:app:console
@@ -90,10 +124,19 @@ cap production nomad:app:console TASK=custom-task-name
 cap production nomad:analytics:grafana:console
 ```
 
+Display logs
+
+```shell
+cap production nomad:app:logs
+cap production nomad:app:stdout
+cap production nomad:app:stderr
+cap production nomad:analytics:grafana:follow
+```
+
 Create missing and delete unused namespaces
 
 ```shell
-cap production nomad:all:replace_namespaces
+cap production nomad:all:modify_namespaces
 ```
 
 ## Development

data/capistrano-nomad.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name = "capistrano-nomad"
-  spec.version = "0.6.4"
+  spec.version = "0.7.0"
   spec.authors = ["James Hu"]
 
   spec.summary = "Capistrano plugin for deploying and managing Nomad jobs"

data/lib/capistrano/nomad/helpers/base.rb

@@ -5,3 +5,7 @@ end
 def capistrano_nomad_deep_symbolize_hash_keys(hash)
   JSON.parse(JSON[hash], symbolize_names: true)
 end
+
+def capistrano_nomad_run_remotely(&block)
+  on(roles(:manager), &block)
+end

data/lib/capistrano/nomad/helpers/docker.rb

@@ -21,7 +21,7 @@ end
 def capistrano_nomad_read_docker_image_types_manifest
   manifest = {}
 
-  on(roles(:manager)) do |_host|
+  capistrano_nomad_run_remotely do
     # Ensure file exists
     execute("mkdir -p #{shared_path}")
     execute("touch #{capistrano_nomad_docker_image_types_manifest_path}")
@@ -37,7 +37,7 @@ def capistrano_nomad_read_docker_image_types_manifest
 end
 
 def capistrano_nomad_update_docker_image_types_manifest(image_type, properties = {})
-  on(roles(:manager)) do |_host|
+  capistrano_nomad_run_remotely do
     # Read and update manifest
     manifest = capistrano_nomad_read_docker_image_types_manifest
     manifest[image_type] = (manifest[image_type] || {}).merge(properties.stringify_keys)
@@ -70,6 +70,9 @@ def capistrano_nomad_build_docker_image_for_type(image_type)
 
   return unless attributes
 
+  # No need to build if there's no path
+  return unless attributes[:path]
+
   args = [
     # Ensure images are built for x86_64 which is production env otherwise it will default to local development env
     # which can be arm64 (Apple Silicon)
@@ -87,12 +90,7 @@ def capistrano_nomad_build_docker_image_for_type(image_type)
     args << "--build-arg #{key}=#{value}"
   end
 
-  run_locally do
-    # If any of these files exist then we're in the middle of rebase so we should interrupt
-    if ["rebase-merge", "rebase-apply"].any? { |f| File.exist?("#{capistrano_nomad_git.dir.path}/.git/#{f}") }
-      raise StandardError, "Still in the middle of git rebase, interrupting docker image build"
-    end
-
+  docker_build_command = lambda do |path|
     image_alias_args = args.dup
 
     [capistrano_nomad_build_docker_image_alias(image_type)]
@@ -101,31 +99,60 @@ def capistrano_nomad_build_docker_image_for_type(image_type)
         image_alias_args << "--tag #{tag}"
       end
 
-    execute("docker build #{image_alias_args.join(' ')} #{capistrano_nomad_root.join(attributes[:path])}")
+    "docker build #{image_alias_args.join(' ')} #{path}"
+  end
+
+  case attributes[:strategy]
+
+  # We need to build Docker container locally
+  when :local_build, :local_push
+    run_locally do
+      # If any of these files exist then we're in the middle of rebase so we should interrupt
+      if ["rebase-merge", "rebase-apply"].any? { |f| File.exist?("#{capistrano_nomad_git.dir.path}/.git/#{f}") }
+        raise StandardError, "Still in the middle of git rebase, interrupting docker image build"
+      end
+
+      execute(docker_build_command.call(capistrano_nomad_root.join(attributes[:path])))
+    end
+
+  # We need to build Docker container remotely
+  when :remote_build, :remote_push
+    remote_path = Pathname.new(release_path).join(attributes[:path])
+    capistrano_nomad_upload(local_path: attributes[:path], remote_path: remote_path)
+
+    capistrano_nomad_run_remotely do
+      execute(docker_build_command.call(remote_path))
+    end
   end
 end
 
 def capistrano_nomad_push_docker_image_for_type(image_type, is_manifest_updated: true)
-  # Allows end user to add hooks
-  invoke("nomad:docker_images:push")
+  attributes = fetch(:nomad_docker_image_types)[image_type]
+  alias_digest = attributes&.dig(:alias_digest)
+
+  return false unless [:local_push, :remote_push].include?(attributes[:strategy])
 
   run_locally do
-    interaction_handler = CapistranoNomadDockerPushImageInteractionHandler.new
-    image_alias = capistrano_nomad_build_docker_image_alias(image_type)
+    # Don't push Docker image if alias digest is already passed in
+    unless alias_digest
+      interaction_handler = CapistranoNomadDockerPushImageInteractionHandler.new
+      image_alias = capistrano_nomad_build_docker_image_alias(image_type)
+
+      # We should not proceed if image cannot be pushed
+      unless execute("docker push #{image_alias}", interaction_handler: interaction_handler)
+        raise StandardError, "Docker image push unsuccessful!"
+      end
 
-    # We should not proceed if image cannot be pushed
-    unless execute("docker push #{image_alias}", interaction_handler: interaction_handler)
-      raise StandardError, "Docker image push unsuccessful!"
-    end
+      return unless is_manifest_updated
 
-    return unless is_manifest_updated
+      # Has the @sha256:xxxx appended so we have the ability to also target by digest
+      alias_digest = "#{image_alias}@#{interaction_handler.last_digest}"
+    end
 
     # Update image type manifest
     capistrano_nomad_update_docker_image_types_manifest(image_type,
       alias: image_alias,
-
-      # Has the @sha256:xxxx appended so we have the ability to also target by digest
-      alias_digest: "#{image_alias}@#{interaction_handler.last_digest}",
+      alias_digest: alias_digest,
     )
   end
 end

data/lib/capistrano/nomad/helpers/dsl.rb

@@ -1,6 +1,14 @@
+require "active_support/core_ext/hash"
+
 def nomad_docker_image_type(image_type, attributes = {})
   docker_image_types = fetch(:nomad_docker_image_types) || {}
-  docker_image_types[image_type] = attributes
+  docker_image_types[image_type] = attributes.reverse_merge(
+    # In case image doesn't get pushed, this will still be populated
+    alias_digest: attributes[:alias],
+
+    # By default build and push Docker image locally
+    strategy: :local_push,
+  )
 
   set(:nomad_docker_image_types, docker_image_types)
 end
@@ -50,12 +58,12 @@ def nomad_job(name, attributes = {})
 
       desc "Run #{description_name} job"
       task :run do
-        capistrano_nomad_run_jobs([name], namespace: namespace)
+        capistrano_nomad_run_jobs([name], namespace: namespace, is_detached: false)
       end
 
       desc "Purge and run #{description_name} job again"
       task :rerun do
-        capistrano_nomad_rerun_jobs([name], namespace: namespace)
+        capistrano_nomad_rerun_jobs([name], namespace: namespace, is_detached: false)
       end
 
       desc "Upload and plan #{description_name} job"
@@ -65,22 +73,17 @@ def nomad_job(name, attributes = {})
 
       desc "Upload and run #{description_name} job"
       task :upload_run do
-        capistrano_nomad_upload_run_jobs([name], namespace: namespace)
+        capistrano_nomad_upload_run_jobs([name], namespace: namespace, is_detached: false)
       end
 
       desc "Upload and re-run #{description_name} job"
       task :upload_rerun do
-        capistrano_nomad_upload_rerun_jobs([name], namespace: namespace)
+        capistrano_nomad_upload_rerun_jobs([name], namespace: namespace, is_detached: false)
       end
 
       desc "Deploy #{description_name} job"
       task :deploy do
-        capistrano_nomad_deploy_jobs([name], namespace: namespace)
-      end
-
-      desc "Restart #{description_name} job"
-      task :restart do
-        capistrano_nomad_restart_jobs([name], namespace: namespace)
+        capistrano_nomad_deploy_jobs([name], namespace: namespace, is_detached: false)
       end
 
       desc "Stop #{description_name} job"
@@ -88,6 +91,11 @@ def nomad_job(name, attributes = {})
         capistrano_nomad_stop_jobs([name], namespace: namespace)
       end
 
+      desc "Restart #{description_name} job"
+      task :restart do
+        capistrano_nomad_restart_jobs([name], namespace: namespace)
+      end
+
       desc "Purge #{description_name} job"
       task :purge do
         capistrano_nomad_purge_jobs([name], namespace: namespace, is_detached: false)
@@ -98,9 +106,32 @@ def nomad_job(name, attributes = {})
         capistrano_nomad_display_job_status(name, namespace: namespace)
       end
 
-      desc "Open console to #{description_name} job. Specify task by passing TASK environment variable"
+      desc "Open console to #{description_name} job. Specify task with TASK, command with CMD"
       task :console do
-        capistrano_nomad_exec_within_job(name, "/bin/bash", namespace: namespace, task: ENV["TASK"].presence)
+        command = ENV["CMD"].presence || "/bin/bash"
+
+        capistrano_nomad_exec_within_job(name, command, namespace: namespace, task: ENV["TASK"])
+      end
+
+      desc "Display stdout and stderr of #{description_name} job. Specify task with TASK"
+      task :logs do
+        capistrano_nomad_tail_job_logs(name, namespace: namespace, stdout: true)
+        capistrano_nomad_tail_job_logs(name, namespace: namespace, stderr: true)
+      end
+
+      desc "Display stdout of #{description_name} job. Specify task with TASK"
+      task :stdout do
+        capistrano_nomad_tail_job_logs(name, namespace: namespace, stdout: true)
+      end
+
+      desc "Display stderr of #{description_name} job. Specify task with TASK"
+      task :stderr do
+        capistrano_nomad_tail_job_logs(name, namespace: namespace, stderr: true)
+      end
+
+      desc "Follow logs of #{description_name} job. Specify task with TASK"
+      task :follow do
+        capistrano_nomad_display_job_logs(name, namespace: namespace, f: true)
       end
     end
   end
@@ -116,3 +147,7 @@ def nomad_job(name, attributes = {})
     end
   end
 end
+
+def nomad_template_helpers(&block)
+  CapistranoNomadErbNamespace.class_eval(&block)
+end

data/lib/capistrano/nomad/helpers/nomad.rb

@@ -10,16 +10,6 @@ class CapistranoNomadErbNamespace
     end
   end
 
-  # Use default value passed in unless `nomad_job_task_cpu_resource` is set
-  def build_nomad_job_task_cpu_resource(default:)
-    nomad_job_task_cpu_resource == "null" ? default : nomad_job_task_cpu_resource
-  end
-
-  # Use default value passed in unless `nomad_job_task_memory_resource` is set
-  def build_nomad_job_task_memory_resource(default:)
-    nomad_job_task_memory_resource == "null" ? default : nomad_job_task_memory_resource
-  end
-
   # rubocop:disable Style/MissingRespondToMissing
   def method_missing(name, *args)
     instance_variable = "@#{name}"
@@ -106,7 +96,7 @@ def capistrano_nomad_run_nomad_command(kind, *args)
 end
 
 def capistrano_nomad_execute_nomad_command(*args)
-  on(roles(:manager)) do |host|
+  capistrano_nomad_run_remotely do |host|
     run_interactively(host) do
       capistrano_nomad_run_nomad_command(:execute, *args)
     end
@@ -116,32 +106,59 @@ end
 def capistrano_nomad_capture_nomad_command(*args)
   output = nil
 
-  on(roles(:manager)) do |_host|
+  capistrano_nomad_run_remotely do
     output = capistrano_nomad_run_nomad_command(:capture, *args)
   end
 
   output
 end
 
-def capistrano_nomad_exec_within_job(name, command, namespace: nil, task: nil)
-  on(roles(:manager)) do
-    task ||= name
-
-    # Find alloc id that contains task
-    output = capistrano_nomad_capture_nomad_command(
-      :job,
-      :allocs,
-      { namespace: namespace, t: "'{{range .}}{{ .ID }},{{ .TaskGroup }}|{{end}}'" },
-      name,
-    )
-    alloc_id = output.split("|").map { |s| s.split(",") }.find { |_, t| t == task.to_s }&.first
+def capistrano_nomad_find_job_task_details(name, namespace: nil, task: nil)
+  task = task.presence || name
+
+  # Find alloc id that contains task
+  allocs_output = capistrano_nomad_capture_nomad_command(
+    :job,
+    :allocs,
+    { namespace: namespace, t: "'{{range .}}{{ .ID }},{{ .TaskGroup }}|{{end}}'" },
+    name,
+  )
+  alloc_id = allocs_output.split("|").map { |s| s.split(",") }.find { |_, t| t == task.to_s }&.first
+
+  # Can't continue if we can't choose an alloc id
+  return unless alloc_id
+
+  tasks_output = capistrano_nomad_capture_nomad_command(
+    :alloc,
+    :status,
+    { namespace: namespace, t: "'{{range $key, $value := .TaskStates}}{{ $key }},{{ .State }}|{{end}}'" },
+    alloc_id,
+  )
+  tasks_by_score = tasks_output.split("|").each_with_object({}) do |task_output, hash|
+    task, state = task_output.split(",")
+
+    score = 0
+    score += 5 if state == "running"
+    score += 5 unless task.match?(/connect-proxy/)
+
+    hash[task] = score
+  end
+  task = tasks_by_score.max_by { |_, v| v }.first
 
-    if alloc_id
+  {
+    alloc_id: alloc_id,
+    name: task,
+  }
+end
+
+def capistrano_nomad_exec_within_job(name, command, namespace: nil, task: nil)
+  capistrano_nomad_run_remotely do
+    if (task_details = capistrano_nomad_find_job_task_details(name, namespace: namespace, task: task))
       capistrano_nomad_execute_nomad_command(
         :alloc,
         :exec,
-        { namespace: namespace, task: task },
-        alloc_id,
+        { namespace: namespace, task: task_details[:name] },
+        task_details[:alloc_id],
         command,
       )
     else
@@ -157,40 +174,55 @@ def capistrano_nomad_exec_within_job(name, command, namespace: nil, task: nil)
   end
 end
 
-def capistrano_nomad_upload_file(local_path:, remote_path:, erb_vars: {})
-  docker_image_types = fetch(:nomad_docker_image_types)
-  docker_image_types_manifest = capistrano_nomad_read_docker_image_types_manifest
-
-  # Merge manifest into image types
-  docker_image_types_manifest.each do |manifest_image_type, manifest_attributes|
-    docker_image_types[manifest_image_type]&.merge!(manifest_attributes) || {}
-  end
-
-  # Parse manifest files using ERB
-  erb = ERB.new(File.open(local_path).read, trim_mode: "-")
-
-  final_erb_vars = {
-    git_commit_id: fetch(:current_revision) || capistrano_nomad_git_commit_id,
-    docker_image_types: docker_image_types,
-  }
-
-  # Add global ERB vars
-  final_erb_vars.merge!(fetch(:nomad_erb_vars) || {})
+def capistrano_nomad_upload(local_path:, remote_path:, erb_vars: {})
+  # If directory upload everything within the directory
+  if File.directory?(local_path)
+    Dir.glob("#{local_path}/*").each do |path|
+      capistrano_nomad_upload(local_path: path, remote_path: "#{remote_path}/#{File.basename(path)}")
+    end
+  else
+    io =
+      if File.extname(local_path) == ".erb"
+        docker_image_types = fetch(:nomad_docker_image_types)
+        docker_image_types_manifest = capistrano_nomad_read_docker_image_types_manifest
+
+        # Merge manifest into image types
+        docker_image_types_manifest.each do |manifest_image_type, manifest_attributes|
+          docker_image_types[manifest_image_type]&.merge!(manifest_attributes) || {}
+        end
+
+        # Parse manifest files using ERB
+        erb = ERB.new(File.open(local_path).read, trim_mode: "-")
+
+        final_erb_vars = {
+          git_commit_id: fetch(:current_revision) || capistrano_nomad_git_commit_id,
+          docker_image_types: docker_image_types,
+        }
+
+        # Add global ERB vars
+        final_erb_vars.merge!(fetch(:nomad_template_vars) || {})
+
+        # Add job-specific ERB vars
+        final_erb_vars.merge!(erb_vars)
+
+        # We use a custom namespace class so that we can include helper methods into the namespace to make them available
+        # for template to access
+        namespace = CapistranoNomadErbNamespace.new(
+          context: self,
+          vars: final_erb_vars,
+        )
+
+        StringIO.new(erb.result(namespace.instance_eval { binding }))
+      else
+        File.open(local_path)
+      end
 
-  # Add job-specific ERB vars
-  final_erb_vars.merge!(erb_vars)
+    capistrano_nomad_run_remotely do
+      # Ensure parent directory exists
+      execute(:mkdir, "-p", File.dirname(remote_path))
 
-  # We use a custom namespace class so that we can include helper methods into the namespace to make them available for
-  # template to access
-  namespace = CapistranoNomadErbNamespace.new(
-    context: self,
-    vars: final_erb_vars,
-  )
-  erb_io = StringIO.new(erb.result(namespace.instance_eval { binding }))
-
-  on(roles(:manager)) do
-    execute(:mkdir, "-p", File.dirname(remote_path))
-    upload!(erb_io, remote_path)
+      upload!(io, remote_path)
+    end
   end
 end
 
@@ -236,7 +268,7 @@ def capistrano_nomad_upload_jobs(names, *args)
   uniq_var_files = names.map { |n| capistrano_nomad_fetch_job_var_files(n, *args) }.flatten.uniq
 
   uniq_var_files.each do |var_file|
-    capistrano_nomad_upload_file(
+    capistrano_nomad_upload(
       local_path: capistrano_nomad_build_local_var_file_path(var_file, *args),
       remote_path: capistrano_nomad_build_release_var_file_path(var_file, *args),
     )
@@ -252,7 +284,7 @@ def capistrano_nomad_upload_jobs(names, *args)
       # Can set a custom template instead
       file_basename = nomad_job_options[:template] || name
 
-      capistrano_nomad_upload_file(
+      capistrano_nomad_upload(
         local_path: capistrano_nomad_build_local_job_path(file_basename, *args),
         remote_path: capistrano_nomad_build_release_job_path(name, *args),
         erb_vars: erb_vars,
@@ -269,11 +301,11 @@ def capistrano_nomad_plan_jobs(names, *args)
   end
 end
 
-def capistrano_nomad_run_jobs(names, namespace: nil)
+def capistrano_nomad_run_jobs(names, namespace: nil, is_detached: true)
   names.each do |name|
     run_options = {
       namespace: namespace,
-      detach: true,
+      detach: is_detached,
 
       # Don't reset counts since they may have been scaled
       preserve_counts: true,
@@ -292,44 +324,52 @@ def capistrano_nomad_run_jobs(names, namespace: nil)
 end
 
 # Remove job and run again
-def capistrano_nomad_rerun_jobs(names)
+def capistrano_nomad_rerun_jobs(names, **options)
+  general_options = options.slice!(:is_detached)
+
   names.each do |name|
     # Wait for jobs to be purged before running again
-    capistrano_nomad_purge_jobs([name], is_detached: false)
+    capistrano_nomad_purge_jobs([name], **general_options.merge(is_detached: false))
 
-    capistrano_nomad_run_jobs([name])
+    capistrano_nomad_run_jobs([name], **general_options.merge(options))
   end
 end
 
-def capistrano_nomad_upload_plan_jobs(names, *args)
-  capistrano_nomad_upload_jobs(names, *args)
-  capistrano_nomad_plan_jobs(names, *args)
+def capistrano_nomad_upload_plan_jobs(names, **options)
+  capistrano_nomad_upload_jobs(names, **options)
+  capistrano_nomad_plan_jobs(names, **options)
 end
 
-def capistrano_nomad_upload_run_jobs(names, *args)
-  capistrano_nomad_upload_jobs(names, *args)
-  capistrano_nomad_run_jobs(names, *args)
+def capistrano_nomad_upload_run_jobs(names, **options)
+  general_options = options.slice!(:is_detached)
+
+  capistrano_nomad_upload_jobs(names, **general_options)
+  capistrano_nomad_run_jobs(names, **general_options.merge(options))
 end
 
-def capistrano_nomad_upload_rerun_jobs(names, *args)
-  capistrano_nomad_upload_jobs(names, *args)
-  capistrano_nomad_rerun_jobs(names, *args)
+def capistrano_nomad_upload_rerun_jobs(names, **options)
+  general_options = options.slice!(:is_detached)
+
+  capistrano_nomad_upload_jobs(names, **general_options)
+  capistrano_nomad_rerun_jobs(names, **general_options.merge(options))
 end
 
-def capistrano_nomad_deploy_jobs(names, *args)
-  capistrano_nomad_assemble_jobs_docker_images(names, *args)
-  capistrano_nomad_upload_run_jobs(names, *args)
+def capistrano_nomad_deploy_jobs(names, **options)
+  general_options = options.slice!(:is_detached)
+
+  capistrano_nomad_assemble_jobs_docker_images(names, **general_options)
+  capistrano_nomad_upload_run_jobs(names, **general_options.merge(options))
 end
 
-def capistrano_nomad_stop_jobs(names, namespace: nil)
+def capistrano_nomad_stop_jobs(names, **options)
   names.each do |name|
-    capistrano_nomad_execute_nomad_command(:job, :stop, { namespace: namespace }, name)
+    capistrano_nomad_execute_nomad_command(:job, :stop, options, name)
   end
 end
 
-def capistrano_nomad_restart_jobs(names, namespace: nil)
+def capistrano_nomad_restart_jobs(names, **options)
   names.each do |name|
-    capistrano_nomad_execute_nomad_command(:job, :restart, { namespace: namespace }, name)
+    capistrano_nomad_execute_nomad_command(:job, :restart, options, name)
   end
 end
 
@@ -339,6 +379,29 @@ def capistrano_nomad_purge_jobs(names, namespace: nil, is_detached: true)
   end
 end
 
-def capistrano_nomad_display_job_status(name, *args)
-  capistrano_nomad_execute_nomad_command(:status, *args, name)
+def capistrano_nomad_display_job_status(name, **options)
+  capistrano_nomad_execute_nomad_command(:status, options, name)
+end
+
+def capistrano_nomad_display_job_logs(name, namespace: nil, **options)
+  if (task_details = capistrano_nomad_find_job_task_details(name, namespace: namespace, task: ENV["TASK"]))
+    capistrano_nomad_execute_nomad_command(
+      :alloc,
+      :logs,
+      options.merge(namespace: namespace, task: task_details[:name]),
+      task_details[:alloc_id],
+    )
+  else
+    # If task can't be determined choose a random allocation
+    capistrano_nomad_execute_nomad_command(
+      :alloc,
+      :logs,
+      options.merge(namespace: namespace, job: true),
+      name,
+    )
+  end
+end
+
+def capistrano_nomad_tail_job_logs(*args, **options)
+  capistrano_nomad_display_job_logs(*args, **options.merge(tail: true, n: 50))
 end

data/lib/capistrano/nomad/tasks/nomad.rake

@@ -55,7 +55,7 @@ namespace :nomad do
     end
 
     desc "Create missing and remove unused namespaces"
-    task :replace_namespaces do
+    task :modify_namespaces do
       output = capistrano_nomad_capture_nomad_command(:namespace, :list, t: "'{{range .}}{{ .Name }}|{{end}}'")
       current_namespaces = output.split("|").compact.map(&:to_sym)
       desired_namespaces = fetch(:nomad_jobs).keys

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-nomad
 version: !ruby/object:Gem::Version
-  version: 0.6.4
+  version: 0.7.0
 platform: ruby
 authors:
 - James Hu
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-11-06 00:00:00.000000000 Z
+date: 2023-11-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport