awes_cli 0.0.22 → 0.0.23

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ed029c05bdeaa99e0163aa5c2e87d9666e24b6007f1b1ce583a24dab8ba695bd
-  data.tar.gz: 7fce07a3fe1255d784255844bd4b7476fb57c2e36dab632a703ff99ba12efdc2
+  metadata.gz: 7c56c2fd31133f9f9099a96c29b891f1cf30e206b9c4c5380e15d434037b2d9e
+  data.tar.gz: e87b7f33e19820c970ea30217cbf5d0e0e972b8428b76bcd2eeb81f572e9f1cc
 SHA512:
-  metadata.gz: 3f444a6751fda720334e53a5bba33a394fbbccc55686fbdd9ebe4b6d174fca3d6a4b07efd38183f691f281ea90a60ff93194cd699efd13baf1d1c481b67f0113
-  data.tar.gz: 9044707bcab97af14587cdd8e4bf996bfae89ab2769c042891afd4f4f5cd5488cecc17fb7681339840501ce6e3fabc81511667c332bfb3a37fc5fcd4587c3779
+  metadata.gz: 31091850270124ce71745f6c5b56e93329a668c2135c1a2c2b5fbd4fdd49de76904746e5247d3e9771d5a13de8011cb2c24c6e1f2833e3d82683eddc59523c79
+  data.tar.gz: 3c6c403da51706902d567c2f000912573c4fb07188bec6d3c61490d4059ea6468552dae6e5388c505daca3994f996044228b2f414b08ed401d241ee8df5a5db2

data/lib/deploy_app.rb

@@ -2,12 +2,17 @@
 require 'optparse'
 
 class DeployApp
+  attr_accessor :workload_hash
+
   def main
     options = get_options
     take_approval(options)
 
     gcr_url = "asia.gcr.io/#{options[:project_id]}/#{options[:image_repo] || options[:app_name]}"
 
+    self.workload_hash = fetch_worload_yaml(options[:app_name], options[:namespace])
+    throw "Workload does not exists name: #{option[:app_name]} namespace: #{option[:namespace]}" if workload_hash.nil?
+
     if(!options[:image_tag].nil?)
       image_url = validate_image_tag_exists?(gcr_url, options[:image_tag])
     else
@@ -130,10 +135,17 @@ class DeployApp
       )
 
       app_name = options[:app_name]
-      shell_cmd("kubectl set image deployment #{app_name} #{app_name}=#{image_url} -n #{options[:namespace]}")
+      shell_cmd("kubectl set image #{workload_hash['kind']} #{app_name} #{app_name}=#{image_url} -n #{options[:namespace]}")
 
-      shell_cmd("kubectl rollout status deployment/#{app_name} -n #{options[:namespace]}")
+      shell_cmd("kubectl rollout status #{workload_hash['kind']} #{app_name} -n #{options[:namespace]}")
 
       shell_cmd("gcloud container images add-tag --quiet #{image_url} #{gcr_url}:current")
     end
+
+    def fetch_worload_yaml(app_name, namespace)
+      json_deploy = `kubectl get deployment -n #{namespace} #{app_name} -o json`
+      return JSON.parse(json_deploy) if(json_deploy != "")
+      json_statefulset = `kubectl get statefulset -n #{namespace} #{app_name} -o json`
+      return JSON.parse(json_statefulset) if(json_statefulset != "")
+    end
 end

data/lib/deploy_config.rb

@@ -6,6 +6,8 @@ require 'yaml'
 require 'fileutils'
 
 class DeployConfig
+  attr_accessor :workload_hash
+
   def main
     options = get_options
 
@@ -162,7 +164,7 @@ class DeployConfig
         yaml_file.gsub!(/\$#{env_key}\s*$/, "'#{env_val}'")
       end
 
-      gcr_url = "gcr.io/#{options[:project_id]}/#{options[:image_repo] || options[:app_name]}"
+      gcr_url = "asia.gcr.io/#{options[:project_id]}/#{options[:image_repo] || options[:app_name]}"
       opt_image_url = options[:image_tag] && "#{gcr_url}:#{options[:image_tag]}"
 
       image_url = "#{gcr_url}:current"
@@ -183,7 +185,9 @@ class DeployConfig
 
       shell_cmd("kubectl apply -f ./tmp/final.yaml")
 
-      shell_cmd("kubectl rollout status deployment/#{options[:app_name]} -n #{options[:namespace]}")
+      self.workload_hash = fetch_worload_yaml(options[:app_name], options[:namespace])
+
+      shell_cmd("kubectl rollout status #{workload_hash['kind']}/#{options[:app_name]} -n #{options[:namespace]}")
     end
 
     def override_envs(options)
@@ -215,7 +219,8 @@ class DeployConfig
     end
 
     def app_container_spec(app_name, namespace)
-      deploy_containers = `kubectl get deployment -n #{namespace} #{app_name} --output=jsonpath={.spec.template.spec.containers}`
+      return nil if workload_hash.nil?
+      deploy_containers = `kubectl get #{workload_hash['kind']} -n #{namespace} #{app_name} --output=jsonpath={.spec.template.spec.containers}`
       return nil if deploy_containers == ''
       JSON.parse(deploy_containers, symbolize_names: true).find { |container| container[:name] == app_name }
     end
@@ -229,4 +234,11 @@ class DeployConfig
 
       env_export_cmd
     end
+
+    def fetch_worload_yaml(app_name, namespace)
+      json_deploy = `kubectl get deployment -n #{namespace} #{app_name} -o json`
+      return JSON.parse(json_deploy) if(json_deploy != "")
+      json_statefulset = `kubectl get statefulset -n #{namespace} #{app_name} -o json`
+      return JSON.parse(json_statefulset) if(json_statefulset != "")
+    end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awes_cli
 version: !ruby/object:Gem::Version
-  version: 0.0.22
+  version: 0.0.23
 platform: ruby
 authors:
 - Umar Siddiqui
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-06-03 00:00:00.000000000 Z
+date: 2022-08-03 00:00:00.000000000 Z
 dependencies: []
 description: Ather cli tool
 email: umar.siddiqui@atherenergy.com