lib/core/controlplane.rb in cpl-0.4.1 vs lib/core/controlplane.rb in cpl-0.5.0

@@ -5,77 +5,137 @@
 
   def initialize(config)
     @config = config
     @api = ControlplaneApi.new
     @gvc = config.app
-    @org = config[:cpln_org]
+    @org = config.org
   end
 
+  # profile
+
+  def profile_switch(profile)
+    ENV["CPLN_PROFILE"] = profile
+  end
+
+  def profile_exists?(profile)
+    cmd = "cpln profile get #{profile} -o yaml"
+    perform_yaml(cmd).length.positive?
+  end
+
+  def profile_create(profile, token)
+    cmd = "cpln profile create #{profile} --token #{token}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
+    perform!(cmd)
+  end
+
+  def profile_delete(profile)
+    cmd = "cpln profile delete #{profile}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
+    perform!(cmd)
+  end
+
   # image
 
   def image_build(image, dockerfile:, push: true)
     cmd = "cpln image build --org #{org} --name #{image} --dir #{config.app_dir} --dockerfile #{dockerfile}"
     cmd += " --push" if push
     perform!(cmd)
   end
 
-  def image_query(app_name = config.app)
-    cmd = "cpln image query --org #{org} -o yaml --max -1 --prop repository=#{app_name}"
+  def image_query(app_name = config.app, org_name = config.org)
+    cmd = "cpln image query --org #{org_name} -o yaml --max -1 --prop repository=#{app_name}"
     perform_yaml(cmd)
   end
 
   def image_delete(image)
     api.image_delete(org: org, image: image)
   end
 
+  def image_login(org_name = config.org)
+    cmd = "cpln image docker-login --org #{org_name}"
+    cmd += " > /dev/null 2>&1" if Shell.tmp_stderr
+    perform!(cmd)
+  end
+
+  def image_pull(image)
+    cmd = "docker pull #{image}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
+    perform!(cmd)
+  end
+
+  def image_tag(old_tag, new_tag)
+    cmd = "docker tag #{old_tag} #{new_tag}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
+    perform!(cmd)
+  end
+
+  def image_push(image)
+    cmd = "docker push #{image}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
+    perform!(cmd)
+  end
+
   # gvc
 
+  def fetch_gvcs
+    api.gvc_list(org: org)
+  end
+
   def gvc_query(app_name = config.app)
     # When `match_if_app_name_starts_with` is `true`, we query for any gvc containing the name,
     # otherwise we query for a gvc with the exact name.
     op = config.current[:match_if_app_name_starts_with] ? "~" : "="
 
     cmd = "cpln gvc query --org #{org} -o yaml --prop name#{op}#{app_name}"
     perform_yaml(cmd)
   end
 
-  def fetch_gvc(a_gvc = gvc)
-    api.gvc_get(gvc: a_gvc, org: org)
+  def fetch_gvc(a_gvc = gvc, a_org = org)
+    api.gvc_get(gvc: a_gvc, org: a_org)
   end
 
   def fetch_gvc!(a_gvc = gvc)
     gvc_data = fetch_gvc(a_gvc)
     return gvc_data if gvc_data
 
-    Shell.abort("Can't find GVC '#{gvc}', please create it with 'cpl setup gvc -a #{config.app}'.")
+    raise "Can't find GVC '#{gvc}', please create it with 'cpl setup gvc -a #{config.app}'."
   end
 
   def gvc_delete(a_gvc = gvc)
     api.gvc_delete(gvc: a_gvc, org: org)
   end
 
   # workload
 
+  def fetch_workloads(a_gvc = gvc)
+    api.workload_list(gvc: a_gvc, org: org)
+  end
+
+  def fetch_workloads_by_org(a_org = org)
+    api.workload_list_by_org(org: a_org)
+  end
+
   def fetch_workload(workload)
     api.workload_get(workload: workload, gvc: gvc, org: org)
   end
 
   def fetch_workload!(workload)
     workload_data = fetch_workload(workload)
     return workload_data if workload_data
 
-    Shell.abort("Can't find workload '#{workload}', please create it with 'cpl setup #{workload} -a #{config.app}'.")
+    raise "Can't find workload '#{workload}', please create it with 'cpl setup #{workload} -a #{config.app}'."
   end
 
   def workload_get_replicas(workload, location:)
     cmd = "cpln workload get-replicas #{workload} #{gvc_org} --location #{location} -o yaml"
     perform_yaml(cmd)
   end
 
   def workload_set_image_ref(workload, container:, image:)
     cmd = "cpln workload update #{workload} #{gvc_org}"
-    cmd += " --set spec.containers.#{container}.image=/org/#{config[:cpln_org]}/image/#{image}"
+    cmd += " --set spec.containers.#{container}.image=/org/#{config.org}/image/#{image}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
     perform!(cmd)
   end
 
   def workload_set_suspend(workload, value)
     data = fetch_workload!(workload)
@@ -83,10 +143,11 @@
     apply(data)
   end
 
   def workload_force_redeployment(workload)
     cmd = "cpln workload force-redeployment #{workload} #{gvc_org}"
+    cmd += " > /dev/null" if Shell.tmp_stderr
     perform!(cmd)
   end
 
   def workload_delete(workload)
     cmd = "cpln workload delete #{workload} #{gvc_org}"
@@ -124,16 +185,21 @@
     api.log_get(org: org, gvc: gvc, workload: workload, from: from, to: to)
   end
 
   # apply
 
-  def apply(data)
+  def apply(data) # rubocop:disable Metrics/MethodLength
     Tempfile.create do |f|
       f.write(data.to_yaml)
       f.rewind
       cmd = "cpln apply #{gvc_org} --file #{f.path} > /dev/null"
-      perform!(cmd)
+      if Shell.tmp_stderr
+        cmd += " 2> #{Shell.tmp_stderr.path}"
+        perform(cmd)
+      else
+        perform!(cmd)
+      end
     end
   end
 
   private
 
@@ -145,10 +211,10 @@
     system(cmd) || exit(false)
   end
 
   def perform_yaml(cmd)
     result = `#{cmd}`
-    $?.success? ? YAML.safe_load(result) : exit(false) # rubocop:disable Style/SpecialGlobalVars
+    $CHILD_STATUS.success? ? YAML.safe_load(result) : exit(false)
   end
 
   def gvc_org
     "--gvc #{gvc} --org #{org}"
   end