rockette 0.0.2 → 0.0.6

data/lib/rockette/controller/exporter.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+require_relative "../text_helper"
+require_relative "../commands/export"
+
+module Rockette
+  # Export and download APEX application
+  class Exporter
+    include TextHelper
+
+    def initialize
+      @pastel = Pastel.new
+      @prompt = TTY::Prompt.new
+      @spinner = TTY::Spinner.new
+    end
+
+    def launch!
+      @conf = Psych.load(File.read(CONF))
+      puts padder("Choose an application to download.")
+      puts
+
+      # input/action loop
+      loop do
+        enviros = Rockette::Viewer.new.environments
+        list = list_builder(enviros)
+        action = @prompt.select("Which environment is the app found?", list)
+        break if action == list.length
+
+        apps_url = enviros[action - 1]["deployment_api"]
+        cred = enviros[action - 1]["web_cred"]
+        choose_app(apps_url, cred)
+      end
+    end
+
+    def do_export(app_id, apps_url, cred)
+      response = @prompt.yes?("Would you like to enter a filename for the export?")
+      if response == true
+        file = @prompt.ask("Please enter your desired filename:")
+        options = Thor::CoreExt::HashWithIndifferentAccess.new "app_id" => app_id, "url" => apps_url, "cred" => cred,
+                                                               "file" => file, "force" => true
+      else
+        puts padder("Saving under default file name: f#{app_id}.sql")
+        options = Thor::CoreExt::HashWithIndifferentAccess.new "app_id" => app_id, "url" => apps_url, "cred" => cred, "force" => true
+      end
+      Rockette::Commands::Export.new(options).execute
+      puts
+    end
+
+    protected
+
+    def choose_app(apps_url, cred)
+      loop do
+        apps = Rockette::Viewer.new.applications(apps_url, cred)
+        list = list_builder(apps)
+        action = @prompt.slider("Download application => ", list, default: 1)
+        break if action == list.length
+
+        app_id = apps[action - 1]["application_id"]
+        do_export(app_id, apps_url, cred)
+      end
+    end
+
+    def list_builder(array)
+      names = [] # Start building selection list
+      if array[0].key?("name")
+        array.each { |n| names << n["name"] }
+      else
+        array.each { |n| names << "#{n["application_name"]}  App ID: #{n["application_id"]}" }
+      end
+      names << "Go Back"
+      names.map.with_index { |n, x| [n, x + 1] }.to_h
+    end
+  end
+end

data/lib/rockette/controller/viewer.rb

