trucker-cli 0.0.1

data/lib/cf/cli/user/create.rb

@@ -0,0 +1,51 @@
+require "cf/cli/user/base"
+
+module CF::User
+  class Create < Base
+    desc "Create a user"
+    group :admin, :user
+    input :email, :desc => "User email", :argument => :optional
+    input :password, :desc => "User password"
+    input :verify, :desc => "Repeat password"
+    input :organization, :desc => "User organization",
+      :aliases => %w{--org -o},
+      :default => proc { client.current_organization },
+      :from_given => by_name(:organization)
+
+    def create_user
+      org = CF::Populators::Organization.new(input).populate_and_save!
+      email = input[:email]
+      password = input[:password]
+
+      if !force? && password != input[:verify]
+        fail "Passwords don't match."
+      end
+
+      user = nil
+      with_progress("Creating user") do
+        user = client.register(email, password)
+      end
+
+      with_progress("Adding user to #{org.name}") do
+        user.audited_organizations = user.managed_organizations = user.organizations = [org]
+        user.update!
+      end
+    end
+
+    alias_command :add_user, :create_user
+
+    private
+
+    def ask_email
+      ask("Email")
+    end
+
+    def ask_password
+      ask("Password", :echo => "*", :forget => true)
+    end
+
+    def ask_verify
+      ask("Verify Password", :echo => "*", :forget => true)
+    end
+  end
+end

data/lib/cf/cli/user/passwd.rb

@@ -0,0 +1,36 @@
+require "cf/cli/user/base"
+
+module CF::User
+  class Passwd < Base
+    desc "Update the current user's password"
+    group :admin, :user
+    input :password, :desc => "Current password"
+    input :new_password, :desc => "New password"
+    input :verify, :desc => "Repeat new password"
+
+    def passwd
+      password = input[:password]
+      new_password = input[:new_password]
+
+      validate_password! new_password
+
+      with_progress("Changing password") do
+        client.current_user.change_password!(new_password, password)
+      end
+    end
+
+    private
+
+    def ask_password
+      ask("Current Password", :echo => "*", :forget => true)
+    end
+
+    def ask_new_password
+      ask("New Password", :echo => "*", :forget => true)
+    end
+
+    def ask_verify
+      ask("Verify Password", :echo => "*", :forget => true)
+    end
+  end
+end

data/lib/cf/cli/user/register.rb

@@ -0,0 +1,42 @@
+require "cf/cli/user/base"
+
+module CF::User
+  class Register < Base
+    def precondition; end
+
+    desc "Create a user and log in"
+    group :admin, :user
+    input :email, :desc => "Desired email", :argument => :optional
+    input :password, :desc => "Desired password"
+    input :verify, :desc => "Repeat password"
+    input :login, :desc => "Automatically log in?", :default => true
+    def register
+      email = input[:email]
+      password = input[:password]
+
+      validate_password!(password)
+
+      with_progress("Creating user") do
+        client.register(email, password)
+      end
+
+      if input[:login]
+        invoke :login, :username => email, :password => password
+      end
+    end
+
+    private
+
+    def ask_email
+      ask("Email")
+    end
+
+    def ask_password
+      ask("Password", :echo => "*", :forget => true)
+    end
+
+    def ask_verify
+      ask("Confirm Password", :echo => "*", :forget => true)
+    end
+  end
+end

data/lib/cf/cli/user/users.rb

@@ -0,0 +1,32 @@
+require "cf/cli/user/base"
+
+module CF::User
+  class Users < Base
+    desc "List all users"
+    group :admin
+    def users
+      users =
+        with_progress("Getting users") do
+          client.users(:depth => 0)
+        end
+
+      spaced(users) do |u|
+        display_user(u)
+      end
+    end
+
+    private
+
+    def display_user(u)
+      if quiet?
+        puts u.email
+      else
+        line "#{c(u.email, :name)}:"
+
+        indented do
+          line "admin?: #{c(u.admin?, u.admin? ? :yes : :no)}"
+        end
+      end
+    end
+  end
+end

data/lib/cf/constants.rb

@@ -0,0 +1,10 @@
+module CF
+  CONFIG_DIR = "~/.truck".freeze
+
+  LOGS_DIR        = "#{CONFIG_DIR}/logs".freeze
+  PLUGINS_FILE    = "#{CONFIG_DIR}/plugins.yml".freeze
+  TARGET_FILE     = "#{CONFIG_DIR}/target".freeze
+  TOKENS_FILE     = "#{CONFIG_DIR}/tokens.yml".freeze
+  COLORS_FILE     = "#{CONFIG_DIR}/colors.yml".freeze
+  CRASH_FILE      = "#{CONFIG_DIR}/crash".freeze
+end

data/lib/cf/errors.rb

