bbcloud 0.6.2

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

@@ -0,0 +1,13 @@
+desc 'show details on Cloud IPs'
+arg_name 'cloudip-id...'
+command [:show] do |c|
+
+  c.action do |global_options,options,args|
+
+    ips = CloudIP.find args
+
+    fields = [:id, :status, :public_ip, :reverse_dns, :server_id, :interface_id]
+
+    render_table(ips.compact, global_options.merge({ :vertical => true, :fields => fields}))                                              
+  end
+end

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

@@ -0,0 +1,32 @@
+desc 'unmap Cloud IPs'
+arg_name 'cloudip-id...'
+command [:unmap] do |c|
+
+  c.action do |global_options,options,args|
+
+    raise "you must specify the cloud ip ids to unmap as arguments" if args.empty?
+
+    ips = CloudIP.find args
+
+    ips.compact!
+
+    ips.each do |ip|
+      if ip.mapped?
+        info "Unmapping cloud ip #{ip}"
+        ip.unmap
+        ip.reload
+      else
+        warn "Cloud ip #{ip} already unmapped"
+      end
+    end
+
+    # Wait up to 3 seconds for unmapping to complete
+    3.times do
+      break unless ips.find { |ip| ip.mapped? }
+      sleep 1
+      ips.each { |ip| ip.reload if ip.mapped? }
+    end
+
+    render_table(ips, global_options)
+  end
+end

data/lib/bbcloud/commands/config-client-add.rb

@@ -0,0 +1,45 @@
+desc 'Add new api client details to config'
+arg_name 'client-id secret [api_url, auth_url]'
+command [:client_add] do |c|
+
+  c.desc "api client alias, for local reference (defaults to client-id)"
+  c.flag [:a, "alias"]
+
+  c.desc "auth url"
+  c.flag [:t, "auth-url"]
+
+  c.action do |global_options, options, args|
+
+    info "Using config file #{CONFIG.config_filename}"
+
+    client_id = args.shift
+    secret = args.shift
+    api_url = args.shift || "https://api.gb1.brightbox.com"
+    auth_url = args.shift || api_url
+    calias = options[:a] || client_id
+
+    if client_id.nil?
+      raise "You must specify the api client-id"
+    end
+
+    if secret.nil?
+      raise "You must specify the api secret"
+    end
+
+    client_config = CONFIG[calias]
+    unless client_config.empty?
+      raise "An api client with the id or alias #{calias} already exists"
+    end
+
+    info "Creating new api client config #{calias}"
+
+    client_config['alias'] = calias    
+    client_config['client_id'] = client_id
+    client_config['secret'] = secret
+    client_config['api_url'] = api_url
+    client_config['auth_url'] = auth_url
+
+    CONFIG.save!
+
+  end
+end

data/lib/bbcloud/commands/config-client-default.rb

@@ -0,0 +1,24 @@
+desc 'Set the default api client in the config'
+arg_name 'alias'
+command [:client_default] do |c|
+
+  c.action do |global_options, options, args|
+
+    info "Using config file #{CONFIG.config_filename}"
+    calias = args.shift
+
+    if calias.nil?
+      raise "You must specify the api alias you want to set as the default"
+    end
+
+    client_config = CONFIG[calias]
+    if client_config.empty?
+      raise "An api client with the alias #{calias} does not exist in the config"
+    end
+
+    info "Setting #{calias} as default api client"
+    CONFIG['core']['default_client'] = calias
+    CONFIG.save!
+
+  end
+end

data/lib/bbcloud/commands/config-client-remove.rb

@@ -0,0 +1,25 @@
+desc 'Remove api client details from config'
+arg_name 'alias'
+command [:client_remove] do |c|
+
+  c.action do |global_options, options, args|
+
+    info "Using config file #{CONFIG.config_filename}"
+
+    calias = args.shift
+
+    if calias.nil?
+      raise "You must specify the api alias you want to remove"
+    end
+
+    client_config = CONFIG[calias]
+    if client_config.empty?
+      raise "An api client with alias #{calias} does not exist in the config"
+    end
+
+    info "Removing api client #{calias}"
+    CONFIG.delete_section calias
+    CONFIG.save!
+
+  end
+end

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

@@ -0,0 +1,18 @@
+desc 'Destroy images'
+arg_name 'image-id...'
+command [:destroy] do |c|
+
+  c.action do |global_options,options,args|
+
+    images = Image.find(args).compact
+
+    images.each do |i|
+      info "Destroying image #{i}"
+      i.destroy
+      i.reload
+    end
+
+    render_table(images)
+
+  end
+end

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

