awx 0.1.0 → 0.2.0

data/lib/routes/aws_cloudformation_delete.rb

@@ -0,0 +1,37 @@
+module AppCommand
+
+    class AWSCloudFormationDelete < ::Convoy::ActionCommand::Base
+
+        def execute
+
+            begin
+
+                @opts = command_options
+                @args = arguments
+
+                opts_validate
+                opts_routing
+
+            rescue => e
+
+                Blufin::Terminal::print_exception(e)
+
+            end
+
+        end
+
+        def opts_validate
+
+        end
+
+        def opts_routing
+
+            # TODO - IMPLEMENT
+
+            raise RuntimeError, 'Not yet implemented! Delete.'
+
+        end
+
+    end
+
+end

data/lib/routes/aws_cloudformation_detect_drift.rb

@@ -0,0 +1,44 @@
+module AppCommand
+
+    class AWSCloudFormationDetectDrift < ::Convoy::ActionCommand::Base
+
+        def execute
+
+            begin
+
+                @opts = command_options
+                @args = arguments
+
+                @regions = []
+                @threads = []
+
+                opts_validate
+                opts_routing
+
+            rescue => e
+
+                Blufin::Terminal::print_exception(e)
+
+            end
+
+        end
+
+        def opts_validate
+
+            @regions = App::AWSValidator::get_and_validate_regions(@opts[:region_given])
+
+        end
+
+        def opts_routing
+
+            # TODO - IMPLEMENT
+
+            raise RuntimeError, 'Not yet implemented! Detect Drift.'
+
+
+
+        end
+
+    end
+
+end

data/lib/routes/aws_lambda.rb

