viaduct-toolkit 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a7eceb02225b5d2bdba774374a4b0cc66ad28d43
+  data.tar.gz: 3af40448a600abcf1708c9e15d05bbdeed9d898f
+SHA512:
+  metadata.gz: bfd1069aada93499f1ede946a415bce09be8e1733e2b9acdabd29b632dc4c8748fb445daa78243e59b5c9c5aaf696e46fe49d8a1193b07924158bc8084eb247b
+  data.tar.gz: da954154572bede916028769b4dac835d2976c54fb13504281e2d0f35544948ff025c877567328bace915fdf8c6014d0f4fe520e91e3edac1e0bbafa177177cb

data/lib/viaduct/toolkit.rb

@@ -0,0 +1,76 @@
+require 'yaml'
+require 'colorize'
+require 'commander'
+require 'viaduct/toolkit/version'
+require 'viaduct/api'
+
+# Hardcoded for this application.
+Viaduct::API.application_token = '3148984b-8a50-424c-98f5-117e8dea2971'
+
+# Add our helpers to Commander
+require 'viaduct/toolkit/helpers'
+Commander::Methods.send :include, Viaduct::Toolkit::Helpers
+
+module Viaduct
+  module Toolkit
+    
+    class Error < StandardError
+    end
+    
+    class << self
+      
+      def binary
+        File.expand_path(File.join('..', '..', '..', 'bin', 'viaduct'), __FILE__)
+      end
+      
+      def cli
+        @cli ||= begin
+          c = Commander::Runner.instance
+          c.program :name, "Viaduct Toolkit"
+          c.program :version, Viaduct::Toolkit::VERSION
+          c.program :description, "A CLI toolkit for Viaduct developers"
+          c.global_option('-c', '--config FILE', 'The config file to store local credentials within') { |file| $config_file_path = file }
+          c.default_command :help
+          c
+        end
+      end
+    
+      def add_commands
+        Dir[File.expand_path(File.join('..', 'toolkit', 'commands', '*.rb'), __FILE__)].each do |file|
+          require file
+        end
+      end
+    
+      def api
+        @api ||= begin
+          Viaduct::API::Client.new(config['token'], config['secret'])
+        end
+      end
+      
+      def reset_api
+        @api = nil
+      end
+    
+      def config_file_path
+        $config_file_path || File.join(ENV['HOME'], '.viaduct')
+      end
+    
+      def config
+        @config ||= begin
+          if File.exist?(config_file_path)
+            YAML.load_file(config_file_path)
+          else
+            {}
+          end
+        end
+      end
+    
+      def save_config
+        File.open(config_file_path, 'w') do |f|
+          f.write self.config.to_yaml
+        end
+      end
+      
+    end
+  end
+end

data/lib/viaduct/toolkit/commands/app_console.rb