@@ -0,0 +1,19 @@
+desc 'List available images'
+arg_name '[image-id...]'
+command [:list] do |c|
+  c.action do |global_options, options, args|
+    images = Image.find args
+
+    snapshots = images.select { |i| i.source_type == 'snapshot' }
+
+    images = images - snapshots
+
+    images.sort! do |a,b|
+      [a.official ? 0 : 1, a.name, a.arch] <=> [b.official ? 0 : 1, b.name, b.arch]
+    end
+
+    snapshots.sort! { |a, b| a.created_at <=> b.created_at }
+    
+    render_table(images + snapshots, global_options)
+  end
+end

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

@@ -0,0 +1,17 @@
+desc 'Show detailed image info'
+arg_name 'image-id...'
+command [:show] do |c|
+
+  c.action do |global_options,options,args|
+
+    images = Image.find(args).compact
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :type, :owner, :created_at, :status, :arch, :name, :description, :virtual_size, :disk_size, :ancestor_id ]
+    })
+
+    render_table(images, table_opts)
+
+  end
+end

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

@@ -0,0 +1,65 @@
+desc 'Create servers'
+arg_name 'image_id'
+command [:create] do |c|
+  c.desc "Number of servers to create"
+  c.default_value 1
+  c.flag [:i, "server-count"]
+
+  c.desc "Zone to create the servers in"
+  c.flag [:z, "zone"]
+
+  c.desc "Type of server to create"
+  c.default_value "nano"
+  c.flag [:t, :type]
+
+  c.desc "Friendly name of server"
+  c.flag [:n, :name]
+
+
+  c.action do |global_options, options, args|
+
+    if args.empty?
+      raise "You must specify the image_id as the first argument"
+    end
+
+    if options[:i].to_s !~ /^[0-9]+$/
+      raise "server-count must be a number"
+    end
+
+    options[:i] = options[:i].to_i
+
+    image_id = args.shift
+    image = Image.find image_id
+    raise "Couldn't find image #{image_id}" unless image
+    
+    type_id = options[:t]
+    if type_id =~ /^typ\-/
+      type = Type.find type_id
+    else
+      type = Type.find_by_handle type_id
+    end
+    raise "Couldn't find server type #{type_id}" unless type
+
+    if options[:z]
+      zone = options[:z]
+      if zone =~ /^typ\-/
+        zone = Zone.find zone
+      else
+        zone = Zone.find_by_handle zone
+      end
+    end
+    raise "Couldn't find server type #{type_id}" unless type
+
+    msg = "Creating #{options[:i]} '#{type.handle}' (#{type.id}) server#{options[:i] > 1 ? 's' : ''} with image #{image.name.strip} (#{image.id})"
+    msg << " in zone #{zone.handle} (#{zone})" if zone
+    info msg
+    servers = []
+    options[:i].times do
+      servers << Server.create(:image_id => image.id,
+                               :flavor_id => type.id,
+                               :zone_id => zone.to_s,
+                               :name => options[:n])
+    end
+    render_table(servers, global_options)
+  end
+end

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

@@ -0,0 +1,25 @@
+desc 'Destroy servers'
+arg_name '[server-id...]'
+command [:destroy] do |c|
+  c.action do |global_options, options, args|
+
+    if args.empty?
+      raise "you must specify the id of the server(s) you want to destroy"
+    end
+
+
+    servers = args.collect do |sid|
+      info "Destroying server #{sid}"
+      server = Server.find sid
+      begin
+        server.destroy
+      rescue Brightbox::Api::Conflict => e
+        error "Could not destroy #{sid}"
+      end
+      server.reload
+      server
+    end
+
+    render_table(servers, global_options)
+  end
+end

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

@@ -0,0 +1,13 @@
+desc 'List servers'
+arg_name '[server-id...]'
+command [:list] do |c|
+  c.action do |global_options,options,args|
+
+    Type.cache_all!
+    Zone.cache_all!
+
+    servers = Server.find(args).compact
+
+    render_table(servers, global_options)
+  end
+end

data/lib/bbcloud/commands/servers-restart.rb

