artofmission-heroku 1.6.3

data/lib/heroku/commands/domains.rb

@@ -0,0 +1,31 @@
+module Heroku::Command
+	class Domains < BaseWithApp
+		def list
+			domains = heroku.list_domains(app)
+			if domains.empty?
+				display "No domain names for #{app}.#{heroku.host}"
+			else
+				display "Domain names for #{app}.#{heroku.host}:"
+				display domains.map { |d| d[:domain] }.join("\n")
+			end
+		end
+		alias :index :list
+
+		def add
+			domain = args.shift.downcase rescue nil
+			heroku.add_domain(app, domain)
+			display "Added #{domain} as a custom domain name to #{app}.#{heroku.host}"
+		end
+
+		def remove
+			domain = args.shift.downcase rescue nil
+			heroku.remove_domain(app, domain)
+			display "Removed #{domain} as a custom domain name to #{app}.#{heroku.host}"
+		end
+
+		def clear
+			heroku.remove_domains(app)
+			display "Removed all domain names for #{app}.#{heroku.host}"
+		end
+	end
+end

data/lib/heroku/commands/help.rb

@@ -0,0 +1,148 @@
+module Heroku::Command
+	class Help < Base
+		class HelpGroup < Array
+			attr_reader :title
+
+			def initialize(title)
+				@title = title
+			end
+
+			def command(name, description)
+				self << [name, description]
+			end
+
+			def space
+				self << ['', '']
+			end
+		end
+
+		def self.groups
+			@groups ||= []
+		end
+
+		def self.group(title, &block)
+			groups << begin
+				group = HelpGroup.new(title)
+				group.instance_eval(&block)
+				group
+			end
+		end
+
+		def self.create_default_groups!
+			group('General Commands') do
+				command 'help',                         'show this usage'
+				command 'version',                      'show the gem version'
+				space
+				command 'list',                         'list your apps'
+				command 'create [<name>]',              'create a new app'
+				space
+				command 'keys',                         'show your user\'s public keys'
+				command 'keys:add [<path to keyfile>]', 'add a public key'
+				command 'keys:remove <keyname> ',       'remove a key by name (user@host)'
+				command 'keys:clear',                   'remove all keys'
+				space
+				command 'info',                         'show app info, like web url and git repo'
+				command 'open',                         'open the app in a web browser'
+				command 'rename <newname>',             'rename the app'
+				space
+				command 'dynos <qty>',                  'scale to qty web processes'
+				command 'workers <qty>',                'scale to qty background processes'
+				space
+				command 'sharing:add <email>',          'add a collaborator'
+				command 'sharing:remove <email>',       'remove a collaborator'
+				command 'sharing:transfer <email>',     'transfers the app ownership'
+				space
+				command 'domains:add <domain>',         'add a custom domain name'
+				command 'domains:remove <domain>',      'remove a custom domain name'
+				command 'domains:clear',                'remove all custom domains'
+				space
+				command 'ssl:add <pem> <key>',          'add SSL cert to the app'
+				command 'ssl:remove <domain>',          'removes SSL cert from the app domain'
+				space
+				command 'rake <command>',               'remotely execute a rake command'
+				command 'console <command>',            'remotely execute a single console command'
+				command 'console',                      'start an interactive console to the remote app'
+				space
+				command 'restart',                      'restart app servers'
+				command 'logs',                         'fetch recent log output for debugging'
+				command 'logs:cron',                    'fetch cron log output'
+				space
+				command 'maintenance:on',               'put the app into maintenance mode'
+				command 'maintenance:off',              'take the app out of maintenance mode'
+				space
+				command 'config',                       'display the app\'s config vars (environment)'
+				command 'config:add key=val [...]',     'add one or more config vars'
+				command 'config:remove key [...]',      'remove one or more config vars'
+				command 'config:clear',                 'clear user-set vars and reset to default'
+				space
+				command 'db:pull [<database_url>]',     'pull the app\'s database into a local database'
+				command 'db:push [<database_url>]',     'push a local database into the app\'s remote database'
+				command 'db:reset',                     'reset the database for the app'
+				space
+				command 'bundles',                      'list bundles for the app'
+				command 'bundles:capture [<bundle>]',   'capture a bundle of the app\'s code and data'
+				command 'bundles:download',             'download most recent app bundle as a tarball'
+				command 'bundles:download <bundle>',    'download the named bundle'
+				command 'bundles:animate <bundle>',     'animate a bundle into a new app'
+				command 'bundles:destroy <bundle>',     'destroy the named bundle'
+				space
+				command 'addons',                       'list installed addons'
+				command 'addons:info',                  'list all available addons'
+				command 'addons:add name [key=value]',  'install addon (with zero or more config vars)'
+				command 'addons:remove name',           'uninstall an addons'
+				command 'addons:clear',                 'uninstall all addons'
+				space
+				command 'destroy',                      'destroy the app permanently'
+			end
+
+			group('Plugins') do
+				command 'plugins',                      'list installed plugins'
+				command 'plugins:install <url>',        'install the plugin from the specified git url'
+				command 'plugins:uninstall <url/name>', 'remove the specified plugin'
+			end
+		end
+
+		def index
+			display usage
+		end
+
+		def version
+			display Heroku::Client.version
+		end
+
+		def usage
+			longest_command_length = self.class.groups.map do |group|
+				group.map { |g| g.first.length }
+			end.flatten.max
+
+			self.class.groups.inject(StringIO.new) do |output, group|
+				output.puts "=== %s" % group.title
+				output.puts
+
+				group.each do |command, description|
+					if command.empty?
+						output.puts
+					else
+						output.puts "%-*s # %s" % [longest_command_length, command, description]
+					end
+				end
+
+				output.puts
+				output
+			end.string + <<-EOTXT
+=== Example:
+
+ rails myapp
+ cd myapp
+ git init
+ git add .
+ git commit -m "my new app"
+ heroku create
+ git push heroku master
+
+EOTXT
+		end
+	end
+end
+
+Heroku::Command::Help.create_default_groups!

