pogo 2.31.2

data/lib/heroku/command/plugins.rb

@@ -0,0 +1,110 @@
+require "heroku/command/base"
+
+module Heroku::Command
+
+  # manage plugins to the heroku gem
+  class Plugins < Base
+
+    # plugins
+    #
+    # list installed plugins
+    #
+    #Example:
+    #
+    # $ heroku plugins
+    # === Installed Plugins
+    # heroku-accounts
+    #
+    def index
+      validate_arguments!
+
+      plugins = ::Heroku::Plugin.list
+
+      if plugins.length > 0
+        styled_header("Installed Plugins")
+        styled_array(plugins)
+      else
+        display("You have no installed plugins.")
+      end
+    end
+
+    # plugins:install URL
+    #
+    # install a plugin
+    #
+    #Example:
+    #
+    # $ heroku plugins:install https://github.com/ddollar/heroku-accounts.git
+    # Installing heroku-accounts... done
+    #
+    def install
+      plugin = Heroku::Plugin.new(shift_argument)
+      validate_arguments!
+
+      action("Installing #{plugin.name}") do
+        if plugin.install
+          unless Heroku::Plugin.load_plugin(plugin.name)
+            plugin.uninstall
+            exit(1)
+          end
+        else
+          error("Could not install #{plugin.name}. Please check the URL and try again")
+        end
+      end
+    end
+
+    # plugins:uninstall PLUGIN
+    #
+    # uninstall a plugin
+    #
+    #Example:
+    #
+    # $ heroku plugins:uninstall heroku-accounts
+    # Uninstalling heroku-accounts... done
+    #
+    def uninstall
+      plugin = Heroku::Plugin.new(shift_argument)
+      validate_arguments!
+
+      action("Uninstalling #{plugin.name}") do
+        plugin.uninstall
+      end
+    end
+
+    # plugins:update [PLUGIN]
+    #
+    # updates all plugins or a single plugin by name
+    #
+    #Example:
+    #
+    # $ heroku plugins:update
+    # Updating heroku-accounts... done
+    #
+    # $ heroku plugins:update heroku-accounts
+    # Updating heroku-accounts... done
+    #
+    def update
+      plugins = if plugin = shift_argument
+        [plugin]
+      else
+        ::Heroku::Plugin.list
+      end
+      validate_arguments!
+
+      plugins.each do |plugin|
+        begin
+          action("Updating #{plugin}") do
+            begin
+              Heroku::Plugin.new(plugin).update
+            rescue Heroku::Plugin::ErrorUpdatingSymlinkPlugin
+              status "skipped symlink"
+            end
+          end
+        rescue SystemExit
+          # ignore so that other plugins still update
+        end
+      end
+    end
+
+  end
+end

data/lib/heroku/command/ps.rb

