awes_cli 0.0.0 → 0.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 77e5eefdce88cb9bf280df1c12b03072a83c0e435383410e1a41b1074caacc94
-  data.tar.gz: 63a84f12bdf2874d0fa5a9ab5b8a205a3f04208284d9c918ecea658354526732
+  metadata.gz: be520b19b3caa21d29f8208eab535c95e728180b7c74e77feb477263933587f4
+  data.tar.gz: 1233834f30464477ec72c618857d980bd68c2d750ce5c8cbd508a98db8876371
 SHA512:
-  metadata.gz: 28699c477813def96fef6259122461590ebea3786b9fecfa41561b7cccc8a13263fdb6edb395b8710a79661a71073ccf1484b21414d85da1ffd9fcb34ca731ea
-  data.tar.gz: c1b67ca8631e92766835a395e76c50163b7394115cebaf6e89b8a0b1bf4ad971e64e84861a4ff8cc5e3c7c154c594bd0dd3e22567f9e8bc55df3313d140965a8
+  metadata.gz: 655d71c2e1f7563b8bd6f414f2e3ca73ee5a4c50713ab9009dc9a5709a30866682494cd1d2ddf6edd8bc8d8fa2e47d39cf2ae2c57c12bdf76ed45accd4671fa4
+  data.tar.gz: e4133b8c0233f5f89352e968daeaad5f31f837b1f0adeed939b83bd986e9d476dcf5e6f6ab8b52d6c56d4ad559754e44e656cad31dbdb21ae1203f15fc39d4b8

data/bin/ather

@@ -1,5 +1,25 @@
 #!/usr/bin/env ruby
 require 'build_app_image'
+require 'deploy_app'
+require 'deploy_config'
 
-build = BuildAppImage.new
-build.main
+primary_cmd = ARGV[0]
+secondary_cmd = ARGV[1]
+
+if primary_cmd.nil?
+  puts 'Run `ather build|deploy`'
+  exit 1
+end
+
+if primary_cmd == 'deploy' && secondary_cmd.nil?
+  puts 'Run `ather deploy app|config`'
+  exit 1
+end
+
+if primary_cmd == 'build'
+  BuildAppImage.new.main
+elsif primary_cmd == 'deploy' && secondary_cmd == 'app'
+  DeployApp.new.main
+elsif primary_cmd == 'deploy' && secondary_cmd == 'config'
+  DeployConfig.new.main
+end

data/lib/build_app_image.rb

@@ -17,7 +17,11 @@ class BuildAppImage
       require_options_keys = [:project_id, :app_name]
 
       opt_parser = OptionParser.new do |opts|
-        opts.banner = "Usage: ./deploy/build.rb [options]"
+        opts.banner = "Usage: ather build [options]"
+
+        opts.on("--image-tag IMAGE_TAG", "Image tag to deploy [optional]") do |val|
+          options[:image_tag] = val
+        end
 
         opts.on("--project-id PROJECT_ID", "GCP project id [required or set env var PROJECT_ID]") do |val|
           options[:project_id] = val
@@ -60,13 +64,19 @@ class BuildAppImage
     def build_image(gcr_url, options)
       commit_image_tag = ENV['CI_COMMIT_SHORT_SHA'] || `git rev-parse --short HEAD`
       time_image_tag=`TZ=IST-5:30 date +'%Y.%m.%d.%HH.%MM.%SS'`
-      image_url = "#{gcr_url}:#{commit_image_tag}".strip
+      commit_image_url = "#{gcr_url}:#{commit_image_tag}".strip
+      time_image_url = "#{gcr_url}:#{time_image_tag}".strip
+
       project_id = options[:project_id]
 
-      shell_cmd("gcloud builds submit --project #{project_id} --timeout=20m --tag #{image_url} || true")
+      shell_cmd("gcloud builds submit --project #{project_id} --timeout=20m --tag #{time_image_url} || true")
+
+      shell_cmd("gcloud container images add-tag --quiet #{time_image_url} #{commit_image_url}")
+
+      opt_image_url = options[:image_tag] && "#{gcr_url}:#{options[:image_tag]}"
 
-      shell_cmd("gcloud container images add-tag --quiet #{image_url} #{gcr_url}:#{time_image_tag}")
+      opt_image_url && shell_cmd("gcloud container images add-tag --quiet #{time_image_url} #{opt_image_url}")
 
