scalarium 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in sshalarium.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,7 @@
+require "bundler/gem_tasks"
+
+
+task :run do
+  exec "bundle exec ruby -I lib bin/scalarium #{ARGV[1..-1]}"
+end
+

data/bin/scalarium

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require 'scalarium'
+require 'scalarium/cli'
+
+Scalarium::CLI.start

data/lib/scalarium.rb

@@ -0,0 +1,68 @@
+require "scalarium/version"
+require 'rest-client'
+require 'json'
+require 'dispatch_queue'
+require 'ostruct'
+
+class Scalarium
+  class Instance < OpenStruct ; end
+  class Rol     < OpenStruct ; end
+  class Cloud    < OpenStruct ; end
+
+  attr_reader :clouds
+  def initialize(token, cloud = nil)
+    @token = token
+    @clouds = []
+
+    @clouds = get('clouds').map{|c| Cloud.new(c) }
+
+    return if cloud == false
+
+    if cloud != nil
+      filtered_clouds = @clouds.select do |c|
+        (c.name =~ /#{cloud}/i) || c.name.downcase.include?(cloud.downcase)
+      end
+      @clouds = filtered_clouds || []
+    end
+    DQ[*(@clouds.map {  |cloud| lambda { process_cloud(cloud) } })]
+  end
+
+  private
+
+  def get(resource)
+    $stderr.puts "Getting #{resource}" if $DEBUG
+    url = "https://manage.scalarium.com/api/#{resource}"
+    headers = {
+      'X-Scalarium-Token' => @token,
+      'Accept' => 'application/vnd.scalarium-v1+json'
+    }
+    response = RestClient.get(url, headers)
+    JSON.parse(response)
+  end
+
+  def get_roles_proc(cloud_id)
+    lambda do
+      get("clouds/#{cloud_id}/roles").map { |hash|
+        Rol.new(hash)
+      }
+    end
+  end
+
+  def get_instances_proc(cloud_id)
+    lambda do
+      get("clouds/#{cloud_id}/instances").map { |hash|
+        Instance.new(hash)
+      }
+    end
+  end
+
+  def process_cloud(cloud)
+    roles, instances = DQ[ get_roles_proc(cloud.id) , get_instances_proc(cloud.id)  ]
+    cloud.roles = roles
+    cloud.instances = instances
+    roles.each do |rol|
+      rol.instances = instances.select{|instance| instance.role_ids.include?(rol.id)}
+    end
+  end
+end
+

data/lib/scalarium/cli.rb

@@ -0,0 +1,113 @@
+require 'thor'
+
+trap("INT"){ exit -1 }
+
+class Scalarium
+  class CLI < Thor
+    include Thor::Actions
+    desc 'update_sshconfig', "Create ssh hosts for your infraestructure in ~/.ssh/config"
+    method_option :cloud, :aliases => "-c", :desc => "only for clouds matchin this regexp"
+    method_option :certificate, :aliases => "-i", :desc => "specify alternate certificate for login"
+    def update_sshconfig
+      capture_exceptions do
+        scalarium = ::Scalarium.new( get_token, options[:cloud] )
+
+        config_file = File.expand_path("~/.ssh/config")
+        header = "# Added by scalarium"
+        tail = "# Do not modify"
+        if File.exists?(config_file)
+          config = File.read(config_file)
+          config.gsub!(/#{header}.*#{tail}/mi,'')
+        else
+          config = ""
+        end
+
+        config  << header << "\n"
+        scalarium.clouds.each do |cloud|
+          say "Adding hosts from #{cloud.name}"
+          cloud.instances.each do |instance|
+            config << format_ssh_config_host(instance)
+          end
+        end
+        config << "\n" << tail << "\n"
+
+        File.open(config_file,'w'){|f|
+          f.rewind
+          f.write config
+        }
+        say("Configuration file updated")
+      end
+
+
+
+    end
+
+    desc 'inspect', "Show Clouds, Roles and Instance names"
+    method_option :cloud, :aliases => "-c", :desc => "only for clouds matchin this regexp"
+    def inspect
+      capture_exceptions do
+        scalarium = ::Scalarium.new( get_token, options[:cloud] )
+
+        scalarium.clouds.each do |cloud|
+          print_cloud(cloud)
+        end
+      end
+    end
+
+    desc 'clouds', "List the clouds"
+    def clouds
+      capture_exceptions do
+        scalarium = ::Scalarium.new( get_token, false )
+        scalarium.clouds.each do |cloud|
+          puts cloud.name
+        end
+      end
+    end
+
+
+    protected
+
+    def format_ssh_config_host(instance)
+      return "" if instance.ip.to_s.strip.empty?
+      host = "\nHost #{instance.nickname}\n" << "    Hostname #{instance.ip}\n"
+      host << "    IdentityFile #{options[:certificate]}" if options[:certificate]
+      host
+    end
+
+    def capture_exceptions
+      yield
+    rescue ::RestClient::Unauthorized
+      say("The token is not valid", Color::RED)
+      File.unlink token_path
+      retry
+    rescue ::OpenSSL::SSL::SSLError
+      say("There were problems with ssl. Pleas retry")
+      exit -2
+    end
+
+    def get_token
+      if File.exists?(token_path)
+        token = File.read(token_path).strip
+      else
+        token = ask("I need your token:").strip
+        File.open(token_path, 'w'){|f| f.write token }
+      end
+      token
+    end
+
+    def token_path
+      File.expand_path("~/.scalarium_token")
+    end
+
+    def print_cloud(cloud)
+      say "#{cloud.name} (#{cloud.id[0..6]})", Color::GREEN
+      cloud.roles.each do |rol|
+        say "  #{rol.name}", Color::BLUE
+        rol.instances.each do |instance|
+          puts "    #{instance.nickname} (#{instance.ip})"
+        end
+      end
+    end
+
+  end
+end

data/lib/scalarium/version.rb

@@ -0,0 +1,3 @@
+class Scalarium
+  VERSION = "0.0.1"
+end

data/scalarium.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "scalarium/version"
+
+Gem::Specification.new do |s|
+  s.name        = "scalarium"
+  s.version     = Scalarium::VERSION
+  s.authors     = ["Guillermo Álvarez"]
+  s.email       = ["guillermo@cientifico.net"]
+  s.homepage    = ""
+  s.summary     = %q{Scalarium console access}
+  s.description = %q{Access your scalarium clouds from console}
+
+  s.rubyforge_project = "scalarium"
+
+  s.files         = `git ls-files`.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_runtime_dependency 'rest-client'
+	s.add_runtime_dependency 'dispatch_queue'
+  s.add_runtime_dependency 'thor'
+  s.add_development_dependency 'rake'
+end

metadata

@@ -0,0 +1,98 @@
+--- !ruby/object:Gem::Specification
+name: scalarium
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Guillermo Álvarez
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-09-09 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: &2160501020 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2160501020
+- !ruby/object:Gem::Dependency
+  name: dispatch_queue
+  requirement: &2160500600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2160500600
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: &2160500180 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2160500180
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &2160499760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *2160499760
+description: Access your scalarium clouds from console
+email:
+- guillermo@cientifico.net
+executables:
+- scalarium
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/scalarium
+- lib/scalarium.rb
+- lib/scalarium/cli.rb
+- lib/scalarium/version.rb
+- scalarium.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: scalarium
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Scalarium console access
+test_files: []