mccloud 0.0.13 → 0.0.14

data/lib/mccloud/provider/libvirt/provider/status.rb

@@ -0,0 +1,43 @@
+module Mccloud::Provider
+  module Libvirt
+    module ProviderCommand
+
+      def status(selection=nil,options=nil)
+
+       env.ui.info "" 
+        env.ui.info "Server(s) - provider #{@name}"
+
+        printf "%-10s %-12s %-20s %-15s %-8s\n", "Name", "Instance Id", "IP", "Type","Status"
+        80.times { |i| printf "=" } ; env.ui.info ""
+
+        raw.servers.each do |vm|
+          if vm.name.start_with?(self.filter)
+            unless self.filter==""
+              vm.name[self.filter]=""
+              printf "%-10s %-12s %-20s %-20s %-15s %-8s\n",vm.name,vm.mac, vm.public_ip_address, vm.cpus,vm.memory_size,vm.state
+            else
+              env.ui.info "Name: #{vm.name}"
+              env.ui.info "Instance Id: #{vm.uuid}"
+              env.ui.info "Public Ip: #{vm.public_ip_address}"
+              env.ui.info "Cpus: #{vm.cpus}"
+              env.ui.info "State: #{vm.state}"
+              80.times { |i| printf "=" } ; env.ui.info ""
+            end
+          end
+        end
+
+
+        env.ui.info "Volume(s) - provider #{@name}"
+        80.times { |i| printf "=" } ; env.ui.info ""
+        volume_list=raw.volumes.all() 
+        volume_list.each do |volume|
+          printf "%-20s: %-10s %-10s \n",volume.name,volume.id,volume.key
+          printf "%-20s: %-10s %-20s %-10s\n","",volume.pool_name, volume.path, volume.format_type
+          printf "%-20s: %-10s %-10s\n","",volume.capacity,volume.allocation
+        end
+
+      end
+
+    end #module  
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/provider.rb