@@ -0,0 +1,30 @@
+desc 'Restart the specified servers'
+arg_name 'server-id...'
+command [:restart] do |c|
+  c.desc "Soft restart (default)"
+  c.switch [:s, "soft"]
+
+  c.desc "Hard restart"
+  c.default_value true
+  c.switch [:d, "hard"]
+
+  c.action do |global_options,options,args|
+
+    if options[:soft] and options[:hard]
+      raise "Cannot do both hard and soft restart at the same time"
+    end
+
+    type = options[:hard] ? "hard" : "soft"
+
+    servers = Server.find(args).compact
+
+    servers.each do |s|
+      info "Restarting server #{s}"
+      s.reboot(type)
+      s.reload
+    end
+
+    render_table(servers)
+
+  end
+end

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

@@ -0,0 +1,47 @@
+desc 'Show detailed server info'
+arg_name 'server-id...'
+command [:show] do |c|
+
+  c.action do |global_options,options,args|
+    
+    servers = Server.find(args)
+    rows = []
+    servers.each do |s|
+      if s.is_a? String
+        error "Could not find server #{s}"
+        next
+      end
+      o = s.to_row
+      if s.server_type.exists?
+        o[:type] = s.server_type.id
+        o[:type_handle] = s.server_type.handle
+        o[:type_name] = s.server_type.name
+        o[:ram] = s.server_type.ram
+        o[:cores] = s.server_type.cores
+        o[:disk] = s.server_type.disk.to_i
+      end
+
+      if s.image.exists? and false
+        o[:image_name] = s.image.name
+        o[:arch] = s.image.arch
+      end
+      o[:private_ips] = s.interfaces.collect { |i| i['ipv4_address'] }.join(", ")
+      o[:cloud_ip_ids] = s.cloud_ips.collect { |i| i['id'] }.join(", ")
+      o[:cloud_ips] = s.cloud_ips.collect { |i| i['public_ip'] }.join(", ")
+      o[:snapshots] = s.snapshots.collect { |i| i['id'] }.join(", ")
+      rows << o
+    end
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :status, :name, :description, :created_at, :deleted_at, 
+                  :zone, :type, :type_name, :type_handle, :ram, :cores, 
+                  :disk, :image, :image_name, :private_ips, :cloud_ips, 
+                  :cloud_ip_ids, :hostname, :public_hostname, :snapshots
+                 ]
+    })
+
+    render_table(rows, table_opts)
+
+  end
+end

data/lib/bbcloud/commands/servers-shutdown.rb

@@ -0,0 +1,17 @@
+desc 'Shutdown the specified servers'
+arg_name 'server-id...'
+command [:shutdown] do |c|
+  c.action do |global_options,options,args|
+
+    servers = Server.find(args).compact
+
+    servers.each do |s|
+      info "Shutting down server #{s}"
+      s.shutdown
+      s.reload
+    end
+
+    render_table(servers)
+
+  end
+end

data/lib/bbcloud/commands/servers-snapshot.rb

@@ -0,0 +1,17 @@
+desc 'Snapshot the specified servers'
+arg_name 'server-id...'
+command [:snapshot] do |c|
+
+  c.action do |global_options,options,args|
+
+    servers = Server.find(args).compact
+
+    servers.each do |s|
+      info "Snapshotting server #{s}"
+      s.snapshot
+    end
+
+    render_table(servers)
+
+  end
+end

data/lib/bbcloud/commands/servers-start.rb

@@ -0,0 +1,18 @@
+desc 'Start the specified servers'
+arg_name 'server-id...'
+command [:start] do |c|
+
+  c.action do |global_options,options,args|
+    
+    servers = Server.find(args).compact
+
+    servers.each do |s|
+      info "Starting server #{s}"
+      s.start
+      s.reload
+    end
+
+    render_table(servers)
+
+  end
+end

data/lib/bbcloud/commands/servers-stop.rb

@@ -0,0 +1,18 @@
+desc 'Stop the specified servers'
+arg_name 'server-id...'
+command [:stop] do |c|
+
+  c.action do |global_options,options,args|
+    
+    servers = Server.find(args).compact
+
+    servers.each do |s|
+      info "Stopping server #{s}"
+      s.stop
+      s.reload
+    end
+
+    render_table(servers)
+
+  end
+end

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

@@ -0,0 +1,10 @@
+desc 'List types'
+arg_name '[type-id...]'
+command [:list] do |c|
+  c.action do |global_options,options,args|
+
+    types = Type.find args, :order => :ram
+
+    render_table(types, global_options)
+  end
+end

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

