pogo 2.31.2

data/lib/heroku/command/run.rb

@@ -0,0 +1,179 @@
+require "readline"
+require "heroku/command/base"
+
+# run one-off commands (console, rake)
+#
+class Heroku::Command::Run < Heroku::Command::Base
+
+  # run COMMAND
+  #
+  # run an attached process
+  #
+  #Example:
+  #
+  # $ heroku run bash
+  # Running `bash` attached to terminal... up, run.1
+  # ~ $
+  #
+  def index
+    command = args.join(" ")
+    error("Usage: heroku run COMMAND")if command.empty?
+    run_attached(command)
+  end
+
+  # run:detached COMMAND
+  #
+  # run a detached process, where output is sent to your logs
+  #
+  #Example:
+  #
+  # $ heroku run:detached ls
+  # Running `ls` detached... up, run.1
+  # Use `heroku logs -p run.1` to view the output.
+  #
+  def detached
+    command = args.join(" ")
+    error("Usage: heroku run COMMAND")if command.empty?
+    opts = { :attach => false, :command => command }
+    process_data = action("Running `#{command}` detached", :success => "up") do
+      process_data = api.post_ps(app, command, { :attach => false }).body
+      status(process_data['process'])
+      process_data
+    end
+    display("Use `heroku logs -p #{process_data['process']}` to view the output.")
+  end
+
+  # run:rake COMMAND
+  #
+  # WARNING: `heroku run:rake` has been deprecated. Please use `heroku run rake` instead."
+  #
+  # remotely execute a rake command
+  #
+  #Example:
+  #
+  # $ heroku run:rake -T
+  # Running `rake -T` attached to terminal... up, run.1
+  # (in /app)
+  # rake test  # run tests
+  #
+  def rake
+    deprecate("`heroku #{current_command}` has been deprecated. Please use `heroku run rake` instead.")
+    command = "rake #{args.join(' ')}"
+    run_attached(command)
+  end
+
+  alias_command "rake", "run:rake"
+
+  # run:console [COMMAND]
+  #
+  # open a remote console session
+  #
+  # if COMMAND is specified, run the command and exit
+  #
+  # NOTE: For Cedar apps, use `heroku run console`
+  #
+  #Examples:
+  #
+  # $ heroku console
+  # Ruby console for myapp.heroku.com
+  # >>
+  #
+  def console
+    cmd = args.join(' ').strip
+    if cmd.empty?
+      console_session(app)
+    else
+      display(heroku.console(app, cmd))
+    end
+  rescue RestClient::RequestFailed => e
+    if e.http_body =~ /For Cedar apps, use: `heroku run console`/
+      deprecate("`heroku #{current_command}` has been deprecated for Cedar apps. Please use: `heroku run console` instead.")
+      command = "console #{args.join(' ')}"
+      run_attached(command)
+    else
+      raise(e)
+    end
+  rescue RestClient::RequestTimeout
+    error("Timed out. Long running requests are not supported on the console.\nPlease consider creating a rake task instead.")
+  rescue Heroku::Client::AppCrashed => e
+    error(e.message)
+  end
+
+  alias_command "console", "run:console"
+
+protected
+
+  def run_attached(command)
+    process_data = action("Running `#{command}` attached to terminal", :success => "up") do
+      process_data = api.post_ps(app, command, { :attach => true, :ps_env => get_terminal_environment }).body
+      status(process_data["process"])
+      process_data
+    end
+    rendezvous_session(process_data["rendezvous_url"])
+  end
+
+  def rendezvous_session(rendezvous_url, &on_connect)
+    begin
+      set_buffer(false)
+      rendezvous = Heroku::Client::Rendezvous.new(
+        :rendezvous_url => rendezvous_url,
+        :connect_timeout => (ENV["HEROKU_CONNECT_TIMEOUT"] || 120).to_i,
+        :activity_timeout => nil,
+        :input => $stdin,
+        :output => $stdout)
+      rendezvous.on_connect(&on_connect)
+      rendezvous.start
+    rescue Timeout::Error
+      error "\nTimeout awaiting process"
+    rescue Errno::ECONNREFUSED, Errno::ECONNRESET, OpenSSL::SSL::SSLError
+      error "\nError connecting to process"
+    rescue Interrupt
+    ensure
+      set_buffer(true)
+    end
+  end
+
+  def console_history_dir
+    FileUtils.mkdir_p(path = "#{home_directory}/.heroku/console_history")
+    path
+  end
+
+  def console_session(app)
+    heroku.console(app) do |console|
+      console_history_read(app)
+
+      display "Ruby console for #{app}.#{heroku.host}"
+      while cmd = Readline.readline('>> ')
+        unless cmd.nil? || cmd.strip.empty?
+          console_history_add(app, cmd)
+          break if cmd.downcase.strip == 'exit'
+          display console.run(cmd)
+        end
+      end
+    end
+  end
+
+  def console_history_file(app)
+    "#{console_history_dir}/#{app}"
+  end
+
+  def console_history_read(app)
+    history = File.read(console_history_file(app)).split("\n")
+    if history.size > 50
+      history = history[(history.size - 51),(history.size - 1)]
+      File.open(console_history_file(app), "w") { |f| f.puts history.join("\n") }
+    end
+    history.each { |cmd| Readline::HISTORY.push(cmd) }
+  rescue Errno::ENOENT
+  rescue Exception => ex
+    display "Error reading your console history: #{ex.message}"
+    if confirm("Would you like to clear it? (y/N):")
+      FileUtils.rm(console_history_file(app)) rescue nil
+    end
+  end
+
+  def console_history_add(app, cmd)
+    Readline::HISTORY.push(cmd)
+    File.open(console_history_file(app), "a") { |f| f.puts cmd + "\n" }
+  end
+end

