knife-azure 1.0.0

data/lib/azure/host.rb

@@ -0,0 +1,90 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Azure
+  class Hosts
+    def initialize(connection)
+      @connection=connection
+    end
+    def all
+      hosted_services = Array.new
+      responseXML = @connection.query_azure('hostedservices')
+      servicesXML = responseXML.css('HostedServices HostedService')
+      servicesXML.each do |serviceXML|
+        host = Host.new(@connection)
+        hosted_services << host.parse(serviceXML)
+      end
+      hosted_services
+    end
+    def exists(name)
+      hostExists = false
+      self.all.each do |host|
+        next unless host.name == name
+        hostExists = true
+      end
+      hostExists
+    end
+    def create(params)
+      host = Host.new(@connection)
+      host.create(params)
+    end
+    def delete(name)
+      if self.exists name
+          servicecall = "hostedservices/" + name
+        @connection.query_azure(servicecall, "delete") 
+      end
+    end
+  end
+end
+
+class Azure
+  class Host
+    include AzureUtility
+    attr_accessor :connection, :name, :url, :label
+    attr_accessor :dateCreated, :description, :location
+    attr_accessor :dateModified, :status
+    def initialize(connection)
+      @connection = connection
+    end
+    def parse(serviceXML)
+      @name = xml_content(serviceXML, 'ServiceName')
+      @url = xml_content(serviceXML, 'Url')
+      @label = xml_content(serviceXML, 'HostedServiceProperties Label')
+      @dateCreated = xml_content(serviceXML, 'HostedServiceProperties DateCreated')
+      @description = xml_content(serviceXML, 'HostedServiceProperties Description')
+      @location = xml_content(serviceXML, 'HostedServiceProperties Location')
+      @dateModified = xml_content(serviceXML, 'HostedServiceProperties DateLastModified')
+      @status = xml_content(serviceXML, 'HostedServiceProperties Status')
+      self
+    end
+    def create(params)
+      builder = Nokogiri::XML::Builder.new do |xml|
+        xml.CreateHostedService('xmlns'=>'http://schemas.microsoft.com/windowsazure') {
+          xml.ServiceName params[:hosted_service_name]
+          xml.Label Base64.encode64(params[:hosted_service_name])
+          xml.Description params['hosted_service_description'] || 'Explicitly created hosted service'
+          xml.Location params['hosted_service_location'] || 'Windows Azure Preview'
+        }
+      end
+      @connection.query_azure("hostedservices", "post", builder.to_xml)
+    end
+    def details
+      response = @connection.query_azure('hostedservices/' + @name + '?embed-detail=true')
+    end
+  end
+end

data/lib/azure/image.rb

@@ -0,0 +1,58 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Azure
+  class Images
+    def initialize(connection)
+      @connection=connection
+    end
+    def all
+      images = Array.new
+      response = @connection.query_azure('images')
+      osimages = response.css('OSImage')
+      osimages.each do |image|
+        item = Image.new(image)
+        images << item
+      end
+      images
+    end
+    def exists(name)
+      imageExists = false
+      self.all.each do |host|
+        next unless host.name == name
+        imageExists = true
+      end
+      imageExists
+    end
+  end
+end
+
+class Azure
+  class Image
+    attr_accessor :category, :label
+    attr_accessor :name, :os, :eula, :description
+    def initialize(image)
+      @category = image.at_css('Category').content
+      @label = image.at_css('Label').content
+      @name = image.at_css('Name').content
+      @os = image.at_css('OS').content
+      @eula = image.at_css('Eula').content
+      @description = image.at_css('Description').content
+    end
+  end
+end

data/lib/azure/rest.rb

