PackerFiles 0.0.1

data/lib/PackerFiles/OS/CentOS/Server.rb

@@ -0,0 +1,170 @@
+# Requires
+require 'OS/Builder'
+require 'Core/Locale'
+require 'Core/Keyboard'
+require 'Core/TimeZone'
+require 'Core/User'
+require 'Core/Disk'
+require 'Core/Network'
+require 'Core/Packages'
+require 'Core/CDImage'
+require 'Virtual/VirtualBoxConverter'
+require 'Virtual/KVMConverter'
+require 'Virtual/VMWareConverter'
+require_relative 'Kickstart'
+require_relative 'CD'
+
+# OS class that handles details for generating Packer files for CentOS
+# Servers.
+module PackerFiles
+ module CentOS
+
+   class Server < Builder
+
+     # List of type accessors used by this class that also gets converted
+     # to a preseed
+     def self.preseed_accessors
+	[Core::Locale, Core::Keyboard, Core::TimeZone, 
+         Core::User,   Core::Disk,     Core::Network, 
+         Core::Packages
+        ]
+     end
+
+     # Create accessors based on Type
+     self.preseed_accessors.each do |type|
+     	type_accessor type
+     end
+
+     # Constructor. Yield self for further initialization
+     def initialize
+	yield self if block_given?
+     end 
+
+     # hook function for generation
+     def generate_hook
+
+	# Construct objects that needs to be converted by using accessors.
+        # The tracker type_accessors tracks the mapping b/w Type names and
+        # method calls.
+	objects =  []
+	objects.push self.Locale, self.Keyboard, self.TimeZone, self.User
+	objects.push self.Disk,   self.Network,  self.Packages
+	objects.each do |obj|
+	   obj.normalize
+        end
+
+	# Call a hook function to customize kickstart generation by
+	# derived classes.
+	kickstart_write_hook(objects)
+     end
+
+     # Hook function called before normalizing CD Image
+     def cd_image_hook
+	self.CDImage.impl = CD.new
+     end 
+
+     # Hook function called for converting this object into a hash for
+     # VirtualBox
+     def vbox_converter_hook(gen)
+        Virtual::VirtualBoxConverter.new(self, gen).value
+     end
+
+     # Hook function called for converting this object into a hash for
+     # KVM
+     def kvm_converter_hook(gen)
+        Virtual::KVMConverter.new(self, gen).value
+     end
+
+     # Hook function called for converting this object into a hash for
+     # VMWare
+     def vmware_converter_hook(gen)
+        Virtual::VMWareConverter.new(self, gen).value
+     end
+
+     # Hook for writing into preseed file.
+     def kickstart_write_hook(objects)
+
+	# Preseed file by evaluating templates @ Debian folder.
+	kickstart = Kickstart.new(kickstart_file)
+
+	# Evaluate Head 
+	kickstart.eval_file('CentOS/Templates/Base_erb.rb', {})
+
+        # Convert objects into Preseed configuration.
+        objects.each do |obj|
+	   kickstart.convert obj, 'CentOS/Templates'
+        end
+
+	# Evaluate tail
+	kickstart.eval_file('CentOS/Templates/Tail_erb.rb', {'obj' => self})
+
+     end
+
+     # Name of the preseed file
+     def kickstart_file
+        rel  = self.CDImage.release
+        arch = self.CDImage.arch
+        name = "kickstart-centos-server-#{rel}-#{arch}.cfg"
+	File.join(http_dir, name)
+     end
+
+     # Boot Command Override
+     def boot_command
+        file  = PackerFiles.DirPath('CentOS/Templates/boot_command_erb.rb')
+	hash  = { 'obj' => self, 'network' => self.Network }
+        value = PackerFiles.evaluate_erb(file.first, hash)
+	value.split("\n")
+     end
+
+     # Shutdown command Override
+     def shutdown_command
+        file  = PackerFiles.DirPath('CentOS/Templates/shutdown_command_erb.rb')
+        hash  = { 'obj' => self.User }
+        value = PackerFiles.evaluate_erb(file.first, hash)
+	value.chop
+     end
+
+     # Guest OS Type override
+     def guest_os_type
+	return "RedHat_64" if (self.CDImage.arch == "x86_64")
+	return "RedHat"
+     end
+
+     # VMWare Guest OS Type override
+     def vmware_guest_os_type
+	return "rhel4-64" if (self.CDImage.arch == "x86_64")
+	return "rhel4"
+     end
+
+     # VMWare Guest Flavor override
+     def vmware_guest_os_flavor
+	return "linux"
+     end
+
+     # Run any given command as root
+     def run_command_as_root
+        file  = PackerFiles.DirPath('CentOS/Templates/command_as_root_erb.rb')
+        hash  = { 'obj' => self }
+        value = PackerFiles.evaluate_erb(file.first, hash)
+	value.chop
+     end 
+
+     # Enable root user command w/o a password
+     def enable_root_for_user
+        file  = PackerFiles.DirPath('CentOS/Templates/enable_root_erb.rb')
+        hash  = { 'obj' => self }
+        value = PackerFiles.evaluate_erb(file.first, hash)
+	value.chop
+     end
+
+     # Disable root user command w/o a password
+     def disable_root_for_user
+        file  = PackerFiles.DirPath('CentOS/Templates/disable_root_erb.rb')
+        hash  = { 'obj' => self }
+        value = PackerFiles.evaluate_erb(file.first, hash)
+	value.chop
+     end
+
+  end
+ end
+end 

