cpflow 3.0.0

data/lib/command/ps.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Command
+  class Ps < Base
+    NAME = "ps"
+    OPTIONS = [
+      app_option(required: true),
+      location_option,
+      workload_option
+    ].freeze
+    DESCRIPTION = "Shows running replicas in app"
+    LONG_DESCRIPTION = <<~DESC
+      - Shows running replicas in app
+    DESC
+    EXAMPLES = <<~EX
+      ```sh
+      # Shows running replicas in app, for all workloads.
+      cpflow ps -a $APP_NAME
+
+      # Shows running replicas in app, for a specific workload.
+      cpflow ps -a $APP_NAME -w $WORKLOAD_NAME
+      ```
+    EX
+    WITH_INFO_HEADER = false
+
+    def call
+      cp.fetch_gvc!
+
+      location = config.location
+
+      workloads = [config.options[:workload]] if config.options[:workload]
+      workloads ||= config[:app_workloads] + config[:additional_workloads]
+      workloads.each do |workload|
+        cp.fetch_workload!(workload)
+
+        result = cp.fetch_workload_replicas(workload, location: location)
+        result["items"].each { |replica| puts replica }
+      end
+    end
+  end
+end

data/lib/command/ps_restart.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Command
+  class PsRestart < Base
+    NAME = "ps:restart"
+    OPTIONS = [
+      app_option(required: true),
+      workload_option
+    ].freeze
+    DESCRIPTION = "Forces redeploy of workloads in app"
+    LONG_DESCRIPTION = <<~DESC
+      - Forces redeploy of workloads in app
+    DESC
+    EXAMPLES = <<~EX
+      ```sh
+      # Forces redeploy of all workloads in app.
+      cpflow ps:restart -a $APP_NAME
+
+      # Forces redeploy of a specific workload in app.
+      cpflow ps:restart -a $APP_NAME -w $WORKLOAD_NAME
+      ```
+    EX
+
+    def call
+      workloads = [config.options[:workload]] if config.options[:workload]
+      workloads ||= config[:app_workloads] + config[:additional_workloads]
+
+      workloads.each do |workload|
+        step("Restarting workload '#{workload}'") do
+          cp.fetch_workload!(workload)
+
+          cp.workload_force_redeployment(workload)
+        end
+      end
+    end
+  end
+end

data/lib/command/ps_start.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module Command
+  class PsStart < Base
+    NAME = "ps:start"
+    OPTIONS = [
+      app_option(required: true),
+      workload_option,
+      location_option,
+      wait_option("workload to be ready")
+    ].freeze
+    DESCRIPTION = "Starts workloads in app"
+    LONG_DESCRIPTION = <<~DESC
+      - Starts workloads in app
+    DESC
+    EXAMPLES = <<~EX
+      ```sh
+      # Starts all workloads in app.
+      cpflow ps:start -a $APP_NAME
+
+      # Starts a specific workload in app.
+      cpflow ps:start -a $APP_NAME -w $WORKLOAD_NAME
+      ```
+    EX
+
+    def call
+      @workloads = [config.options[:workload]] if config.options[:workload]
+      @workloads ||= config[:app_workloads] + config[:additional_workloads]
+
+      @workloads.reverse_each do |workload|
+        step("Starting workload '#{workload}'") do
+          cp.set_workload_suspend(workload, false)
+        end
+      end
+
+      wait_for_ready if config.options[:wait]
+    end
+
+    private
+
+    def wait_for_ready
+      progress.puts
+
+      @workloads.reverse_each do |workload|
+        step("Waiting for workload '#{workload}' to be ready", retry_on_failure: true) do
+          cp.workload_deployments_ready?(workload, location: config.location, expected_status: true)
+        end
+      end
+    end
+  end
+end

data/lib/command/ps_stop.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+module Command
+  class PsStop < Base
+    NAME = "ps:stop"
+    OPTIONS = [
+      app_option(required: true),
+      workload_option,
+      replica_option,
+      location_option,
+      wait_option("workload to not be ready")
+    ].freeze
+    DESCRIPTION = "Stops workloads in app"
+    LONG_DESCRIPTION = <<~DESC
+      - Stops workloads in app
+    DESC
+    EXAMPLES = <<~EX
+      ```sh
+      # Stops all workloads in app.
+      cpflow ps:stop -a $APP_NAME
+
+      # Stops a specific workload in app.
+      cpflow ps:stop -a $APP_NAME -w $WORKLOAD_NAME
+
+      # Stops a specific replica of a workload.
+      cpflow ps:stop -a $APP_NAME -w $WORKLOAD_NAME -r $REPLICA_NAME
+      ```
+    EX
+
+    def call
+      workload = config.options[:workload]
+      replica = config.options[:replica]
+      if replica
+        stop_replica(workload, replica)
+      else
+        workloads = [workload] if workload
+        workloads ||= config[:app_workloads] + config[:additional_workloads]
+
+        stop_workloads(workloads)
+      end
+    end
+
+    private
+
+    def stop_workloads(workloads)
+      workloads.each do |workload|
+        step("Stopping workload '#{workload}'") do
+          cp.set_workload_suspend(workload, true)
+        end
+      end
+
+      wait_for_workloads_not_ready(workloads) if config.options[:wait]
+    end
+
+    def stop_replica(workload, replica)
+      step("Stopping replica '#{replica}'", retry_on_failure: true) do
+        cp.stop_workload_replica(workload, replica, location: config.location)
+      end
+
+      wait_for_replica_not_ready(workload, replica) if config.options[:wait]
+    end
+
+    def wait_for_workloads_not_ready(workloads)
+      progress.puts
+
+      workloads.each do |workload|
+        step("Waiting for workload '#{workload}' to not be ready", retry_on_failure: true) do
+          cp.workload_deployments_ready?(workload, location: config.location, expected_status: false)
+        end
+      end
+    end
+
+    def wait_for_replica_not_ready(workload, replica)
+      progress.puts
+
+      step("Waiting for replica '#{replica}' to not be ready", retry_on_failure: true) do
+        result = cp.fetch_workload_replicas(workload, location: config.location)
+        !result["items"].include?(replica)
+      end
+    end
+  end
+end

data/lib/command/ps_wait.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module Command
+  class PsWait < Base
+    NAME = "ps:wait"
+    OPTIONS = [
+      app_option(required: true),
+      workload_option,
+      location_option
+    ].freeze
+    DESCRIPTION = "Waits for workloads in app to be ready after re-deployment"
+    LONG_DESCRIPTION = <<~DESC
+      - Waits for workloads in app to be ready after re-deployment
+    DESC
+    EXAMPLES = <<~EX
+      ```sh
+      # Waits for all workloads in app.
+      cpflow ps:wait -a $APP_NAME
+
+      # Waits for a specific workload in app.
+      cpflow ps:swait -a $APP_NAME -w $WORKLOAD_NAME
+      ```
+    EX
+
+    def call # rubocop:disable Metrics/MethodLength
+      @workloads = [config.options[:workload]] if config.options[:workload]
+      @workloads ||= config[:app_workloads] + config[:additional_workloads]
+
+      @workloads.reverse_each do |workload|
+        if cp.workload_suspended?(workload)
+          progress.puts("Workload '#{workload}' is suspended. Skipping...")
+        else
+          step("Waiting for workload '#{workload}' to be ready", retry_on_failure: true) do
+            cp.workload_deployments_ready?(workload, location: config.location, expected_status: true)
+          end
+        end
+      end
+    end
+  end
+end