rightimage_tools 0.1.0

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -0,0 +1,14 @@
+source "http://rubygems.org"
+
+gem "rest_connection"
+gem "ruby-debug"
+
+# Add dependencies to develop your gem here.
+# Include everything needed to run rake, tests, features, etc.
+group :development do
+  gem "rspec", "~> 2.3.0"
+  gem "bundler", "~> 1.0.0"
+  gem "jeweler", "~> 1.6.4"
+  gem "rcov", ">= 0"
+  gem "flexmock"
+end

data/Gemfile.lock

@@ -0,0 +1,50 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    activesupport (2.3.10)
+    columnize (0.3.4)
+    diff-lcs (1.1.3)
+    flexmock (0.9.0)
+    git (1.2.5)
+    highline (1.6.2)
+    jeweler (1.6.4)
+      bundler (~> 1.0)
+      git (>= 1.2.5)
+      rake
+    json (1.6.1)
+    linecache (0.46)
+      rbx-require-relative (> 0.0.4)
+    net-ssh (2.1.4)
+    rake (0.9.2.2)
+    rbx-require-relative (0.0.5)
+    rcov (1.0.0)
+    rest_connection (0.1.2)
+      activesupport (= 2.3.10)
+      highline
+      json
+      net-ssh (= 2.1.4)
+    rspec (2.3.0)
+      rspec-core (~> 2.3.0)
+      rspec-expectations (~> 2.3.0)
+      rspec-mocks (~> 2.3.0)
+    rspec-core (2.3.1)
+    rspec-expectations (2.3.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.3.0)
+    ruby-debug (0.10.4)
+      columnize (>= 0.1)
+      ruby-debug-base (~> 0.10.4.0)
+    ruby-debug-base (0.10.4)
+      linecache (>= 0.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.0.0)
+  flexmock
+  jeweler (~> 1.6.4)
+  rcov
+  rest_connection
+  rspec (~> 2.3.0)
+  ruby-debug

data/LICENSE.txt

@@ -0,0 +1,5 @@
+# RightScale Tools
+
+# Copyright RightScale, Inc. All rights reserved.  All access and use subject to the
+# RightScale Terms of Service available at http://www.rightscale.com/terms.php and,
+# if applicable, other agreements such as a RightScale Master Subscription Agreement.

data/README.rdoc

@@ -0,0 +1,19 @@
+= rightimage_tools
+
+RightImage tools is a collection of miscellaneous scripts and libraries useful to the creation of RightImages
+
+== Usage
+require 'rightimage_tools'
+
+# This will create a random mci if it doesn't exist.
+mci_tool = RightImageTools::MCI.new
+mci_tool.add_image_to_mci(
+  :name => "DEV: RightImage_CentOS_5.6_x64_v5.7",
+  :cloud_id => 6, #Oregon,
+  :image_id => "aws-xxxxxx")
+
+
+== Copyright
+
+Copyright (c) 2012 RightScale, Inc. See LICENSE.txt for further details.
+

data/Rakefile

@@ -0,0 +1,49 @@
+# encoding: utf-8
+
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'rake'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
+  gem.name = "rightimage_tools"
+  gem.homepage = "http://github.com/rightscale/rightimage_tools"
+  gem.license = "MIT"
+  gem.summary = %Q{A set of tools to support the building of RightImages}
+  gem.description = gem.summary
+  gem.email = "peter.schroeter@rightscale.com"
+  gem.authors = ["Peter Schroeter"]
+  # dependencies defined in Gemfile
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/**/*_spec.rb']
+end
+
+RSpec::Core::RakeTask.new(:rcov) do |spec|
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "rightimage_tools #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/lib/mci.rb