data/lib/PackerFiles/OS/CentOS/Templates/Base_erb.rb

@@ -0,0 +1,4 @@
+# Install from CDROM sources
+install
+cdrom
+

data/lib/PackerFiles/OS/CentOS/Templates/Disk_erb.rb

@@ -0,0 +1,35 @@
+# Disk and Partition related
+bootloader --location=mbr
+zerombr
+clearpart --all --initlabel
+
+# Case 1: When just a disk is given, always use the regular atomic
+# recipe, unless indicated otherwise.
+<% if @obj.is_raw? -%>
+autopart
+<% end -%>
+
+# Case 2: A disk with multiple partitions
+<% if @obj.only_partitions? -%>
+  <% @obj.partitions.each do |part| -%>
+      <% if part.file_system == 'linux-swap' -%>
+         partition swap --recommended --size=<%= part.size -%> --grow
+      <% else -%>
+         partition <%= part.mount_point -%> --size=<%= part.size -%> --fstype=<%= part.file_system %>
+      <% end -%>
+  <% end -%>
+<% end -%>
+
+# Case 3: A disk with Logical Volumes
+<% if @obj.only_vg? -%>
+   partition pv.01 --size=<%= @obj.size -%> --grow
+   volgroup <%= @obj.vg.name -%> pv.01
+   <% @obj.vg.lvs.each do |lv| -%>
+       <% if lv.file_system == 'linux-swap' -%>
+         logvol swap --recommended --size=<%= lv.size -%> --grow --name=<%= lv.name -%> --vgname=<%= lv.vg_name %>
+      <% else -%>
+         logvol <%= lv.mount_point -%> --size=<%= lv.size -%> --fstype=<%= lv.file_system -%> --name=<%= lv.name -%> --vgname=<%= lv.vg_name %>
+      <% end -%>
+   <% end -%> 
+<% end -%>
+

data/lib/PackerFiles/OS/CentOS/Templates/Keyboard_erb.rb

@@ -0,0 +1,4 @@
+# Specify the Keyboard to use
+keyboard <%= @obj.layout -%>
+
+

data/lib/PackerFiles/OS/CentOS/Templates/Locale_erb.rb

@@ -0,0 +1,4 @@
+# Specify locale
+lang <%= @obj.default -%>
+
+

data/lib/PackerFiles/OS/CentOS/Templates/Network_erb.rb

@@ -0,0 +1,17 @@
+# Network Configuration
+<% if (@obj.is_static?) && ((@obj.type & PackerFiles::Core::Network::IPV4) > 0) -%>
+network --device=<%= @obj.interface -%> --bootproto=static --ip=<%= @obj.ip -%> --netmask=<%= @obj.mask -%> --gateway=<%= @obj.gateway -%> --nameserver=<%= @obj.dns.join(',') -%>
+<% elsif (@obj.is_static?) && ((@obj.type & PackerFiles::Core::Network::IPV6) > 0) -%>
+network --device=<%= @obj.interface -%> --bootproto=static --ipv6=<%= @obj.ip -%> --netmask=<%= @obj.mask -%> --ipv6gateway=<%= @obj.gateway -%> --nameserver=<%= @obj.dns.join(',') -%>
+<% elsif (!@obj.is_static?) && ((@obj.type & PackerFiles::Core::Network::IPV6) > 0) -%>
+network --device=<%= @obj.interface -%> --bootproto=dhcp --noipv4
+<% elsif (!@obj.is_static?) && ((@obj.type & PackerFiles::Core::Network::IPV4) > 0) -%>
+network --device=<%= @obj.interface -%> --bootproto=dhcp --noipv6
+<% end -%>
+
+# Enable Firewall
+firewall --enabled --trust=<%= @obj.interface -%> --ssh 
+
+# Run SELINux in permissive mode
+selinux --permissive
+