data/lib/heroku/commands/keys.rb

@@ -0,0 +1,49 @@
+module Heroku::Command
+	class Keys < Base
+		def list
+			long = args.any? { |a| a == '--long' }
+			keys = heroku.keys
+			if keys.empty?
+				display "No keys for #{heroku.user}"
+			else
+				display "=== #{keys.size} key#{'s' if keys.size > 1} for #{heroku.user}"
+				keys.each do |key|
+					display long ? key.strip : format_key_for_display(key)
+				end
+			end
+		end
+		alias :index :list
+
+		def add
+			keyfile = args.first || find_key
+			key = File.read(keyfile)
+
+			display "Uploading ssh public key #{keyfile}"
+			heroku.add_key(key)
+		end
+
+		def remove
+			heroku.remove_key(args.first)
+			display "Key #{args.first} removed."
+		end
+
+		def clear
+			heroku.remove_all_keys
+			display "All keys removed."
+		end
+
+		protected
+			def find_key
+				%w(rsa dsa).each do |key_type|
+					keyfile = "#{home_directory}/.ssh/id_#{key_type}.pub"
+					return keyfile if File.exists? keyfile
+				end
+				raise CommandFailed, "No ssh public key found in #{home_directory}/.ssh/id_[rd]sa.pub.  You may want to specify the full path to the keyfile."
+			end
+
+			def format_key_for_display(key)
+				type, hex, local = key.strip.split(/\s/)
+				[type, hex[0,10] + '...' + hex[-10,10], local].join(' ')
+			end
+	end
+end

data/lib/heroku/commands/logs.rb

@@ -0,0 +1,11 @@
+module Heroku::Command
+	class Logs < BaseWithApp
+		def index
+			display heroku.logs(app)
+		end
+
+		def cron
+			display heroku.cron_logs(app)
+		end
+	end
+end

data/lib/heroku/commands/maintenance.rb

