groove-ey-core 3.6.1

data/lib/ey-core/associations.rb

@@ -0,0 +1,155 @@
+module Ey::Core::Associations
+  def assoc_coverage(options)
+    if defined? Cistern::Coverage
+      call = Cistern::Coverage.find_caller_before("ey-core/associations.rb")
+
+      if call and call.label.start_with?("<class:")
+        options[:coverage_file] = call.absolute_path
+        options[:coverage_line] = call.lineno
+        options[:coverage_hits] = 0
+      end
+    end
+  end
+
+  def associations
+    @associations ||= {}
+  end
+
+  def assoc_reader(name, options={})
+    assoc_key  = options[:key]        || "#{name}_id"
+    collection = options[:collection] || "#{name}s"
+    resource   = options[:resource]   || name
+    assoc_name = options[:assoc_name] || resource
+
+    # handle polymorphic assocations
+    #
+    # really only used for connector sources, as they may be cluster_components or addons
+    disambiguate = Proc.new do |object, url, index=0|
+      if object.is_a?(Array) && index > (object.size - 1)
+        raise "No matching object found in #{object}"
+      elsif object.is_a?(Array) && url
+        url_pieces = URI.parse(url).path.split("/").map { |piece| piece.gsub(/-|_/,'') } # "/cluster-components/x" => ["clustercomponents", "x"]
+
+        object.find do |model|
+          normalized_model = model.to_s.gsub(/-|_/, '') # [:cluster_components, :addons] => ["clustercomponents", "addons"]
+          url_pieces.find { |piece| piece.include?(normalized_model) }
+        end
+      elsif object.is_a?(Array)
+        object[index]
+      else
+        object
+      end
+    end
+
+    define_method(name) do
+      url = self.send("#{name}_url")
+      id  = send(assoc_key)
+
+      if id || url
+        begin
+          index ||= 0
+          this_resource   = disambiguate[resource, url, index]
+          this_assoc_name = disambiguate[assoc_name, url, index]
+          this_collection = disambiguate[collection, url, index]
+
+          fetched = self.connection.send("get_#{this_assoc_name}", {"id" => id, "url" => url}).body
+        rescue RuntimeError => e
+          if e.message.match(/url needed/i)
+            index += 1
+            retry if index < 3
+          else
+            raise e
+          end
+        rescue Ey::Core::Response::NotFound
+          index += 1
+          index < 3 ? retry : raise
+        end
+
+        unless fetched[this_resource.to_s]
+          raise "Expected a '#{this_resource.to_s}' in '#{fetched.inspect}'"
+        end
+
+        attributes = fetched[this_resource.to_s]
+
+        if self.connection.respond_to?(this_collection)
+          attributes = attributes.merge(collection: self.connection.send(this_collection))
+        end
+
+        self.connection.send(this_assoc_name, attributes)
+      end
+    end
+  end
+
+  def assoc_writer(name, options={})
+    assoc_key = options[:key] || "#{name}_id"
+    url_key   = options[:key] || "#{name}_url"
+
+    attr_accessor assoc_key
+
+    define_method("#{name}=") do |assoc|
+      if assoc.respond_to?(:identity)
+        self.send("#{assoc_key}=", assoc.identity)
+      elsif assoc.is_a?(String) && assoc.match(URI::regexp)
+        self.send("#{url_key}=", assoc)
+      else
+        self.send("#{assoc_key}=", assoc)
+      end
+    end
+  end
+
+  def assoc_accessor(name, options={})
+    assoc_coverage(options)
+
+    if self.associations[name]
+      raise ArgumentError, "#{self.name} association[#{name}] specified more than once"
+    else
+      self.associations[name] = options
+    end
+
+    aliases = Array(options[:aliases] || name.to_s)
+
+    attribute(:"#{name}_url", aliases: aliases, parser: lambda { |v,_| v.is_a?(String) && v })
+
+    assoc_reader(name, options)
+    assoc_writer(name, options)
+  end
+
+  alias has_one assoc_accessor
+
+  def collection_reader(name, options={})
+    assoc_coverage(options)
+
+    if self.associations[name]
+      raise ArgumentError, "#{self.name} association[#{name}] specified more than once"
+    else
+      self.associations[name] = options
+    end
+
+    attribute(:"#{name}_url", aliases: options[:aliases] || [name.to_s])
+
+    assoc_key = options[:key] || name
+    collection_key = options[:model] || assoc_key
+
+    define_method(name) do
+      value = self.send("#{name}_url")
+
+      if self.identity && !value
+        raise "Cannot load #{name} association: #{name}_url is not set"
+      end
+
+      # @fixme this allows collections to be stored in a url variable on create before being replaced
+      # by a url entry after create upon {#merge_attributes}
+      if value.nil?
+        []
+      elsif value.is_a?(String) && value.match(URI::regexp)
+        self.connection.send(collection_key, {"url" => value})
+      elsif value.is_a?(Hash)
+        [value]
+      else
+        Array(value)
+      end
+    end
+  end
+
+  alias has_many collection_reader
+end