data/lib/PackerFiles/OS/CentOS/Templates/Packages_erb.rb

@@ -0,0 +1,34 @@
+# Install the following packages
+%packages --nobase --ignoremissing --excludedocs
+<%= @obj.list.join("\n") -%>
+
+# unnecessary firmware
+-aic94xx-firmware
+-atmel-firmware
+-b43-openfwwf
+-bfa-firmware
+-ipw2100-firmware
+-ipw2200-firmware
+-ivtv-firmware
+-iwl100-firmware
+-iwl1000-firmware
+-iwl3945-firmware
+-iwl4965-firmware
+-iwl5000-firmware
+-iwl5150-firmware
+-iwl6000-firmware
+-iwl6000g2a-firmware
+-iwl6050-firmware
+-libertas-usb8388-firmware
+-ql2100-firmware
+-ql2200-firmware
+-ql23xx-firmware
+-ql2400-firmware
+-ql2500-firmware
+-rt61pci-firmware
+-rt73usb-firmware
+-xorg-x11-drv-ati-firmware
+-zd1211-firmware
+%end
+
+

data/lib/PackerFiles/OS/CentOS/Templates/Tail_erb.rb

@@ -0,0 +1,15 @@
+# Misc. Parameters that are required for installation.
+unsupported_hardware
+text
+firstboot --disabled
+reboot
+
+# Post Install Script
+%post
+
+# enable sudo for first user
+<% str="%#{@obj.User.login} ALL=(ALL) NOPASSWD: ALL" -%>
+echo "<%= str -%>" >> /etc/sudoers.d/<%= @obj.User.login %>
+sed -i "s/^.*requiretty/#Defaults requiretty/" /etc/sudoers
+
+%end

data/lib/PackerFiles/OS/CentOS/Templates/TimeZone_erb.rb

@@ -0,0 +1,10 @@
+# Set the time zone configuration
+<% if @obj.utc 
+      utc = '--utc'
+   else
+      utc = ''
+   end
+-%>
+timezone <%= utc -%> <%= @obj.zone -%>
+
+

data/lib/PackerFiles/OS/CentOS/Templates/User_erb.rb

@@ -0,0 +1,20 @@
+# Initial User configuration
+<% if @obj.uid.nil?
+      uid = ''
+   else
+      uid = "--uid=#{@obj.uid}"
+   end
+   if @obj.groups.nil? || @obj.groups.empty?
+      groups = ''
+   else
+      groups = "--groups=" + @obj.groups.join(",")
+   end
+-%>
+user --name=<%= @obj.login -%> --password=<%= @obj.password -%> <%= uid -%> <%= groups %>
+
+# Also specify the root password along with it.
+rootpw <%= @obj.password -%>
+
+# Enable shadow passwords with SHA-512 hash
+auth --enableshadow --passalgo=sha512 
+

data/lib/PackerFiles/OS/CentOS/Templates/boot_command_erb.rb

@@ -0,0 +1,16 @@
+<% kickstart = @obj.kickstart_file.gsub(@obj.http_dir,'').gsub('/', '') -%>
+<% if @network.nil? -%>
+<tab> text ks=http://{{user `host`}}:{{user `port`}}/<%= kickstart -%> <wait>
+<% elsif !@network.is_static? -%>
+<tab> text ks=http://{{ .HTTPIP }}:{{ .HTTPPort }}/<%= kickstart -%> <wait>
+<% else -%>
+<tab> text ks=http://{{user `host`}}:{{user `port`}}/<%= kickstart -%> <wait>
+<% end  -%>
+<% if !@network.nil? && @network.is_static? -%>
+ip=<%= @network.ip %> <wait>
+netmask=<%= @network.mask %> <wait>
+gateway=<%= @network.gateway %>  <wait>
+dns=<%= @network.dns.join(" ") %> <wait>
+ksdevice=<%= @network.interface %> <wait>
+<% end -%>
+<enter><wait>