@@ -0,0 +1,13 @@
+module Heroku::Command
+	class Maintenance < BaseWithApp
+		def on
+			heroku.maintenance(app, :on)
+			display "Maintenance mode enabled."
+		end
+
+		def off
+			heroku.maintenance(app, :off)
+			display "Maintenance mode disabled."
+		end
+	end
+end

data/lib/heroku/commands/plugins.rb

@@ -0,0 +1,25 @@
+module Heroku::Command
+	class Plugins < Base
+		def list
+			::Heroku::Plugin.list.each do |plugin|
+				display plugin
+			end
+		end
+		alias :index :list
+
+		def install
+			plugin = Heroku::Plugin.new(args.shift)
+			if plugin.install
+				display "#{plugin} installed"
+			else
+				error "Could not install #{plugin}. Please check the URL and try again"
+			end
+		end
+
+		def uninstall
+			plugin = Heroku::Plugin.new(args.shift)
+			plugin.uninstall
+			display "#{plugin} uninstalled"
+		end
+	end
+end

data/lib/heroku/commands/ps.rb

@@ -0,0 +1,37 @@
+module Heroku::Command
+	class Ps < Base
+		def index
+			ps = heroku.ps(extract_app)
+
+			output = []
+			output << "UPID     Slug          Command                     State       Since"
+			output << "-------  ------------  --------------------------  ----------  ---------"
+
+			ps.each do |p|
+				output << "%-7s  %-12s  %-26s  %-10s  %-9s" %
+					[p['upid'], p['slug'], truncate(p['command'], 22), p['state'], time_ago(p['elapsed'])]
+			end
+
+			display output.join("\n")
+		end
+
+	private
+		def time_ago(elapsed)
+			if elapsed < 60
+				"#{elapsed.floor}s ago"
+			elsif elapsed < (60 * 60)
+				"#{(elapsed / 60).floor}m ago"
+			else
+				"#{(elapsed / 60 / 60).floor}h ago"
+			end
+		end
+
+		def truncate(text, length)
+			if text.size > length
+				text[0, length - 2] + '..'
+			else
+				text
+			end
+		end
+	end
+end

data/lib/heroku/commands/service.rb

@@ -0,0 +1,23 @@
+module Heroku::Command
+	class Service < Base
+		def start
+			error "heroku service:start is defunct.  Use heroku workers +1 instead."
+		end
+
+		def up
+			error "heroku service:up is defunct.  Use heroku workers +1 instead."
+		end
+
+		def down
+			error "heroku service:down is defunct.  Use heroku workers -1 instead."
+		end
+
+		def bounce
+			error "heroku service:bounce is defunct.  Use heroku restart instead."
+		end
+
+		def status
+			error "heroku service:status is defunct.  Use heroku ps instead."
+		end
+	end
+end

data/lib/heroku/commands/sharing.rb

@@ -0,0 +1,29 @@
+module Heroku::Command
+	class Sharing < BaseWithApp
+		def list
+			list = heroku.list_collaborators(app)
+			display list.map { |c| c[:email] }.join("\n")
+		end
+		alias :index :list
+
+		def add
+			email = args.shift.downcase rescue ''
+			raise(CommandFailed, "Specify an email address to share the app with.") if email == ''
+			display heroku.add_collaborator(app, email)
+		end
+
+		def remove
+			email = args.shift.downcase rescue ''
+			raise(CommandFailed, "Specify an email address to remove from the app.") if email == ''
+			heroku.remove_collaborator(app, email)
+			display "Collaborator removed."
+		end
+
+		def transfer
+			email = args.shift.downcase rescue ''
+			raise(CommandFailed, "Specify the email address of the new owner") if email == ''
+			heroku.update(app, :transfer_owner => email)
+			display "App ownership transfered. New owner is #{email}"
+		end
+	end
+end

data/lib/heroku/commands/ssl.rb

