vmit 0.0.3

data/lib/vmit/bootstrap.rb

@@ -0,0 +1,273 @@
+#
+# Copyright (C) 2013 Duncan Mac-Vicar P. <dmacvicar@suse.de>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+require 'vmit'
+require 'vmit/autoyast'
+require 'vmit/kickstart'
+require 'vmit/debian_preseed'
+
+module Vmit
+
+  module Bootstrap
+
+    def self.arch
+      Cheetah.run('arch', :stdout => :capture).strip
+    end
+
+    def self.bootstrapper_for(location)
+      case
+      when local_iso
+      end
+    end
+
+    module MethodDebianPreseed
+      # @param [Hash] args Arguments for 1st stage
+      def execute_autoinstall(args)
+        auto_install_args = {}
+        preseed = Vmit::DebianPreseed.new
+
+        Dir.mktmpdir do |floppy_dir|
+          qemu_args = {:floppy => floppy_dir,
+                      :append => "preseed/file=/floppy/preseed.cfg auto=true priority=critical",
+                      :reboot => false}
+          qemu_args.merge!(auto_install_args)
+          # transform duplicates into an array
+          qemu_args.merge!(args) do |key, oldv, newv|
+            case key
+              when :append then [oldv, newv].flatten
+              else newv
+            end
+          end
+
+          # Configure the autoinstallation profile to persist eth0
+          # for the current MAC address
+          # The interface will be setup with DHCP by default.
+          # TODO: make this more flexible in the future?
+          #autoyast.name_network_device(vm[:mac_address], 'eth0')
+          File.write(File.join(floppy_dir, 'preseed.cfg'), preseed.to_txt)
+          Vmit.logger.info "Preseed: 1st stage."
+          vm.run(qemu_args)
+          Vmit.logger.info "Preseed: 2st stage."
+          # 2nd stage
+          vm.run(:reboot => false)
+        end
+      end
+    end
+
+    module MethodKickstart
+      # @param [Hash] args Arguments for 1st stage
+      def execute_autoinstall(args)
+        auto_install_args = {}
+        kickstart = Vmit::Kickstart.new
+
+        case media
+          when Vmit::VFS::URI
+            kickstart.install = location
+          when Vmit::VFS::ISO
+            kickstart.install = :cdrom
+            auto_install_args.merge!(:cdrom => location.to_s)
+          else raise ArgumentError.new("Unsupported autoinstallation: #{location}")
+        end
+
+        Dir.mktmpdir do |floppy_dir|
+          qemu_args = {:floppy => floppy_dir,
+                      :append => "ks=floppy repo=#{kickstart.install}",
+                      :reboot => false}
+          qemu_args.merge!(auto_install_args)
+          # transform duplicates into an array
+          qemu_args.merge!(args) do |key, oldv, newv|
+            case key
+              when :append then [oldv, newv].flatten
+              else newv
+            end
+          end
+
+          File.write(File.join(floppy_dir, 'ks.cfg'), kickstart.to_ks_script)
+          Vmit.logger.info "Kickstart: 1st stage."
+          vm.run(qemu_args)
+        end
+      end
+    end
+
+    module MethodAutoYaST
+      # @param [Hash] args Arguments for 1st stage
+      def execute_autoinstall(args)
+        auto_install_args = {}
+        auto_install_args.merge!(args)
+        kernel_append_arg = case media
+          when Vmit::VFS::URI then "install=#{location}"
+          when Vmit::VFS::ISO then 'install=cdrom'
+          else raise ArgumentError.new("Unsupported autoinstallation: #{location}")
+        end
+        auto_install_args.merge!(:append => kernel_append_arg)
+        if media.is_a?(Vmit::VFS::ISO)
+          auto_install_args.merge!(:cdrom => location.to_s)
+        end
+
+        Dir.mktmpdir do |floppy_dir|
+          qemu_args = {:floppy => floppy_dir,
+                      :append => "autoyast=device://fd0/autoinst.xml",
+                      :reboot => false}
+          # transform duplicates into an array
+          qemu_args.merge!(auto_install_args) do |key, oldv, newv|
+            case key
+              when :append then [oldv, newv].flatten
+              else newv
+            end
+          end
+
+          autoyast = Vmit::AutoYaST.new
+
+          # WTF SLE and openSUSE have different
+          # base pattern names
+          media.open('/content') do |content_file|
+            content_file.each_line do |line|
+              case line
+                when /^DISTRIBUTION (.+)$/
+                  case $1
+                    when /SUSE_SLE/ then autoyast.minimal_sle!
+                    when /openSUSE/ then autoyast.minimal_opensuse!
+                  end
+              end
+            end
+          end
+
+          # Configure the autoinstallation profile to persist eth0
+          # for the current MAC address
+          # The interface will be setup with DHCP by default.
+          # TODO: make this more flexible in the future?
+          #autoyast.name_network_device(vm[:mac_address], 'eth0')
+          File.write(File.join(floppy_dir, 'autoinst.xml'), autoyast.to_xml)
+          Vmit.logger.info "AutoYaST: 1st stage."
+          vm.run(qemu_args)
+          Vmit.logger.info "AutoYaST: 2st stage."
+          # 2nd stage
+          vm.run(:reboot => false)
+        end
+      end
+    end
+
+    module SUSEMedia
+      include MethodAutoYaST
+
+      def get_initrd
+        media.open("/boot/#{Vmit::Bootstrap.arch}/loader/initrd")
+      end
+
+      def get_kernel
+        media.open("/boot/#{Vmit::Bootstrap.arch}/loader/linux")
+      end
+    end
+
+    module FedoraMedia
+      include MethodKickstart
+
+      def get_initrd
+        media.open("/images/pxeboot/initrd.img")
+      end
+
+      def get_kernel
+        media.open("/images/pxeboot/vmlinuz")
+      end
+    end
+
+    module DebianMedia
+      include MethodDebianPreseed
+
+      def get_initrd
+        arch = Vmit::Bootstrap.arch.gsub(/x86_64/, 'amd64')
+        media.open("/main/installer-#{arch}/current/images/netboot/debian-installer/#{arch}/initrd.gz")
+      end
+
+      def get_kernel
+        arch = Vmit::Bootstrap.arch.gsub(/x86_64/, 'amd64')
+        media.open("/main/installer-#{arch}/current/images/netboot/debian-installer/#{arch}/linux")
+      end
+    end
+
+    # Boostraps a vm from a SUSE repository
+    class FromMedia
+
+      attr_reader :vm
+      attr_reader :media
+      attr_reader :location
+
+      # @param [URI] location
+      def self.accept?(location)
+        # either a local ISO or a remote repository
+        # (and not a remote file, but we don't have
+        # a good way to check)
+        Vmit::VFS::ISO.accept?(location) ||
+          (Vmit::VFS::URI.accept?(location) &&
+            File.extname(location.to_s) == '')
+      end
+
+      def initialize(vm, location)
+        @location = location
+        @vm = vm
+        @media = Vmit::VFS.from(location)
+
+        # TODO FIXME we need a clever way to detect the
+        # location distro type. I could uname the kernel, but
+        # I need the type to know the location.
+        media_handler = case location.to_s.downcase
+          when /fedora|redhat|centos/ then FedoraMedia
+          when /suse/ then SUSEMedia
+          when /debian/ then DebianMedia
+          else
+            raise "Don't know how to bootstrap media #{location}"
+        end
+        self.extend media_handler
+
+        @boot_kernel = get_kernel
+        @boot_initrd = get_initrd
+      end
+
+      def execute
+        args = {}
+        args.merge!({:kernel => @boot_kernel.path, :initrd => @boot_initrd.path})
+        execute_autoinstall(args)
+      end
+    end
+
+    class FromImage
+
+       # @param [URI] location
+      def self.accept?(location)
+        uri = case location
+          when ::URI then location
+          else ::URI.parse(location.to_s)
+        end
+        return false unless ['.raw', '.qcow2'].include?(File.extname(uri.to_s))
+        uri.scheme == 'http' || File.exist?(uri.to_s)
+      end
+
+      def initialize(vm, location)
+        @location = location
+        #http://download.suse.de/ibs/Devel:/Galaxy:/Manager:/1.7:/Appliance/images/SUSE_Manager_Server_pg_Devel.x86_64-1.7.0-Build3.126.raw.xz
+      end
+
+      def execute
+        Vmit.logger.info "From media!!!"
+      end
+    end
+
+  end
+end