data/lib/PackerFiles/OS/CentOS/Templates/command_as_root_erb.rb

@@ -0,0 +1 @@
+echo '<%= @obj.User.password -%>' | {{ .Vars }} sudo -E -S sh '{{ .Path }}'

data/lib/PackerFiles/OS/CentOS/Templates/disable_root_erb.rb

@@ -0,0 +1 @@
+rm -rf /etc/sudoers/<%= @obj.User.login %>

data/lib/PackerFiles/OS/CentOS/Templates/enable_root_erb.rb

@@ -0,0 +1 @@
+echo "<%= @obj.User.login -%> ALL=NOPASSWD: ALL" >> /etc/sudoers.d/<%= @obj.User.login %>

data/lib/PackerFiles/OS/CentOS/Templates/shutdown_command_erb.rb

@@ -0,0 +1 @@
+sudo /sbin/halt -h -p

data/lib/PackerFiles/OS/Debian/Apt.rb

@@ -0,0 +1,42 @@
+# Requires
+require 'Core/Base'
+require 'Core/Exceptions'
+
+# Apt Setup handles all apt-related questions during the initial
+# phase of OS installation.
+module PackerFiles
+ module Debian
+   class Apt < Core::Base
+
+     # Accessors
+     attr_accessor :tasks
+     attr_accessor :non_free
+     attr_accessor :contrib
+     attr_accessor :upgrades
+     attr_accessor :kernel
+
+     # Documentation for this class
+     def self.doc_file
+        PackerFiles.DirPath('Debian/example/Apt.txt').first
+     end
+
+     # Constructor
+     def initialize
+	@tasks    = []
+	@non_free = false
+	@contrib  = false
+	@upgrades = 'none'
+        @kernel   = 'linux-server'
+	yield self if block_given?
+     end
+
+     # Normalize
+     def normalize
+       raise Core::NilException.new(self, 'tasks') if @tasks.nil?
+       raise Core::NilException.new(self, 'kernel') if @kernel.nil?
+       raise Core::EmptyArrayException.new(self, 'tasks') if @tasks.empty?
+     end
+
+   end
+ end
+end

data/lib/PackerFiles/OS/Debian/CD.rb

@@ -0,0 +1,65 @@
+# Requires
+require 'uri'
+require 'open-uri'
+require 'nokogiri'
+require "Core/CDImage"
+require "Utils/AutoZone"
+
+# An implementation for CDImage which lists all available Debian CD Mirrors
+module PackerFiles
+   module Debian
+      class CD < Core::CDImageImpl
+ 
+         # Index URL which contains all the Debian CD Mirror list
+	 @@index  = 'https://www.debian.org/CD/http-ftp/'
+
+         # Override method for getting the ISO name
+	 def get_iso_name_pattern
+	    'debian-@release@-@arch@-CD-1.iso'
+         end
+
+	 # Override method for getting checksums
+	 def get_check_sum_pattern
+	    'MD5SUMS'
+	 end
+
+	 # Override method for checksum type
+	 def get_check_sum_type
+	    'md5'
+	 end
+
+         # Override method for getting index URLs.
+         def index_urls
+	    master_index_urls + mirror_index_urls
+         end
+
+         # The Index URLs from the main debian CD Image server.
+         private
+         def master_index_urls
+	 %w[
+           http://cdimage.debian.org/debian-cd/@release@/@arch@/iso-cd/
+           http://cdimage.debian.org/mirror/cdimage/archive/@release@/@arch@/iso-cd/
+         ]
+         end
+
+	 # The Index URLs from other debian CD mirrors. The CD Mirrors only support CD
+         # images and not DVD Images.
+         private
+         def mirror_index_urls
+            
+            # Use nokogiri to parse the index page, to get all the URLs that contain
+            # country name [as detected by AutoZone]. Unfortunately this code can only
+	    # be understood if HTML structure is visually inspected first. In short, 
+            # debian mirrors for all the countries in the index file, looks like
+            # <li> "Country" <a rel="nofollow" href="http://debian.xfree.com.ar/debian-cd/"> </li>
+            page    = Nokogiri::HTML(open(@@index))
+            country = Utils::AutoZone.new.JSON['country']
+            nodes   = page.css('li').select {|node| node.text.include?(country)}
+            arefs   = nodes.map {|node| node.css('a')}.flatten
+	    mirrors = arefs.map {|aref| aref.attribute('href').value}
+	    mirrors.map! {|m| URI::join(m, '@release@/@arch@/iso-cd/').to_s}
+         end
+
+      end
+   end
+end

