bbcloud 0.6.2

data/.gitignore

@@ -0,0 +1,4 @@
+pkg/*
+*.gem
+.bundle
+lib/bbcloud/vendor/

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2010 Brightbox Systems Ltd.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README

@@ -0,0 +1,34 @@
+= Brightbox Cloud Command Line Interface
+
+bbcloud is a cli tool to interact with the (forthcoming) Brightbox
+cloud API.
+
+http://www.brightbox.co.uk
+
+== Packaging
+
+=== Vendoring libraries
+
+gems can be vendored into lib/bbcloud/vendor for packaging and will be
+prioritized over any gems.
+
+=== GEM packaging
+
+The gem currently has our branch of fog vendored, as the official fog
+gem doesn't have our work merged in yet.
+
+=== Debian/Ubuntu packaging
+
+Any gems that have equivalent packages available are set as
+dependencies to the Debian package.  Any other gems are vendored prior
+to packaging.  These are:
+
+* formatador (required by hirb)
+* excon (required by fog)
+* gli
+* fog
+* hirb
+* ini
+
+Rubygems support can be disabled by defining the DISABLE_RUBYGEMS
+constant in lib/bbcloud/os_config.rb

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/bbcloud.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "bbcloud/version"
+
+Gem::Specification.new do |s|
+  s.name        = "bbcloud"
+  s.version     = Brightbox::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["John Leach"]
+  s.email       = ["john@brightbox.co.uk"]
+  s.homepage    = "http://docs.brightbox.com/cli"
+  s.summary     = %q{The Brightbox cloud management system}
+  s.description = %q{Scripts to interact with the Brightbox cloud API}
+
+  s.rubyforge_project = "bbcloud"
+
+  s.files         = `git ls-files`.split("\n") + `find lib/bbcloud/vendor`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+	s.add_dependency 'gli', '1.1.2'
+#	unless File.exists?(File.dirname(__FILE__) + 'lib/bbcloud/vendor/hirb')
+#		s.add_dependency 'hirb', '0.3.4'
+#	end
+	s.add_dependency 'brightbox-fog', '0.3.16'
+	s.add_dependency 'excon', '>=0.2.4'
+	s.add_dependency 'ini', '0.1.1'
+end

data/bin/brightbox-accounts

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-cloudips

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-config

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-images

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-servers

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-types

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-users

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/bin/brightbox-zones

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+begin
+  require "bbcloud/cli"
+rescue LoadError
+  bbcloud_path = File.expand_path('../../lib', __FILE__)
+  $:.unshift(bbcloud_path)
+  require "bbcloud/cli"
+end

data/lib/bbcloud/accounts.rb

@@ -0,0 +1,39 @@
+module Brightbox
+  class Account < Api
+
+    def ram_free
+      [ram_limit.to_i - ram_used.to_i, 0].max
+    end
+
+    def to_row
+      attributes.merge({ :ram_free => ram_free, 
+                         :cloud_ip_limit => limits_cloudips
+                       })
+    end
+
+    def self.all
+      a = conn.account
+      a.connection = conn
+      [new(a)]
+    end
+
+    def self.get(id)
+      a = conn.account
+      a.connection = conn
+      if a.id == id
+        new(a)
+      else
+        nil
+      end
+    end
+
+    def self.default_field_order
+      [:id, :name, :cloud_ip_limit, :ram_limit, :ram_used, :ram_free]
+    end
+
+    def to_s
+      @id
+    end
+
+  end
+end

data/lib/bbcloud/api.rb

@@ -0,0 +1,107 @@
+module Brightbox
+  class Api
+    attr_reader :id
+    class ApiError < StandardError ; end
+    class NotFound < ApiError ; end
+    class Conflict < ApiError ; end
+    class InvalidRecord < ApiError ; end
+    class Forbidden < ApiError ; end
+
+    @@api = nil
+
+    def self.conn
+      if @@api
+        @@api
+      else
+        @@api = Fog::Brightbox::Compute.new CONFIG.to_fog
+      end
+    end
+
+    def initialize(m = nil)
+      if m.is_a? String
+        @id = m
+      elsif !m.nil?
+        @fog_model = m
+        @id = m.id
+      end
+    end
+
+    def fog_model
+      @fog_model ||= self.class.find(@id)
+    end
+
+    def exists?
+      !fog_model.nil?
+    rescue NotFound
+      false
+    end
+
+    def to_s
+      @id
+    end
+
+    def self.find(args = :all, options = {})
+      return nil if args.nil?
+      options = {
+        :order => :created_at,
+      }.merge options
+      objects = []
+      # get the data from the Api
+      if args == :all or (args.respond_to?(:empty?) and args.empty?)
+        objects = all
+      elsif args.respond_to? :collect
+        objects = args.collect do |arg|
+          cached_get(arg)
+        end
+      end
+      # wrap in our objects
+      objects.collect! { |o| new(o) unless o.nil? }
+      # Sort
+      objects.sort! do |a,b| 
+        sort_method = options[:order]
+        begin
+          a.send(sort_method) <=> b.send(sort_method)
+        rescue NoMethodError
+          0
+        end
+      end
+      if objects.size <= 1 and args.is_a? String
+        # This was a single lookup
+        objects.first
+      else
+        objects
+      end
+    end
+
+    def method_missing(m, *args)
+      if fog_model
+        fog_model.send(m, *args)
+      else
+        raise NoMethodError
+      end
+    end
+
+    def self.cached_get(id)
+      @cache = {} if @cache.nil?
+      value = @cache[id]
+      if value
+        value
+      else
+        debug "writing cache entry #{id}"
+        @cache[id] = get(id)
+      end
+    end
+
+    def self.find_by_handle(h)
+      find(:all).find { |o| o.handle == h }
+    end
+
+    def self.cache_all!
+      @cache = {}
+      all.each do |f|
+        @cache[f.id] = f
+      end
+    end
+
+  end
+end

data/lib/bbcloud/cli.rb

@@ -0,0 +1,127 @@
+os_config = File.join(File.dirname(__FILE__), 'os_config.rb')
+require os_config if File.exist? os_config
+
+require "rubygems" unless defined?(DISABLE_RUBYGEMS)
+
+# Add any vendored libraries into search path
+Dir.glob(File.join(File.dirname(__FILE__), 'vendor/*')).each do |f|
+  $:.unshift File.join(f, 'lib')
+end
+
+begin
+  require 'json/ext'
+rescue LoadError
+  require 'json/pure'
+end
+
+require 'date'
+require 'gli'
+require 'bbcloud/tables'
+require 'brightbox-fog'
+
+%w{api servers images types zones cloud_ips users accounts config version}.each do |f|
+  require File.join(File.dirname(__FILE__), f)
+end
+
+CONFIG = BBConfig.new
+
+include Brightbox
+include GLI
+
+class Time
+  def rfc8601
+    self.strftime("%Y-%m-%dT%H:%M")
+  end
+  def to_s
+    rfc8601
+  end
+end
+
+def error(s='')
+  STDERR.write s
+  STDERR.write "\n"
+  STDERR.flush
+end
+
+def info(s='')
+  STDERR.write s
+  STDERR.write "\n"
+  STDERR.flush
+end
+
+def warn(s='')
+  info "WARN: #{s}"
+end
+
+def data(s)
+  STDOUT.write s
+  STDOUT.write "\n"
+end
+
+def debug(s)
+  if ENV['DEBUG']
+    STDERR.write "DEBUG: "
+    STDERR.write s
+    STDERR.write "\n"
+  end
+end
+
+# Global options
+desc "Simple output (tab separated, don't draw fancy tables)"
+switch [:s, :simple]
+
+desc "Set the api client to use (named in #{CONFIG.config_filename})"
+flag [:c, :client]
+
+# Load the command libraries for the current group
+cmd_group_name = File.basename($0).gsub(/brightbox\-/,'')
+cmd_group_files = File.join(File.dirname(__FILE__), "commands/#{cmd_group_name}*.rb")
+Dir.glob(cmd_group_files).each do |f|
+  load f
+end
+
+pre do |global_options,command,options,args|
+  CONFIG.client_name = global_options[:c] if global_options[:c]
+  info "INFO: client_id: #{CONFIG.client_name}" if CONFIG.clients.size > 1
+  true
+end
+
+on_error do |e|
+  case e
+  when Excon::Errors::ServiceUnavailable
+    error "Api currently unavailable"
+  when Excon::Errors::Error
+    if e.respond_to?(:response) and e.response.respond_to?(:body)
+      r = JSON.parse(e.response.body) rescue {}
+      if r['error']
+        if r['error'].is_a? Hash
+          # Most API errors
+          r = r['error']
+          error "ERROR: #{e.class}: #{r['name']}: #{r['summary']}"
+        else
+          # Auth error
+          error "ERROR: #{e.class}: #{r['error']}: #{r['error_description']}"
+        end
+      else
+        error "ERROR: #{e}"
+      end
+    else
+      error "ERROR: #{e}"
+    end           
+  else
+    error "ERROR: #{e}"
+  end
+  debug e
+  debug e.class.to_s
+  debug e.backtrace.join("\n")
+  exit 1
+end
+
+desc 'Display version information'
+command [:version] do |c|
+  c.action do |global_options, options, args|
+    info "Brightbox CLI version: #{Brightbox::VERSION}"
+  end
+end
+
+run ARGV

data/lib/bbcloud/cloud_ips.rb

@@ -0,0 +1,40 @@
+module Brightbox
+  class CloudIP < Api
+    def self.get id
+      conn.cloud_ips.get id
+    end
+
+    def self.all
+      conn.cloud_ips
+    end
+
+    def self.create
+      r = conn.create_cloud_ip
+      new(JSON.parse(r.body)["id"])
+    rescue Excon::Errors::Forbidden => e
+      response = JSON.parse(e.response.body) rescue {}
+      response = response.fetch("error", {})
+      raise Forbidden, "#{response["details"]}: #{response["summary"]}"
+    end
+
+    def attributes
+      fog_model.attributes
+    end
+
+    def to_row
+      attributes
+    end
+
+    def mapped?
+      status == 'mapped'
+    end
+
+    def self.default_field_order
+      [:id, :status, :public_ip, :server_id, :interface_id, :reverse_dns]
+    end
+
+    def <=>(b)
+      self.status <=> b.status
+    end
+  end
+end

data/lib/bbcloud/commands/accounts-list.rb

@@ -0,0 +1,9 @@
+desc 'List accounts'
+arg_name '[account-id...]'
+command [:list] do |c|
+  c.action do |global_options, options, args|
+
+    accounts = Account.find(args).compact
+    render_table(accounts, global_options)
+  end
+end

data/lib/bbcloud/commands/accounts-reset-ftp-password.rb

@@ -0,0 +1,27 @@
+desc 'Reset FTP Library password'
+arg_name 'account-id...'
+command [:reset_ftp_password] do |c|
+
+  c.action do |global_options,options,args|
+
+    accounts = Account.find(args)
+
+    rows = []
+
+    accounts.each do |s|
+      info "Resetting ftp password for #{s}"
+      new_password = s.reset_ftp_password
+      o = s.to_row
+      o[:library_ftp_password] = new_password
+      rows << o
+    end
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :name, :library_ftp_host, :library_ftp_user, :library_ftp_password ]
+    })
+
+    render_table(rows, table_opts)
+
+  end
+end

data/lib/bbcloud/commands/accounts-show.rb

@@ -0,0 +1,18 @@
+desc 'Show detailed account info'
+arg_name 'account-id...'
+command [:show] do |c|
+
+  c.action do |global_options,options,args|
+
+    accounts = Account.find(args).compact
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :name, :cloud_ip_limit, :ram_limit, :ram_used, 
+                  :ram_free, :library_ftp_host, :library_ftp_user ]
+    })
+
+    render_table(accounts, table_opts)
+
+  end
+end

data/lib/bbcloud/commands/cloudips-create.rb

@@ -0,0 +1,21 @@
+desc 'Create a new cloud ip'
+command [:create] do |c|
+  c.desc "Number of cloud ips to create"
+  c.default_value 1
+  c.flag [:i, "count"]
+
+  c.action do |global_options, options, args|
+
+    if options[:i].to_s !~ /^[0-9]+$/
+      raise "count must be a number"
+    end
+
+    options[:i] = options[:i].to_i
+
+    ips = []
+    options[:i].times do
+      ips << CloudIP.create
+    end
+    render_table(ips, global_options)
+  end
+end

data/lib/bbcloud/commands/cloudips-destroy.rb

@@ -0,0 +1,37 @@
+desc 'destroy Cloud IPs'
+arg_name '[cloudip-id...]'
+command [:destroy] do |c|
+  c.desc "Unmap mapped ips before destroying them"
+  c.switch [:u, "unmap"]
+
+  c.action do |global_options,options,args|
+
+    if args.size < 1
+      raise "You must specify the cloud ip id you want to destroy"
+    end
+
+    args.each do |ip_id|
+
+      ip = CloudIP.find ip_id
+
+      raise "Cannot find cloud ip #{ip_id}" if ip.nil?
+
+      if ip.mapped?
+        if options[:u]
+          info "Unmapping cloud ip #{ip}"
+          ip.unmap
+          3.times do
+            break unless ip.mapped?
+            sleep 1
+            ip.reload
+          end
+        else
+          raise "Cannot destroy mapped cloud ip #{ip}"
+        end
+      end
+      info "Destroying cloud ip #{ip.public_ip} (#{ip})"
+      ip.destroy
+    end
+
+  end
+end

data/lib/bbcloud/commands/cloudips-list.rb

@@ -0,0 +1,11 @@
+desc 'List CloudIPs'
+arg_name '[cloudip-id...]'
+command [:list] do |c|
+
+  c.action do |global_options,options,args|
+
+    ips = CloudIP.find args
+
+    render_table(ips.compact.sort, global_options)
+  end
+end

data/lib/bbcloud/commands/cloudips-map.rb

@@ -0,0 +1,55 @@
+desc 'map Cloud IPs'
+arg_name 'cloudip-id server-id'
+command [:map] do |c|
+  c.desc "Unmap mapped ips before remapping them"
+  c.switch [:u, "unmap"]
+
+  c.action do |global_options,options,args|
+
+    if args.size > 2
+      raise "Too many arguments"
+    end
+
+    if args.size < 2
+      raise "You must specify the cloud ip id and the server id"
+    end
+
+    ip_id = args.first
+
+    ip = CloudIP.find ip_id
+
+    raise "Cannot find ip #{ip_id}" if ip.nil?
+
+    if ip.mapped?
+      if options[:u]
+        info "Unmapping ip #{ip}"
+        ip.unmap
+        3.times do
+          break unless ip.mapped?
+          sleep 1
+          ip.reload
+        end
+      else
+        raise "Refusing to map already mapped ip #{ip}"
+      end
+    end
+
+    server_id = args.last
+    server = Server.find server_id
+    raise "Cannot find server #{server_id}" if server.nil?
+    
+    interface_id = server.interfaces.first["id"]
+    info "Mapping #{ip} to interface #{interface_id} on #{server}"
+
+    ip.map interface_id
+
+    # Wait up to 3 seconds for mapping to complete
+    3.times do
+      sleep 1
+      ip.reload
+      break if ip.mapped?
+    end
+
+    render_table([ip], global_options)
+  end
+end