jdc 0.2.1 → 0.2.2.pre

data/lib/jdc/cli/app/push/sync.rb

@@ -0,0 +1,57 @@
+module JDC::App
+  module Sync
+    def apply_changes(app)
+      app.memory = megabytes(input[:memory]) if input.has?(:memory)
+      app.total_instances = input[:instances] if input.has?(:instances)
+      app.command = input[:command] if input.has?(:command)
+      app.buildpack = input[:buildpack] if input.has?(:buildpack)
+    end
+
+    def display_changes(app)
+      return unless app.changed?
+
+      line "Changes:"
+
+      indented do
+        app.changes.each do |attr, (old, new)|
+          line "#{c(attr, :name)}: #{diff_str(attr, old)} -> #{diff_str(attr, new)}"
+        end
+      end
+    end
+
+    def commit_changes(app)
+      if app.changed?
+        with_progress("Updating #{c(app.name, :name)}") do
+          wrap_message_format_errors do
+            app.update!
+          end
+        end
+      end
+
+      if input[:restart] && app.started?
+        invoke :restart, :app => app
+      end
+    end
+
+    private
+
+    def diff_str(attr, val)
+      case attr
+      when :memory
+        human_mb(val)
+      when :command, :buildpack
+        "'#{val}'"
+      else
+        val
+      end
+    end
+
+    def bool(b)
+      if b
+        c("true", :yes)
+      else
+        c("false", :no)
+      end
+    end
+  end
+end

data/lib/jdc/cli/app/rename.rb

@@ -0,0 +1,35 @@
+require "jdc/cli/app/base"
+
+module JDC::App
+  class Rename < Base
+    desc "Rename an application"
+    group :apps, :manage
+    input :app, :desc => "Application to rename", :argument => :optional,
+          :from_given => by_name(:app)
+    input :name, :desc => "New application name", :argument => :optional
+    def rename
+      app = input[:app]
+      name = input[:name]
+
+      app.name = name
+
+      with_progress("Renaming to #{c(name, :name)}") do
+        app.update!
+      end
+    end
+
+    private
+
+    def ask_app
+      apps = client.apps
+      fail "No applications." if apps.empty?
+
+      ask("Rename which application?", :choices => apps.sort_by(&:name),
+          :display => proc(&:name))
+    end
+
+    def ask_name
+      ask("New name")
+    end
+  end
+end

data/lib/jdc/cli/app/restart.rb

@@ -0,0 +1,31 @@
+require "jdc/cli/app/base"
+
+module JDC::App
+  class Restart < Base
+    desc "Stop and start an application"
+    group :apps, :manage
+    input :apps, :desc => "Applications to start", :argument => :splat,
+          :singular => :app, :from_given => by_name(:app)
+    input :debug_mode, :desc => "Debug mode to start in", :aliases => "-d"
+    input :all, :desc => "Restart all applications", :default => false
+
+    ############# Uncomment to complete 50543607
+    #input :command, :desc => "Command to restart application", :default => nil
+
+    def restart
+      invoke :stop, :all => input[:all], :apps => input[:apps]
+
+      line unless quiet?
+
+      input[:apps].each do |app|
+        unless input[:command].nil?
+          app.command = input[:command]
+        end
+        app.update!
+      end
+
+      invoke :start, :all => input[:all], :apps => input[:apps],
+        :debug_mode => input[:debug_mode]
+    end
+  end
+end

data/lib/jdc/cli/app/scale.rb