-      image_url
+      opt_image_url
     end
 end

data/lib/deploy_app.rb

@@ -0,0 +1,135 @@
+#!/usr/bin/env ruby
+require 'optparse'
+
+class DeployApp
+  def main
+    options = get_options
+    take_approval(options)
+
+    gcr_url = "gcr.io/#{options[:project_id]}/#{options[:app_name]}"
+
+    if(!options[:image_tag].nil?)
+      image_url = validate_image_tag_exists?(gcr_url, options[:image_tag])
+    else
+      image_url = build_image(gcr_url, options)
+    end
+
+    deploy_image(gcr_url, image_url, options)
+  end
+
+  private
+    def get_options
+      options = {}
+
+      require_options_keys = [:project_id, :app_name, :namespace, :cluster, :cluster_location]
+
+      opt_parser = OptionParser.new do |opts|
+        opts.banner = "Usage: ather deploy app [options]"
+
+        opts.on("-y", "Auto approve") do |flag|
+          options[:approve] = flag
+        end
+
+        opts.on("--image-tag IMAGE_TAG", "Image tag to deploy [optional]") do |val|
+          options[:image_tag] = val
+        end
+
+        opts.on("--project-id PROJECT_ID", "GCP project id [required or set env var PROJECT_ID]") do |val|
+          options[:project_id] = val
+        end
+
+        opts.on("--app-name APP_NAME", "K8s app name [required or set env var APP_NAME]") do |val|
+          options[:app_name] = val
+        end
+
+        opts.on("--namespace NAMESPACE", "K8s app namespace [required or set env var NAMESPACE]") do |val|
+          options[:namespace] = val
+        end
+
+        opts.on("--cluster CLUSTER", "App name [required or set env var CLUSTER]") do |val|
+          options[:cluster] = val
+        end
+
+        opts.on("--cluster-location CLUSTER_LOCATION", "App name [required or set env var CLUSTER_LOCATION]") do |val|
+          options[:cluster_location] = val
+        end
+
+        opts.on("-h", "--help", "Prints this help") do
+          puts opts
+          exit
+        end
+      end
+
+      opt_parser.parse!
+
+      require_options_keys.each do |key|
+        options[key] ||= ENV[key.to_s.upcase]
+        if options[key].nil?
+          puts "ERROR: Required option --#{key.to_s.gsub('_', '-')} or set env var #{key.to_s.upcase} \n"
+          opt_parser.parse! %w[--help]
+          exit 1
+        end
+      end
+
+      options
+    end
+
+    def shell_cmd(cmd)
+      puts "EXECUTING : \e[32m\e[1m#{cmd}\e[22m\e[0m"
+      op = system(cmd)
+      if !op
+        puts "FAILED : \e[31m\e[1m#{cmd}\e[22m\e[0m"
+        exit 1
+      end
+      puts "SUCCESS : \e[32m\e[1m#{cmd}\e[22m\e[0m"
+    end
+
+    def take_approval(options)
+      puts "\e[32m\e[1mDeployment parameter details \e[22m\e[0m"
+      options.each do |key, val|
+        puts "\e[1m#{key} = \e[31m#{val}\e[0m \e[22m"
+      end
+
+      return if(options[:approve])
+      puts 'Are you sure to want to deploy as per the above deployment parameters?'
+      confirmation = STDIN.gets.strip
+      return if(confirmation == 'y' || confirmation == 'Y')
+      exit
+    end
+
+    def validate_image_tag_exists?(gcr_url, image_tag)
+      image_url = "#{gcr_url}:#{image_tag}"
+      shell_cmd("gcloud container images describe #{image_url}")
+      image_url
+    end
+
+    def build_image(gcr_url, options)
+      commit_image_tag = ENV['CI_COMMIT_SHORT_SHA'] || `git rev-parse --short HEAD`
+      time_image_tag=`TZ=IST-5:30 date +'%Y.%m.%d.%HH.%MM.%SS'`
+      commit_image_url = "#{gcr_url}:#{commit_image_tag}".strip
+      time_image_url = "#{gcr_url}:#{time_image_tag}".strip
+
+      project_id = options[:project_id]
+
+      shell_cmd("gcloud builds submit --project #{project_id} --timeout=20m --tag #{time_image_url} || true")
+
+      shell_cmd("gcloud container images add-tag --quiet #{time_image_url} #{commit_image_url}")
+
+      commit_image_url
+    end
+
+    def deploy_image(gcr_url, image_url, options)
+      shell_cmd(
+        "gcloud container clusters get-credentials #{options[:cluster]} " \
+        "--region #{options[:cluster_location]} " \
+        "--project #{options[:project_id]} "
+      )
+
+      app_name = options[:app_name]
+      shell_cmd("kubectl set image deployment #{app_name} #{app_name}=#{image_url} -n #{options[:namespace]}")
+
+      shell_cmd("kubectl rollout status deployment/#{app_name} -n #{options[:namespace]}")
+
+      shell_cmd("gcloud container images add-tag --quiet #{image_url} #{gcr_url}:current")
+    end
+end

