trucker-cli 0.0.1

data/lib/cf/cli/app/push/interactions.rb

@@ -0,0 +1,90 @@
+module CF::App
+  module PushInteractions
+    def ask_name
+      ask("Name")
+    end
+
+    def ask_host(name)
+      # Use .dup here because when we pass app.name deep into interactive,
+      # it needs an unfrozen String because the cli allows people to change
+      # this value.
+      host = name.dup
+      ask "Subdomain", :choices => [host, "none"],
+        :default => host,
+        :allow_other => true
+    end
+
+    def ask_domain(app)
+      choices = app.space.domains
+
+      options = {
+        :choices => choices + ["none"],
+        :display => proc { |choice| choice.is_a?(String) ? choice : choice.name },
+        :allow_other => true
+      }
+
+      options[:default] = choices.first
+
+      ask "Domain", options
+    end
+
+    def ask_memory(default)
+      ask("Memory Limit",
+          :choices => memory_choices,
+          :allow_other => true,
+          :default => default || "128M")
+    end
+
+    def ask_instances
+      ask("Instances", :default => 1)
+    end
+
+    def ask_command
+      command = ask("Custom startup command", :default => "none")
+
+      if command != "none"
+        command
+      end
+    end
+
+    def ask_create_services
+      line unless quiet?
+      ask "Create services for application?", :default => false
+    end
+
+    def ask_bind_services
+      return if all_instances.empty?
+
+      ask "Bind other services to application?", :default => false
+    end
+
+    private
+
+    def ask_with_other(message, all, choices, default, other)
+      choices = choices.sort_by(&:name)
+      choices << other if other
+
+      opts = {
+        :choices => choices,
+        :display => proc { |x|
+          if other && x == other
+            "other"
+          else
+            x.name
+          end
+        }
+      }
+
+      opts[:default] = default if default
+
+      res = ask(message, opts)
+
+      if other && res == other
+        opts[:choices] = all
+        res = ask(message, opts)
+      end
+
+      res
+    end
+  end
+end

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

@@ -0,0 +1,57 @@
+module CF::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/cf/cli/app/rename.rb

@@ -0,0 +1,35 @@
+require "cf/cli/app/base"
+
+module CF::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/cf/cli/app/restart.rb

@@ -0,0 +1,31 @@
+require "cf/cli/app/base"
+
+module CF::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/cf/cli/app/scale.rb

@@ -0,0 +1,63 @@
+require "cf/cli/app/base"
+
+module CF::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/cf/cli/app/start.rb

@@ -0,0 +1,175 @@
+require "cf/cli/app/base"
+
+module CF::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 CFoundry::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."
+            line "#{c("TIP: The system will continue to attempt restarting all requested app instances that have crashed. Try 'truck app' to monitor app status. To troubleshoot crashes, try 'truck events' and 'truck crashlogs'.", :warning)}"
+            return
+          end
+
+          begin
+            return unless instances = app.instances
+
+            indented { print_instances_summary(instances) }
+
+            if one_instance_running?(instances)
+              line "#{c("Push successful! App '#{app.name}' available at #{app.host}.#{app.domain}", :good)}"
+              unless all_instances_running?(instances)
+                line "#{c("TIP: The system will continue to start all requested app instances. Try 'truck app' to monitor app status.", :warning)}"
+              end
+              return
+            end
+          rescue CFoundry::NotStaged
+            print (".")
+          end
+
+          sleep 1
+          seconds += 1
+        end
+      rescue CFoundry::StagingError
+        err "Application failed to stage"
+      end
+    end
+
+    def one_instance_running?(instances)
+      instances.any? { |i| i.state == "RUNNING" }
+    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 FLAPPING DOWN}.each do |state|
+        if (num = counts[state]) > 0
+          states << "#{b(num)} #{c(for_output(state), 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
+
+    private
+    def for_output(state)
+      state = "CRASHING" if state == "FLAPPING"
+      state.downcase
+    end
+  end
+end