data/lib/ey-core/cli.rb

@@ -0,0 +1,8 @@
+module Ey
+  module Core
+
+    # The overall namespace for CLI-related code
+    module Cli
+    end
+  end
+end

data/lib/ey-core/cli/accounts.rb

@@ -0,0 +1,17 @@
+require 'ey-core/cli/subcommand'
+
+module Ey
+  module Core
+    module Cli
+      class Accounts < Subcommand
+        title "accounts"
+        summary "Retrieve a list of Engine Yard accounts that you have access to."
+
+        def handle
+          table_data = TablePrint::Printer.new(core_accounts, [{id: {width: 36}}, :name])
+          puts table_data.table_print
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/applications.rb

@@ -0,0 +1,36 @@
+require 'ey-core/cli/subcommand'
+
+module Ey
+  module Core
+    module Cli
+      class Applications < Subcommand
+        title "applications"
+        summary "Retrieve a list of Engine Yard applications that you have access to."
+
+        option :account,
+          short: 'c',
+          long: 'account',
+          description: 'Filter by account name or id',
+          argument: 'Account'
+
+        def handle
+          table_data = TablePrint::Printer.new(
+            applications,
+            [{id: {width: 10}}, :name]
+          )
+
+          puts table_data.table_print
+        end
+
+        private
+        def applications
+          if option(:account)
+            core_account.applications.all
+          else
+            core_accounts.map(&:applications).flatten.sort_by(&:id)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/console.rb

@@ -0,0 +1,28 @@
+require 'ey-core/cli/subcommand'
+require 'pry'
+
+module Ey
+  module Core
+    module Cli
+      class Console < Subcommand
+        title "console"
+        summary "Start an interactive console"
+
+        option :execute_command,
+          short: "e",
+          long: "command",
+          description: "Command to execute",
+          argument: "command"
+
+        def handle
+          if command = option(:execute_command)
+            core_client.instance_eval(command)
+          else
+            Pry.config.prompt = proc { |obj, nest_level, _| "ey-core:> " }
+            core_client.pry
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/current_user.rb

@@ -0,0 +1,16 @@
+require 'ey-core/cli/subcommand'
+
+module Ey
+  module Core
+    module Cli
+      class CurrentUser < Subcommand
+        title "current_user"
+        summary "Print the current user information"
+
+        def handle
+          ap core_client.users.current
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/deploy.rb

@@ -0,0 +1,123 @@
+require 'ey-core/cli/subcommand'
+require 'ey-core/cli/helpers/log_streaming'
+
+module Ey
+  module Core
+    module Cli
+      class Deploy < Subcommand
+        include Ey::Core::Cli::Helpers::LogStreaming
+        title "deploy"
+        summary "Deploy your application"
+
+        option :environment,
+          short: "e",
+          long: "environment",
+          description: "Name or id of the environment to deploy to.",
+          argument: "Environment"
+
+        option :account,
+          short: 'c',
+          long: 'account',
+          description: 'Name or ID of the account that the environment resides in.  If no account is specified, the app will deploy to the first environment that meets the criteria, in the accounts you have access to.',
+          argument: 'Account name or id'
+
+        option :ref,
+          short: "r",
+          long: "ref",
+          description: "A git reference to deploy.",
+          argument: "ref"
+
+        option :migrate,
+          short: "m",
+          long: "migrate",
+          description: "The migration command to run.  This option has a 50 character limit.",
+          argument: "migrate"
+
+        option :app,
+          short: "a",
+          long: "app",
+          description: "Application name or ID to deploy.  If :account is not specified, this will be the first app that matches the criteria in the accounts you have access to.",
+          argument: "app"
+
+        switch :no_wait,
+          long: "no-wait",
+          description: "Don't wait for deploy to finish, exit after deploy is started"
+
+        switch :verbose,
+          short: "v",
+          long: "verbose",
+          description: "Deploy with verbose output"
+
+        switch :verbose,
+          short: "v",
+          long: "verbose",
+          description: "Stream deploy output to this console. Alias to stream for backwards compatibility."
+
+        switch :no_migrate,
+          long: "no-migrate",
+          description: "Skip migration."
+
+        def handle
+          operator, environment = core_operator_and_environment_for(self.options)
+          if operator.is_a?(Ey::Core::Client::Account)
+            abort <<-EOF
+Found account #{operator.name} but requested account #{option(:account)}.
+Use the ID of the account instead of the name.
+This can be retrieved by running "ey accounts".
+EOF
+            .red unless operator.name == option(:account) || operator.id == option(:account)
+          end
+
+          unless environment
+            abort "Unable to locate environment #{option[:environment]} in #{operator.name}".red
+          end
+
+          unless option(:account)
+            self.options.merge!(environment: environment)
+          end
+
+          app = core_application_for(environment, self.options)
+
+          deploy_options = {verbose: options[:verbose], cli_args: ARGV}
+          latest_deploy = nil
+          if options[:ref]
+            deploy_options.merge!(ref: option(:ref))
+          else
+            puts "--ref not provided, checking latest deploy...".yellow
+            latest_deploy ||= environment.latest_deploy(app)
+            if latest_deploy && latest_deploy.ref
+              deploy_options[:ref] = latest_deploy.ref
+            else
+              raise "--ref is required (HEAD is the typical choice)"
+            end
+          end
+          if (option(:migrate) || switch_active?(:no_migrate))
+            deploy_options.merge!(migrate_command: option(:migrate)) if option(:migrate)
+            deploy_options.merge!(migrate_command: '') if switch_active?(:no_migrate)
+          else
+            puts "missing migrate option (--migrate or --no-migrate), checking latest deploy...".yellow
+            latest_deploy ||= environment.latest_deploy(app)
+            if latest_deploy
+              deploy_options.merge!(migrate_command: (latest_deploy.migrate && latest_deploy.migrate_command) || '')
+            else
+              raise "either --migrate or --no-migrate needs to be specified"
+            end
+          end
+          request = environment.deploy(app, deploy_options)
+
+          puts <<-EOF
+Deploy started to environment: #{environment.name} with application: #{app.name}
+Request ID: #{request.id}
+EOF
+          ap request
+          ap request.resource
+          if switch_active?(:no_wait)
+            puts("Deploy started".green + " (use status command with --tail to view output)")
+          else
+            stream_deploy_log(request)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/docker_registry_login.rb

@@ -0,0 +1,29 @@
+require 'ey-core/cli/subcommand'
+
+module Ey
+  module Core
+    module Cli
+      class DockerRegistryLogin < Subcommand
+        title "get-docker-registry-login"
+        summary "Prints the docker login command to authorize the Docker Engine with the AWS ECR registry"
+
+        option :account,
+               short:       'c',
+               long:        'account',
+               description: 'Name or ID of the account that the environment resides in.',
+               argument:    'Account name or id'
+
+        option :location,
+               short:       'l',
+               long:        'location',
+               description: 'ECR availability regions',
+               argument:    'Location name'
+
+        def handle
+          credentials = core_account.retrieve_docker_registry_credentials(option(:location))
+          stdout.puts "docker login -u #{credentials.username} -p #{credentials.password} #{credentials.registry_endpoint}"
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/environment_variables.rb

@@ -0,0 +1,71 @@
+require 'ey-core/cli/subcommand'
+require 'ey-core/cli/helpers/stream_printer'
+
+module Ey
+  module Core
+    module Cli
+      class EnvironmentVariables < Subcommand
+        MASK = '****'.freeze
+        SYMBOLS_TO_DISPLAY = 4
+        MAX_LENGTH_TO_DISPLAY = 30
+
+        include Ey::Core::Cli::Helpers::StreamPrinter
+
+        title "environment_variables"
+        summary "Retrieve a list of Engine Yard environment variables for environments that you have access to."
+
+        option :environment,
+          short: 'e',
+          long: 'environment',
+          description: 'Filter by environmeent name or id',
+          argument: 'Environment'
+
+        option :application,
+          short: 'a',
+          long: 'application',
+          description: 'Filter by application name or id',
+          argument: 'Application'
+
+        switch :display_sensitive,
+          short: 's',
+          long: 'display_sensitive',
+          description: 'Determines whether values of sensitive variables should be printed',
+          argument: 'Display Sensitive'
+
+        def handle
+          environment_variables = if option(:application)
+                                    core_applications(option(:application)).flat_map(&:environment_variables)
+                                  elsif option(:environment)
+                                    core_environments(option(:environment)).flat_map(&:environment_variables)
+                                  else
+                                    core_environment_variables
+                                  end
+
+          stream_print("ID" => 10, "Name" => 30, "Value" => 50, "Environment" => 30, "Application" => 30) do |printer|
+            environment_variables.each_entry do |ev|
+              printer.print(ev.id, ev.name, print_variable_value(ev), ev.environment_name, ev.application_name)
+            end
+          end
+        end
+
+        private
+
+        def print_variable_value(environment_variable)
+          if environment_variable.sensitive && !switch_active?(:display_sensitive)
+            hide_sensitive_data(environment_variable.value)
+          else
+            environment_variable.value
+          end
+        end
+
+        def hide_sensitive_data(value)
+          if value.length > SYMBOLS_TO_DISPLAY
+            MASK + value[-SYMBOLS_TO_DISPLAY, SYMBOLS_TO_DISPLAY]
+          else
+            MASK
+          end
+        end
+      end
+    end
+  end
+end