@@ -0,0 +1,96 @@
+require 'mccloud/provider/libvirt/provider/status'
+require 'mccloud/provider/libvirt/vm'
+require 'mccloud/provider/fog/provider'
+
+module Mccloud
+  module Provider
+    module Libvirt
+      class Provider  < ::Mccloud::Provider::Fog::Provider
+
+        attr_accessor :name
+        attr_accessor :flavor
+
+        attr_accessor :options
+
+        attr_accessor :vms
+
+        include Mccloud::Provider::Libvirt::ProviderCommand
+
+
+        def initialize(name,options,env)
+
+          super
+
+          @vms=Hash.new
+
+          @options=options
+          @flavor=self.class.to_s.split("::")[-2]
+          @name=name
+
+          required_gems=%w{ruby-libvirt fog}
+          check_gem_availability(required_gems)
+          require 'libvirt'
+          require 'fog'
+        end
+
+
+        def raw
+          if @raw.nil?
+            begin
+              @raw=::Fog::Compute.new({:provider => "Libvirt"}.merge(@options))
+            rescue ArgumentError => e
+              env.ui.error "Error loading raw provider : #{e.to_s} #{$!}"
+              @raw=nil
+            end
+          end
+          return @raw
+        end
+
+        def up(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm.up(options)
+          end
+        end
+
+        def bootstrap(selection,script,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm._bootstrap(script,options)
+          end
+        end
+
+        def destroy(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm.destroy(options)
+          end
+        end
+
+        def ssh(selection,command,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm.ssh(command,options)
+          end
+        end
+
+
+
+        def provision(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm._provision(options)
+          end
+        end
+
+        def halt(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm.halt(options)
+          end
+        end
+
+        def reload(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            vm.reload(options)
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/mccloud/provider/libvirt/vm/bootstrap.rb

@@ -0,0 +1,17 @@
+require 'mccloud/util/platform'
+
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+      def _bootstrap(command,options=nil)
+        if raw.ready?
+          env.ui.info "[#{@name}] - Waiting for an ip-address "
+          raw.wait_for {   printf "."; STDOUT.flush; !public_ip_address.nil? }
+        end
+        ssh_bootstrap(command,options)
+      end
+
+    end #module
+  end #Module
+end #module

data/lib/mccloud/provider/libvirt/vm/destroy.rb

@@ -0,0 +1,33 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+      def destroy(options)
+        fullname="#{@provider.filter}#{@name}"
+        server=@provider.raw.servers.all(:name => fullname)
+        volname="#{fullname}.img"
+        #TODO use the creation options
+        volume=@provider.raw.volumes.all(:name => volname)
+
+        unless server.nil?
+          env.ui.info "[#{@name}] - Destroying machine #{@provider.namespace}::#{@name}"
+
+          server.first.destroy
+        else
+          env.ui.info "[#{@name}] - Server #{@provider.namespace}::#{@name} does not exist"
+        end
+
+        unless volume.nil?
+          env.ui.info "[#{@name}] - Destroying volume #{@provider.namespace}::#{@name}.img"
+
+          volume.first.destroy
+        else
+          env.ui.info "[#{@name}] - Volume #{@provider.namespace}::#{@name}.img does not exist"
+
+        end
+
+      end
+
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm/forward.rb

@@ -0,0 +1,12 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+        def forward(command,options={})
+          @forward_threads=Array.new
+          return self.ssh_forward(options)
+        end
+
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm/halt.rb

@@ -0,0 +1,20 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+      def halt(options)
+
+        if self.running?
+          env.ui.info "Halting machine #{@name}(#{@raw.id})"
+          raw.shutdown
+          raw.wait_for { printf "."; STDOUT.flush; state=="stopped"||state=="crashed"}
+          env.ui.info ""
+        else
+          env.ui.info "#{@name}(#{raw.id}) is already halted."
+        end
+
+      end
+
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm/provision.rb

@@ -0,0 +1,25 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+      def _provision(options)
+
+        unless raw.nil?
+
+          if raw.ready?
+            @provisioners.each do |provisioner|
+              env.ui.info "[#{@name}] - starting provisioning with #{provisioner.name} as provisioner"
+              provisioner.run(self)
+            end
+          else
+            env.ui.info "[#{@name}] - machine is not running, skipping provisioning"
+          end
+        else
+          env.ui.info "[#{@name}] - machine doesn't exit yet"
+        end
+
+      end
+
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm/reload.rb

@@ -0,0 +1,13 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+      def reload(options)
+         env.logger.info "Checking if #{@name} is running: #{self.running?}"
+        if self.running?
+          self.halt(options)
+        end
+        self.up(options)
+      end
+    end
+  end
+end

data/lib/mccloud/provider/libvirt/vm/rsync.rb

@@ -0,0 +1,11 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+        def rsync(path,options)
+          super(path,options)
+        end
+ 
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm/scp.rb

@@ -0,0 +1,15 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+      def transfer(src,dest,options = {})
+        scp(src,dest,options)
+      end
+
+       def scp(src,dest, options = {})
+         raw.scp(src,dest,options)
+      end
+
+       end #module
+      end #module
+    end #module

data/lib/mccloud/provider/libvirt/vm/ssh.rb

@@ -0,0 +1,11 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+        def ssh(command,options={})
+          super(command,options)
+        end
+
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm/up.rb

@@ -0,0 +1,65 @@
+module Mccloud::Provider
+  module Libvirt
+    module VmCommand
+
+      def up(options)
+
+        env.ui.info "Upping of libvirt vm #{@name}"
+
+        # There is no existing machine yet, so we create it
+        if raw.nil?
+          enhanced_create_options=@create_options
+          enhanced_create_options[:name]="#{@provider.filter}#{@name}"
+          env.ui.info "[#{@name}] - Creating machine #{@provider.namespace}::#{@name}"
+          @raw=@provider.raw.servers.create(enhanced_create_options)
+          raw.start
+
+          env.ui.info "[#{@name}] - Waiting for the machine to become accessible"
+          raw.wait_for { printf "."; STDOUT.flush;  ready?}
+
+          # Wait for ssh to become available ...
+          env.ui.info "[#{@name}] - Waiting for ip address"
+          #env.ui.info instance.console_output.body["output"]
+          raw.wait_for { printf "."; STDOUT.flush;  !public_ip_address.nil?}
+
+          env.ui.info "[#{@name}] - Waiting for ssh on #{self.ip_address} to become available"
+          Mccloud::Util::Ssh.execute_when_tcp_available(self.ip_address, { :port => @port, :timeout => 6000 }) do
+            env.ui.info "[#{@name}] - Ssh is available , proceeding with bootstrap"
+          end
+
+          # Because it's a new machine we bootstrap it to
+          self._bootstrap(nil,options)
+
+        else
+          state=raw.state
+          if state !="running"
+            env.ui.info "Booting up machine #{@name}"
+            raw.start
+            raw.wait_for { printf ".";STDOUT.flush;  ready?}
+            env.ui.info ""
+          else
+            unless raw.state == "shutting-down"
+              env.ui.info "[#{@name}] - already running."
+            else
+              env.ui.info "[#{@name}] - can't start machine because it is in state #{raw.state}"
+              return
+            end
+          end
+
+        end
+
+        unless options["noprovision"]
+          env.ui.info "[#{@name}] - Waiting for ssh to become available"
+          Mccloud::Util::Ssh.execute_when_tcp_available(self.ip_address, { :port => @port, :timeout => 6000 }) do
+            env.ui.info "[#{@name}] - Ssh is available , proceeding with provisioning"
+          end
+
+          env.ui.info "[#{@name}] - provision step #{@name}"
+          self._provision(options)
+        end
+
+      end
+
+    end #module
+  end #module
+end #module

data/lib/mccloud/provider/libvirt/vm.rb

@@ -0,0 +1,67 @@
+require 'mccloud/provider/core/vm'
+
+require 'mccloud/provider/libvirt/vm/up'
+require 'mccloud/provider/libvirt/vm/bootstrap'
+require 'mccloud/provider/libvirt/vm/ssh'
+require 'mccloud/provider/libvirt/vm/scp'
+require 'mccloud/provider/libvirt/vm/rsync'
+require 'mccloud/provider/libvirt/vm/halt'
+require 'mccloud/provider/libvirt/vm/provision'
+require 'mccloud/provider/libvirt/vm/destroy'
+require 'mccloud/provider/libvirt/vm/reload'
+require 'mccloud/provider/libvirt/vm/forward'
+
+
+module Mccloud::Provider
+  module Libvirt
+
+    class Vm < ::Mccloud::Provider::Core::Vm
+
+      include Mccloud::Provider::Libvirt::VmCommand
+
+      def ip_address
+        return self.public_ip_address
+      end
+
+      def public_ip_address
+        unless raw.nil?
+          ip=raw.public_ip_address
+        else
+          ip=nil
+        end
+        return ip
+      end
+
+      def running?
+        if raw.nil?
+          return false
+        else
+          return raw.ready?
+        end
+      end
+
+      def private_ip_address
+        unless raw.nil?
+          ip=raw.private_ip_address
+        else
+          ip=nil
+        end
+        return ip
+      end
+
+      def raw
+        if @raw.nil?
+          found=@provider.raw.servers.all(:name => "#{@provider.filter}#{@name}")
+          unless found.nil?
+            @raw=found.first
+          end
+        else
+          @raw.private_key_path=@private_key_path
+          @raw.username = @user
+        end
+
+        return @raw
+      end
+    end
+  end
+end

data/lib/mccloud/provider/script/provider/status.rb

@@ -0,0 +1,12 @@
+module Mccloud::Provider
+  module Script
+    module ProviderCommand
+
+    def status(selection=nil,options=nil)
+
+    end
+
+    end #module
+  end #module
+end #module
+

data/lib/mccloud/provider/script/provider.rb

@@ -0,0 +1,125 @@
+require 'mccloud/provider/script/vm'
+require 'mccloud/util/platform'
+require 'mccloud/provider/core/provider'
+
+module Mccloud
+  module Provider
+    module Script
+      class Provider  < ::Mccloud::Provider::Core::Provider
+
+        attr_accessor :name
+        attr_accessor :flavor
+
+        attr_accessor :options
+
+        attr_accessor :vms
+
+        attr_accessor :script_dir
+        attr_accessor :variables
+
+        #include Mccloud::Provider::Script::ProviderCommand
+
+
+        def initialize(name,options,env)
+
+          super(name,options,env)
+
+          @vms=Hash.new
+          @variables=Hash.new
+
+          @options=options
+          @flavor=self.class.to_s.split("::")[-2]
+          @name=name
+        end
+
+        def up(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            script_exec("up.sh",vm,options)
+          end
+        end
+
+        def status(selection,options)
+          unless selection.nil?
+            on_selected_components("vm",selection) do |id,vm|
+              script_exec("status.sh",vm,options)
+            end
+          else
+            script_exec("status.sh",nil,options)
+          end
+        end
+
+        def bootstrap(selection,script,options)
+          on_selected_components("vm",selection) do |id,vm|
+            script_exec("bootstrap.sh",vm,options)
+          end
+
+        end
+
+        def destroy(selection,options)
+
+          on_selected_components("vm",selection) do |id,vm|
+            script_exec("destroy.sh",vm,options)
+          end
+
+        end
+
+        def ssh(selection,command,options)
+
+          on_selected_components("vm",selection) do |id,vm|
+            script_exec("ssh.sh",vm,options)
+          end
+
+        end
+
+        def provision(selection,options)
+
+          on_selected_components("vm",selection) do |id,vm|
+            script_exec("provision.sh",vm,options)
+          end
+
+        end
+
+        def halt(selection,options)
+          on_selected_components("vm",selection) do |id,vm|
+            script_exec("halt.sh",vm,options)
+          end
+
+        end
+
+        private
+        def script_exec(filename,vm,options)
+
+          # Set provider variables
+          unless vm.nil?
+            vm.variables.each do |name,value|
+              env.logger.info "setting environment: #{name}=#{value}"
+              ENV["MC_"+name.to_s]=value
+            end
+            ENV["MC_BOXNAME"]=vm.name
+          end
+
+          # Set vm variables
+          env.logger.info "script_dir : #{File.expand_path(@script_dir)}"
+          ENV["MC_SCRIPT_DIR"]=File.expand_path(@script_dir)
+          @variables.each do |name,value|
+            env.logger.info "setting environment: #{name}=#{value}"
+            ENV["MC_"+name.to_s]=value
+          end
+
+          # Some hackery going on here. On Mac OS X Leopard (1.5), exec fails
+          # (GH-51). As a workaround, we fork and wait. On all other platforms,
+          # we simply exec.
+          pid = nil
+          pid = fork if ::Mccloud::Util::Platform.leopard? || ::Mccloud::Util::Platform.tiger?
+
+          env.ui.info "Executing internal ssh command"
+          script_command="#{File.join(self.script_dir,filename)}"
+          env.ui.info script_command
+          Kernel.exec "sh #{script_command}" if pid.nil?
+          Process.wait(pid) if pid
+        end
+
+      end
+    end
+  end
+end

data/lib/mccloud/provider/script/vm.rb

@@ -0,0 +1,13 @@
+require 'mccloud/provider/core/vm'
+
+module Mccloud::Provider
+  module Script
+
+    class Vm < ::Mccloud::Provider::Core::Vm
+
+      attr_accessor :ip_address
+      attr_accessor :variables
+
+    end
+  end
+end

data/lib/mccloud/provider/vagrant/provider/status.rb

@@ -0,0 +1,14 @@
+module Mccloud::Provider
+  module Vagrant
+    module ProviderCommand
+
+    def status(selection=nil,options=nil)
+
+      self.raw.cli(['status',selection])
+
+    end
+
+    end #module
+  end #module
+end #module
+