ey-core 3.1.2 → 3.1.3

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

@@ -1,17 +1,39 @@
-class Ey::Core::Cli::Environments < Ey::Core::Cli::Subcommand
-  title "environments"
-  summary "Retrieve a list of Engine Yard environments that you have access to."
-  option :account, short: 'c', long: 'account', description: 'Filter by account name or id', argument: 'Account'
+require 'ey-core/cli/subcommand'
+require 'ey-core/cli/helpers/stream_printer'
 
-  def handle
-    environments = if option(:account)
-                     core_account_for(options).environments.all
-                   else
-                     current_accounts.map(&:environments).flatten.sort_by(&:id)
-                   end
+module Ey
+  module Core
+    module Cli
+      class Environments < Subcommand
+        include Ey::Core::Cli::Helpers::StreamPrinter
+        title "environments"
+        summary "Retrieve a list of Engine Yard environments that you have access to."
 
+        option :account,
+          short: 'c',
+          long: 'account',
+          description: 'Filter by account name or id',
+          argument: 'Account'
 
-    table_data = TablePrint::Printer.new(environments, [{id: {width: 10}}, :name])
-    puts table_data.table_print
+        def handle
+          if option(:account)
+            stream_print("ID" => 10, "Name" => 50) do |printer|
+              core_account.environments.each_entry do |env|
+                printer.print(env.id, env.name)
+              end
+            end
+          else
+            stream_print("ID" => 10, "Name" => 50, "Account" => 50) do |printer|
+              core_accounts.each_entry do |account|
+                account.environments.each_entry do |env|
+                  printer.print(env.id, env.name, account.name)
+                end
+              end
+            end
+          end
+        end
+
+      end
+    end
   end
 end

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

@@ -1,7 +1,11 @@
-class Ey::Core::Cli::Errors
-  ::Ey::Core::Cli::RecipesNotFound = Class.new(ArgumentError)
-  ::Ey::Core::Cli::NoCommand       = Class.new(ArgumentError)
-  ::Ey::Core::Cli::NoRepository    = Class.new(ArgumentError)
-  ::Ey::Core::Cli::RecipesExist    = Class.new(ArgumentError)
-  ::Ey::Core::Cli::AmbiguousSearch = Class.new(ArgumentError)
+module Ey
+  module Core
+    module Cli
+      RecipesNotFound = Class.new(ArgumentError)
+      NoCommand       = Class.new(ArgumentError)
+      NoRepository    = Class.new(ArgumentError)
+      RecipesExist    = Class.new(ArgumentError)
+      AmbiguousSearch = Class.new(ArgumentError)
+    end
+  end
 end

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

@@ -0,0 +1,30 @@
+require 'belafonte'
+
+module Ey
+  module Core
+    module Cli
+      class Help < Belafonte::App
+        title 'help'
+        summary 'Print out the help docs'
+
+        arg :command_line,
+          times: :unlimited
+
+        def handle
+          Main.new(
+            command_line.unshift('-h'),
+            stdin,
+            stdout,
+            stderr,
+            kernel
+          ).execute!
+        end
+
+        private
+        def command_line
+          arg(:command_line)
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/helpers/archive.rb

@@ -0,0 +1,70 @@
+require 'stringio'
+require 'rubygems/package'
+require 'zlib'
+
+module Ey
+  module Core
+    module Cli
+      module Helpers
+        module Archive
+          def gzip(tarfile)
+            gz = StringIO.new("")
+            zipper = Zlib::GzipWriter.new(gz)
+            zipper.write tarfile.string
+            zipper.close # this is necessary!
+
+            # z was closed to write the gzip footer, so
+            # now we need a new StringIO
+            StringIO.new gz.string
+          end
+
+          def archive_directory(path)
+            tarfile = StringIO.new("")
+            Gem::Package::TarWriter.new(tarfile) do |tar|
+              Dir[File.join(path, "**/*")].each do |file|
+                mode          = File.stat(file).mode
+                relative_file = "cookbooks/#{file.sub(/^#{Regexp::escape path}\/?/, '')}"
+
+                if File.directory?(file)
+                  tar.mkdir relative_file, mode
+                else
+                  tar.add_file relative_file, mode do |tf|
+                    File.open(file, "rb") { |f| tf.write f.read }
+                  end
+                end
+              end
+            end
+
+            tarfile.rewind
+            gzip(tarfile)
+          end
+
+          def ungzip(tarfile)
+            zipped = Zlib::GzipReader.new(tarfile)
+            unzipped = StringIO.new(zipped.read)
+            zipped.close
+            unzipped
+          end
+
+          def untar(io, destination)
+            Gem::Package::TarReader.new io do |tar|
+              tar.each do |tarfile|
+                destination_file = File.join destination, tarfile.full_name
+
+                if tarfile.directory?
+                  FileUtils.mkdir_p destination_file
+                else
+                  destination_directory = File.dirname(destination_file)
+                  FileUtils.mkdir_p destination_directory unless File.directory?(destination_directory)
+                  File.open destination_file, "wb" do |file|
+                    file.print tarfile.read
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/helpers/chef.rb