data/lib/heroku/command/sharing.rb

@@ -0,0 +1,89 @@
+require "heroku/command/base"
+
+module Heroku::Command
+
+  # manage collaborators on an app
+  #
+  class Sharing < Base
+
+    # sharing
+    #
+    # list collaborators on an app
+    #
+    #Example:
+    #
+    # $ heroku sharing
+    # === myapp Collaborators
+    # collaborator@example.com
+    # email@example.com
+    #
+    def index
+      validate_arguments!
+
+      # this is never empty, as it always includes the owner
+      collaborators = api.get_collaborators(app).body
+      styled_header("#{app} Collaborators")
+      styled_array(collaborators.map {|collaborator| collaborator["email"]})
+    end
+
+    # sharing:add EMAIL
+    #
+    # add a collaborator to an app
+    #
+    #Example:
+    #
+    # $ heroku sharing:add collaborator@example.com
+    # Adding collaborator@example.com to myapp collaborators... done
+    #
+    def add
+      unless email = shift_argument
+        error("Usage: heroku sharing:add EMAIL\nMust specify EMAIL to add sharing.")
+      end
+      validate_arguments!
+
+      action("Adding #{email} to #{app} collaborators") do
+        api.post_collaborator(app, email)
+      end
+    end
+
+    # sharing:remove EMAIL
+    #
+    # remove a collaborator from an app
+    #
+    #Example:
+    #
+    # $ heroku sharing:remove collaborator@example.com
+    # Removing collaborator@example.com to myapp collaborators... done
+    #
+    def remove
+      unless email = shift_argument
+        error("Usage: heroku sharing:remove EMAIL\nMust specify EMAIL to remove sharing.")
+      end
+      validate_arguments!
+
+      action("Removing #{email} from #{app} collaborators") do
+        api.delete_collaborator(app, email)
+      end
+    end
+
+    # sharing:transfer EMAIL
+    #
+    # transfer an app to a new owner
+    #
+    #Example:
+    #
+    # $ heroku sharing:transfer collaborator@example.com
+    # Transferring myapp to collaborator@example.com... done
+    #
+    def transfer
+      unless email = shift_argument
+        error("Usage: heroku sharing:transfer EMAIL\nMust specify EMAIL to transfer an app.")
+      end
+      validate_arguments!
+
+      action("Transferring #{app} to #{email}") do
+        api.put_app(app, "transfer_owner" => email)
+      end
+    end
+  end
+end

data/lib/heroku/command/ssl.rb

@@ -0,0 +1,61 @@
+require "heroku/command/base"
+
+module Heroku::Command
+
+  # manage ssl certificates for an app
+  #
+  class Ssl < Base
+
+    # ssl
+    #
+    # list certificates for an app
+    #
+    def index
+      api.get_domains(app).body.each do |domain|
+        if cert = domain['cert']
+          display "#{domain['domain']} has a SSL certificate registered to #{cert['subject']} which expires on #{format_date(cert['expires_at'])}"
+        else
+          display "#{domain['domain']} has no certificate"
+        end
+      end
+    end
+
+    # ssl:add PEM KEY
+    #
+    # add an ssl certificate to an app
+    #
+    def add
+      pem_file = args.shift
+      key_file = args.shift
+      fail "Usage: heroku ssl:add PEM KEY" unless pem_file && key_file
+      raise CommandFailed, "Missing pem file." unless pem_file
+      raise CommandFailed, "Missing key file." unless key_file
+      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 at #{info['expires_at']}"
+    end
+
+    # ssl:remove DOMAIN
+    #
+    # remove an ssl certificate from an app
+    #
+    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
+
+    # ssl:clear
+    #
+    # remove all ssl certificates from an app
+    #
+    def clear
+      heroku.clear_ssl(app)
+      display "Cleared certificates for #{app}"
+    end
+  end
+end