@@ -0,0 +1,136 @@
+Viaduct::Toolkit.cli.command "app:console" do |c|
+  
+  c.syntax = "app:console SUBDOMAIN_OF_APP"
+  c.description = "Start a new SSH-based console session"
+  c.option "--disable", "Disable access to the SSH console"
+  c.option "--enable", "Enable access to the SSH console"
+  c.option "--status", "Display the status of the port forward"
+  
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if app = find_application(args[0])
+      
+      # Check the user's SSH keys
+      response = Viaduct::Toolkit.api.ssh_keys.all
+      if response.success?
+        
+        if response.data.empty?
+          puts "You haven't uploaded any SSH keys to your Viaduct user account.".red
+          puts "You cannot use SSH console access without them.".red
+          puts
+          puts "Upload your key using the command below".blue
+          puts
+          puts "    $ vdt ssh_keys:add".blue
+          puts
+          exit(1)
+        else
+          stdout, stderr, status = run("ssh-add -l")
+          if status == 0
+            remote_fingerprints = response.data.map { |d| d['fingerprint'] }
+            local_fingerprints = stdout.split(/\n/).map { |l| l.split(/\s+/)[1] }
+            unless remote_fingerprints.any? { |f| local_fingerprints.include?(f) }
+              puts "Warning: it doesn't seem as though your SSH key has been uploaded".yellow
+              puts "to your Viaduct account. This session may not succeed. If it doesn't".yellow
+              puts "ensure that you have uploaded to your SSH key to your Viaduct account.".yellow
+            end
+          end
+        end
+        
+      else
+        error "Couldn't verify remote SSH keys. Please try again."
+      end
+      
+      # Get the console port forward for the application
+      response = Viaduct::Toolkit.api.port_forwards.all(:application => app['subdomain'])
+      if response.success?
+        unless console = response.data.select { |c| c['mode'] == 'console'}.first
+          error "Console access is not supported by this application. Please contact support."
+        end
+      else
+        error "Couldn't get port forward information from API for application."
+      end
+      
+      if opts.status
+        details do
+          heading "SSH console access"
+          field "Status", console['enabled'] ? "Enabled" : "Disabled"
+          field "Auto disable?", console['auto_disable_at'] ? time(console['auto_disable_at']) : ''
+          field "Connection", "#{console['ip_address']}:#{console['port']}"
+          console['allowed_ips'].each_with_index do |ip, i|
+            field i == 0 ? 'Allowed IPs' : '', ip
+          end
+        end
+      
+      elsif opts.disable
+        
+        # 
+        # Just disable console access
+        #
+        
+        if console['enabled']
+          response = Viaduct::Toolkit.api.port_forwards.save(:id => console['id'], :enabled => 0)
+          if response.success?
+            puts "Console access has been disabled.".green
+          else
+            error "We couldn't disable console access at this time. Please try later."
+          end
+        else
+          puts "Console access isn't currently enabled.".yellow
+        end
+        
+      elsif opts.enable
+        
+        #
+        # Just enable console access
+        #
+        
+        if console['enabled']
+          puts "Console is already enabled.".yellow
+        else
+          auto_disable = agree("Would you like to disable this again after an hour?".blue)
+          response = Viaduct::Toolkit.api.port_forwards.save(:id => console['id'], :enabled => 1, :auto_disable_at => (auto_disable ? '1 hour from now' : nil))
+          if response.success?
+            if auto_disable
+              puts "Console access has been enaled and will be automatically disabled again in 1 hour.".green
+            else
+              puts "Console access has been enabled.".green
+            end
+          else
+            error "We couldn't enable console access at this time. Please try later."
+          end
+        end
+        
+      else
+        
+        #
+        # Enable if needed and connect.
+        #
+        unless console['enabled']
+          puts "SSH Console access is not currently enabled for this application.".magenta
+          if enable =  agree("Would you like to enable it now?".blue)
+            disable_after_hour = agree("Would you like to disable this again after an hour?".blue)
+            response = Viaduct::Toolkit.api.port_forwards.save(:id => console['id'], :enabled => 1, :auto_disable_at => (disable_after_hour ? "1 hour from now" : nil))
+            unless response.success?
+              puts response.inspect
+              error "We couldn't enable console access at this time. Please try later."
+            end
+          else
+            puts "Unfortunately, unless enabled you cannot use this command.".red
+            exit(1)
+          end
+        end
+        
+        command = "ssh vdt@#{console['ip_address']} -p #{console['port']}"
+        
+        exec(command)
+      end
+      
+      
+    end
+  end
+  
+end
+
+
+Viaduct::Toolkit.cli.alias_command "app:ssh", "app:console"

data/lib/viaduct/toolkit/commands/app_deployments.rb

@@ -0,0 +1,25 @@
+Viaduct::Toolkit.cli.command "app:deployments" do |c|
+  
+  c.syntax = "app:deployments SUBDOMAIN_OF_APP"
+  c.description = "List all deployments for an application"
+  c.option "--page PAGE", Integer, "The page of deployments to return"
+  
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if application = find_application(args[0])
+      response = Viaduct::Toolkit.api.applications.deployments(:application => application['subdomain'], :page => opts.page ? opts.page.to_i : 1)
+      if response.success?
+
+        require 'terminal-table'
+        rows = response.data.map do |d|
+          [d['number'], d['status'], d['version']['description'].to_s[0,30], d['user'] ? d['user']['username'] : '', d['timing']['created_at']]
+        end
+        puts Terminal::Table.new :rows => rows, :headings => ['#', 'Status', 'Description', 'User', 'Time']
+        
+      else
+        error "Couldn't get application deployment list"
+      end
+    end
+  end
+end