@@ -0,0 +1,232 @@
+require "heroku/command/base"
+
+# manage processes (dynos, workers)
+#
+class Heroku::Command::Ps < Heroku::Command::Base
+
+  # ps:dynos [QTY]
+  #
+  # DEPRECATED: use `heroku ps:scale dynos=N`
+  #
+  # scale to QTY web processes
+  #
+  # if QTY is not specified, display the number of web processes currently running
+  #
+  #Example:
+  #
+  # $ heroku ps:dynos 3
+  # Scaling dynos... done, now running 3
+  #
+  def dynos
+    # deprecation notice added to v2.21.3 on 03/16/12
+    display("~ `heroku ps:dynos QTY` has been deprecated and replaced with `heroku ps:scale dynos=QTY`")
+
+    dynos = shift_argument
+    validate_arguments!
+
+    if dynos
+      action("Scaling dynos") do
+        new_dynos = api.put_dynos(app, dynos).body["dynos"]
+        status("now running #{new_dynos}")
+      end
+    else
+      app_data = api.get_app(app).body
+      if app_data["stack"] == "cedar"
+        raise(Heroku::Command::CommandFailed, "For Cedar apps, use `heroku ps`")
+      else
+        display("#{app} is running #{quantify("dyno", app_data["dynos"])}")
+      end
+    end
+  end
+
+  alias_command "dynos", "ps:dynos"
+
+  # ps:workers [QTY]
+  #
+  # DEPRECATED: use `heroku ps:scale workers=N`
+  #
+  # scale to QTY background processes
+  #
+  # if QTY is not specified, display the number of background processes currently running
+  #
+  #Example:
+  #
+  # $ heroku ps:dynos 3
+  # Scaling workers... done, now running 3
+  #
+  def workers
+    # deprecation notice added to v2.21.3 on 03/16/12
+    display("~ `heroku ps:workers QTY` has been deprecated and replaced with `heroku ps:scale workers=QTY`")
+
+    workers = shift_argument
+    validate_arguments!
+
+    if workers
+      action("Scaling workers") do
+        new_workers = api.put_workers(app, workers).body["workers"]
+        status("now running #{new_workers}")
+      end
+    else
+      app_data = api.get_app(app).body
+      if app_data["stack"] == "cedar"
+        raise(Heroku::Command::CommandFailed, "For Cedar apps, use `heroku ps`")
+      else
+        display("#{app} is running #{quantify("worker", app_data["workers"])}")
+      end
+    end
+  end
+
+  alias_command "workers", "ps:workers"
+
+  # ps
+  #
+  # list processes for an app
+  #
+  #Example:
+  #
+  # $ heroku ps
+  # === run: one-off processes
+  # run.1: up for 5m: `bash`
+  #
+  # === web: `bundle exec thin start -p $PORT`
+  # web.1: created for 30s
+  #
+  def index
+    validate_arguments!
+    processes = api.get_ps(app).body
+
+    processes_by_command = Hash.new {|hash,key| hash[key] = []}
+    processes.each do |process|
+      name    = process["process"].split(".").first
+      elapsed = time_ago(Time.now - process['elapsed'])
+
+      if name == "run"
+        key  = "run: one-off processes"
+        item = "%s: %s %s: `%s`" % [ process["process"], process["state"], elapsed, process["command"] ]
+      else
+        key  = "#{name}: `#{process["command"]}`"
+        item = "%s: %s %s" % [ process["process"], process["state"], elapsed ]
+      end
+
+      processes_by_command[key] << item
+    end
+
+    processes_by_command.keys.each do |key|
+      processes_by_command[key] = processes_by_command[key].sort do |x,y|
+        x.match(/\.(\d+):/).captures.first.to_i <=> y.match(/\.(\d+):/).captures.first.to_i
+      end
+    end
+
+    processes_by_command.keys.sort.each do |key|
+      styled_header(key)
+      styled_array(processes_by_command[key], :sort => false)
+    end
+  end
+
+  # ps:restart [PROCESS]
+  #
+  # restart an app process
+  #
+  # if PROCESS is not specified, restarts all processes on the app
+  #
+  #Examples:
+  #
+  # $ heroku ps:restart web.1
+  # Restarting web.1 process... done
+  #
+  # $ heroku ps:restart web
+  # Restarting web processes... done
+  #
+  # $ heroku ps:restart
+  # Restarting processes... done
+  #
+  def restart
+    process = shift_argument
+    validate_arguments!
+
+    message, options = case process
+    when NilClass
+      ["Restarting processes", {}]
+    when /.+\..+/
+      ps = args.first
+      ["Restarting #{ps} process", { :ps => ps }]
+    else
+      type = args.first
+      ["Restarting #{type} processes", { :type => type }]
+    end
+
+    action(message) do
+      api.post_ps_restart(app, options)
+    end
+  end
+
+  alias_command "restart", "ps:restart"
+
+  # ps:scale PROCESS1=AMOUNT1 [PROCESS2=AMOUNT2 ...]
+  #
+  # scale processes by the given amount
+  #
+  #Examples:
+  #
+  # $ heroku ps:scale web=3 worker+1
+  # Scaling web processes... done, now running 3
+  # Scaling worker processes... done, now running 1
+  #
+  def scale
+    changes = {}
+    args.each do |arg|
+      if arg =~ /^([a-zA-Z0-9_]+)([=+-]\d+)$/
+        changes[$1] = $2
+      end
+    end
+
+    if changes.empty?
+      error("Usage: heroku ps:scale PROCESS1=AMOUNT1 [PROCESS2=AMOUNT2 ...]\nMust specify PROCESS and AMOUNT to scale.")
+    end
+
+    changes.keys.sort.each do |process|
+      amount = changes[process]
+      action("Scaling #{process} processes") do
+        amount.gsub!("=", "")
+        new_qty = api.post_ps_scale(app, process, amount).body
+        status("now running #{new_qty}")
+      end
+    end
+  end
+
+  alias_command "scale", "ps:scale"
+
+  # ps:stop PROCESS
+  #
+  # stop an app process
+  #
+  # Examples:
+  #
+  # $ heroku stop run.3
+  # Stopping run.3 process... done
+  #
+  # $ heroku stop run
+  # Stopping run processes... done
+  #
+  def stop
+    process = shift_argument
+    validate_arguments!
+
+    message, options = case process
+    when NilClass
+      error("Usage: heroku ps:stop PROCESS\nMust specify PROCESS to stop.")
+    when /.+\..+/
+      ps = args.first
+      ["Stopping #{ps} process", { :ps => ps }]
+    else
+      type = args.first
+      ["Stopping #{type} processes", { :type => type }]
+    end
+
+    action(message) do
+      api.post_ps_stop(app, options)
+    end
+  end
+
+  alias_command "stop", "ps:stop"
+end