@@ -0,0 +1,63 @@
+require "jdc/cli/app/base"
+
+module JDC::App
+  class Scale < Base
+    desc "Update the instances/memory limit for an application"
+    group :apps, :info
+    input :app, :desc => "Application to update", :argument => true,
+          :from_given => by_name(:app)
+    input :instances, :desc => "Number of instances to run",
+          :type => :numeric
+    input :memory, :desc => "Memory limit"
+    input :disk, :desc => "Disk quota"
+    input :restart, :desc => "Restart app after updating?", :default => true
+
+    def scale
+      app = input[:app]
+
+      if input.has?(:instances)
+        instances = input[:instances, app.total_instances]
+      end
+
+      if input.has?(:memory)
+        memory = input[:memory, app.memory]
+      end
+
+      if input.has?(:disk)
+        disk = input[:disk, human_mb(app.disk_quota)]
+      end
+
+      unless instances || memory || disk
+        instances = input[:instances, app.total_instances]
+        memory = input[:memory, app.memory]
+      end
+
+      app.total_instances = instances if input.has?(:instances)
+      app.memory = megabytes(memory) if input.has?(:memory)
+      app.disk_quota = megabytes(disk) if input.has?(:disk)
+
+      fail "No changes!" unless app.changed?
+
+      with_progress("Scaling #{c(app.name, :name)}") do
+        app.update!
+      end
+
+      needs_restart = app.changes.key?(:memory) || app.changes.key?(:disk_quota)
+
+      if needs_restart && app.started? && input[:restart]
+        invoke :restart, :app => app
+      end
+    end
+
+    private
+
+    def ask_instances(default)
+      ask("Instances", :default => default)
+    end
+
+    def ask_memory(default)
+      ask("Memory Limit", :choices => memory_choices,
+          :default => human_mb(default), :allow_other => true)
+    end
+  end
+end

data/lib/jdc/cli/app/start.rb

@@ -0,0 +1,161 @@
+require "jdc/cli/app/base"
+
+module JDC::App
+  class Start < Base
+    APP_CHECK_LIMIT = 60
+
+    desc "Start an application"
+    group :apps, :manage
+    input :apps, :desc => "Applications to start", :argument => :splat,
+          :singular => :app, :from_given => by_name(:app)
+    input :debug_mode, :desc => "Debug mode to start in", :aliases => "-d"
+    input :all, :desc => "Start all applications", :default => false
+    def start
+      apps = input[:all] ? client.apps : input[:apps]
+      fail "No applications given." if apps.empty?
+
+      spaced(apps) do |app|
+        app = filter(:start_app, app)
+
+        switch_mode(app, input[:debug_mode])
+
+        if app.started?
+          err "Application #{b(app.name)} is already started."
+          next
+        end
+
+        log = start_app(app)
+        stream_start_log(log) if log
+        check_application(app)
+
+        if !app.debug_mode.nil? && app.debug_mode != "none" && !quiet?
+          line
+          invoke :instances, :app => app
+        end
+      end
+    end
+
+    private
+
+    def start_app(app)
+      log = nil
+      with_progress("Preparing to start #{c(app.name, :name)}") do
+        app.start! do |url|
+          log = url
+        end
+      end
+      log
+    end
+
+    def stream_start_log(log)
+      offset = 0
+
+      while true
+        begin
+          client.stream_url(log + "&tail&tail_offset=#{offset}") do |out|
+            offset += out.size
+            print out
+          end
+        rescue Timeout::Error
+        end
+      end
+    rescue JFoundry::APIError
+    end
+
+    # set app debug mode, ensuring it's valid, and shutting it down
+    def switch_mode(app, mode)
+      mode = "run" if mode == "" # no value given
+
+      return false if app.debug == mode
+
+      if mode == "none"
+        with_progress("Removing debug mode") do
+          app.debug = mode
+          app.stop! if app.started?
+        end
+
+        return true
+      end
+
+      with_progress("Switching mode to #{c(mode, :name)}") do |s|
+        app.debug = mode
+        app.stop! if app.started?
+      end
+    end
+
+    def check_application(app)
+      if app.debug == "suspend"
+        line "Application is in suspended debugging mode."
+        line "It will wait for you to attach to it before starting."
+        return
+      end
+
+      print("Checking status of app '#{c(app.name, :name)}'...")
+
+      seconds = 0
+      @first_time_after_staging_succeeded = true
+
+      begin
+        instances = []
+        while true
+          if any_instance_flapping?(instances) || seconds == APP_CHECK_LIMIT
+            err "Push unsuccessful."
+            return
+          end
+
+          begin
+            return unless instances = app.instances
+
+            indented { print_instances_summary(instances) }
+
+            if all_instances_running?(instances)
+              line "#{c("Push successful! App '#{app.name}' available at http://#{app.host}.#{app.domain}", :good)}"
+              return
+            end
+          rescue JFoundry::NotStaged
+            print (".")
+          end
+
+          sleep 1
+          seconds += 1
+        end
+      rescue JFoundry::StagingError
+        err "Application failed to stage"
+      end
+    end
+
+    def all_instances_running?(instances)
+      instances.all? { |i| i.state == "RUNNING" }
+    end
+
+    def any_instance_flapping?(instances)
+      instances.any? { |i| i.state == "FLAPPING" }
+    end
+
+    def print_instances_summary(instances)
+
+      if @first_time_after_staging_succeeded
+        line
+        @first_time_after_staging_succeeded = false
+      end
+
+      counts = Hash.new { 0 }
+      instances.each do |i|
+        counts[i.state] += 1
+      end
+
+      states = []
+      %w{RUNNING STARTING DOWN FLAPPING}.each do |state|
+        if (num = counts[state]) > 0
+          states << "#{b(num)} #{c(state.downcase, state_color(state))}"
+        end
+      end
+
+      total = instances.count
+      running = counts["RUNNING"].to_s.rjust(total.to_s.size)
+
+      ratio = "#{running}#{d(" of ")}#{total} instances running"
+      line "#{ratio} (#{states.join(", ")})"
+    end
+  end
+end