data/lib/viaduct/toolkit/commands/app_env.rb

@@ -0,0 +1,35 @@
+Viaduct::Toolkit.cli.command "app:env" do |c|
+  
+  c.syntax = "app:env SUBDOMAIN_OF_APP"
+  c.description = "Show all environment variables for an application"
+  c.option "--simple", "Return environment variables "
+  c.option "--yaml", "Return environment variables as YAML file"
+  c.option "--export", "Return environment variables as export lines"
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if application = find_application(args[0])
+      response = Viaduct::Toolkit.api.applications.environment_variables(:application => application['subdomain'])
+      if response.success?
+        if opts.simple
+          response.data.each do |key, value|
+            puts "#{key}: #{value}"
+          end
+        elsif opts.yaml
+          require 'yaml'
+          puts response.data.to_yaml
+        elsif opts.export
+          response.data.each do |key, value|
+            puts "export #{key}=\"#{value}\""
+          end
+        else
+          require 'terminal-table'
+          puts Terminal::Table.new :rows => response.data.to_a
+        end
+      else
+        error "Couldn't get environment variables"
+      end
+    end
+  end
+
+end

data/lib/viaduct/toolkit/commands/app_info.rb

@@ -0,0 +1,42 @@
+Viaduct::Toolkit.cli.command "app:info" do |c|
+  
+  c.syntax = "app:info SUBDOMAIN_OF_APP"
+  c.description = "Show details of an existing application"
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if application = find_application(args[0])
+      details do
+        heading "Application Details"
+        field "Name", application['name']
+        field "Viaduct Domain", application['viaduct_domain']
+        field "Status", application['status']
+        field "Maintenance", application['maintenance?'] ? 'Yes' : 'No'
+        field 'Owner', "#{application['user']['name']} (#{application['user']['username']})"
+        field "Zone", application['zone']['name']
+        field "Platform", "#{application['platform']['name']} (on #{application['platform']['stack']['name']} stack)"
+        field "Subnet(s)", application['subnets'].map { |s| s['description'] }.join(', ')
+        if application['repository'] && application['source_backend_module'] == 'Viaduct::SourceBackends::Repository'
+          heading "Repository"
+          field 'URL', application['repository']['repository']['url']
+          field 'Status', application['repository']['repository']['status']
+          field 'Last update', application['repository']['repository']['last_updated_at']
+          field 'Username', application['repository']['repository']['username']
+          field 'Branch', application['repository']['branch']
+        end
+        
+        if application['deployment']
+          heading "Active Deployment"
+          field 'Number', application['deployment']['number']
+          field 'Started at', application['deployment']['timing']['created_at']
+          field 'Time', application['deployment']['timing']['time'].round(1).to_s + "s"
+          field 'Description', application['deployment']['version']['description']
+          if application['deployment']['user']
+            field 'Deployer', "#{application['deployment']['user']['name']} (#{application['deployment']['user']['username']})"
+          end
+        end
+      end
+    end
+  end
+  
+end

data/lib/viaduct/toolkit/commands/app_list.rb

@@ -0,0 +1,36 @@
+Viaduct::Toolkit.cli.command "app:list" do |c|
+  c.syntax = "app:list"
+  c.description = "Show a list of all applications you have access to"
+  c.option "--simple", "Return just a list of subdomains"
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    
+    pages_seen = 0
+    applications = []
+    loop do
+      response = Viaduct::Toolkit.api.applications.all(:page => pages_seen + 1)
+      if response.success?
+        applications = applications | response.data
+      else
+        error "Couldn't get list of applications."
+      end
+      pages_seen += 1
+      if pages_seen == response.flags['paginated']['page']
+        break
+      end
+    end
+
+    if opts.simple
+      applications.each do |application|
+        puts application['subdomain']
+      end
+    else
+      rows = applications.map do |app|
+        [app['name'], app['subdomain'], app['status'], app['user']['name']]
+      end
+      table ['Name', 'Subdomain', 'Status', 'Owner'], rows
+    end
+    
+  end
+end