@@ -0,0 +1,97 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 "net/https"
+require "uri"
+require "nokogiri"
+
+module AzureAPI
+
+  class Rest
+    def initialize(params)
+      @subscription_id = params[:azure_subscription_id]
+      @pem_file = File.read find_pem(params[:azure_pem_file])
+      @host_name = params[:azure_host_name]
+    end
+    def find_pem(name)
+      config_dir = Chef::Knife.chef_config_dir
+      if File.exist? name
+        pem_file = name
+      elsif config_dir && File.exist?(File.join(config_dir, name))
+        pem_file = File.join(config_dir, name)
+      elsif File.exist?(File.join(ENV['HOME'], '.chef', name))
+        pem_file = File.join(ENV['HOME'], '.chef', name)
+      else
+        raise 'Unable to find certificate pem file - ' + name
+      end
+      pem_file
+    end
+    def query_azure(service_name, verb = 'get', body = '')
+      request_url = "https://#{@host_name}/#{@subscription_id}/services/#{service_name}"
+      print '.'
+      uri = URI.parse(request_url)
+      http = http_setup(uri)
+      request = request_setup(uri, verb, body)
+      response = http.request(request)
+      @last_request_id = response['x-ms-request-id']
+      response
+    end
+    def query_for_completion()
+      request_url = "https://#{@host_name}/#{@subscription_id}/operations/#{@last_request_id}"
+      uri = URI.parse(request_url)
+      http = http_setup(uri)
+      request = request_setup(uri, 'get', '')
+      response = http.request(request)
+    end
+    def http_setup(uri)
+      http = Net::HTTP.new(uri.host, uri.port)
+      store = OpenSSL::X509::Store.new
+      store.add_cert(OpenSSL::X509::Certificate.new(File.read(find_pem("cacert.pem"))))
+      http.cert_store = store
+      http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+      http.use_ssl = true
+      http.cert = OpenSSL::X509::Certificate.new(@pem_file)
+      http.key = OpenSSL::PKey::RSA.new(@pem_file)
+      http
+    end
+    def request_setup(uri, verb, body)
+      if verb == 'get'
+        request = Net::HTTP::Get.new(uri.request_uri)
+      elsif verb == 'post'
+        request = Net::HTTP::Post.new(uri.request_uri)
+      elsif verb == 'delete'
+        request = Net::HTTP::Delete.new(uri.request_uri)
+      end
+      request["x-ms-version"] = "2012-03-01"
+      request["content-type"] = "application/xml"
+      request["accept"] = "application/xml"
+      request["accept-charset"] = "utf-8"
+      request.body = body
+      request
+    end
+    def showResponse(response)
+      puts "=== response body ==="
+      puts response.body
+      puts "=== response.code ==="
+      puts response.code
+      puts "=== response.inspect ==="
+      puts response.inspect
+      puts "=== all of the headers ===" 
+      puts response.each_header { |h, j| puts h.inspect + ' : ' + j.inspect}
+    end
+  end
+end

data/lib/azure/role.rb