@@ -0,0 +1,204 @@
+require 'rubygems'
+require 'logger'
+require 'ruby-debug'
+require 'rest_connection'
+
+module RightImageTools
+  class MCI
+
+    def initialize(options = {})
+      @options = options
+      @logger = (@options.key?(:logger)) ? @options[:logger] : Logger.new(STDOUT)
+      @api_url = Tag.connection.settings[:api_url]
+    end
+
+    def default_description(mci_name, is_ec2 = false)
+      # 32-Bit Image description
+      #   OsName NN.NN Hypervisor with 32-bit architecture (i386)
+      # 64-Bit image description
+      #   OsName NN.NN Hypervisor with 64-bit architecture (x64)
+      #
+
+      os = (mci_name =~ /(centos|ubuntu|windows|rhel)/i) ? $1 : ""
+      os_version = ""
+      unless os.empty?
+        os_version = (mci_name =~ /#{os}[\s_](\d+\.\d+)/) ? $1 : ""
+      end
+      os_arch = (mci_name =~ /(i[3|6]86|x64|x86_64|amd64)/i) ? $1 : ""
+      hypervisor = (mci_name =~ /(xenserver|xen|vmware|esxi|kvm)/i) ? $1 : ""
+      rl_version = (mci_name =~ /v(\d+\.\d+)/) ? $1 : ""
+
+      description = "Development build:"
+      description << " #{os}" if os
+      description << " #{os_version}" if os_version
+      description << " #{hypervisor}" if !hypervisor.empty? && !is_ec2
+      if os_arch
+        if os_arch =~ /i[3|6]86/
+          description << " with 32-bit architecture (#{os_arch})"
+        else
+          description << " with 64-bit architecture (#{os_arch})"
+        end
+      end
+      description << " and RightLink #{rl_version}" if rl_version
+      description << "."
+    end
+
+    # Adds an image to an mci, if the image is not already attached to the mci
+    #
+    # === Parameters
+    # image_id(String)
+    # cloud_id(Number)
+    # mci_name(String) 
+    #
+    # === Return
+    # mci(MultiCloudImageCloudSettingInternal):: new or existing MCI setting
+    def add_image_to_mci(options)
+      cloud_id = options[:cloud_id].to_i 
+      image_id = options[:image_id]
+      mci_name = options[:name]
+      description = options[:description]
+
+      raise ArgumentError, ":cloud_id not supplied" unless cloud_id > 0
+      raise ArgumentError, ":image_id not supplied" unless image_id =~ /./
+      raise ArgumentError, "MCI name (:name) not supplied" unless mci_name =~ /./
+
+      image_id = image_id.split("/").last
+
+      if api_version(cloud_id) == "1.0"
+        unless image_id =~ /^ami-[0-9a-z]+$/
+          raise ArgumentError, "image_id #{image_id} doesn't look like an amazon ami"
+        end
+      end
+
+      mci = find_mci(cloud_id, mci_name) 
+      #validate_mci_name(mci_name) unless mci
+      mci = create_mci(cloud_id, mci_name, description) unless mci
+      add_rightlink_tag(mci)
+      mci_setting = find_or_create_cloud_setting(mci, image_id, cloud_id) 
+
+      # Create the MCIs, if they don't exist.
+
+      return mci.href
+    end
+
+    def validate_mci_name(name)
+    end
+
+    def guess_instance_type(mci_name)
+      mci_name =~ /i386/ ? "m1.small" : "m1.large"
+    end
+
+    def find_cloud_setting(mci,cloud_id)
+      mci_setting = nil
+      mci_setting = mci.multi_cloud_image_cloud_settings.select { |setting| setting.cloud_id.to_i == cloud_id.to_i }.first
+    end
+
+    def find_or_create_cloud_setting(mci, image_id, cloud_id)
+      mci_setting = nil;
+      if api_version(cloud_id) == "1.0"
+        mci_setting = find_cloud_setting(mci, cloud_id)
+        instance_type = guess_instance_type(mci.name)
+
+        if mci_setting
+          unless mci_setting.image_href.include?(image_id)
+            @logger.warn("Replacing image for cloud #{cloud_id} for MCI #{mci.rs_id}")
+            res = mci_setting.destroy
+            raise "Non success code returned #{res.inspect} " if res.code.to_s !~ /^2\d\d$/
+            create_cloud_setting(mci, image_id, cloud_id, instance_type)
+          end
+        else
+          @logger.info("Adding cloud images to MCI #{mci.href}")
+          mci_setting = create_cloud_setting(mci, image_id, cloud_id, instance_type)
+          #returns nil
+        end
+      else 
+        raise NotImplementedError, "API 1.5 not supported yet"
+      end
+      mci_setting
+    end
+
+    def create_cloud_setting(mci, image_id, cloud_id, instance_type = nil)
+      mci_setting = nil
+      if api_version(cloud_id) == "1.0"
+        # create the setting
+        #update_connection_settings(MultiCloudImageInternal)
+        image_href = "#{@api_url}/ec2_images/#{image_id}?cloud_id=#{cloud_id}"
+        mci_setting = MultiCloudImageCloudSettingInternal.create(
+          :multi_cloud_image_href  => mci.href,
+          :cloud_id                => cloud_id.to_i,
+          :ec2_image_href          => image_href, 
+          :aws_instance_type       => instance_type)
+      else 
+        raise NotImplementedError, "API 1.5 not supported yet"
+      end
+      mci_setting
+    end
+
+    # return MultiCloudImageInternal, or nil
+    def find_mci(cloud_id, mci_name)
+      mcis = []
+      if api_version(cloud_id) == "1.0"
+        #update_connection_settings(MultiCloudImage)
+        mcis = MultiCloudImage.find_all.
+          select {|n| n.is_head_version && n.name == mci_name }
+        mci_ids = mcis.map { |m| m.rs_id }
+        @logger.warn("Found multiple MCIs with name #{mci_name}: #{mci_ids.join(', ')}") if mcis.length > 1
+        if mcis.length > 0
+          existing_mci = MultiCloudImageInternal.find(mcis.first.rs_id.to_i)
+        end
+      else
+        # Double filter, find_with_filter is a wildcard ended match
+        # Also revision 0 is HEAD
+        #update_connection_settings(McMultiCloudImage)
+        mcis = McMultiCloudImage.
+          find_with_filter(:name=>mci_name).
+          select {|mci| mci.name == mci_name}.
+          select {|mci| mci.revision == 0}
+        @logger.warn("Found multiple MCIs with name #{mci_name}") if mcis.length > 1
+        existing_mci = mcis.first
+      end
+
+      @logger.info("Found mci #{existing_mci.rs_id}") if existing_mci
+      existing_mci
+    end
+
+    def create_mci(cloud_id, mci_name, description = nil)
+      description ||= default_description(mci_name, api_version(cloud_id) == "1.0")
+      @logger.info("Creating mci #{mci_name}")
+      mci = nil
+
+      #update_connection_settings(MultiCloudImageInternal)
+      mci = MultiCloudImageInternal.create(
+        :name        => mci_name,
+        :description => description)
+
+      raise "Could not create MCI" unless mci
+      mci
+    end
+
+    def add_rightlink_tag(mci)
+      tag = "provides:rs_agent_type=right_link"
+      add_tag_to_mci(mci, tag)
+    end 
+
+    # mci(MultiCloudImageInternal):: the MCI to add the tag to
+    # Tag an mci or raise an error on failure
+    def add_tag_to_mci(mci, tag)
+      begin
+        @logger.info "Adding tag #{tag} to #{mci.href}"
+        href = mci.href
+        result = Tag.set(href, ["#{tag}"])
+        #update_connection_settings(Tag)
+        @logger.debug("Successfully tagged MCI. Code: #{result.code}")
+      rescue Exception => e
+        @logger.error("Failed to tag MCI #{mci.name}! #{e.inspect}")
+        raise e
+      end 
+    end
+
+    def api_version(cloud_id)
+      cloud_id.to_i < 50 ? "1.0" : "1.5"
+    end
+
+  end
+end

data/lib/rightimage_tools.rb

@@ -0,0 +1,5 @@
+module RightImageTools
+  VERSION = File.read(File.join(File.dirname(__FILE__), "..", "VERSION")).chomp
+end
+
+require File.join(File.dirname(__FILE__), 'mci')

data/spec/mci_spec.rb

@@ -0,0 +1,97 @@
+require 'flexmock/rspec'
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe RightImageTools::MCI do
+
+  before :each do 
+    @mci = RightImageTools::MCI.new
+
+    @mock_mci_name = "Mock_RightImage_Ubuntu_10.04_x64_v5.7" 
+    @mock_ami = "ami-2ehahaha"
+
+    @mock_mci = flexmock(MultiCloudImage, 
+      :is_head_version => true, 
+      :name => @mock_mci_name,
+      :rs_id => "10000", 
+      :version=>0, 
+      :href=>"https://my.rightscale.com/api/acct/99999/multi_cloud_images/10000")
+
+    @mock_mci_setting1 = flexmock(MultiCloudImageCloudSettingInternal,
+      :href=>"https://my.rightscale.com/api/acct/99999/multi_cloud_image_cloud_settings/20000", 
+      :aws_instance_type =>"m1.large", 
+      :image_name => @mock_mci_name,
+      :image_href => "https://my.rightscale.com/api/acct/0/ec2_images/#{@mock_ami}?cloud_id=5",
+      :cloud_id=>5,
+      :cloud=>"AWS AP-Tokyo")
+
+    @mock_mci_i = flexmock(MultiCloudImageInternal, 
+      :is_head_version => @mock_mci.is_head_version, 
+      :name=>@mock_mci.name,
+      :rs_id => @mock_mci.rs_id, 
+      :version=>@mock_mci.version, 
+      :href=>@mock_mci.href,
+      :multi_cloud_image_cloud_settings=>[@mock_mci_setting1])
+  end
+
+#  it "finds a MCI for gateway cloud (cloudstack)" do
+#    pending "TODO"
+#  end
+#  it "creates a MCI for gateway cloud (cloudstack)" do
+#    pending "TODO"
+#  end
+#  it "associates an image for a gateway cloud (cloudstack)" do 
+#    pending "TODO"
+#  end
+#  it "replaces an image association for a gateway cloud (cloudstack)" do 
+#    pending "TODO"
+#  end
+#
+  it "finds a MCI for EC2 cloud" do
+    flexmock(MultiCloudImage).should_receive(:find_all).and_return([@mock_mci])
+    flexmock(MultiCloudImageInternal).should_receive(:find).with(10000).and_return(@mock_mci_i)
+  
+    @mci.find_mci(6, @mock_mci_name).rs_id.should == "10000"
+  end
+  it "creates a MCI for EC2 mock cloud" do
+    flexmock(MultiCloudImageInternal).should_receive(:create).with(
+      :name=>@mock_mci_name,
+      :description => "blah"
+    ).and_return(flexmock(MultiCloudImageInternal))
+    @mci.create_mci(6, @mock_mci_name, "blah")
+  end
+
+  it "associates an image to an EC2 cloud" do 
+    flexmock(MultiCloudImage).should_receive(:find_all).and_return([@mock_mci])
+    flexmock(MultiCloudImageInternal).should_receive(:find).with(10000).and_return(@mock_mci_i)
+    flexmock(Tag).should_receive(:set).and_return(flexmock(Tag, :code=>"204"))
+    flexmock(MultiCloudImageCloudSettingInternal).should_receive(:create)
+
+    @mci.add_image_to_mci(
+      :cloud_id=>6,
+      :name=>@mock_mci_name,
+      :image_id=>"ami-b1ahb1ah")
+  end
+
+  it "replaces an image association for an EC2 cloud" do 
+    flexmock(MultiCloudImage).should_receive(:find_all).and_return([@mock_mci])
+    flexmock(MultiCloudImageInternal).should_receive(:find).with(10000).and_return(@mock_mci_i)
+    flexmock(Tag).should_receive(:set).and_return(flexmock(Tag, :code=>"204"))
+    flexmock(MultiCloudImageCloudSettingInternal).should_receive(:destroy).and_return(flexmock("Response",:code=>"200"))
+    flexmock(MultiCloudImageCloudSettingInternal).should_receive(:create)
+
+    @mci.add_image_to_mci(
+      :cloud_id=>5,
+      :name=>@mock_mci_name,
+      :image_id=>"ami-b1ahb1ah")
+  end
+
+  it "sets a default description correctly" do
+    flexmock(MultiCloudImageInternal).should_receive(:create).with(
+      :name=>@mock_mci_name,
+      :description => "Development build: Ubuntu 10.04 with 64-bit architecture (x64) and RightLink 5.7." 
+    ).and_return(@mock_mci)
+
+    @mci.create_mci(5, @mock_mci_name)
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'rspec'
+require 'rightimage_tools'
+
+
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+  config.mock_with :flexmock
+end

metadata

@@ -0,0 +1,180 @@
+--- !ruby/object:Gem::Specification 
+name: rightimage_tools
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Peter Schroeter
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-02-07 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id001
+  prerelease: false
+  name: rest_connection
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id002
+  prerelease: false
+  name: ruby-debug
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 2
+        - 3
+        - 0
+        version: 2.3.0
+  requirement: *id003
+  prerelease: false
+  name: rspec
+  type: :development
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  requirement: *id004
+  prerelease: false
+  name: bundler
+  type: :development
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 1
+        - 6
+        - 4
+        version: 1.6.4
+  requirement: *id005
+  prerelease: false
+  name: jeweler
+  type: :development
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id006
+  prerelease: false
+  name: rcov
+  type: :development
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  requirement: *id007
+  prerelease: false
+  name: flexmock
+  type: :development
+description: A set of tools to support the building of RightImages
+email: peter.schroeter@rightscale.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.rdoc
+files: 
+- .document
+- .rspec
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/mci.rb
+- lib/rightimage_tools.rb
+- spec/mci_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/rightscale/rightimage_tools
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: A set of tools to support the building of RightImages
+test_files: []
+