@@ -0,0 +1,154 @@
+# frozen_string_literal: true
+
+require_relative "../text_helper"
+
+module Rockette
+  # View resources
+  class Viewer
+    include TextHelper
+
+    def initialize
+      @config = TTY::Config.new
+      @config.append_path APP_PATH
+      @config.read
+      @conf = Psych.load(File.read(CONF))
+      @body = @conf["token_body"]
+      @hdrs = @conf["token_hdrs"]
+      #@hdrs["Authorization"] = @conf["web_creds"]["controller_cred"]
+      token_url = @conf["rockette"]["controller_url"].sub!('deploy/', '')
+      @token = get_token(token_url, "controller_cred")
+      @hdrs["Authorization"] = "Bearer " + @token
+      @pastel = Pastel.new
+      @prompt = TTY::Prompt.new
+      @spinner = TTY::Spinner.new # ("[:spinner] Loading APEX environments ...", format: pulse_2)
+      @view_actions = { "🏔  APEX Environments" => 1, "🎭 Registered Applications" => 2,
+                        "🌎 Applications by Environment" => 3, "⬅️  Go Back" => 4 }
+    end
+
+    def self.config
+      @config ||= self.class.new.config
+    end
+
+    def launch!
+      puts padder("You can view environments or registered applications")
+      puts
+
+      # input/action loop
+      loop do
+        action = @prompt.select("Which would you like to see?", @view_actions)
+        break if action == 4
+
+        do_action(action)
+      end
+    end
+
+    def do_action(action)
+      case action
+      when 1
+        puts
+        environments unless @table_env
+        spinner(0.5)
+        puts @table_env.render(:unicode, resize: true, border: { style: :yellow })
+        puts
+      when 2
+        puts
+        registered unless @table_reg
+        puts @table_reg.render(:unicode, resize: true, border: { style: :yellow })
+        puts
+      when 3
+        puts
+        puts "This can take a while...hang tight!"
+        puts
+        all_apps unless @table_all_apps
+        puts @table_all_apps.render(:unicode, resize: true, border: { style: :yellow })
+        puts
+      else
+        puts "\nI don't understand that command.\n\n"
+      end
+    end
+
+    def add_web_cred(env_cred, env_name)
+      puts "You are attempting to access a resource that requires authentication."
+      puts "Please enter the OAuth client credentials for the #{env_name} environment."
+      user = @prompt.ask("User:")
+      pass = @prompt.ask("Pass:")
+      basey = 'Basic ' + Base64.encode64(user + ":" + pass).tr("\n", "")
+      @config.set(:web_creds, env_cred.to_sym, value: basey)
+      @config.write(force: true)
+      refresh_conf
+    end
+
+    def ape_e_i(uri, headers = @hdrs)
+      response = Rester.new(url: uri, headers: headers).rest_try
+      bail unless response
+      abort padder("#{uri} didn't work. Received: #{response.code}") unless response.code == 200
+      response
+    end
+
+    def applications(url, cred)
+      headers = @hdrs
+      #@hdrs["Authorization"] = @conf["web_creds"][cred]
+      @token = get_token(url, cred)
+      @hdrs["Authorization"] = "Bearer " + @token
+      uri = "#{url}deploy/apps/"
+
+      response = ape_e_i(uri, @hdrs)
+      @hdrs = headers
+      JSON.parse(response.body)["items"]
+    end
+
+    def environments
+      uri = "#{@conf["rockette"]["controller_url"]}deploy/environments/"
+      response = ape_e_i(uri)
+      @table_env = TTY::Table.new(header: ["Environment Name", "API", "Domain", "Owner", "Workspace", "Web Cred"])
+      items = JSON.parse(response.body)["items"]
+      items.each { |h| @table_env << [h["name"], h["deployment_api"], h["domain"], h["owner"], h["workspace"], h["web_cred"]] }
+    end
+
+    def get_token(url, cred)
+      @hdrs["Authorization"] = @conf["web_creds"][cred]
+      token_url = url + "oauth/token"
+      response = Rester.new(headers: @hdrs, meth: "Post", params: @body, url: token_url).rest_try
+      return JSON.parse(response.body)["access_token"]
+    end
+
+    def registered
+      uri = "#{@conf["rockette"]["controller_url"]}deploy/registered_apps/"
+      response = ape_e_i(uri)
+      @table_reg = TTY::Table.new(header: ["Registered Name", "Source App ID", "Source URI", "Target App ID",
+                                           "Target URI"])
+      JSON.parse(response.body)["items"].each do |h|
+        @table_reg << [h["registered_name"], h["src_app_id"], h["src_url"], h["tgt_app_id"], h["tgt_url"]]
+      end
+    end
+
+    def all_apps
+      environments unless @table_env
+      @table_all_apps = TTY::Table.new(header: ["Environment Name", "Application Name", "Application ID"])
+      @table_env.each do |env|
+        next if env[0] == "Environment Name"
+
+        creds = Hash.new
+        creds = @conf["web_creds"]
+        add_web_cred(env[5], env[0]) unless creds.has_key?(env[5])
+
+        apps = applications(env[1], env[5])
+        apps.each do |app|
+          @table_all_apps << [env[0], app["application_name"], app["application_id"]]
+        end
+      end
+    end
+
+    def refresh_conf
+      @config.read
+      @conf = Psych.load(File.read(CONF))
+    end
+
+    def spinner(dur=1)
+      @spinner.auto_spin
+      sleep(dur)
+      @spinner.stop
+    end
+
+  end
+end

data/lib/rockette/controller.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+require_relative "controller/configurator"
+require_relative "controller/deployer"
+require_relative "controller/exporter"
+require_relative "controller/viewer"
+require_relative "text_helper"
+
+module Rockette
+  MAIN_ACTIONS = { "🔭  View Resources" => 1, "🚀  Deploy" => 2, "📥  Export" => 3,
+                   "🛠   Configure" => 4, "❌  Quit" => 5 }.freeze
+  # Manage Rockette in interactive mode
+  class Controller
+    def initialize
+      @conf = Psych.load(File.read(CONF))
+      @pastel = Pastel.new
+      @prompt = TTY::Prompt.new
+    end
+
+    def launch!
+      introduction
+
+      if @conf["rockette"]["check_for_url"] && @conf["rockette"]["controller_url"].length < 10
+        configurer = Rockette::Configurator.new
+        configurer.first_run
+      end
+
+      # input/action loop
+      loop do
+        action = actions
+        break if action == 5
+
+        do_action(action)
+      end
+      conclusion
+    end
+
+    private
+
+    def introduction
+      font = TTY::Font.new(:starwars)
+      puts "-" * 85
+      puts
+      puts @pastel.yellow(font.write("Rockette"))
+      puts
+      puts "-" * 85
+      puts "Version: " + VERSION
+      puts "-" * 14
+      puts
+      puts "Rockette helps export and deploy APEX applications."
+      puts
+    end
+
+    def conclusion
+      puts
+      puts "-" * 85
+      puts @pastel.yellow("Have a good one!".upcase.center(85))
+      puts "-" * 85
+      puts
+    end
+
+    def actions
+      @prompt.select("What would you like to do?", MAIN_ACTIONS)
+    end
+
+    def do_action(action)
+      case action
+      when 1
+        viewer = Rockette::Viewer.new
+        viewer.launch!
+      when 2
+        deployer = Rockette::Deployer.new
+        deployer.launch!
+      when 3
+        exporter = Rockette::Exporter.new
+        exporter.launch!
+      when 4
+        configurer = Rockette::Configurator.new
+        configurer.launch!
+      else
+        puts "\nI don't understand that command.\n\n"
+      end
+    end
+  end
+end