@@ -0,0 +1,22 @@
+desc 'Show detailed type info'
+arg_name 'type-id...'
+command [:show] do |c|
+
+  c.action do |global_options,options,args|
+    
+    types = Type.find(args)
+    rows = []
+    types.each do |t|
+      next if t.nil?
+      rows << t
+    end
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :handle, :status, :name, :ram, :disk, :cores, :description]
+    })
+
+    render_table(rows, table_opts)
+
+  end
+end

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

@@ -0,0 +1,10 @@
+desc 'List users'
+arg_name '[user-id...]'
+command [:list] do |c|
+  c.action do |global_options, options, args|
+
+    users = User.find args
+
+    render_table(users, global_options)
+  end
+end

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

@@ -0,0 +1,27 @@
+desc 'Show detailed user info'
+arg_name 'user-id...'
+command [:show] do |c|
+
+  c.action do |global_options,options,args|
+
+    users = User.find(args)
+
+    rows = []
+
+    users.each do |s|
+      s.reload # to get ssh_key
+      o = s.to_row
+      o[:ssh_key] = s.ssh_key.to_s.strip
+      o[:accounts] = s.accounts.collect { |a| a.id }.join(", ")
+      rows << o
+    end
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :name, :email_address, :accounts, :ssh_key ]
+    })
+
+    render_table(rows, table_opts)
+
+  end
+end

data/lib/bbcloud/commands/users-update.rb

@@ -0,0 +1,39 @@
+desc 'Update user details'
+arg_name 'user-id...'
+command [:update] do |c|
+  c.desc "Path to public ssh key file"
+  c.long_desc "This is the path to the public ssh key that you'd like to use for new servers. You can specify '-' to read from stdin"
+  c.flag [:f, "ssh-key"]
+
+  c.desc "Name"
+  c.flag [:n, "name"]
+
+  c.action do |global_options, options, args|
+
+    raise "You must specify the user id as the first argument" if args.empty?
+
+    user = User.find args.first
+
+    raise "Could not find user #{args.first}" if user.nil?
+      
+    if options[:f] == '-'
+      user.ssh_key = STDIN.read
+    elsif options[:f]
+      File.open(File.expand_path(options[:f])) { |f| user.ssh_key = f.read }
+    end
+
+    if options[:n]
+      user.name = options[:n]
+    end
+
+    user.save
+
+    table_opts = global_options.merge({
+      :vertical => true,
+      :fields => [:id, :name, :email_address, :ssh_key ]
+    })
+
+    render_table([user], table_opts)
+
+  end
+end

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

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

data/lib/bbcloud/config.rb

@@ -0,0 +1,92 @@
+class BBConfigError < StandardError ; end
+
+class BBConfig
+  require 'fileutils'
+  require 'ini'
+
+  attr_writer :client_name
+
+  def initialize(options = {})
+    @options = options
+  end
+
+  def dir
+    return @dir if @dir
+    return nil if @dir == false
+    @dir = File.expand_path(@options[:dir] || "~/.brightbox")
+    @config_filename = File.join(@dir, 'config')
+    # Make the directory if necessary
+    unless File.exists? @dir
+      begin
+        FileUtils.mkdir @dir
+      rescue Errno::EEXIST
+      end
+    end
+    # 
+    if File.directory? @dir
+      @dir
+    else
+      @dir = false
+      nil
+    end
+  end
+
+  def config_filename
+    dir
+    @config_filename
+  end
+
+  def config
+    return @config if @config
+    return {} if @config == false
+    @config ||= Ini.new config_filename
+    @config
+  rescue Ini::Error => e
+    raise BBConfigError, "Config problem in #{config_filename}: #{e}"
+  end
+
+  def [](k)
+    config[k]
+  end
+
+  def delete_section(name)
+    config.delete_section name
+  end
+
+  def save!
+    if @config.is_a? Ini
+      @config.write
+    end
+  end
+
+  def clients
+    config.sections.find_all { |s| s != 'core' }
+  end
+
+  def client_name
+    if @client_name
+      @client_name
+    else
+      default_client = config['core']['default_client']
+      @client_name = default_client || clients.first
+    end
+  end
+
+  def to_fog
+    raise Ini::Error, "No api client configured" if clients.empty?
+    c = config[client_name]
+    %w{api_url client_id secret}.each do |k|
+      if c[k].to_s.empty?
+        raise BBConfigError, "#{k} option missing from config in section #{client_name}"
+      end
+    end
+    { 
+      :brightbox_api_url => c['api_url'],
+      :brightbox_auth_url => c['auth_url'] || c['api_url'],
+      :brightbox_client_id => c['client_id'],
+      :brightbox_secret => c['secret']
+    }
+  end
+
+
+end