data/lib/heroku/command/stack.rb

@@ -0,0 +1,62 @@
+require "heroku/command/base"
+
+module Heroku::Command
+
+  # manage the stack for an app
+  class Stack < Base
+
+    # stack
+    #
+    # show the list of available stacks
+    #
+    #Example:
+    #
+    # $ heroku stack
+    # === myapp Available Stacks
+    #   bamboo-mri-1.9.2
+    #   bamboo-ree-1.8.7
+    # * cedar
+    #
+    def index
+      validate_arguments!
+
+      stacks_data = api.get_stack(app).body
+
+      styled_header("#{app} Available Stacks")
+      stacks = stacks_data.map do |stack|
+        row = [stack['current'] ? '*' : ' ', stack['name']]
+        row << '(beta)' if stack['beta']
+        row << '(deprecated)' if stack['deprecated']
+        row << '(prepared, will migrate on next git push)' if stack['requested']
+        row.join(' ')
+      end
+      styled_array(stacks)
+    end
+
+    # stack:migrate STACK
+    #
+    # prepare migration of this app to a new stack
+    #
+    #Example:
+    #
+    # $ heroku stack:migrate cedar
+    # -----> Preparing to migrate evening-warrior-2345
+    #        bamboo-mri-1.9.2 -> bamboo-ree-1.8.7
+    #
+    #        NOTE: You must specify ALL gems (including Rails) in manifest
+    #
+    #        Please read the migration guide:
+    #        http://devcenter.heroku.com/articles/bamboo
+    #
+    # -----> Migration prepared.
+    #        Run 'git push heroku master' to execute migration.
+    #
+    def migrate
+      unless stack = shift_argument
+        error("Usage: heroku stack:migrate STACK.\nMust specify target stack.")
+      end
+
+      display(api.put_stack(app, stack).body)
+    end
+  end
+end

data/lib/heroku/command/status.rb

@@ -0,0 +1,51 @@
+require "heroku/command/base"
+
+# check status of heroku platform
+#
+class Heroku::Command::Status < Heroku::Command::Base
+
+  # status
+  #
+  # display current status of heroku platform
+  #
+  #Example:
+  #
+  # $ heroku status
+  # === Heroku Status
+  # Development: No known issues at this time.
+  # Production:  No known issues at this time.
+  #
+  def index
+    validate_arguments!
+
+    heroku_status_host = ENV['HEROKU_STATUS_HOST'] || "status.heroku.com"
+    require('excon')
+    status = json_decode(Excon.get("https://#{heroku_status_host}/api/v3/current-status.json", :nonblock => false).body)
+
+    styled_header("Heroku Status")
+
+    status['status'].each do |key, value|
+      if value == 'green'
+        status['status'][key] = 'No known issues at this time.'
+      end
+    end
+    styled_hash(status['status'])
+
+    unless status['issues'].empty?
+      display
+      status['issues'].each do |issue|
+        duration = time_ago(issue['created_at']).gsub(' ago', '+')
+        styled_header("#{issue['title']}  #{duration}")
+        changes = issue['updates'].map do |issue|
+          [
+            time_ago(Time.now - Time.parse(issue['created_at'])),
+            issue['update_type'],
+            issue['contents']
+          ]
+        end
+        styled_array(changes, :sort => false)
+      end
+    end
+  end
+
+end

data/lib/heroku/command/update.rb

@@ -0,0 +1,47 @@
+require "heroku/command/base"
+require "heroku/updater"
+
+# update the heroku client
+#
+class Heroku::Command::Update < Heroku::Command::Base
+
+  # update
+  #
+  # update the heroku client
+  #
+  # Example:
+  #
+  # $ heroku update
+  # Updating from v1.2.3... done, updated to v2.3.4
+  #
+  def index
+    validate_arguments!
+    update_from_url("https://toolbelt.herokuapp.com/download/zip")
+  end
+
+  # update:beta
+  #
+  # update to the latest beta client
+  #
+  # $ heroku update
+  # Updating from v1.2.3... done, updated to v2.3.4.pre
+  #
+  def beta
+    validate_arguments!
+    update_from_url("https://toolbelt.herokuapp.com/download/beta-zip")
+  end
+
+private
+
+  def update_from_url(url)
+    Heroku::Updater.check_disabled!
+    action("Updating from #{Heroku::VERSION}") do
+      if new_version = Heroku::Updater.update(url)
+        status("updated to #{new_version}")
+      else
+        status("nothing to update")
+      end
+    end
+  end
+
+end