ubuntu_ami 0.2 → 0.4.0

data/README.md

@@ -0,0 +1,53 @@
+This gem provides a library for retrieving Canonical's Ubuntu AMI
+release list for Amazon EC2.
+
+It also includes a knife plugin to easily retrieve AMIs for launching
+instances with Chef.
+
+# Usage
+
+Find an AMI based on criteria:
+
+    ami = Ubuntu.release("lucid").amis.find do |ami|
+      ami.arch == "amd64" and
+      ami.root_store == "instance-store" and
+      ami.region == "us-east-1"
+    end
+
+Return certain information about the AMIs:
+
+    Ubuntu.release("lucid").amis.each do |ami|
+      puts "#{ami.region} #{ami.name} (#{ami.arch}, #{ami.root_store})"
+    end
+
+Use the knife plugin with Knife, from Chef. Pass just a distro name
+for the list of available AMIs by type:
+
+    knife ec2 amis ubuntu lucid
+
+The first column will be the available types, and the second is the
+associated AMI. Specify the type to return just the AMI.
+
+    knife ec2 amis ubuntu lucid us_east_small
+
+The type is made up of the region, the architecture size (small for 32
+bit, large for 64 bit). If the AMI is an EBS root store, that will be indicated.
+
+# License and Authors
+
+Author:: Joshua Timberman (<joshua@housepub.org>)
+Author:: Michael Hale (<mike@hales.ws>)
+
+Copyright:: 2011, Joshua Timberman
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

data/lib/chef/knife/ec2_amis_ubuntu.rb

@@ -0,0 +1,66 @@
+#
+# Author:: Joshua Timberman (<joshua@housepub.org>)
+# Description:: Retrieves AMI information from Canonical's AMI release list.
+#
+# Copyright:: 2011, Joshua Timberman
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'chef/knife'
+
+module KnifePlugins
+  class Ec2AmisUbuntu < Chef::Knife
+
+    deps do
+      begin
+        require 'ubuntu_ami'
+      rescue LoadError
+        Chef::Log.error("Could not load Ubuntu AMI library.")
+      end
+    end
+
+    banner "knife ec2 amis ubuntu DISTRO [TYPE] (options)"
+
+    def region_fix(region)
+      region.gsub(/-1/,'').gsub(/-/,'_')
+    end
+
+    def disk_store(store)
+      store =~ /ebs/ ? "_ebs" : ''
+    end
+
+    def size_of(arch)
+      String(arch) =~ /(amd64|x86_64|large|^64)/ ? "large" : "small"
+    end
+
+    def build_type(region, arch, root_store)
+      "#{region_fix(region)}_#{size_of(arch)}#{disk_store(root_store)}"
+    end
+
+    def list_amis(distro)
+      amis = Hash.new
+      Ubuntu.release(distro).amis.each do |ami|
+        amis[build_type(ami.region, ami.arch, ami.root_store)] = ami.name
+      end
+      amis
+    end
+
+    def run
+      distro = name_args[0]
+      type = name_args[1]
+      ami_list = list_amis(distro)[type] || list_amis(distro)
+      output(format_for_display(ami_list))
+    end
+
+  end
+end

data/lib/ubuntu_ami.rb

@@ -1,5 +1,5 @@
 #
-# Author:: Joshua Timberman (<joshua@opscode.com>)
+# Author:: Joshua Timberman (<joshua@housepub.org>), Michael Hale (<mike@hales.ws>)
 # Description:: Retrieves AMI information from Canonical's AMI release list.
 #
 # Copyright:: 2011, Joshua Timberman
@@ -16,33 +16,56 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-require 'pp'
 require 'open-uri'
 
-class UbuntuAmi
-  def initialize(release)
-    @uri = "http://uec-images.ubuntu.com/query/#{release}/server/released.current.txt"
+class Ubuntu
+  attr_reader :release_name
+
+  def self.release(release_name)
+    instance_for_release(release_name)
+  end
+
+  def self.instance_for_release(release_name)
+    @releases ||= {}
+    @releases[release_name] ||= new(release_name)
+  end
+
+  def initialize(release_name)
+    @release_name = release_name
   end
 