@@ -0,0 +1,182 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 Azure
+  class Roles
+    attr_accessor :connection, :roles
+    def initialize(connection)
+      @connection = connection
+      @roles = nil
+    end
+    def all
+      @roles = Array.new
+      @connection.deploys.all.each do |deploy|
+        deploy.roles.each do |role|
+          @roles << role
+        end
+      end
+      @roles
+    end
+    def find(name)
+      if @roles == nil
+        all
+      end
+      @roles.each do |role|
+        if(role.name == name)
+          return role 
+        end
+      end
+      nil   
+    end
+    def alone_on_host(name)
+      found_role = find(name)
+      @roles.each do |role|
+        if (role.name != found_role.name && 
+            role.deployname == found_role.deployname && 
+            role.hostedservicename == found_role.hostedservicename)
+          return false;
+        end
+      end
+      true
+    end
+    def exists(name)
+      find(name) != nil
+    end
+    def delete(name)
+      role = find(name)
+      if role != nil
+        if alone_on_host(name)
+          servicecall = "hostedservices/#{role.hostedservicename}/deployments" +
+          "/#{role.deployname}"
+        else
+          servicecall = "hostedservices/#{role.hostedservicename}/deployments" +
+          "/#{role.deployname}/roles/#{role.name}"
+        end
+        @connection.query_azure(servicecall, "delete") 
+      end
+      #@connection.disks.clear_unattached
+    end
+  end
+  class Role
+    include AzureUtility
+    attr_accessor :connection, :name, :status, :size, :ipaddress
+    attr_accessor :sshport, :sshipaddress, :hostedservicename, :deployname
+    attr_accessor :hostname, :tcpports, :udpports
+    def initialize(connection)
+      @connection = connection
+    end
+    def parse(roleXML, hostedservicename, deployname)
+      @name = xml_content(roleXML, 'RoleName')
+      @status = xml_content(roleXML, 'InstanceStatus')
+      @size = xml_content(roleXML, 'InstanceSize')
+      @ipaddress = xml_content(roleXML, 'IpAddress')
+      @hostname = xml_content(roleXML, 'HostName')
+      @hostedservicename = hostedservicename
+      @deployname = deployname
+      @tcpports = Array.new
+      @udpports = Array.new
+      endpoints = roleXML.css('InstanceEndpoint')
+      endpoints.each do |endpoint|
+        if xml_content(endpoint, 'Name').downcase == 'ssh'
+          @sshport = xml_content(endpoint, 'PublicPort')
+          @sshipaddress = xml_content(endpoint, 'Vip')
+        else
+          hash = Hash.new
+          hash['Name'] = xml_content(endpoint, 'Name')
+          hash['Vip'] = xml_content(endpoint, 'Vip')
+          hash['PublicPort'] = xml_content(endpoint, 'PublicPort')
+          hash['LocalPort'] = xml_content(endpoint, 'LocalPort')
+          if xml_content(endpoint, 'Protocol') == 'tcp'
+            @tcpports << hash
+          else # == 'udp'
+            @udpports << hash
+          end
+        end
+      end
+    end
+    def setup(params)
+      builder = Nokogiri::XML::Builder.new do |xml|
+        xml.PersistentVMRole(
+          'xmlns'=>'http://schemas.microsoft.com/windowsazure',
+          'xmlns:i'=>'http://www.w3.org/2001/XMLSchema-instance'
+        ) {
+          xml.RoleName {xml.text params[:role_name]}
+          xml.OsVersion('i:nil' => 'true')
+          xml.RoleType 'PersistentVMRole'
+          xml.ConfigurationSets {
+            xml.ConfigurationSet('i:type' => 'LinuxProvisioningConfigurationSet') {
+            xml.ConfigurationSetType 'LinuxProvisioningConfiguration'
+            xml.HostName params[:host_name] 
+            xml.UserName params[:ssh_user]
+            xml.UserPassword params[:ssh_password]
+            xml.DisableSshPasswordAuthentication 'false'
+          }
+          xml.ConfigurationSet('i:type' => 'NetworkConfigurationSet') {
+            xml.ConfigurationSetType 'NetworkConfiguration'
+            xml.InputEndpoints {
+              xml.InputEndpoint {
+              xml.LocalPort '22' 
+              xml.Name 'SSH'
+              xml.Protocol 'TCP'
+            }
+            if params[:tcp_endpoints]
+              params[:tcp_endpoints].split(',').each do |endpoint|
+                ports = endpoint.split(':')
+                xml.InputEndpoint {
+                  xml.LocalPort ports[0]
+                  xml.Name 'tcpport_' + ports[0] + '_' + params[:host_name]
+                  if ports.length > 1
+                    xml.Port ports[1]
+                  end 
+                  xml.Protocol 'TCP'
+                }
+              end
+            end
+            if params[:udp_endpoints]
+              params[:udp_endpoints].split(',').each do |endpoint|
+                ports = endpoint.split(':')
+                xml.InputEndpoint {
+                  xml.LocalPort ports[0]
+                  xml.Name 'udpport_' + ports[0] + '_' + params[:host_name]
+                  if ports.length > 1
+                    xml.Port ports[1]
+                  end 
+                  xml.Protocol 'UDP'
+                }
+              end
+            end
+            }
+          }
+          }
+          xml.Label Base64.encode64(params[:role_name]).strip
+          xml.OSVirtualHardDisk {
+            xml.MediaLink 'http://' + params[:media_location_prefix] + 'imagestore.blob.core.azure-preview.com/os-disks/' + (params[:os_disk_name] || Time.now.strftime('disk_%Y_%m_%d_%H_%M'))
+            xml.SourceImageName params[:source_image]
+          }
+          xml.RoleSize params[:role_size]
+        }
+      end 
+      builder.doc
+    end
+    def create(params, roleXML)
+      servicecall = "hostedservices/#{params[:hosted_service_name]}/deployments" +
+      "/#{params['deploy_name']}/roles"
+      @connection.query_azure(servicecall, "post", roleXML.to_xml) 
+    end
+  end
+end