data/lib/deploy_config.rb

@@ -0,0 +1,210 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'json'
+require 'yaml'
+require 'fileutils'
+
+class DeployConfig
+  def main
+    options = get_options
+
+    take_approval_deploy_params(options)
+
+    take_approval_app_env(options)
+
+    gcr_url = "gcr.io/#{options[:project_id]}/#{options[:app_name]}"
+
+    deploy_config(gcr_url, options)
+  end
+
+  private
+
+    def get_options
+      options = {}
+
+      require_options_keys = [:project_id, :app_name, :namespace, :cluster, :cluster_location, :kuztomize_env]
+
+      opt_parser = OptionParser.new do |opts|
+        opts.banner = "Usage: ather deploy config [options]"
+
+        opts.on("-y", "Auto approve") do |flag|
+          options[:approve] = flag
+        end
+
+        opts.on("--image-tag IMAGE_TAG", "Image tag to deploy [optional]") do |val|
+          options[:image_tag] = val
+        end
+
+        opts.on("--project-id PROJECT_ID", "GCP project id [required or set env var PROJECT_ID]") do |val|
+          options[:project_id] = val
+        end
+
+        opts.on("--app-name APP_NAME", "K8s app name [required or set env var APP_NAME]") do |val|
+          options[:app_name] = val
+        end
+
+        opts.on("--namespace NAMESPACE", "K8s app namespace [required or set env var NAMESPACE]") do |val|
+          options[:namespace] = val
+        end
+
+        opts.on("--cluster CLUSTER", "App name [required or set env var CLUSTER]") do |val|
+          options[:cluster] = val
+        end
+
+        opts.on("--cluster-location CLUSTER_LOCATION", "App name [required or set env var CLUSTER_LOCATION]") do |val|
+          options[:cluster_location] = val
+        end
+
+        opts.on("--kuztomize-env KUZTOMIZE_ENV", "App name [required or set env var KUZTOMIZE_ENV]") do |val|
+          options[:kuztomize_env] = val
+        end
+
+        opts.on("-h", "--help", "Prints this help") do
+          puts opts
+          exit
+        end
+      end
+
+      opt_parser.parse!
+
+      require_options_keys.each do |key|
+        options[key] ||= ENV[key.to_s.upcase]
+        if options[key].nil?
+          puts "ERROR: Required option --#{key.to_s.gsub('_', '-')} or set env var #{key.to_s.upcase} \n"
+          opt_parser.parse! %w[--help]
+          exit 1
+        end
+      end
+
+      options
+    end
+
+    def shell_cmd(cmd)
+      puts "EXECUTING : \e[32m\e[1m#{cmd}\e[22m\e[0m"
+      op = system(cmd)
+      if !op
+        puts "FAILED : \e[31m\e[1m#{cmd}\e[22m\e[0m"
+        exit 1
+      end
+      puts "SUCCESS : \e[32m\e[1m#{cmd}\e[22m\e[0m"
+    end
+
+    def take_approval_deploy_params(options)
+      puts "\e[32m\e[1mDeployment parameter details \e[22m\e[0m"
+      options.each do |key, val|
+        puts "\e[1m#{key} = \e[31m#{val}\e[0m \e[22m"
+      end
+
+      return if(options[:approve])
+      puts 'Are you sure to want to deploy as per the above deployment parameters? (y/N)'
+      confirmation = STDIN.gets.strip
+      return if(confirmation == 'y' || confirmation == 'Y')
+      exit
+    end
+
+    def take_approval_app_env(options)
+      override_env_filepath = "./deploy/env/#{options[:kuztomize_env]}.sh"
+      return if !File.file?(override_env_filepath)
+      puts "\e[32m\e[1mOverriding Application enviroment variables \e[22m\e[0m"
+
+      app_envs = `cat #{override_env_filepath}`.gsub('export ', "\n")
+      puts "\e[34m\e[1m#{app_envs}\e[22m\e[0m"
+
+      return if(options[:approve])
+      puts 'Are you sure to want to deploy the above application enviroment variables? (y/N)'
+      confirmation = STDIN.gets.strip
+      return if(confirmation == 'y' || confirmation == 'Y')
+      exit
+    end
+
+    def validate_image_tag_exists?(image_url)
+      shell_cmd("gcloud container images describe #{image_url}")
+      image_url
+    end
+
+    def deploy_config(gcr_url, options)
+      shell_cmd(
+        "gcloud container clusters get-credentials #{options[:cluster]} " \
+        "--region #{options[:cluster_location]} " \
+        "--project #{options[:project_id]} "
+      )
+
+      spec = app_container_spec(options[:app_name], options[:namespace]) || {}
+
+      override_envs = override_envs(options)
+      k8s_deployment_envs = k8s_deployment_envs(options)
+
+      yaml_file = `kustomize build deploy/overlays/#{options[:kuztomize_env]}`
+
+      yaml_file.scan(/\$[A-Z_]+/).each do |env_name|
+        env_key = env_name[1..-1]
+        env_val = override_envs[env_key] || ENV[env_key] || k8s_deployment_envs[env_key]
+        next if env_val.nil?
+        yaml_file[env_name] = env_val
+      end
+
+      gcr_url = "gcr.io/#{options[:project_id]}/#{options[:app_name]}"
+
+      opt_image_url = options[:image_tag] && "#{gcr_url}:#{options[:image_tag]}"
+
+      image_url = opt_image_url || spec[:image] || "#{gcr_url}:current"
+      validate_image_tag_exists?(image_url)
+
+      yaml_file['$IMAGE_URL'] = image_url
+
+      FileUtils.mkdir_p('./tmp')
+
+      File.write('./tmp/final.yaml', yaml_file)
+
+      puts "\e[36m\e[1m#{yaml_file}\e[22m\e[0m"
+
+      shell_cmd("kubectl apply -f ./tmp/final.yaml")
+
+      shell_cmd("kubectl rollout status deployment/#{options[:app_name]} -n #{options[:namespace]}")
+    end
+
+    def override_envs(options)
+      override_env_filepath = "./deploy/env/#{options[:kuztomize_env]}.yaml"
+      override_envs = {}
+
+      if File.file?(override_env_filepath)
+        env = YAML.load_file(override_env_filepath, symbolize_names: true)
+        env[:env].each do |env|
+          override_envs[env[:name]] = env[:value]
+        end
+      end
+
+      override_envs
+    end
+
+    def k8s_deployment_envs(options)
+      spec = app_container_spec(options[:app_name], options[:namespace])
+
+      return {} if spec.nil?
+
+      k8s_deployment_envs = {}
+
+      spec[:env].each do |env|
+        k8s_deployment_envs[env[:name]] = env[:value]
+      end
+
+      k8s_deployment_envs
+    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 deploy_containers == ''
+      JSON.parse(deploy_containers, symbolize_names: true).find { |container| container[:name] == app_name }
+    end
+
+    def app_container_env_vars_cmd(spec)
+      env_export_cmd = ''
+      spec[:env].each do |env|
+        env_var_val = ENV.fetch(env[:name]) { env[:value] }
+        env_export_cmd += "export #{env[:name]}=#{env_var_val} && "
+      end
+
+      env_export_cmd
+    end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awes_cli
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.1
 platform: ruby
 authors:
 - Umar Siddiqui
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-07-20 00:00:00.000000000 Z
+date: 2021-07-22 00:00:00.000000000 Z
 dependencies: []
 description: Ather cli tool
 email: umar.siddiqui@atherenergy.com
@@ -20,6 +20,8 @@ files:
 - bin/ather
 - lib/awes_cli.rb
 - lib/build_app_image.rb
+- lib/deploy_app.rb
+- lib/deploy_config.rb
 homepage: https://rubygems.org/gems/awes_cli
 licenses:
 - MIT