-  def arch_size(arch)
-    arch =~ /amd64/ ? "large" : "small"
+  def amis
+    content.map do |line|
+      Ami.new(
+              line.split[7],
+              line.split[4],
+              line.split[5],
+              line.split[6],
+              line.split[8..9].last)
+    end
   end
 
-  def disk_store(store)
-    "_ebs" if store =~ /ebs/
+  def url
+    "http://uec-images.ubuntu.com/query/#{release_name}/server/released.current.txt"
   end
 
-  def region_fix(region)
-    region.gsub(/-1/,'').gsub(/-/,'_')
+  def content
+    begin
+      @content ||= open(url).read.split("\n")
+    rescue
+      raise "Could not find AMI list for distro release '#{release_name}', did you specify it correctly? (e.g., 'lucid')"
+    end
   end
 
-  def run
-    amis = {}
-    open(@uri).each do |a|
-      key = a.split[4..6]
-      ami = a.split[7]
-      amis["#{region_fix(key[2])}_#{arch_size(key[1])}#{disk_store(key[0])}"] = ami
+  class Ami
+    attr_reader :name, :root_store, :arch, :region, :virtualization_type
+
+    def initialize(name, root_store, arch, region, virtualization_type)
+      @name = name
+      @root_store = root_store
+      @arch = arch
+      @region = region
+      @virtualization_type = virtualization_type
     end
-    amis
   end
 end

data/lib/ubuntu_ami/version.rb

@@ -0,0 +1,22 @@
+# Author:: Joshua Timberman (<joshua@housepub.org>)
+# Description:: Retrieves AMI information from Canonical's AMI release list.
+#
+# Copyright:: 2011, Joshua Timberman
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+class Ubuntu
+  class Ami
+    VERSION = '0.4.0'
+  end
+end

metadata

@@ -1,59 +1,49 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: ubuntu_ami
-version: !ruby/object:Gem::Version 
+version: !ruby/object:Gem::Version
+  version: 0.4.0
   prerelease: 
-  version: "0.2"
 platform: ruby
-authors: 
+authors:
 - Joshua Timberman
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2011-04-23 00:00:00 -06:00
-default_executable: 
+date: 2011-11-11 00:00:00.000000000Z
 dependencies: []
-
 description: Retrieves AMI information from Canonical's Ubuntu release list.
 email: joshua@opscode.com
 executables: []
-
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - LICENSE
 - README.md
-- lib/chef/knife/ubuntu_amis.rb
+- lib/chef/knife/ec2_amis_ubuntu.rb
+- lib/ubuntu_ami/version.rb
 - lib/ubuntu_ami.rb
-has_rdoc: true
 homepage: http://github.com/jtimberman/ubuntu_ami
 licenses: []
-
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      version: "0"
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
 requirements: []
-
 rubyforge_project: 
-rubygems_version: 1.5.2
+rubygems_version: 1.8.10
 signing_key: 
 specification_version: 3
 summary: Retrieves AMI information from Canonical's Ubuntu release list.
 test_files: []
-

data/lib/chef/knife/ubuntu_amis.rb

@@ -1,24 +0,0 @@
-require 'chef/knife'
-
-module KnifePlugins
-  class Ec2AmisUbuntu < Chef::Knife
-
-    deps do
-      begin
-        require 'ubuntu_ami'
-      rescue LoadError
-        Chef::Log.error("Could not load Ubuntu AMI library.")
-      end
-    end
-
-    banner "knife ec2 amis ubuntu DISTRO [TYPE]"
-
-    def run
-      distro = name_args[0]
-      type = name_args[1]
-      ami_list = UbuntuAmi.new(distro).run[type] || UbuntuAmi.new(distro).run
-      output(format_for_display(ami_list))
-    end
-
-  end
-end