data/lib/vmit/debian_preseed.rb

@@ -0,0 +1,64 @@
+#
+# Copyright (C) 2013 Duncan Mac-Vicar P. <dmacvicar@suse.de>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+require 'erb'
+
+module Vmit
+
+  # Some good references here:
+  # http://www.hps.com/~tpg/notebook/autoinstall.php
+  class DebianPreseed
+
+    def to_txt
+      template = ERB.new <<-EOF
+d-i debian-installer/locale string en_US
+d-i console-tools/archs select at
+d-i console-keymaps-at/keymap select American English
+d-i debian-installer/keymap string us
+d-i netcfg/get_hostname string unassigned-hostname
+d-i netcfg/get_hostname seen true
+d-i netcfg/get_domain string unassigned-domain
+d-i netcfg/get_domain seen true
+d-i mirror/protocol string ftp
+d-i mirror/ftp/hostname string ftp.de.debian.org
+d-i mirror/ftp/directory string /debian/
+d-i mirror/ftp/proxy string
+
+d-i partman-auto/method string regular
+d-i partman-partitioning/confirm_write_new_label boolean true
+d-i partman/choose_partition select finish
+d-i partman/confirm boolean true
+d-i partman/confirm_nooverwrite boolean true
+
+d-i clock-setup/utc boolean true
+d-i time/zone string US/Eastern
+d-i clock-setup/ntp boolean true
+popularity-contest popularity-contest/participate boolean false
+d-i pkgsel/include string ssh rsync initrd-tools cramfsprogs lzop
+d-i passwd/root-password password linux
+d-i passwd/root-password-again password linux
+d-i passwd/make-user boolean false
+d-i grub-installer/only_debian boolean true
+EOF
+      return template.result(binding)
+    end
+
+  end
+end

