trucker-cli 0.0.1

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

@@ -0,0 +1,43 @@
+require "cf/cli/app/base"
+
+module CF::App
+  class Apps < Base
+    IS_UTF8 = !!(ENV["LC_ALL"] || ENV["LC_CTYPE"] || ENV["LANG"] || "")["UTF-8"].freeze
+
+    desc "Show app information"
+    group :apps
+    input :app, :desc => "App to show", :argument => :required,
+          :from_given => by_name(:app)
+    def app
+      app = input[:app]
+
+      if quiet?
+        line app.name
+      else
+        display_app(app)
+      end
+    end
+
+    def display_app(a)
+      status = app_status(a)
+
+      line "#{c(a.name, :name)}: #{status}"
+
+      indented do
+        start_line "usage: #{b(human_mb(a.memory))}"
+        print " #{d(IS_UTF8 ? "\xc3\x97" : "x")} #{b(a.total_instances)}"
+        print " instance#{a.total_instances == 1 ? "" : "s"}"
+
+        line
+
+        unless a.urls.empty?
+          line "urls: #{a.urls.collect { |u| b(u) }.join(", ")}"
+        end
+
+        unless a.services.empty?
+          line "services: #{a.services.collect { |s| b(s.name) }.join(", ")}"
+        end
+      end
+    end
+  end
+end

data/lib/cf/cli/app/apps.rb

@@ -0,0 +1,87 @@
+require "cf/cli/app/base"
+
+module CF::App
+  class Apps < Base
+    desc "List your applications"
+    group :apps
+    input :space, :desc => "Show apps in given space",
+          :default => proc { client.current_space },
+          :from_given => by_name(:space)
+    input :name, :desc => "Filter by name regexp"
+    input :url, :desc => "Filter by url regexp"
+    input :full, :desc => "Verbose output format", :default => false
+    def apps
+      if space = input[:space]
+        begin
+          space.summarize!
+        rescue CFoundry::APIError
+        end
+
+        apps =
+          with_progress("Getting applications in #{c(space.name, :name)}") do
+            space.apps
+          end
+      else
+        apps =
+          with_progress("Getting applications") do
+            client.apps(:depth => 2)
+          end
+      end
+
+      line unless quiet?
+
+      if apps.empty? and !quiet?
+        line "No applications."
+        return
+      end
+
+      apps.reject! do |a|
+        !app_matches?(a, input)
+      end
+
+      apps = apps.sort_by(&:name)
+
+      if input[:full]
+        spaced(apps) do |a|
+          invoke :app, :app => a
+        end
+      elsif quiet?
+        apps.each do |a|
+          line a.name
+        end
+      else
+        display_apps_table(apps)
+      end
+    end
+
+    def display_apps_table(apps)
+      table(
+        ["name", "status", "usage", "url"],
+        apps.collect { |a|
+          [ c(a.name, :name),
+            app_status(a),
+            "#{a.total_instances} x #{human_mb(a.memory)}",
+            if a.urls.empty?
+              d("none")
+            elsif a.urls.size == 1
+              a.url
+            else
+              "#{a.url}, ..."
+            end
+          ]
+        })
+    end
+
+    def app_matches?(a, options)
+      if name = options[:name]
+        return false if a.name !~ /#{name}/
+      end
+
+      if url = options[:url]
+        return false if a.urls.none? { |u| u =~ /#{url}/ }
+      end
+
+      true
+    end
+  end
+end

data/lib/cf/cli/app/base.rb

@@ -0,0 +1,72 @@
+require "cf/cli"
+
+module CF
+  module App
+    class Base < CLI
+      include LoginRequirements
+
+      # choose the right color for app/instance state
+      def state_color(s)
+        case s
+        when "STARTING"
+          :neutral
+        when "STARTED", "RUNNING"
+          :good
+        when "DOWN"
+          :bad
+        when "FLAPPING"
+          :error
+        when "N/A"
+          :unknown
+        else
+          :warning
+        end
+      end
+
+      def app_status(a)
+        health = a.health
+
+        if a.debug_mode == "suspend" && health == "0%"
+          c("suspended", :neutral)
+        else
+          c(health.downcase, state_color(health))
+        end
+      end
+
+      def memory_choices
+        [128, 256, 512, 1024].map{|n| human_mb(n)}
+      end
+
+      def human_mb(num)
+        human_size(num * 1024 * 1024, 0)
+      end
+
+      def human_size(num, precision = 1)
+        sizes = %w(G M K)
+        sizes.each.with_index do |suf, i|
+          pow = sizes.size - i
+          unit = 1024.0 ** pow
+          if num >= unit
+            return format("%.#{precision}f%s", num / unit, suf)
+          end
+        end
+
+        format("%.#{precision}fB", num)
+      end
+
+      def megabytes(str)
+        if str =~ /T$/i
+          str.to_i * 1024 * 1024
+        elsif str =~ /G$/i
+          str.to_i * 1024
+        elsif str =~ /M$/i
+          str.to_i
+        elsif str =~ /K$/i
+          str.to_i / 1024
+        else # assume megabytes
+          str.to_i
+        end
+      end
+    end
+  end
+end

data/lib/cf/cli/app/delete.rb

@@ -0,0 +1,95 @@
+require "set"
+
+require "cf/cli/app/base"
+
+module CF::App
+  class Delete < Base
+    desc "Delete an application"
+    group :apps, :manage
+    input :apps, :desc => "Applications to delete", :argument => :splat,
+          :singular => :app, :from_given => by_name(:app)
+    input :routes, :desc => "Delete associated routes", :default => false
+    input :delete_orphaned, :desc => "Delete orphaned services",
+          :aliases => "-o", :default => proc { force? ? false : interact },
+          :forget => true
+    input :all, :desc => "Delete all applications", :default => false
+    input :really, :type => :boolean, :forget => true, :hidden => true,
+          :default => proc { force? || interact }
+    def delete
+      apps = client.apps
+
+      if input[:all]
+        return unless input[:really, "ALL APPS", :bad]
+
+        to_delete = apps
+        others = []
+      else
+        to_delete = input[:apps]
+        others = apps - to_delete
+      end
+
+      all_services = apps.collect(&:services).flatten
+      deleted_app_services = []
+
+      spaced(to_delete) do |app|
+        really = input[:all] || input[:really, app.name, :name]
+        next unless really
+
+        deleted_app_services += app.services
+
+        with_progress("Deleting #{c(app.name, :name)}") do
+          app.routes.collect(&:delete!) if input[:routes]
+          app.delete!
+        end
+      end
+
+      delete_orphaned_services(
+        find_orphaned_services(deleted_app_services, all_services))
+
+      to_delete
+    end
+
+    def find_orphaned_services(deleted, all)
+      orphaned = []
+
+      leftover = all.dup
+      deleted.each do |svc|
+        leftover.slice!(leftover.index(svc))
+        orphaned << svc unless leftover.include?(svc)
+      end
+
+      # clear out the relationships as the apps are now deleted
+      orphaned.each(&:invalidate!)
+    end
+
+    def delete_orphaned_services(orphans)
+      return if orphans.empty?
+
+      line unless quiet? || force?
+
+      orphans.select { |o| input[:delete_orphaned, o] }.each do |service|
+        # TODO: splat
+        invoke :delete_service, :service => service, :really => true
+      end
+    end
+
+    private
+
+    def ask_apps
+      apps = client.apps
+      fail "No applications." if apps.empty?
+
+      [ask("Delete which application?", :choices => apps.sort_by(&:name),
+           :display => proc(&:name))]
+    end
+
+    def ask_really(name, color)
+      ask("Really delete #{c(name, color)}?", :default => false)
+    end
+
+    def ask_delete_orphaned(service)
+      ask("Delete orphaned service #{c(service.name, :name)}?",
+          :default => false)
+    end
+  end
+end

data/lib/cf/cli/app/deprecated.rb

@@ -0,0 +1,11 @@
+require "cf/cli/app/base"
+
+module CF::App
+  class Deprecated < Base
+    desc "DEPRECATED. Use 'push' instead."
+    input :app, :argument => :optional
+    def update
+      fail "The 'update' command is no longer needed; use 'push' instead."
+    end
+  end
+end

data/lib/cf/cli/app/env.rb

@@ -0,0 +1,99 @@
+require "cf/cli/app/base"
+
+module CF::App
+  class Env < Base
+    VALID_ENV_VAR = /^[a-zA-Za-z_][[:alnum:]_]*$/
+
+    desc "Show all environment variables set for an app"
+    group :apps, :info
+    input :app, :desc => "Application to inspect the environment of",
+          :argument => true, :from_given => by_name(:app)
+    def env
+      app = input[:app]
+
+      vars =
+        with_progress("Getting env for #{c(app.name, :name)}") do |s|
+          app.env
+        end
+
+      line unless quiet?
+
+      vars.each do |name, val|
+        line "#{c(name, :name)}: #{val}"
+      end
+    end
+
+    desc "Set an environment variable"
+    group :apps, :info
+    input :app, :desc => "Application to set the variable for",
+          :argument => true, :from_given => by_name(:app)
+    input :name, :desc => "Variable name", :argument => true
+    input :value, :desc => "Variable value", :argument => :optional
+    input :restart, :desc => "Restart app after updating?", :default => false
+
+    def set_env
+      app = input[:app]
+      name, value = parse_name_and_value!
+
+      with_progress("Updating env variable #{c(name, :name)} for app #{c(app.name, :name)}") do
+        app.env[name] = value
+        app.update!
+      end
+
+      restart_if_necessary(app)
+    end
+
+    desc "Remove an environment variable"
+    group :apps, :info
+    input :app, :desc => "Application to set the variable for",
+          :argument => true, :from_given => by_name(:app)
+    input :name, :desc => "Variable name", :argument => true
+    input :restart, :desc => "Restart app after updating?", :default => false
+    def unset_env
+      app = input[:app]
+      name = input[:name]
+
+      with_progress("Unsetting #{c(name, :name)} for app #{c(app.name, :name)}") do
+        app.env.delete(name)
+        app.update!
+      end
+
+      restart_if_necessary(app)
+    end
+
+    private
+
+    def restart_if_necessary(app)
+      unless input[:restart]
+        line c("TIP: Use 'truck push' to ensure your env variable changes take effect.", :warning)
+        return
+      end
+
+      if app.started?
+        invoke :restart, :app => app
+      else
+        line "Your app was unstarted. Starting now."
+        invoke :start, :app => app
+      end
+    end
+
+    def parse_name_and_value!
+      name = input[:name]
+      value = input[:value]
+
+      if name["="]
+        name, new_value, extra_values = name.split("=")
+        fail "You attempted to specify the value of #{name} twice." if value
+        fail "Invalid format: environment variable definition contains too many occurences of '='" if extra_values
+        value = new_value
+      end
+
+      unless name =~ VALID_ENV_VAR
+        fail "Invalid format: environment variable names cannot start with a number" if name[0] =~ /\d/
+        fail "Invalid format: environment variable names can only contain alphanumeric characters and underscores"
+      end
+
+      return name, value
+    end
+  end
+end

data/lib/cf/cli/app/events.rb

@@ -0,0 +1,45 @@
+require "cf/cli/app/base"
+
+module CF::App
+  class Events < Base
+    desc "Display application events"
+    group :apps, :info
+    input :app, :desc => "Application to get the events for",
+          :argument => true, :from_given => by_name(:app)
+    def events
+      app = input[:app]
+
+      events =
+        with_progress("Getting events for #{c(app.name, :name)}") do
+          format_events(app.events)
+        end
+
+      line unless quiet?
+      table(%w{time instance\ index description exit\ status}, events)
+
+    end
+
+    private
+
+    def sort_events(events)
+      events.sort_by { |event| DateTime.parse(event.timestamp) }
+    end
+
+    def format_events(events)
+      sort_events(events).map do |event|
+        [event.timestamp,
+         c(event.instance_index.to_s, :warning),
+         event.exit_description,
+         format_status(event)]
+      end
+    end
+
+    def format_status(event)
+      if event.exit_status == 0
+        c("Success (#{event.exit_status})", :good)
+      else
+        c("Failure (#{event.exit_status})", :bad)
+      end
+    end
+  end
+end