@@ -0,0 +1,33 @@
+module Heroku::Command
+	class Ssl < BaseWithApp
+		def list
+			heroku.list_domains(app).each do |d|
+				if cert = d[:cert]
+					display "#{d[:domain]} has a SSL certificate registered to #{cert[:subject]} which expires on #{cert[:expires_at].strftime("%b %d, %Y")}"
+				else
+					display "#{d[:domain]} has no certificate"
+				end
+			end
+		end
+		alias :index :list
+
+		def add
+			usage  = 'heroku ssl:add <pem> <key>'
+			raise CommandFailed, "Missing pem file. Usage:\n#{usage}" unless pem_file = args.shift
+			raise CommandFailed, "Missing key file. Usage:\n#{usage}" unless key_file = args.shift
+			raise CommandFailed, "Could not find pem in #{pem_file}"  unless File.exists?(pem_file)
+			raise CommandFailed, "Could not find key in #{key_file}"  unless File.exists?(key_file)
+
+			pem  = File.read(pem_file)
+			key  = File.read(key_file)
+			info = heroku.add_ssl(app, pem, key)
+			display "Added certificate to #{info['domain']}, expiring in #{info['expires_at']}"
+		end
+
+		def remove
+			raise CommandFailed, "Missing domain. Usage:\nheroku ssl:remove <domain>" unless domain = args.shift
+			heroku.remove_ssl(app, domain)
+			display "Removed certificate from #{domain}"
+		end
+	end
+end

data/lib/heroku/commands/version.rb

@@ -0,0 +1,7 @@
+module Heroku::Command
+	class Version < Base
+		def index
+			display Heroku::Client.gem_version_string
+		end
+	end
+end

data/lib/heroku/helpers.rb

@@ -0,0 +1,23 @@
+module Heroku
+	module Helpers
+		def home_directory
+			running_on_windows? ? ENV['USERPROFILE'] : ENV['HOME']
+		end
+
+		def running_on_windows?
+			RUBY_PLATFORM =~ /mswin32|mingw32/
+		end
+
+		def running_on_a_mac?
+			RUBY_PLATFORM =~ /-darwin\d/
+		end
+	end
+end
+
+unless String.method_defined?(:shellescape)
+	class String
+		def shellescape
+			empty? ? "''" : gsub(/([^A-Za-z0-9_\-.,:\/@\n])/n, '\\\\\\1').gsub(/\n/, "'\n'")
+		end
+	end
+end

data/lib/heroku/plugin.rb

@@ -0,0 +1,65 @@
+# based on the Rails Plugin
+
+module Heroku
+	class Plugin
+		class << self
+			include Heroku::Helpers
+		end
+
+		attr_reader :name, :uri
+
+		def self.directory
+			"#{home_directory}/.heroku/plugins"
+		end
+
+		def self.list
+			Dir["#{directory}/*"].map do |folder|
+				File.basename(folder)
+			end
+		end
+
+		def self.load!
+			list.each do |plugin|
+				folder = "#{self.directory}/#{plugin}"
+				$: << "#{folder}/lib"    if File.directory? "#{folder}/lib"
+				load "#{folder}/init.rb" if File.exists?    "#{folder}/init.rb"
+			end
+		end
+
+		def initialize(uri)
+			@uri = uri
+			guess_name(uri)
+		end
+
+		def to_s
+			name
+		end
+
+		def path
+			"#{self.class.directory}/#{name}"
+		end
+
+		def install
+			FileUtils.mkdir_p(path)
+			Dir.chdir(path) do
+				system("git init > /dev/null 2>&1")
+				if !system("git pull --depth 1 #{uri}  > /dev/null 2>&1")
+					FileUtils.rm_rf path
+					return false
+				end
+			end
+			true
+		end
+
+		def uninstall
+			FileUtils.rm_r path if File.directory?(path)
+		end
+
+		private 
+			def guess_name(url)
+				@name = File.basename(url)
+				@name = File.basename(File.dirname(url)) if @name.empty?
+				@name.gsub!(/\.git$/, '') if @name =~ /\.git$/
+			end
+	end
+end