data/lib/PackerFiles/OS/Debian/Mirrors.rb

@@ -0,0 +1,73 @@
+# Requires
+require 'open-uri'
+require 'benchmark'
+require 'uri'
+require 'Utils/AutoZone'
+require 'Utils/FastestURL'
+
+# Auto configures the debian mirror list. This is useful in pre-seeding
+# the fastest possible mirror.
+module PackerFiles
+ module Debian
+   class Mirrors
+
+      # default debian mirror
+      @@default = 'http://mirrors.kernel.org/debian'
+
+      # Default debian mirror list
+      @@mirrors = 'https://www.debian.org/mirror/list'
+
+      # accessors
+      attr_reader :hosts
+      attr_reader :best
+      attr_reader :proxy
+
+      # Constructor
+      def initialize
+
+         # A Single web page get more than this (in seconds) is SLOW  
+         limit   = 1 
+
+         # No. of trials for a web page
+         trials  = 4  
+   
+	 # Get the name of the http proxy
+	 @proxy   = ENV['http_proxy'] 
+
+         # Get the host list of mirrors for this country
+         code     = Utils::AutoZone.new.country_code.downcase
+         @hosts   = country_mirror(code)
+	 if @hosts.empty?
+	    @hosts   = [@@default] if @hosts.empty?
+	    @best    = URI::parse(@@default)
+	 else
+	    f        = Utils::FastestURL.new(limit, trials)
+	    all      = f.best_urls(@hosts).flatten
+	    begin 
+	       @best    = URI::parse(all.first)
+	    rescue Exception => e
+	       puts "Exception: #{e}, All: #{all.to_s}"
+	       raise e
+	    end
+	 end
+      end
+
+      # Given the country code, get a list of mirror sites that match
+      # the country code.
+      private
+      def country_mirror(code)
+        arr = []
+        regex = 'href=\"(http://.+\.' + code + '\/.+)"'
+        open(@@mirrors) {|f| 
+          f.each_line do |line|
+            if match = line.match(regex)
+	       arr.push(match[1])
+            end
+          end
+        }
+        return arr
+      end
+
+   end
+ end
+end 

data/lib/PackerFiles/OS/Debian/Preseed.rb

@@ -0,0 +1,64 @@
+# Requires
+require 'rubygems'
+
+# OS class that handles details for generating Preseed files for Debian
+# Servers.
+module PackerFiles
+ module Debian
+
+   class Preseed 
+
+     # Constructor. The preseed file name is the name of the file
+     # that will be written to. 
+     def initialize(filename)
+        @preseed = filename
+	create_file
+     end 
+
+     # Serializes the given object and write out the content
+     # in the output file.
+     def convert(obj, os_name)
+        file     = erb_file(obj, os_name)
+	content  = PackerFiles.evaluate_erb(file, { "obj" => obj })
+	add_content(content)
+     end
+
+     # Evaluate a partially specified file name (should be in the Gem's
+     # file list though) and add the content to the @preseed file
+     def eval_file(file_name, hash)
+       erb_file = PackerFiles.DirPath(file_name).first
+       content  = PackerFiles.evaluate_erb(erb_file, hash)
+       add_content(content)
+     end 
+
+     # Given an object, figure out the class name and return back the correct
+     # erb file name.
+     private
+     def erb_file(obj, os_name)
+
+       erb_file = File.join(os_name, obj.class.name.gsub(/.*::/,'') + '_erb.rb')
+       erb_path = PackerFiles.DirPath(erb_file)
+       
+       if erb_path.nil? || erb_path.empty?
+          raise RuntimeError.new("Template Path #{erb_file} does not exist")
+       end
+       return erb_path.first
+     end
+
+     # Create a new file with no content
+     private
+     def create_file
+       basename = File.dirname(@preseed)
+       FileUtils.mkdir_p(basename) if !Dir.exist?(basename)
+       File.open(@preseed, 'w') {|f| f.write ''}
+     end
+
+     # Add content to the preseed file
+     private
+     def add_content(content)
+       File.open(@preseed, 'a') {|f| f.write(content) }
+     end
+
+   end
+ end
+end