data/lib/viaduct/toolkit/commands/app_public_key.rb

@@ -0,0 +1,12 @@
+Viaduct::Toolkit.cli.command "app:public_key" do |c|
+  c.syntax = "app:public_key NAME_OF_APP"
+  c.description = "Return the public key for a given application"
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if app = find_application(args[0])
+      puts app['public_key'] + " Viaduct key for #{app['subdomain']}"
+    end
+  end
+end
+Viaduct::Toolkit.cli.alias_command "app:pk", "app:public_key"

data/lib/viaduct/toolkit/commands/app_status.rb

@@ -0,0 +1,118 @@
+Viaduct::Toolkit.cli.command "app:status" do |c|
+  c.syntax = "app:status NAME_OF_APP"
+  c.description = "Return current status of an application"
+  c.option "--process PROCESS_TYPE", String, "A process to show expanded information for"
+  c.option "--database DATABASE_ID", String, "The ID of the database to display further information for"
+  
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if app = find_application(args[0])
+      response = Viaduct::Toolkit.api.applications.status(:application => app['subdomain'])
+      if response.success?
+        
+        if opts.process
+          
+          if process = response.data['processes'].select { |p| p['label'] == opts.process.to_s }.first
+            details do
+              heading "Details"
+              field "Label", process['label']
+              field "Command", process['container']['command']
+              field "Public?", process['container']['public'] ? 'Yes' : 'No'
+              field "Status", process['process']['status'].capitalize
+              process_details(process['process'])
+            end
+          else
+            puts "No process found matching '#{opts.process}'".red
+            exit 1
+          end
+          
+          
+        elsif opts.database
+          
+          if db = response.data['databases'][(opts.database.to_i - 1).abs]
+            details do
+              heading "Database Details"
+              field "Label", db['label']
+              field "Database", db['stack']['name']
+              process_details(db['process'])
+            end
+          else
+            puts "No database found with ID '#{opts.database}'".red
+            exit 1
+          end
+          
+        else
+        
+          puts "Deployment Details".yellow
+          details do
+            puts "+" + '-' * 78 + '+'
+            field "Number", response.data['deployment']['number']
+            field 'Version', response.data['deployment']['version']['id']
+            field "Description", response.data['deployment']['version']['description']
+            field "Deployer", response.data['deployment']['user'] ? response.data['deployment']['user']['full_name'] : ''
+            field "Started", time(response.data['deployment']['timing']['started_at'])
+          end
+        
+          processes = response.data['processes'].map do |p|
+            memory_used = (p['process']['resources']['memory_usage'] / 1024 / 1024).to_s + "MB"
+            memory_available = (p['process']['resources']['max_memory'] / 1024 / 1024).to_s + "MB"
+            memory = "#{memory_used}/#{memory_available}"
+          
+            respawns = "#{p['process']['respawning']['current']}/#{p['process']['respawning']['maximum']}"
+            [p['label'], p['process']['status'].capitalize, memory, p['container']['command'], respawns]
+          end
+          puts
+          puts "Processes".yellow
+          table ['Label', 'Status', 'Memory', 'Command', 'Respawns'], processes
+          
+          count = 0
+          databases = response.data['databases'].map do |d|
+            count += 1
+            memory_used = (d['process']['resources']['memory_usage'] / 1024 / 1024).to_s + "MB"
+            memory_available = (d['process']['resources']['max_memory'] / 1024 / 1024).to_s + "MB"
+            memory = "#{memory_used}/#{memory_available}"
+            respawns = "#{d['process']['respawning']['current']}/#{d['process']['respawning']['maximum']}"
+          
+            [count, d['label'], d['process']['status'].capitalize, d['stack']['name'], memory, respawns, d['process']['networking']['ip_address']]
+          end
+          unless databases.empty?
+            puts
+            puts "Dedicated Databases".yellow
+            table ['#', 'Label', 'Status', 'Type', 'Memory', 'Respawns', 'IP Address'], databases
+          end
+        end
+        
+      else
+        error "Couldn't get application status"
+      end
+    end
+  end
+end
+
+def process_details(p)
+  if p['host'] && p['lxc_name']
+    heading "Host (admin only)"
+    field "Host", p['host']
+    field "LXC Name", p['lxc_name']
+  end
+  
+  heading "Networking"
+  field "IP Address", p['networking']['ip_address']
+  field "MAC Address", p['networking']['mac_address']
+  field "Data", "RX: #{p['networking']['rx']}   TX: #{p['networking']['tx']}"
+  
+  heading "Timing"
+  field "Started", time(p['timing']['started'])
+  field "Run time", length_of_time(p['timing']['run_time'])
+  field "Last seen", time(p['timing']['last_seen_at'])
+  
+  heading "Resources"
+  field "Memory", "#{p['resources']['memory_usage'] /1024/1024}MB of #{p['resources']['max_memory']/1024/1024}MB"
+  field "CPU Usage", p['resources']['cpu_usage']
+  
+  heading "Respawning"
+  field "Respawns", "#{p['respawning']['current']} / #{p['respawning']['maximum']}"
+  field "Last respawn", time(p['respawning']['last'])
+
+end