@@ -0,0 +1,122 @@
+require 'json'
+require 'uri'
+
+module AppCommand
+
+    class AWSLambda < ::Convoy::ActionCommand::Base
+
+        VALID_METHODS = %w(GET POST PUT DELETE)
+
+        def execute
+
+            begin
+
+                @opts = command_options
+                @args = arguments
+
+                @project      = @args[0]
+                @project_path = nil
+                @uri          = @args[1]
+                @method       = @args[2].nil? ? nil : @args[2].upcase
+                @file_payload = @opts[:payload_file]
+                @file_headers = @opts[:headers_file].nil? ? 'default' : @opts[:headers_file]
+                @headers      = nil
+                @payload      = nil
+
+                @valid_projects = {
+                    'bb' => App::Config.param(ConfigUnique::PATH_TO_LAMBDA_BERT),
+                    'ew' => App::Config.param(ConfigUnique::PATH_TO_LAMBDA_EWORLD)
+                }
+
+                opts_validate
+                opts_routing
+
+            rescue => e
+
+                Blufin::Terminal::print_exception(e)
+
+            end
+
+        end
+
+        def opts_validate
+
+            nil_errors = []
+            nil_errors << "#{Blufin::Terminal::format_highlight('1st')} parameter must be #{Blufin::Terminal::format_action('project')}. Valid projects are: #{@valid_projects.keys.join(', ')}" if @project.nil?
+            nil_errors << "#{Blufin::Terminal::format_highlight('2nd')} parameter must be #{Blufin::Terminal::format_action('uri')} aka: route to Lambda handler" if @uri.nil? || @uri.strip == ''
+            nil_errors << "#{Blufin::Terminal::format_highlight('3rd')} parameter must be #{Blufin::Terminal::format_action('method')}. Valid methods are: #{VALID_METHODS.join(', ')}" if @uri.nil? || @uri.strip == ''
+
+            # Catch NULL errors first.
+            Blufin::Terminal::error('Please fix the following errors:', nil_errors) if nil_errors.any?
+
+            # Validate project alias.
+            Blufin::Terminal::error("#{Blufin::Terminal::format_action('project')} #{Blufin::Terminal::format_highlight(@project.nil? || @project.strip == '' ? 'nil' : @project)} is invalid. Valid project(s) are:", @valid_projects.keys, true) unless @valid_projects.keys.include?(@project)
+
+            # Validate project path.
+            @project_path = @valid_projects[@project]
+            Blufin::Terminal::error("#{Blufin::Terminal::format_action('project path')} doesn't exist: #{Blufin::Terminal::format_directory(@project_path)}") unless Blufin::Files::path_exists(@project_path)
+
+            # Validate HTTP Method.
+            Blufin::Terminal::error("#{Blufin::Terminal::format_action('http')} method #{Blufin::Terminal::format_highlight(@method.nil? || @method.strip == '' ? 'nil' : @method)} is invalid. Valid methods are:", VALID_METHODS, true) unless VALID_METHODS.include?(@method)
+
+            # Validate headers file.
+            unless Blufin::Files::file_exists(@file_headers)
+                file_headers_alt = "#{@project_path}/json/headers/#{@file_headers.strip.gsub(/(\.json)$/i, '')}.json"
+                if Blufin::Files::file_exists(file_headers_alt)
+                    @file_headers = file_headers_alt
+                    h_buffer      = ''
+                    Blufin::Files::read_file(@file_headers).each { |line| h_buffer = "#{h_buffer}#{line.gsub("\n", '').strip}" }
+                    begin
+                        JSON.parse(h_buffer)
+                        @headers = h_buffer
+                    rescue => e
+                        Blufin::Terminal::error("Failed to parse JSON from #{Blufin::Terminal::format_action('header file')}: #{Blufin::Terminal::format_directory(@file_headers)}", e.message)
+                    end
+                else
+                    Blufin::Terminal::error("Neither of the following #{Blufin::Terminal::format_action('headers files')} exist:", [@file_headers, file_headers_alt], true)
+                end
+            end
+
+            # Validate payload file.
+            unless @file_payload.nil?
+                unless Blufin::Files::file_exists(@file_payload)
+                    file_payload_alt = "#{@project_path}/json/payload/#{@file_payload.strip.gsub(/(\.json)$/i, '')}.json"
+                    if Blufin::Files::file_exists(file_payload_alt)
+                        @file_payload = file_payload_alt
+                        p_buffer      = ''
+                        Blufin::Files::read_file(@file_payload).each { |line| p_buffer = "#{p_buffer}#{line.gsub("\n", '').strip}" }
+                        begin
+                            JSON.parse(p_buffer)
+                            @payload = p_buffer
+                        rescue => e
+                            Blufin::Terminal::error("Failed to parse JSON from #{Blufin::Terminal::format_action('payload file')}: #{Blufin::Terminal::format_directory(@file_payload)}", e.message)
+                        end
+                    else
+                        Blufin::Terminal::error("Neither of the following #{Blufin::Terminal::format_action('headers files')} exist:", [@file_payload, file_payload_alt], true)
+                    end
+                end
+            end
+
+        end
+
+        def opts_routing
+
+            headers = @headers.nil? ? nil : ",\"headers\":#{@headers}"
+            payload = @payload.nil? ? nil : ",\"body\":\"#{URI::encode(@payload)}\""
+
+            cmd = "sls invoke local -f main -d '{\"resource\":\"/#{@uri}\",\"path\":\"/#{@uri}\",\"httpMethod\":\"#{@method}\"#{headers}#{payload},\"pathParameters\":{\"hash\":\"cae695cfa2c33ad92ea17e1a03fba99cb9e734dbd4cc4eb8f143bf45e06e81a7rHhP4G070Rzke2O5Vg12A==\"}}'"
+            res = Blufin::Terminal::command_capture(cmd, @project_path, true, false)
+
+            begin
+                jsn = JSON.parse(res[0].strip.gsub(/\n$/, '').gsub('\"', '"').gsub('"{', '{').gsub('}"', '}'))
+                sts = jsn['statusCode'].to_i
+                Blufin::Terminal::custom('Lambda Response', 54, "Status: \x1B[38;5;#{sts == 200 ? 46 : 196}m#{sts}\x1B[0m", JSON.pretty_generate(jsn['body']).split("\n"), true)
+            rescue
+                Blufin::Terminal::custom('Lambda Response', 54, "Detected #{Blufin::Terminal::format_highlight('console.log()')} or unparseable output:", res[0].split("\n"), true)
+            end
+
+        end
+
+    end
+
+end