data/lib/rockette/rester.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+module Rockette
+  # configurable rest-client calls with error handling and auto retries
+  class Rester
+    VERBS = {
+      "delete" => :Delete,
+      "get" => :Get,
+      "post" => :Post,
+      "put" => :Put
+    }.freeze
+
+    def initialize(headers: {}, meth: "Get", params: {}, url: "https://array/", config: {})
+      @headers = headers
+      @meth    = meth
+      @params  = params
+      @url     = url
+      @config = config
+      @config["timeout"] = @config["timeout"] ||= 30
+    end
+
+    def make_call
+      response = RestClient::Request.execute(headers: @headers,
+                                             method: VERBS[@meth.downcase], payload: @params,
+                                             timeout: @config["timeout"], url: @url, verify_ssl: false)
+    rescue SocketError, IOError => e
+      puts "#{e.class}: #{e.message}"
+      nil
+    rescue StandardError => e
+      e.response
+    else
+      response
+    end
+
+    def cookie
+      if @url =~ %r{auth/session}
+        response = make_call
+        raise "There was an issue getting a cookie!" unless response.code == 200
+
+        (response.cookies.map { |key, val| "#{key}=#{val}" })[0]
+      else
+        error_text("cookie", @url.to_s, "auth/session")
+      end
+    end
+
+    # use rest-client with retry
+    def rest_try(tries = 3)
+      tries.times do |i|
+        response = make_call
+        unless response.nil?
+          break response if (200..299).include? response.code
+          break response if i > 1
+        end
+        puts "Failed #{@meth} on #{@url}, retry...#{i + 1}"
+        sleep 3 unless i > 1
+        return nil if i > 1 # Handles socket errors, etc. where there is no response.
+      end
+    end
+
+    private
+
+    def error_text(method_name, url, wanted)
+      {
+        "response" =>
+          "ERROR: Wrong url for the #{method_name} method.\n"\
+          "Sent: #{url}\n"\
+          "Expected: \"#{wanted}\" as part of the url.",
+        "status" => 400
+      }
+    end
+
+    def responder(response)
+      {
+        "response" => JSON.parse(response.body),
+        "status" => response.code.to_i
+      }
+    end
+  end
+end

data/lib/rockette/text_helper.rb

@@ -19,6 +19,12 @@ module TextHelper
     exit
   end
 
+  def no_rockette_dir
+    puts "Unable to locate a suitable place for the .rockette directory."
+    puts "Please double check your home directory..."
+    exit
+  end
+
   def padder(str)
     "\n#{str}\n"
   end

data/lib/rockette/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Rockette
-  VERSION = "0.0.2"
+  VERSION = "0.0.6"
 end

data/lib/rockette.rb

@@ -1,91 +1,35 @@
 # frozen_string_literal: true
 
-require           "erb"
-require           "json"
-require           "psych"
-require           "rest-client"
-require           "thor"
-require_relative  "rockette/cli"
-require_relative  "rockette/text_helper"
-require_relative  "rockette/version"
+require_relative "requirements"
+
+# Set paths
+APP_PATH = if File.exist?(File.join("/", "usr", "app"))
+             File.join("/", "usr", "app", ".rockette")
+           elsif File.exist?(Dir.home)
+             File.join(Dir.home, ".rockette")
+           else
+             no_rockette_dir
+           end
+lib_path = File.expand_path("../lib", __dir__)
+tem_path = File.expand_path("../templates", __dir__)
+$LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+
+# Create directories and config file if needed
+Dir.mkdir(APP_PATH) unless File.exist?(APP_PATH)
+unless File.exist?(File.join(APP_PATH, "config.yml"))
+  template = File.read(File.join(tem_path, "config.yml.erb"))
+  data = ERB.new(template).result(binding)
+  File.write(File.join(APP_PATH, "config.yml"), data)
+end
+Dir.mkdir(File.join(APP_PATH, "exports")) unless File.exist?(File.join(APP_PATH, "exports"))
+
+# Set config and export directory paths
+CONF = File.join(APP_PATH, "config.yml")
+ENV["THOR_SILENCE_DEPRECATION"] = "true"
+EXPORT_DIR = File.join(APP_PATH, "exports")
 
 # APEX deployment
 module Rockette
   class Error < StandardError; end