data/lib/viaduct/toolkit/commands/domain_list.rb

@@ -0,0 +1,19 @@
+Viaduct::Toolkit.cli.command "domain:list" do |c|
+  c.syntax = "domain:list NAME_OF_APP"
+  c.description = "Return a list of domains for an application"
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    if app = find_application(args[0])
+      response = Viaduct::Toolkit.api.domains.all(:application => app['subdomain'])
+      if response.success?
+        rows = response.data.map do |d|
+          [d['name'], boolean(d['verification']['verified?']), d['ssl'], d['routing']['cname'], d['routing']['ip']]
+        end
+        table ['Name', 'Verified', 'SSL', 'CNAME', 'IP'], rows
+      else
+        error "Couldn't get domain list"
+      end
+    end
+  end
+end

data/lib/viaduct/toolkit/commands/login.rb

@@ -0,0 +1,70 @@
+Viaduct::Toolkit.cli.command "login" do |c|
+  c.syntax = "login"
+  c.description = "Authenticate this computer with your Viaduct account"
+  c.action do |args, options|
+    
+    Viaduct::Toolkit.config['token'] = nil
+    Viaduct::Toolkit.config['secret'] = nil
+    Viaduct::Toolkit.save_config
+    
+    response = Viaduct::Toolkit.api.authentication.create_login_token
+    if response.success?
+      
+      puts "To log you in we need to open a browser window to allow".magenta
+      puts "you to enter your login details. ".magenta
+      puts
+      if agree("Shall we open this for you?".blue)
+        system("open", response.data['url'])
+      else
+        puts
+        puts "That's fine. Just go to the URL below in your browser.".magenta
+        puts "This command will continue to run until you complete this".magenta
+        puts "action.".magenta
+        puts
+        puts response.data['url']
+      end
+      
+      puts
+      puts "Please wait while we verify your login...".magenta
+      puts
+      check_response = nil
+      100.times do
+        sleep 3
+        check_response = Viaduct::Toolkit.api.authentication.exchange(:token => response.data['token'])
+        if check_response.success?
+          if check_response.data['status'] == 'approved'
+            Viaduct::Toolkit.reset_api
+            @authorised = true
+            break
+          elsif check_response.data['status'] == 'denied'
+            error "The login request was rejected. Ensure that you approve the login request."
+          end
+        else
+          error "Couldn't successfully exchange login token for an API token. Please try again later."
+        end
+      end
+      
+      if @authorised
+        Viaduct::Toolkit.config['token'] = check_response.data['token']['token']
+        Viaduct::Toolkit.config['secret'] = check_response.data['token']['secret']
+
+        user_check = Viaduct::Toolkit.api.user.details
+        if user_check.success?
+          Viaduct::Toolkit.save_config
+          puts "Hello #{user_check.data['name']}!".green
+          puts "Your user account is now authorised. Your login details are".magenta
+          puts "stored in a .viaduct file in your home directory.".magenta
+          puts
+        else
+          error "We couldn't verify your user details. Please try again."
+        end
+      else
+        error "We didn't receive a login response in a timely manner. Please try again."
+      end
+      
+    else
+      error "Couldn't generate a remote login token. Please try again."
+    end
+    
+  end
+end