@@ -0,0 +1,19 @@
+module CF
+  class UserFriendlyError < RuntimeError
+    def initialize(msg)
+      @message = msg
+    end
+
+    def to_s
+      @message
+    end
+  end
+
+  class UserError < UserFriendlyError; end
+
+  class NotAuthorized < UserError
+    def initialize
+      @message = "Not authorized."
+    end
+  end
+end

data/lib/cf/plugin.rb

@@ -0,0 +1,56 @@
+require "set"
+require "yaml"
+
+require "cf/constants"
+require "cf/cli"
+
+module CF
+  module Plugin
+    @@plugins = []
+
+    def self.load_all
+      # auto-load gems with 'cf-plugin' in their name
+      matching =
+        if Gem::Specification.respond_to? :find_all
+          Gem::Specification.find_all do |s|
+            s.name =~ /cf-plugin/
+          end
+        else
+          Gem.source_index.find_name(/cf-plugin/)
+        end
+
+      enabled = Set.new(matching.collect(&:name))
+
+      cf_gems = Gem.loaded_specs["cf"]
+      ((cf_gems && cf_gems.dependencies) || Gem.loaded_specs.values).each do |dep|
+        if dep.name =~ /cf-plugin/
+          require "#{dep.name}/plugin"
+          enabled.delete dep.name
+        end
+      end
+
+      # allow explicit enabling/disabling of gems via config
+      plugins = File.expand_path(CF::PLUGINS_FILE)
+      if File.exists?(plugins) && yaml = YAML.load_file(plugins)
+        enabled += yaml["enabled"] if yaml["enabled"]
+        enabled -= yaml["disabled"] if yaml["disabled"]
+      end
+
+      # load up each gem's 'plugin' file
+      #
+      # we require this file specifically so people can require the gem
+      # without it plugging into CF
+      enabled.each do |gemname|
+        begin
+          require "#{gemname}/plugin"
+        rescue Gem::LoadError => e
+          puts "Failed to load #{gemname}:"
+          puts "  #{e}"
+          puts
+          puts "You may need to update or remove this plugin."
+          puts
+        end
+      end
+    end
+  end
+end

data/lib/cf/spacing.rb