data/lib/azure/utility.rb

@@ -0,0 +1,29 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+module AzureUtility
+  def xml_content(xml, key, default='')
+    content = default
+    node = xml.at_css(key)
+    if node
+      content = node.content
+    end
+    content
+  end
+end
+

data/lib/chef/knife/azure_base.rb

@@ -0,0 +1,102 @@
+
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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'
+require File.expand_path('../../../azure/connection', __FILE__)
+
+class Chef
+  class Knife
+    module AzureBase
+
+      # :nodoc:
+      # Would prefer to do this in a rational way, but can't be done b/c of
+      # Mixlib::CLI's design :(
+      def self.included(includer)
+        includer.class_eval do
+
+          deps do
+            require 'readline'
+            require 'chef/json_compat'
+          end
+
+          option :azure_subscription_id,
+            :short => "-S ID",
+            :long => "--azure-subscription-id ID",
+            :description => "Your Azure subscription ID",
+            :proc => Proc.new { |key| Chef::Config[:knife][:azure_subscription_id] = key }
+
+          option :azure_pem_file,
+            :short => "-p FILENAME",
+            :long => "--azure-pem-filename FILENAME",
+            :description => "Your Azure PEM file name",
+            :proc => Proc.new { |key| Chef::Config[:knife][:azure_pem_file] = key }
+
+          option :azure_host_name,
+            :short => "-H HOSTNAME",
+            :long => "--azure_host_name HOSTNAME",
+            :description => "Your Azure host name",
+            :proc => Proc.new { |key| Chef::Config[:knife][:azure_host_name] = key }
+
+        end
+      end
+
+      def connection
+        @connection ||= begin
+                          connection = Azure::Connection.new(
+                            :azure_subscription_id => 
+                            config[:azure_subscription_id] || Chef::Config[:knife][:azure_subscription_id],
+                              :azure_pem_file => 
+                            config[:azure_pem_file] || Chef::Config[:knife][:azure_pem_file],
+                              :azure_host_name => 
+                            config[:azure_host_name] || Chef::Config[:knife][:azure_host_name]
+                          )
+                        end        
+      end
+
+      def locate_config_value(key)
+        key = key.to_sym
+        config[key] || Chef::Config[:knife][key]
+      end
+
+      def msg_pair(label, value, color=:cyan)
+        if value && !value.to_s.empty?
+          puts "#{ui.color(label, color)}: #{value}"
+        end
+      end
+
+      def validate!(keys=[:azure_subscription_id, :azure_pem_file, :azure_host_name])
+        errors = []
+
+        keys.each do |k|
+          pretty_key = k.to_s.gsub(/_/, ' ').gsub(/\w+/){ |w| (w =~ /(ssh)|(aws)/i) ? w.upcase  : w.capitalize }
+          if Chef::Config[:knife][k].nil?
+            errors << "You did not provide a valid '#{pretty_key}' value."
+          end
+        end
+
+        if errors.each{|e| ui.error(e)}.any?
+          exit 1
+        end
+      end
+
+    end
+  end
+end
+
+

data/lib/chef/knife/azure_image_list.rb

@@ -0,0 +1,58 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 File.expand_path('../azure_base', __FILE__)
+
+class Chef
+  class Knife
+    class AzureImageList < Knife
+
+      include Knife::AzureBase
+
+      banner "knife azure image list (options)"
+
+      def run
+        $stdout.sync = true
+
+        validate!
+
+        image_list = [
+          ui.color('Name', :bold),
+          #ui.color('Category', :bold),
+          #ui.color('Label', :bold),
+          #ui.color('OS', :bold),
+          #ui.color('Eula', :bold),
+        ]
+        items = connection.images.all
+        items.each do |image|
+          if image.os == 'Linux'
+            image_list << image.name.to_s
+            #image_list << image.category.to_s
+            #image_list << image.label.to_s
+            #image_list << image.os.to_s
+            #image_list << image.eula.to_s
+          end
+        end
+        puts ''
+        puts ui.list(image_list, :columns_across, 1)
+      end
+    end
+  end
+end