-
-  # rest-client calls with error handling and auto retries
-  class Rester
-    VERBS = {
-      "delete" => :Delete,
-      "get" => :Get,
-      "post" => :Post,
-      "put" => :Put
-    }.freeze
-
-    def initialize(headers: {}, meth: "Get", params: {}, url: "https://array/", config: {})
-      @headers = headers
-      @meth    = meth
-      @params  = params
-      @url     = url
-      @config = config
-      @config["timeout"] = @config["timeout"] ||= 30
-    end
-
-    def make_call
-      response = RestClient::Request.execute(headers: @headers,
-                                             method: VERBS[@meth.downcase], payload: @params,
-                                             timeout: @config["timeout"], url: @url, verify_ssl: false)
-    rescue SocketError, IOError => e
-      puts "#{e.class}: #{e.message}"
-      nil
-    rescue StandardError => e
-      e.response
-    else
-      response
-    end
-
-    def cookie
-      if @url =~ %r{auth/session}
-        response = make_call
-        raise "There was an issue getting a cookie!" unless response.code == 200
-
-        (response.cookies.map { |key, val| "#{key}=#{val}" })[0]
-      else
-        error_text("cookie", @url.to_s, "auth/session")
-      end
-    end
-
-    # use rest-client with retry
-    def rest_try
-      3.times do |i|
-        response = make_call
-        unless response.nil?
-          break response if (200..299).include? response.code
-          break response if i > 1
-        end
-        puts "Failed #{@meth} on #{@url}, retry...#{i + 1}"
-        sleep 3 unless i > 1
-        return nil if i > 1 # Handles socket errors, etc. where there is no response.
-      end
-    end
-
-    private
-
-    def error_text(method_name, url, wanted)
-      {
-        "response" =>
-          "ERROR: Wrong url for the #{method_name} method.\n"\
-          "Sent: #{url}\n"\
-          "Expected: \"#{wanted}\" as part of the url.",
-        "status" => 400
-      }
-    end
-
-    def responder(response)
-      {
-        "response" => JSON.parse(response.body),
-        "status" => response.code.to_i
-      }
-    end
-  end
+  # Code in lib/rockette
 end

data/rockette.gemspec

@@ -32,10 +32,15 @@ Gem::Specification.new do |spec|
   # Dev dependencies
   spec.add_development_dependency "pry", "~> 0.0"
 
-  # Uncomment to register a new dependency of your gem
+  # Dependencies
+  spec.add_dependency "oci", "~> 2.0"
+  spec.add_dependency "pastel", "~> 0.0"
   spec.add_dependency "rest-client", "~> 2.0"
   spec.add_dependency "thor", "~> 1.0"
-
-  # For more information and examples about making a new gem, checkout our
-  # guide at: https://bundler.io/guides/creating_gem.html
+  spec.add_dependency "tty-config", "~> 0.0"
+  spec.add_dependency "tty-editor", "~> 0.0"
+  spec.add_dependency "tty-font", "~> 0.0"
+  spec.add_dependency "tty-prompt", "~> 0.0"
+  spec.add_dependency "tty-spinner", "~> 0.0"
+  spec.add_dependency "tty-table", "~> 0.0"
 end

data/templates/config.yml.erb

@@ -1,17 +1,9 @@
 ---
-push_hdrs:
-  Content-Type: application/json
-  created_by: dcs_automation
-  message_date: ''
-  msg_from_addr: ''
-  msg_from_name: ''
-  file_charset: UTF-8
-  file_mimetype: text/plain
-  file_name: ''
-  file_type: sql
-  tags: ''
-  target_use: push_deployment
-deploy_body:
-  app_id_src: ''
-  app_id_tgt: ''
-  blob_url: https://local.data_loader
+token_body:
+  grant_type: client_credentials
+token_hdrs:
+  Content-Type: application/x-www-form-urlencoded
+rockette:
+  check_for_url: true
+  controller_cred: ''
+  controller_url: ''