data/lib/heroku/command/releases.rb

@@ -0,0 +1,124 @@
+require "heroku/command/base"
+
+# manage app releases
+#
+class Heroku::Command::Releases < Heroku::Command::Base
+
+  # releases
+  #
+  # list releases
+  #
+  #Example:
+  #
+  # $ heroku releases
+  # === myapp Releases
+  # v1 Config add FOO_BAR by email@example.com 0s ago
+  # v2 Config add BAR_BAZ by email@example.com 0s ago
+  # v3 Config add BAZ_QUX by email@example.com 0s ago
+  #
+  def index
+    validate_arguments!
+
+    releases_data = api.get_releases(app).body.sort_by do |release|
+      release["name"][1..-1].to_i
+    end.reverse.slice(0, 15)
+
+    unless releases_data.empty?
+      releases = releases_data.map do |release|
+        [
+          release["name"],
+          truncate(release["descr"], 40),
+          release["user"],
+          time_ago(release['created_at'])
+        ]
+      end
+
+      styled_header("#{app} Releases")
+      styled_array(releases, :sort => false)
+    else
+      display("#{app} has no releases.")
+    end
+  end
+
+  # releases:info RELEASE
+  #
+  # view detailed information for a release
+  # find latest release details by passing 'current' as the release
+  #
+  # -s, --shell  # output config vars in shell format
+  #
+  #Example:
+  #
+  # $ heroku releases:info v10
+  # === Release v10
+  # Addons: deployhooks:http
+  # By:     email@example.com
+  # Change: deploy ABCDEFG
+  # When:   2012-01-01 12:00:00
+  #
+  # === v10 Config Vars
+  # EXAMPLE: foo
+  #
+  def info
+    unless release = shift_argument
+      error("Usage: heroku releases:info RELEASE")
+    end
+    validate_arguments!
+
+    release_data = api.get_release(app, release).body
+
+    data = {
+      'By'     => release_data['user'],
+      'Change' => release_data['descr'],
+      'When'   => time_ago(release_data["created_at"])
+    }
+
+    unless release_data['addons'].empty?
+      data['Addons'] = release_data['addons']
+    end
+
+    styled_header("Release #{release}")
+    styled_hash(data)
+
+    display
+
+    styled_header("#{release} Config Vars")
+    unless release_data['env'].empty?
+      if options[:shell]
+        release_data['env'].keys.sort.each do |key|
+          display("#{key}=#{release_data['env'][key]}")
+        end
+      else
+        styled_hash(release_data['env'])
+      end
+    else
+      display("#{release} has no config vars.")
+    end
+  end
+
+  # releases:rollback [RELEASE]
+  #
+  # roll back to an older release
+  #
+  # if RELEASE is not specified, will roll back one step
+  #
+  #Example:
+  #
+  # $ heroku releases:rollback
+  # Rolling back myapp... done, v122
+  #
+  # $ heroku releases:rollback v42
+  # Rolling back myapp to v42... done
+  #
+  def rollback
+    release = shift_argument
+    validate_arguments!
+
+    action("Rolling back #{app}") do
+      status(api.post_release(app, release).body)
+    end
+  end
+
+  alias_command "rollback", "releases:rollback"
+
+end