data/lib/routes/aws_list.rb

@@ -0,0 +1,234 @@
+module AppCommand
+
+    class AWSList < ::Convoy::ActionCommand::Base
+
+        def execute
+
+            begin
+
+                @opts = command_options
+                @args = arguments
+
+                @resource       = nil
+                @silent         = @opts[:json] || @opts[:json_prompt] || @opts[:yaml]
+                @regions        = App::AWSCli::get_regions
+                @terminal_width = Blufin::Terminal::get_terminal_width
+                @data           = nil
+                @table_widths   = {}
+                @export_map     = {}
+                @columns        = {}
+
+                @columns, @data, @export_map, @table_widths = App::AWSReports::parse_metadata(@regions)
+
+                opts_validate
+                opts_routing
+
+            rescue => e
+
+                Blufin::Terminal::print_exception(e)
+
+            end
+
+        end
+
+        def opts_validate
+
+            Blufin::Routes::only_one_of(@opts, [:json, :json_prompt, :yaml])
+
+            # TODO - Implement!
+            if @opts[:region] || @opts[:environment] || @opts[:project]
+                raise RuntimeError, 'Not yet implemented!'
+            end
+
+            # Throw error if -m is set with any other flas.
+            Blufin::Terminal::error("When #{Blufin::Terminal::format_flag('m')} is set, no other flags can be set.", "Found: #{Blufin::Routes::flags_set(@opts).to_i - 1} other flag(s).", true) if @opts[:metadata] && Blufin::Routes::flags_set(@opts).to_i > 1
+
+            # For resources with exports, this enables you to pass the -r flag as both AcmCertificates or AcmCertificateARN (and it will resolve to the same API call).
+            if @opts[:json_prompt]
+                @opts[:resource] = @export_map[@opts[:resource]] if @export_map.has_key?(@opts[:resource])
+            end
+
+            # Validate that resource exists (if passed from command-line).
+            unless @opts[:resource].nil?
+                if @opts[:resource].downcase == App::AWSReports::CONST_ALL
+                    Blufin::Terminal::error("The #{Blufin::Terminal::format_flag('v')} cannot be set when listing all resources.") if @opts[:verbose]
+                    Blufin::Terminal::error("The #{Blufin::Terminal::format_flag('j')} cannot be set when listing all resources.") if @opts[:json]
+                    Blufin::Terminal::error("The #{Blufin::Terminal::format_flag('j')} cannot be set when listing all resources.") if @opts[:json_prompt]
+                    Blufin::Terminal::error("The #{Blufin::Terminal::format_flag('y')} cannot be set when listing all resources.") if @opts[:yaml]
+                    @resource = App::AWSReports::CONST_ALL
+                else
+                    Blufin::Terminal::error("Invalid: #{Blufin::Terminal::format_invalid(@opts[:resource])} . Available resources are:", @data.keys, true) unless @data.has_key?(@opts[:resource])
+                    @resource = @data[@opts[:resource]]
+                end
+            end
+
+            # If response is JSON/YAML, Resource must be explicitly specified.
+            if @opts[:json] || @opts[:json_prompt] || @opts[:yaml]
+                flag = 'j' if @opts[:json]
+                flag = 'J' if @opts[:json_prompt]
+                flag = 'y' if @opts[:yaml]
+                Blufin::Terminal::error("Must specify #{Blufin::Terminal::format_highlight('resource')} explicitly (using #{Blufin::Terminal::format_flag('r')}) if #{Blufin::Terminal::format_flag(flag)} is set.") if @opts[:resource].nil?
+                if @opts[:json_prompt]
+                    Blufin::Terminal::error("Resource must have #{Blufin::Terminal::format_highlight('export')} defined if #{Blufin::Terminal::format_flag(flag)} is set.") unless @resource.has_key?(App::AWSReports::KEY_EXPORT)
+                end
+            end
+        end
+
+        def opts_routing
+
+            system('clear') unless @silent
+
+            # Display metadata and exit (if -m flag is set)
+            if @opts[:metadata]
+                App::AWSReports::get_auto_fetch_resources(@data).each do |title, data|
+                    output = []
+                    output << "\x1B[38;5;240m    AWS-CLI Command:\x1B[38;5;106m #{data[:resource][App::AWSReports::KEY_CLI]['command']}" if data[:resource].has_key?(App::AWSReports::KEY_CLI) && data[:resource][App::AWSReports::KEY_CLI].has_key?('command')
+                    output << "\x1B[38;5;240m          Region(s):\x1B[38;5;94m #{data[:resource][App::AWSReports::KEY_REGIONS].join(', ')}" if data[:resource].has_key?(App::AWSReports::KEY_REGIONS)
+                    output << "\x1B[38;5;240mPreferred Region(s):\x1B[38;5;94m #{data[:resource][App::AWSReports::KEY_REGIONS_PREFERRED].join(', ')}" if data[:resource].has_key?(App::AWSReports::KEY_REGIONS_PREFERRED)
+                    Blufin::Terminal::custom(data[:resource_title], 55, "Exports: #{Blufin::Terminal::format_highlight(title)}", output)
+                end
+                puts
+                exit
+            end
+
+            # Select Key (if not already specified using the --resource flag)
+            if @resource.nil?
+                resource_title = Blufin::Terminal::prompt_select('Select AWS Resource(s) to list:', @data.keys)
+                @resource      = @data[resource_title]
+            else
+                resource_title = @opts[:resource]
+            end
+
+            # TODO - REMOVE CACHING
+            cache_file = '/tmp/aws-uber-results-cache.txt'
+            if !@opts[:resource].nil? && @opts[:resource].downcase == App::AWSReports::CONST_ALL
+
+                if false && Blufin::Files::file_exists(cache_file)
+
+                    # TODO - REMOVE CACHING
+                    uber_results = {}
+                    puts
+                    puts "  \x1B[38;5;196mGetting regional_response from cache: #{Blufin::Terminal::format_directory(cache_file)}\x1B[0m"
+                    eval("uber_results = #{Blufin::Files::read_file(cache_file)[0]}")
+
+                else
+
+                    # Get and display all resources.
+                    uber_results = {}
+                    cmd_color    = 240
+                    threads      = []
+                    no_fit       = []
+                    puts
+                    @data.each do |resource_title, resource|
+                        # Skip tables that don't fit.
+                        fit_res = fits_in_terminal(resource_title)
+                        unless fit_res[0]
+                            no_fit << "\x1B[38;5;196m#{resource_title}\x1B[38;5;240m (required: #{fit_res[1]} / actual: #{fit_res[2]})"
+                            next
+                        end
+                        sleep(0.01)
+                        cmd_regions = @regions.join(',')
+                        cmd_regions = resource[App::AWSReports::KEY_REGIONS].join(',') if resource.has_key?(App::AWSReports::KEY_REGIONS)
+                        cmd_regions = "\x1B[38;5;#{cmd_color}m[\x1B[38;5;246m#{cmd_regions}\x1B[38;5;#{cmd_color}m]\x1B[0m"
+                        puts "  \x1B[38;5;70m$ \x1B[38;5;#{cmd_color}maws #{resource[App::AWSReports::KEY_CLI][App::AWSReports::KEY_COMMAND]} --region #{cmd_regions}\x1B[0m"
+                        threads << Thread.new {
+                            uber_results[resource_title] = App::AWSReports::get_aws_data(@regions, resource, resource_title, silent: true)
+                        }
+                    end
+                    # If some tables don't fit, output a warning.
+                    sleep(0.01)
+                    if no_fit.any?
+                        Blufin::Terminal::warning("The following resource(s) will be omitted because they won't fit in the terminal window.", no_fit)
+                    else
+                        puts
+                    end
+                    Blufin::Terminal::execute_proc('AWS - Fetching data...', Proc.new {
+                        threads.each { |thread| thread.join }
+                    })
+
+                    # TODO - REMOVE CACHING
+                    Blufin::Files::write_file(cache_file, [uber_results.inspect])
+
+                end
+
+                puts
+                @data.each do |resource_title, resource|
+                    if uber_results.has_key?(resource_title)
+                        output_aws_data(uber_results[resource_title], resource, resource_title, silent: false)
+                    end
+                end
+
+            else
+
+                # Bomb-out if the output does not fit in Terminal.
+                if !@opts[:json] && !@opts[:json_prompt] && !@opts[:yaml] && !fits_in_terminal(resource_title)[0]
+                    Blufin::Terminal::error("Output for #{Blufin::Terminal::format_highlight(resource_title)} \x1B[38;5;240m(#{Blufin::Terminal::format_action(@table_widths[resource_title])}\x1B[38;5;240m-width)\x1B[0m does not fit in Terminal \x1B[38;5;240m(#{Blufin::Terminal::format_action(@terminal_width)}\x1B[38;5;240m-width)\x1B[0m", 'Please make your terminal wider and try again.', true)
+                end
+
+                # TODO - REMOVE CACHING
+                cache_file = "/tmp/aws-results-#{resource_title}-cache.txt"
+                if false && Blufin::Files::file_exists(cache_file)
+
+                    # TODO - REMOVE CACHING
+                    puts
+                    puts "  \x1B[38;5;196mGetting regional_response from cache: #{Blufin::Terminal::format_directory(cache_file)}\x1B[0m"
+                    results = {}
+                    eval("results = #{Blufin::Files::read_file(cache_file)[0]}")
+
+                else
+                    results = App::AWSReports::get_aws_data(@regions, @resource, resource_title, silent: @silent)
+
+                    # TODO - REMOVE CACHING
+                    Blufin::Files::write_file(cache_file, [results.inspect])
+                end
+
+                # Get and display a single resource.
+                output_aws_data(results, @resource, resource_title, silent: @silent)
+            end
+
+        end
+
+        private
+
+        # This outputs the AWS results depending on the flags passed.
+        # @return void
+        def output_aws_data(results, resource, resource_title, silent: false)
+            if @opts[:json]
+                puts JSON.pretty_generate(results)
+            elsif @opts[:json_prompt]
+                puts JSON.pretty_generate(App::AWSReports::parse_results_for_prompt(resource, results))
+            elsif @opts[:yaml]
+                Blufin::Terminal::code_highlight(results.to_yaml, 'yml', 4)
+            else
+                unless silent
+                    if @columns.has_key?(resource_title)
+                        if @opts[:verbose]
+                            display_raw_output(results)
+                            puts
+                            puts
+                        end
+                        App::AWSOutputter::display(resource_title, resource, @columns[resource_title], results, @table_widths[resource_title], @opts, @args)
+                    else
+                        Blufin::Terminal::warning("No column(s) defined for resource: #{Blufin::Terminal::format_highlight(resource_title)}", 'Displaying raw AWS output.')
+                        display_raw_output(results)
+                        puts
+                    end
+                end
+            end
+        end
+
+        # @return void
+        def display_raw_output(output)
+            output.to_yaml.split("\n").each { |line| puts "  \x1B[38;5;240m#{line}\x1B[0m" }
+        end
+
+        # Returns true/false depending if this table fits in the terminal or not.
+        # @return []bool, required_width, actual_width]
+        def fits_in_terminal(resource_title)
+            table_width_adjusted = @table_widths[resource_title] + 2 + 2 + 20 # 2/2 = margins, 20 = * wildcard.
+            [@terminal_width >= table_width_adjusted, table_width_adjusted, @terminal_width]
+        end
+
+    end
+
+end

data/lib/routes/setup.rb

@@ -0,0 +1,31 @@
+module AppCommand
+
+    class Setup < ::Convoy::ActionCommand::Base
+
+        def execute
+
+            begin
+
+                if File.exists?(File.expand_path(App::Config::CONFIG_FILE))
+
+                    App::Config::config_file_edit
+                    App::Config::config_params_get
+                    App::Config::config_params_validate(true)
+
+                else
+
+                    App::Config.initialize
+
+                end
+
+            rescue => e
+
+                Blufin::Terminal::print_exception(e)
+
+            end
+
+        end
+
+    end
+
+end