data/lib/vmit/ext.rb

@@ -0,0 +1,38 @@
+#
+# Copyright (C) 2013 Duncan Mac-Vicar P. <dmacvicar@suse.de>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+
+module Vmit
+  module Ext
+
+    module Hash
+      def symbolize_keys
+        Hash[map{|(k,v)| [k.to_sym,v]}]
+      end
+
+      def symbolize_keys!
+        replace(symbolize_keys)
+      end
+    end
+
+  end
+end
+
+Hash.send(:include, Vmit::Ext::Hash)

data/lib/vmit/kickstart.rb

@@ -0,0 +1,67 @@
+#
+# Copyright (C) 2013 Duncan Mac-Vicar P. <dmacvicar@suse.de>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+require 'vmit'
+require 'erb'
+
+module Vmit
+
+  class Kickstart
+
+    attr_accessor :install
+
+    # ks=floppy
+    def initialize
+    end
+
+    def to_ks_script
+template = ERB.new <<-EOF
+cmdline
+halt
+rootpw linux
+lang en_US.UTF-8
+keyboard us
+timezone --utc America/New_York
+bootloader --location=mbr --driveorder=sda --append="rhgb quiet"
+install
+<% if install.is_a?(String) || install.is_a?(::URI)%>
+url --url=<%= install.to_s.strip %>
+<% else %>
+<%= install %>
+<% end %>
+network --device eth0 --bootproto dhcp
+zerombr yes
+clearpart --all --initlabel
+autopart
+%packages --nobase
+@core
+@server-policy
+wget
+mc
+
+%end
+EOF
+    template.result(binding)
+    end
+
+
+  end
+
+end

data/lib/vmit/logger.rb

@@ -0,0 +1,66 @@
+#
+# Copyright (C) 2013 Duncan Mac-Vicar P. <dmacvicar@suse.de>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+require 'logger'
+require 'cheetah'
+
+module Vmit
+
+  # remote log service for
+  # qemu ifup ifdown scripts
+  class LogServer
+    def method_missing(name, *args)
+      Vmit.logger.send(name, *args)
+    end
+  end
+
+  # Cheetah says it supports a logger
+  # for debugging but then adds entries
+  # with INFO instead of DEBUG
+  class CheetahLoggerAdapter
+    def initialize(logger)
+      @logger = logger
+    end
+
+    def info(message)
+      @logger.debug(message)
+    end
+
+    def error(message)
+      @logger.error(message)
+    end
+  end
+
+  def self.logger=(logger)
+    @logger = logger
+  end
+
+  def self.logger
+    if @logger.nil?
+      logger = Logger.new(STDERR)
+      logger.level = Logger::INFO
+      logger.level = Logger::DEBUG if ENV['DEBUG']
+      Vmit.logger = logger
+      Cheetah.default_options = {:logger => CheetahLoggerAdapter.new(logger)}
+    end
+    @logger
+  end
+
+end