@@ -0,0 +1,89 @@
+module CF
+  module Spacing
+    @@indentation = 0
+
+    def indented
+      @@indentation += 1
+      yield
+    ensure
+      @@indentation -= 1
+    end
+
+    def line(msg = "")
+      return puts "" if msg.empty?
+
+      start_line(msg)
+      puts ""
+    end
+
+    def start_line(msg)
+      print "  " * @@indentation unless quiet?
+      print msg
+    end
+
+    def lines(blob)
+      blob.each_line do |line|
+        start_line(line)
+      end
+
+      line
+    end
+
+    def quiet?
+      false
+    end
+
+    def spaced(vals)
+      num = 0
+      vals.each do |val|
+        line unless quiet? || num == 0
+        yield val
+        num += 1
+      end
+    end
+
+    def tabular(*rows)
+      spacings = []
+      rows.each do |row|
+        next unless row
+
+        row.each.with_index do |col, i|
+          next unless col
+
+          width = text_width(col)
+
+          if !spacings[i] || width > spacings[i]
+            spacings[i] = width
+          end
+        end
+      end
+
+      columns = spacings.size
+      rows.each do |row|
+        next unless row
+
+        row.each.with_index do |col, i|
+          next unless col
+
+          start_line justify(col, spacings[i])
+          print "   " unless i + 1 == columns
+        end
+
+        line
+      end
+    end
+
+    def trim_escapes(str)
+      str.gsub(/\e\[\d+m/, "")
+    end
+
+    def text_width(str)
+      trim_escapes(str).size
+    end
+
+    def justify(str, width)
+      trimmed = trim_escapes(str)
+      str.ljust(width + (str.size - trimmed.size))
+    end
+  end
+end

data/lib/cf/spec_helper.rb

@@ -0,0 +1 @@
+require File.expand_path("../../../spec/spec_helper", __FILE__)

data/lib/cf/test_support.rb

@@ -0,0 +1,6 @@
+module CF::TestSupport
+end
+
+Dir[File.expand_path('../../../spec/support/**/*.rb', __FILE__)].each do |file|
+  require file
+end

data/lib/cf/version.rb

@@ -0,0 +1,3 @@
+module CF
+  VERSION = "0.0.1".freeze
+end

data/lib/console/README.md

@@ -0,0 +1,16 @@
+[![Build Status](https://travis-ci.org/cloudfoundry/console-cf-plugin.png)](https://travis-ci.org/cloudfoundry/console-cf-plugin)
+[![Gem Version](https://badge.fury.io/rb/console-cf-plugin.png)](http://badge.fury.io/rb/console-cf-plugin)
+
+## Console
+### Info
+This plugin lets you connect to a Cloud Foundry application via telnet.
+
+### Installation
+```
+gem install console-cf-plugin
+```
+
+### Usage
+```
+console APP          Open a console connected to your app
+```

data/lib/console/console.rb

@@ -0,0 +1,187 @@
+require "net/telnet"
+require "readline"
+
+require "tunnel/tunnel"
+
+class CFConsole < CFTunnel
+  def initialize(client, app, port = 10000)
+    @client = client
+    @app = app
+    @port = port
+  end
+
+  def get_connection_info(auth)
+    instances = @app.instances
+    if instances.empty?
+      raise "App has no running instances; try starting it."
+    end
+
+    unless console = instances[0].console
+      raise "App does not have console access; try restarting it."
+    end
+
+    { "hostname" => console[:ip],
+      "port" => console[:port]
+    }
+  end
+
+  def get_credentials
+    YAML.load(@app.file("app", "cf-rails-console", ".consoleaccess"))
+  end
+
+  def start_console
+    prompt = login
+
+    init_readline
+
+    run_console prompt
+  end
+
+  def login(auth = get_credentials)
+    if !auth["username"] || !auth["password"]
+      raise "Unable to verify console credentials."
+    end
+
+    @telnet = telnet_client
+
+    prompt = nil
+    err_msg = "Login attempt timed out."
+
+    5.times do
+      begin
+        results = @telnet.login(
+          "Name" => auth["username"],
+          "Password" => auth["password"])
+
+        lines = results.sub("Login: Password: ", "").split("\n")
+
+        last_line = lines.pop
+
+        if last_line =~ /[$%#>] \z/n
+          prompt = last_line
+        elsif last_line =~ /Login failed/
+          err_msg = last_line
+        end
+
+        break
+
+      rescue TimeoutError
+        sleep 1
+
+      rescue EOFError
+        # This may happen if we login right after app starts
+        close_console
+        sleep 5
+        @telnet = telnet_client
+      end
+    end
+
+    unless prompt
+      close_console
+      raise err_msg
+    end
+
+    prompt
+  end
+
+  private
+
+  def init_readline
+    if Readline.respond_to?("basic_word_break_characters=")
+      Readline.basic_word_break_characters= " \t\n`><=;|&{("
+    end
+
+    Readline.completion_append_character = nil
+
+    # Assumes that sending a String ending with tab will return a non-empty
+    # String of comma-separated completion options, terminated by a new line
+    # For example, "app.\t" might result in "to_s,nil?,etc\n"
+    Readline.completion_proc = proc do |s|
+      console_tab_completion_data(s)
+    end
+  end
+
+  def run_console(prompt)
+    prev = trap("INT")  { |x| exit_console; prev.call(x); exit }
+    prev = trap("TERM") { |x| exit_console; prev.call(x); exit }
+
+    loop do
+      cmd = readline_with_history(prompt)
+
+      if cmd == nil
+        exit_console
+        break
+      end
+
+      prompt = send_console_command_display_results(cmd, prompt)
+    end
+  end
+
+  def readline_with_history(prompt)
+    line = Readline::readline(prompt)
+
+    return if line == nil || line == 'quit' || line == 'exit'
+
+    if line !~ /^\s*$/ && Readline::HISTORY.to_a.last != line
+      Readline::HISTORY.push(line)
+    end
+
+    line
+  end
+
+  def send_console_command_display_results(cmd, prompt)
+    begin
+      lines = send_console_command cmd
+
+      # Assumes the last line is a prompt
+      prompt = lines.pop
+
+      lines.each do |line|
+        puts line if line != cmd
+      end
+
+    rescue TimeoutError
+      puts "Timed out sending command to server."
+
+    rescue EOFError
+      raise "The console connection has been terminated. Perhaps the app was stopped or deleted?"
+    end
+
+    prompt
+  end
+
+  def send_console_command(cmd)
+    results = @telnet.cmd(cmd)
+    results.split("\n")
+  end
+
+  def exit_console
+    @telnet.cmd("String" => "exit", "Timeout" => 1)
+  rescue TimeoutError
+    # TimeoutError expected, as exit doesn't return anything
+  ensure
+    close_console
+  end
+
+  def close_console
+    @telnet.close
+  end
+
+  def console_tab_completion_data(cmd)
+    begin
+      results = @telnet.
+          cmd("String" => cmd + "\t", "Match" => /\S*\n$/, "Timeout" => 10)
+      results.chomp.split(",")
+    rescue TimeoutError
+      [] #Just return empty results if timeout occurred on tab completion
+    end
+  end
+
+  def telnet_client
+    Net::Telnet.new(
+      "Port" => @port,
+      "Prompt" => /[$%#>] \z|Login failed/n,
+      "Timeout" => 30,
+      "FailEOF" => true)
+  end
+end