data/lib/viaduct/toolkit/commands/ssh_key_add.rb

@@ -0,0 +1,40 @@
+Viaduct::Toolkit.cli.command "ssh_key:add" do |c|
+  
+  c.syntax = "ssh_keys:add PATH_TO_PUBLIC_KEY"
+  c.description = "Add a new SSH public key to your user account"
+  
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    
+    key_path = args[0]
+    
+    default_key_path = File.join(ENV['HOME'], '.ssh', 'id_rsa.pub')
+    if key_path.nil? && File.exist?(default_key_path)
+      puts "You haven't provided an SSH key to this command.".yellow
+      if agree("Shall we use your default key at #{default_key_path}?".blue)
+        key_path = default_key_path
+      else
+        exit(1)
+      end
+    end
+    
+    if File.exist?(key_path)
+      response = Viaduct::Toolkit.api.ssh_keys.add(:label => "Added from #{`hostname`}", :key => File.read(key_path))
+      if response.success?
+        puts "Key successfully added to your account".green
+      else
+        if response.is_a?(MoonropeClient::Responses::ValidationError)
+          puts "Errors occurred while adding your key:".red
+          validation_errors response.data['errors']
+        else
+          error "Couldn't add your key"
+        end
+      end
+    else
+      puts "No public key found at '#{key_path}'".red
+      exit(1)
+    end
+  end
+  
+end

data/lib/viaduct/toolkit/commands/ssh_key_list.rb

@@ -0,0 +1,26 @@
+Viaduct::Toolkit.cli.command "ssh_key:list" do |c|
+  
+  c.syntax = "ssh_key:list"
+  c.description = "List the keys associated with your account"
+  
+  c.action do |args, opts|
+    include Commander::Methods
+    ensure_logged_in!
+    
+    response = Viaduct::Toolkit.api.ssh_keys.all
+    if response.success?
+      require 'terminal-table'
+      rows = response.data.map do |app|
+        [app['label'], app['fingerprint']]
+      end
+      table = Terminal::Table.new :rows => rows, :headings => ['Label', 'Fingerprint']
+      puts table
+      
+    else
+      error "Couldn't get the SSH key list."
+    end
+    
+  end
+  
+end
+

data/lib/viaduct/toolkit/helpers.rb