data/lib/jdc/cli/app/stats.rb

@@ -0,0 +1,67 @@
+require "jdc/cli/app/base"
+
+module JDC::App
+  class Stats < Base
+    desc "Display application instance status"
+    group :apps, :info
+    input :app, :desc => "Application to get the stats for",
+          :argument => true, :from_given => by_name(:app)
+    def stats
+      app = input[:app]
+
+      stats =
+        with_progress("Getting stats for #{c(app.name, :name)}") do |s|
+          begin
+            app.stats
+          rescue JFoundry::StatsError
+            s.fail do
+              err "Application #{b(app.name)} is not running."
+              return
+            end
+          end
+        end
+
+      line unless quiet?
+
+      table(
+        %w{instance cpu memory disk},
+        stats.sort_by { |idx, _| idx.to_i }.collect { |idx, info|
+          idx = c("\##{idx}", :instance)
+
+          if info[:state] == "DOWN"
+            [idx, c("down", :bad)]
+          else
+            stats = info[:stats]
+            usage = stats[:usage]
+
+            if usage
+              [ idx,
+                "#{percentage(usage[:cpu])}",
+                "#{usage(usage[:mem], stats[:mem_quota])}",
+                "#{usage(usage[:disk], stats[:disk_quota])}"
+              ]
+            else
+              [idx, c("n/a", :neutral)]
+            end
+          end
+        })
+    end
+
+    def percentage(num, low = 50, mid = 70)
+      color =
+        if num <= low
+          :good
+        elsif num <= mid
+          :warning
+        else
+          :bad
+        end
+
+      c(format("%.1f\%", num), color)
+    end
+
+    def usage(used, limit)
+      "#{b(human_size(used))} of #{b(human_size(limit, 0))}"
+    end
+  end
+end

data/lib/jdc/cli/app/stop.rb

@@ -0,0 +1,27 @@
+require "jdc/cli/app/base"
+
+module JDC::App
+  class Stop < Base
+    desc "Stop an application"
+    group :apps, :manage
+    input :apps, :desc => "Applications to start", :argument => :splat,
+          :singular => :app, :from_given => by_name(:app)
+    input :all, :desc => "Stop all applications", :default => false
+    def stop
+      apps = input[:all] ? client.apps : input[:apps]
+      fail "No applications given." if apps.empty?
+
+      apps.each do |app|
+        with_progress("Stopping #{c(app.name, :name)}") do |s|
+          if app.stopped?
+            s.skip do
+              err "Application is not running."
+            end
+          end
+
+          app.stop!
+        end
+      end
+    end
+  end
+end