@@ -0,0 +1,35 @@
+require 'ey-core/cli/errors'
+
+module Ey
+  module Core
+    module Cli
+      module Helpers
+        module Chef
+          def run_chef(type, environment)
+            request = environment.apply(type)
+            puts "Started #{type} chef run".green
+            request.wait_for { |r| r.ready? }
+            if request.successful
+              puts "#{type.capitalize} chef run completed".green
+            else
+              puts "#{type.capitalize} chef run failed".red
+              ap request
+            end
+          end
+
+          def upload_recipes(environment, path="cookbooks/")
+            recipes_path = Pathname.new(path)
+
+            if recipes_path.exist? && recipes_path.to_s.match(/\.(tgz|tar\.gz)/)
+              environment.upload_recipes(recipes_path)
+            elsif recipes_path.exist?
+              environment.upload_recipes(archive_directory(path))
+            else
+              raise RecipesNotFound, "Recipes file not found: #{recipes_path}"
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,195 @@
+require 'colorize'
+require 'ey-core/cli/errors'
+
+module Ey
+  module Core
+    module Cli
+      module Helpers
+        module Core
+          module ClassMethods
+            def core_file
+              @core_file ||= File.expand_path("~/.ey-core")
+            end
+
+            def eyrc
+              @eyrc ||= File.expand_path("~/.eyrc")
+            end
+          end
+
+          def core_url
+            env_url = ENV["CORE_URL"] || ENV["CLOUD_URL"]
+            (env_url && File.join(env_url, '/')) || "https://api.engineyard.com/"
+          end
+
+          def longest_length_by_name(collection)
+            collection.map(&:name).group_by(&:size).max.last.length
+          end
+
+          def core_yaml
+            @core_yaml ||= YAML.load_file(self.class.core_file) || {}
+          rescue Errno::ENOENT => e
+            puts "Creating #{self.class.core_file}".yellow
+            FileUtils.touch(self.class.core_file)
+            retry
+          end
+
+          def operator(options)
+            if options[:account]
+              core_account
+            elsif ENV["STAFF"]
+              core_client
+            else
+              core_client.users.current
+            end
+          end
+
+          def core_operator_and_environment_for(options={})
+            unless options[:environment]
+              raise "--environment is required (for a list of environments, try `ey environments`)"
+            end
+            operator = operator(options)
+            environment = nil
+            if options[:environment].to_i.to_s == options[:environment]
+              environment = operator.environments.get(options[:environment])
+            end
+            unless environment
+              candidate_envs = operator.environments.all(name: options[:environment])
+              if candidate_envs.size > 1
+                raise "Multiple matching environments found named '#{options[:environment]}', please specify --account"
+              else
+                environment = candidate_envs.first
+              end
+            end
+            unless environment
+              raise "environment '#{options[:environment]}' not found (for a list of environments, try `ey environments`)"
+            end
+            [operator, environment]
+          end
+
+          def core_environment_for(options={})
+            core_client.environments.get(options[:environment]) || core_client.environments.first(name: options[:environment])
+          end
+
+          def core_server_for(options={})
+            operator = options.fetch(:operator, core_client)
+            operator.servers.get(options[:server]) || operator.servers.first(provisioned_id: options[:server])
+          end
+
+          def core_application_for(environment, options={})
+            candidate_apps = nil
+            unless options[:app]
+              candidate_apps = environment.applications.map(&:name)
+              if candidate_apps.size == 1
+                options[:app] = candidate_apps.first
+              else
+                raise "--app is required (Candidate apps on environment #{environment.name}: #{candidate_apps.join(', ')})"
+              end
+            end
+
+            app = begin
+                    Integer(options[:app])
+                  rescue
+                    options[:app]
+                  end
+
+            if app.is_a?(Integer)
+              environment.applications.get(app)
+            else
+              applications = environment.applications.all(name: app)
+              if applications.count == 1
+                applications.first
+              else
+                error_msg = [
+                  "Found multiple applications that matched that search.",
+                  "Please be more specific by specifying the account, environment, and application name.",
+                  "Matching applications: #{applications.map(&:name)}.",
+                ]
+                if candidate_apps
+                  error_msg << "applications on this environment: #{candidate_apps}"
+                end
+                raise Ey::Core::Cli::AmbiguousSearch.new(error_msg.join(" "))
+              end
+            end
+          end
+
+          def unauthenticated_core_client
+            @unauthenticated_core_client ||= Ey::Core::Client.new(token: nil, url: core_url)
+          end
+
+          def core_client
+            @core_client ||= begin
+              opts = {url: core_url, config_file: self.class.core_file}
+              if ENV["DEBUG"]
+                opts[:logger] = ::Logger.new(STDOUT)
+              end
+              Ey::Core::Client.new(opts)
+            end
+          rescue RuntimeError => e
+            if legacy_token = e.message.match(/missing token/i) && eyrc_yaml["api_token"]
+              puts "Found legacy .eyrc token.  Migrating to core file".green
+              write_core_yaml(legacy_token)
+              retry
+            elsif e.message.match(/missing token/i)
+              abort "Missing credentials: Run 'ey login' to retrieve your Engine Yard Cloud API token.".yellow
+            else
+              raise e
+            end
+          end
+
+          def core_account
+            @_core_account ||= begin
+              if options[:account]
+                found = core_client.accounts.get(options[:account]) ||
+                        core_client.users.current.accounts.first(name: options[:account])
+                if ENV["STAFF"]
+                  found ||= core_client.accounts.first(name: options[:account])
+                end
+                unless found
+                  account_not_found_error_message = "Couldn't find account '#{options[:account]}'"
+                  if core_client.users.current.staff && !ENV["STAFF"]
+                    account_not_found_error_message += " (set environment variable STAFF=1 to search all accounts)"
+                  end
+                  raise account_not_found_error_message
+                end
+                found
+              else
+                if core_accounts.size == 1
+                  core_accounts.first
+                else
+                  raise "Please specify --account (options: #{core_accounts.map(&:name).join(', ')})"
+                end
+              end
+            end
+          end
+
+          def core_accounts
+            @_core_accounts ||= begin
+              if ENV["STAFF"]
+                core_client.accounts
+              else
+                core_client.users.current.accounts
+              end
+            end
+          end
+
+          def write_core_yaml(token=nil)
+            core_yaml[core_url] = token if token
+            File.open(self.class.core_file, "w") {|file|
+              file.puts core_yaml.to_yaml
+            }
+          end
+
+          def eyrc_yaml
+            @eyrc_yaml ||= YAML.load_file(self.class.eyrc) || {}
+          rescue Errno::ENOENT # we don't really care if this doesn't exist
+            {}
+          end
+
+          def self.included(base)
+            base.extend(ClassMethods)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/helpers/deprecated.rb

@@ -0,0 +1,39 @@
+require 'colorize'
+
+module Ey
+  module Core
+    module Cli
+      module Helpers
+        module Deprecated
+        # Class-level helpers pulled into an including class
+          module ClassMethods
+
+            # A helper for deprecating a Belafonte-based command
+            #
+            # @param deprecated_title [String] the title of the command
+            #
+            # @return [NilClass] this method is not expected to return a value
+            def deprecate(deprecated_title)
+              title deprecated_title
+              summary 'This command has been deprecated'
+              description <<-DESCRIPTION
+The #{meta[:title]} command has been deprecated. We apologize for any inconvenience.
+DESCRIPTION
+            end
+          end
+
+          # Module magic
+          # @api private
+          def self.included(base)
+            base.extend(ClassMethods)
+          end
+
+          # Terminate the deprecated command with a message
+          def handle
+            abort "This command is deprecated".red
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ey-core/cli/helpers/log_streaming.rb

@@ -0,0 +1,41 @@
+module Ey
+  module Core
+    module Cli
+      module Helpers
+        module LogStreaming
+
+          def stream_deploy_log(request)
+            if request.finished_at
+              return finished_request(request)
+            end
+            unless request.read_channel
+              puts "Unable to stream log (streaming not enabled for this deploy)".yellow
+              return
+            end
+            request.subscribe { |m| print m["message"] if m.is_a?(Hash) }
+            puts "" # fix console output from stream
+            finished_request(request)
+          end
+
+          def finished_request(request)
+            if request.successful
+              if request.resource.successful
+                puts "Deploy successful!".green
+              else
+                puts "Deploy failed!".red
+              end
+            else
+              abort <<-EOF
+        Deploy failed!
+        Request output:
+        #{request.message}
+              EOF
+              .red
+            end
+          end
+
+        end
+      end
+    end
+  end
+end