@@ -0,0 +1,111 @@
+require 'open3'
+
+module Viaduct
+  module Toolkit
+    module Helpers
+      def ensure_logged_in!
+        if Viaduct::Toolkit.config['token'].nil? && Viaduct::Toolkit.config['secret'].nil?
+          puts "You need to login before using this toolkit. Use the command below".yellow
+          puts "to login to your Viaduct account.".yellow
+          puts
+          puts "  $ vdt login"
+          puts
+          exit 1
+        end
+      end
+      
+      def length_of_time(seconds)
+        "#{seconds} seconds"
+      end
+      
+      def time(time)
+        require 'time'
+        time = Time.parse(time) rescue nil
+        time ? time.strftime("%d %B %Y at %H:%M:%S UTC") : ''
+      end
+      
+      def boolean(bool)
+        bool ? "\u2713".green : "-".red
+      end
+      
+      def table(headings, rows)
+        require 'terminal-table'
+        puts Terminal::Table.new :rows => rows, :headings => headings.map(&:blue)
+      end
+      
+      def validation_errors(errors)
+        errors.each do |field, messages|
+          messages.each do |message|
+            puts " * #{field} #{message}".red
+          end
+        end
+      end
+      
+      def error(message)
+        raise Viaduct::Toolkit::Error, message
+      end
+      
+      def run(*commands)
+        stdin, stdout, stderr, w = Open3.popen3(*commands)
+        [stdout.gets(nil), stderr.gets(nil), w.value]
+      end
+      
+      def find_application(subdomain)
+        if subdomain.is_a?(String) && subdomain.length > 0
+          app = Viaduct::Toolkit.api.applications.info(:application => subdomain)
+          if app.success?
+            return app.data
+          else
+            puts "Couldn't find application with subdomain matching '#{subdomain}'".red
+            exit(1)
+          end
+        else
+          # Look up from repo
+          out, err, status = run("git", "remote", "-v")
+          if status == 0
+            potential_repositories = out.split("\n").map { |l| l.split(/\s+/)[1] }.uniq
+            app = Viaduct::Toolkit.api.applications.all(:filter => {:repo => potential_repositories})
+            if app.success?
+              if app.data.empty?
+                puts "No Viaduct applications found for any of the following repositories:".red
+                potential_repositories.each do |repo|
+                  puts "  * #{repo}".red
+                end
+                exit(1)
+              elsif app.data.size == 1
+                return find_application(app.data.first['subdomain'])
+              else
+                puts "Multiple applications found matching your repository. Choose an application...".yellow
+                choice = choose('', *app.data.map { |a| "#{a['subdomain']}: #{a['name']}"})
+                choice = choice.split(":", 2).first
+                return find_application(choice)
+              end
+            end
+          end
+        end
+        
+        puts "Couldn't determine a Viaduct application from command.".red
+        exit(1)
+      end
+      
+      def heading(title)
+        puts "+" + ("-" * 78) + "+" 
+        puts "| #{title.ljust(76).yellow} |"
+        puts "+" + ("-" * 78) + "+" 
+      end
+      
+      def field(key, value)
+        key = key[0,16].ljust(16, ' ')
+        value = value.to_s[0,58].ljust(58)
+        
+        puts "| #{key.blue}| #{value} |"
+      end
+      
+      def details(&block)
+        block.call
+        puts "+" + ("-" * 78) + "+" 
+      end
+      
+    end
+  end
+end

data/lib/viaduct/toolkit/version.rb

@@ -0,0 +1,5 @@
+module Viaduct
+  module Toolkit
+    VERSION = '0.0.1'
+  end
+end

metadata

@@ -0,0 +1,121 @@
+--- !ruby/object:Gem::Specification
+name: viaduct-toolkit
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Adam Cooke
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-09-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: commander
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+- !ruby/object:Gem::Dependency
+  name: viaduct-api
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.4.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.4.5
+description: A set of useful tools to help developers use & manage their Viaduct applications.
+email:
+- adam@viaduct.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/viaduct/toolkit.rb
+- lib/viaduct/toolkit/commands/app_console.rb
+- lib/viaduct/toolkit/commands/app_deployments.rb
+- lib/viaduct/toolkit/commands/app_env.rb
+- lib/viaduct/toolkit/commands/app_info.rb
+- lib/viaduct/toolkit/commands/app_list.rb
+- lib/viaduct/toolkit/commands/app_public_key.rb
+- lib/viaduct/toolkit/commands/app_status.rb
+- lib/viaduct/toolkit/commands/domain_list.rb
+- lib/viaduct/toolkit/commands/login.rb
+- lib/viaduct/toolkit/commands/ssh_key_add.rb
+- lib/viaduct/toolkit/commands/ssh_key_list.rb
+- lib/viaduct/toolkit/helpers.rb
+- lib/viaduct/toolkit/version.rb
+homepage: http://viaduct.io
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: A developer toolkit for working with Viaduct from an CLI
+test_files: []
+has_rdoc: