albanpeignier-system-builder 0.0.2

data/History.txt

@@ -0,0 +1,9 @@
+=== 0.0.2 2009-09-02
+
+* 1 major enhancement:
+  * Add rake task
+
+=== 0.0.1 2009-09-01
+
+* 1 major enhancement:
+  * Initial release

data/Manifest.txt

@@ -0,0 +1,31 @@
+History.txt
+Manifest.txt
+PostInstall.txt
+README.rdoc
+Rakefile
+bin/system-builder
+examples/Rakefile
+examples/config.rb
+examples/manifests/classes/test.pp
+examples/manifests/site.pp
+features/development.feature
+features/step_definitions/common_steps.rb
+features/support/common.rb
+features/support/env.rb
+features/support/matchers.rb
+lib/system_builder.rb
+lib/system_builder/boot.rb
+lib/system_builder/cli.rb
+lib/system_builder/configurator.rb
+lib/system_builder/core_ext.rb
+lib/system_builder/image.rb
+lib/system_builder/task.rb
+script/console
+script/destroy
+script/generate
+spec/spec.opts
+spec/spec_helper.rb
+spec/system-builder_cli_spec.rb
+spec/system-builder_spec.rb
+system-builder.gemspec
+tasks/rspec.rake

data/PostInstall.txt

@@ -0,0 +1,7 @@
+
+For more information on system-builder, see http://system-builder.rubyforge.org
+
+NOTE: Change this information in PostInstall.txt 
+You can also delete it if you don't want it.
+
+

data/README.rdoc

@@ -0,0 +1,48 @@
+= system-builder
+
+* http://github.com/#{github_username}/#{project_name}
+
+== DESCRIPTION:
+
+FIX (describe your package)
+
+== FEATURES/PROBLEMS:
+
+* FIX (list of features or problems)
+
+== SYNOPSIS:
+
+  FIX (code sample of usage)
+
+== REQUIREMENTS:
+
+* FIX (list of requirements)
+
+== INSTALL:
+
+* FIX (sudo gem install, anything else)
+
+== LICENSE:
+
+(The MIT License)
+
+Copyright (c) 2009 Alban Peignier
+
+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.

data/Rakefile

@@ -0,0 +1,25 @@
+require 'rubygems'
+gem 'hoe', '>= 2.1.0'
+require 'hoe'
+require 'fileutils'
+require './lib/system_builder'
+
+Hoe.plugin :newgem
+# Hoe.plugin :website
+Hoe.plugin :cucumberfeatures
+
+# Generate all the Rake tasks
+# Run 'rake -T' to see list of generated tasks (from gem root directory)
+$hoe = Hoe.spec 'system-builder' do
+  self.version = SystemBuilder::VERSION
+  self.developer 'Alban Peignier', 'alban.peignier@free.fr'
+  self.rubyforge_name       = self.name # TODO this is default value
+  # self.extra_deps         = [['activesupport','>= 2.0.2']]
+end
+
+require 'newgem/tasks'
+Dir['tasks/**/*.rake'].each { |t| load t }
+
+# TODO - want other tests/tasks run by default? Add them to the list
+# remove_task :default
+# task :default => [:spec, :features]

data/bin/system-builder

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+#
+#  Created by Alban Peignier on 2009-9-1.
+#  Copyright (c) 2009. All rights reserved.
+
+require 'rubygems'
+require File.expand_path(File.dirname(__FILE__) + "/../lib/system_builder")
+require "system_builder/cli"
+
+SystemBuilder::CLI.execute(STDOUT, ARGV)

data/examples/Rakefile

@@ -0,0 +1,18 @@
+require 'system_builder'
+require 'system_builder/task'
+
+def working_dir(name, type)
+  "/var/tmp/system-builder/#{name}/#{type}"
+end
+
+SystemBuilder::Task.new(:simple_test) do
+  SystemBuilder::DiskImage.new(working_dir("simple_test", :disk)).tap do |image|
+    image.boot = SystemBuilder::DebianBoot.new(working_dir("simple_test", :boot)).tap do |boot|
+      boot.mirror = "http://127.0.0.1:9999/debian"
+    end
+    
+    image.boot.configure do |chroot|
+      chroot.apt_install :sudo
+    end
+  end
+end

data/examples/config.rb

@@ -0,0 +1,26 @@
+def working_dir(name, type)
+  "/var/tmp/system-builder/#{name}/#{type}"
+end
+
+SystemBuilder.config(:simple_test) do
+  SystemBuilder::DiskImage.new(working_dir("simple_test", :disk)).tap do |image|
+    image.boot = SystemBuilder::DebianBoot.new(working_dir("simple_test", :boot)).tap do |boot|
+      boot.mirror = "http://127.0.0.1:9999/debian"
+    end
+    
+    image.boot.configure do |chroot|
+      chroot.apt_install :sudo
+    end
+  end
+end
+
+SystemBuilder.config(:puppet_test) do
+  SystemBuilder::DiskImage.new(working_dir("puppet_test", :disk)).tap do |image|
+    image.boot = SystemBuilder::DebianBoot.new(working_dir("puppet_test", :boot)).tap do |boot|
+      boot.mirror = "http://127.0.0.1:9999/debian"
+    end
+
+    # use manifests/site.pp in this directory
+    image.boot.configurators << SystemBuilder::PuppetConfigurator.new(File.dirname(__FILE__))
+  end
+end

data/examples/manifests/classes/test.pp

@@ -0,0 +1 @@
+notice("test")

data/examples/manifests/site.pp

@@ -0,0 +1,3 @@
+notice("puppet site configuration")
+
+import "classes/*"

data/features/development.feature

@@ -0,0 +1,13 @@
+Feature: Development processes of newgem itself (rake tasks)
+
+  As a Newgem maintainer or contributor
+  I want rake tasks to maintain and release the gem
+  So that I can spend time on the tests and code, and not excessive time on maintenance processes
+    
+  Scenario: Generate RubyGem
+    Given this project is active project folder
+    And "pkg" folder is deleted
+    When I invoke task "rake gem"
+    Then folder "pkg" is created
+    And file with name matching "pkg/*.gem" is created else you should run "rake manifest" to fix this
+    And gem spec key "rdoc_options" contains /--mainREADME.rdoc/

data/features/step_definitions/common_steps.rb

@@ -0,0 +1,163 @@
+Given /^this project is active project folder/ do
+  @active_project_folder = File.expand_path(File.dirname(__FILE__) + "/../..")
+end
+
+Given /^env variable \$([\w_]+) set to "(.*)"/ do |env_var, value|
+  ENV[env_var] = value
+end
+
+Given /"(.*)" folder is deleted/ do |folder|
+  in_project_folder { FileUtils.rm_rf folder }
+end
+
+When /^I invoke "(.*)" generator with arguments "(.*)"$/ do |generator, arguments|
+  @stdout = StringIO.new
+  in_project_folder do
+    if Object.const_defined?("APP_ROOT")
+      APP_ROOT.replace(FileUtils.pwd)
+    else 
+      APP_ROOT = FileUtils.pwd
+    end
+    run_generator(generator, arguments.split(' '), SOURCES, :stdout => @stdout)
+  end
+  File.open(File.join(@tmp_root, "generator.out"), "w") do |f|
+    @stdout.rewind
+    f << @stdout.read
+  end
+end
+
+When /^I run executable "(.*)" with arguments "(.*)"/ do |executable, arguments|
+  @stdout = File.expand_path(File.join(@tmp_root, "executable.out"))
+  in_project_folder do
+    system "#{executable} #{arguments} > #{@stdout} 2> #{@stdout}"
+  end
+end
+
+When /^I run project executable "(.*)" with arguments "(.*)"/ do |executable, arguments|
+  @stdout = File.expand_path(File.join(@tmp_root, "executable.out"))
+  in_project_folder do
+    system "ruby #{executable} #{arguments} > #{@stdout} 2> #{@stdout}"
+  end
+end
+
+When /^I run local executable "(.*)" with arguments "(.*)"/ do |executable, arguments|
+  @stdout = File.expand_path(File.join(@tmp_root, "executable.out"))
+  executable = File.expand_path(File.join(File.dirname(__FILE__), "/../../bin", executable))
+  in_project_folder do
+    system "ruby #{executable} #{arguments} > #{@stdout} 2> #{@stdout}"
+  end
+end
+
+When /^I invoke task "rake (.*)"/ do |task|
+  @stdout = File.expand_path(File.join(@tmp_root, "tests.out"))
+  in_project_folder do
+    system "rake #{task} --trace > #{@stdout} 2> #{@stdout}"
+  end
+end
+
+Then /^folder "(.*)" (is|is not) created/ do |folder, is|
+  in_project_folder do
+    File.exists?(folder).should(is == 'is' ? be_true : be_false)
+  end
+end
+
+Then /^file "(.*)" (is|is not) created/ do |file, is|
+  in_project_folder do
+    File.exists?(file).should(is == 'is' ? be_true : be_false)
+  end
+end
+
+Then /^file with name matching "(.*)" is created/ do |pattern|
+  in_project_folder do
+    Dir[pattern].should_not be_empty
+  end
+end
+
+Then /^file "(.*)" contents (does|does not) match \/(.*)\// do |file, does, regex|
+  in_project_folder do
+    actual_output = File.read(file)
+    (does == 'does') ?
+      actual_output.should(match(/#{regex}/)) :
+      actual_output.should_not(match(/#{regex}/))
+  end
+end
+
+Then /gem file "(.*)" and generated file "(.*)" should be the same/ do |gem_file, project_file|
+  File.exists?(gem_file).should be_true
+  File.exists?(project_file).should be_true
+  gem_file_contents = File.read(File.dirname(__FILE__) + "/../../#{gem_file}")
+  project_file_contents = File.read(File.join(@active_project_folder, project_file))
+  project_file_contents.should == gem_file_contents
+end
+
+Then /^(does|does not) invoke generator "(.*)"$/ do |does_invoke, generator|
+  actual_output = File.read(@stdout)
+  does_invoke == "does" ?
+    actual_output.should(match(/dependency\s+#{generator}/)) :
+    actual_output.should_not(match(/dependency\s+#{generator}/))
+end
+
+Then /help options "(.*)" and "(.*)" are displayed/ do |opt1, opt2|
+  actual_output = File.read(@stdout)
+  actual_output.should match(/#{opt1}/)
+  actual_output.should match(/#{opt2}/)
+end
+
+Then /^I should see$/ do |text|
+  actual_output = File.read(@stdout)
+  actual_output.should contain(text)
+end
+
+Then /^I should not see$/ do |text|
+  actual_output = File.read(@stdout)
+  actual_output.should_not contain(text)
+end
+
+Then /^I should see exactly$/ do |text|
+  actual_output = File.read(@stdout)
+  actual_output.should == text
+end
+
+Then /^I should see all (\d+) tests pass/ do |expected_test_count|
+  expected = %r{^#{expected_test_count} tests, \d+ assertions, 0 failures, 0 errors}
+  actual_output = File.read(@stdout)
+  actual_output.should match(expected)
+end
+
+Then /^I should see all (\d+) examples pass/ do |expected_test_count|
+  expected = %r{^#{expected_test_count} examples?, 0 failures}
+  actual_output = File.read(@stdout)
+  actual_output.should match(expected)
+end
+
+Then /^yaml file "(.*)" contains (\{.*\})/ do |file, yaml|
+  in_project_folder do
+    yaml = eval yaml
+    YAML.load(File.read(file)).should == yaml
+  end
+end
+
+Then /^Rakefile can display tasks successfully/ do
+  @stdout = File.expand_path(File.join(@tmp_root, "rakefile.out"))
+  in_project_folder do
+    system "rake -T > #{@stdout} 2> #{@stdout}"
+  end
+  actual_output = File.read(@stdout)
+  actual_output.should match(/^rake\s+\w+\s+#\s.*/)
+end
+
+Then /^task "rake (.*)" is executed successfully/ do |task|
+  @stdout.should_not be_nil
+  actual_output = File.read(@stdout)
+  actual_output.should_not match(/^Don't know how to build task '#{task}'/)
+  actual_output.should_not match(/Error/i)
+end
+
+Then /^gem spec key "(.*)" contains \/(.*)\// do |key, regex|
+  in_project_folder do
+    gem_file = Dir["pkg/*.gem"].first
+    gem_spec = Gem::Specification.from_yaml(`gem spec #{gem_file}`)
+    spec_value = gem_spec.send(key.to_sym)
+    spec_value.to_s.should match(/#{regex}/)
+  end
+end

data/features/support/common.rb

@@ -0,0 +1,29 @@
+module CommonHelpers
+  def in_tmp_folder(&block)
+    FileUtils.chdir(@tmp_root, &block)
+  end
+
+  def in_project_folder(&block)
+    project_folder = @active_project_folder || @tmp_root
+    FileUtils.chdir(project_folder, &block)
+  end
+
+  def in_home_folder(&block)
+    FileUtils.chdir(@home_path, &block)
+  end
+
+  def force_local_lib_override(project_name = @project_name)
+    rakefile = File.read(File.join(project_name, 'Rakefile'))
+    File.open(File.join(project_name, 'Rakefile'), "w+") do |f|
+      f << "$:.unshift('#{@lib_path}')\n"
+      f << rakefile
+    end
+  end
+
+  def setup_active_project_folder project_name
+    @active_project_folder = File.join(@tmp_root, project_name)
+    @project_name = project_name
+  end
+end
+
+World(CommonHelpers)

data/features/support/env.rb

@@ -0,0 +1,14 @@
+require File.dirname(__FILE__) + "/../../lib/system-builder"
+
+gem 'cucumber'
+require 'cucumber'
+gem 'rspec'
+require 'spec'
+
+Before do
+  @tmp_root = File.dirname(__FILE__) + "/../../tmp"
+  @home_path = File.expand_path(File.join(@tmp_root, "home"))
+  FileUtils.rm_rf   @tmp_root
+  FileUtils.mkdir_p @home_path
+  ENV['HOME'] = @home_path
+end

data/features/support/matchers.rb

@@ -0,0 +1,11 @@
+module Matchers
+  def contain(expected)
+    simple_matcher("contain #{expected.inspect}") do |given, matcher|
+      matcher.failure_message = "expected #{given.inspect} to contain #{expected.inspect}"
+      matcher.negative_failure_message = "expected #{given.inspect} not to contain #{expected.inspect}"
+      given.index expected
+    end
+  end
+end
+
+World(Matchers)

data/lib/system_builder.rb

@@ -0,0 +1,23 @@
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+module SystemBuilder
+  VERSION = '0.0.2'
+
+  @@configurations = {}
+  
+  def self.config(name, value = nil, &block)
+    value = (value or block.call)
+    puts "* load configuration #{name}"
+    @@configurations[name.to_s] = value
+  end
+
+  def self.configuration(name)
+    @@configurations[name.to_s]
+  end
+end
+
+require 'system_builder/core_ext'
+require 'system_builder/image'
+require 'system_builder/boot'
+require 'system_builder/configurator'

data/lib/system_builder/boot.rb

@@ -0,0 +1,191 @@
+class SystemBuilder::DebianBoot
+  
+  attr_accessor :version, :mirror, :architecture
+  attr_accessor :exclude, :include
+
+  attr_reader :root
+  attr_reader :configurators
+
+  def initialize(root)
+    @root = root
+
+    @version = :lenny
+    @mirror = 'http://ftp.debian.org/debian'
+    @architecture = :i386
+    @exclude = []
+    @include = []
+
+    # kernel can't be installed by debootstrap
+    @configurators = 
+      [ localhost_configurator, 
+        apt_configurator, 
+        kernel_configurator, 
+        fstab_configurator, 
+        timezone_configurator ]
+  end
+
+  def create
+    unless File.exists?(root)
+      FileUtils::mkdir_p root
+      FileUtils::sudo "debootstrap", debbootstrap_options, version, root, mirror
+    end
+
+    unless @configurators.empty?
+      chroot do |chroot|
+        @configurators.each do |configurator|
+          configurator.configure(chroot)
+        end
+      end
+    end
+  end
+
+  def kernel_configurator
+    SystemBuilder::ProcConfigurator.new do |chroot|
+      chroot.image.open("/etc/kernel-img.conf") do |f|
+        f.puts "do_initrd = yes"
+      end
+      chroot.apt_install %w{linux-image-2.6-686 grub}
+    end
+  end
+
+  def fstab_configurator
+    SystemBuilder::ProcConfigurator.new do |chroot|
+      chroot.image.open("/etc/fstab") do |f|
+        %w{/tmp /var/run /var/log /var/lock /var/tmp}.each do |directory|
+          f.puts "tmpfs #{directory} tmpfs defaults,noatime 0 0"
+        end
+      end
+    end
+  end
+
+  def timezone_configurator
+    SystemBuilder::ProcConfigurator.new do |chroot|
+      # Use same timezone than build machine
+      chroot.image.install "/etc/", "/etc/timezone", "/etc/localtime"
+    end
+  end
+
+  def apt_configurator
+    # TODO see if this step is really needed
+    SystemBuilder::ProcConfigurator.new do |chroot|    
+      chroot.image.install "/etc/apt", "/etc/apt/trusted.gpg"
+      chroot.sudo "apt-get update"
+    end
+  end
+
+  def localhost_configurator
+    SystemBuilder::ProcConfigurator.new do |chroot|
+      chroot.image.open("/etc/hosts") do |f|
+        f.puts "127.0.0.1	localhost"
+        f.puts "::1     localhost ip6-localhost ip6-loopback"
+      end
+    end
+  end
+
+  def configure(&block)
+    @configurators << SystemBuilder::ProcConfigurator.new(block)
+  end
+
+  def debbootstrap_options
+    {
+      :arch => architecture,  
+      :exclude => exclude.join(','),
+      :include => include.join(','),
+      :variant => :minbase
+    }.collect do |k,v| 
+      ["--#{k}", Array(v).join(',')] unless v.blank?
+    end.compact
+  end
+
+  def image(&block)
+    @image ||= Image.new(root)
+
+    if block_given?    
+      yield @image
+    else
+      @image
+    end
+  end
+
+  def chroot(&block)
+    @chroot ||= Chroot.new(image)
+    @chroot.execute &block
+  end
+
+  class Image
+    
+    def initialize(root)
+      @root = root
+    end
+
+    def mkdir(directory)
+      FileUtils::sudo "mkdir -p #{expand_path(directory)}"
+    end
+
+    def install(target, *sources)
+      FileUtils::sudo "cp --preserve=mode,timestamps #{sources.join(' ')} #{expand_path(target)}"
+    end
+
+    def rsync(target, *sources)
+      options = (Hash === sources.last ? sources.pop : {})
+      rsync_options = options.collect { |k,v| "--#{k}=#{v}" }
+      FileUtils::sudo "rsync -av #{rsync_options.join(' ')} #{sources.join(' ')} #{expand_path(target)}"
+    end
+
+    def open(filename, &block) 
+      Tempfile.open(File.basename(filename)) do |f|
+        yield f
+        f.close
+        
+        File.chmod 0644, f.path
+        install filename, f.path
+      end
+    end
+
+    def expand_path(path)
+      File.join(@root,path)
+    end
+
+  end
+
+  class Chroot
+
+    attr_reader :image
+
+    def initialize(image)
+      @image = image
+    end
+
+    def apt_install(*packages)
+      sudo "apt-get install --yes --force-yes #{packages.join(' ')}"
+    end
+
+    def cp(*arguments)
+      sudo "cp #{arguments.join(' ')}"
+    end
+
+    def sh(*arguments)
+      FileUtils::sudo "chroot #{image.expand_path('/')} sh -c \"LC_ALL=C #{arguments.join(' ')}\""
+    end
+    alias_method :sudo, :sh
+
+    def execute(&block)
+      begin
+        prepare_run
+        yield self
+      ensure
+        unprepare_run
+      end
+    end
+
+    def prepare_run
+      FileUtils::sudo "mount proc #{image.expand_path('/proc')} -t proc"
+    end
+
+    def unprepare_run
+      FileUtils::sudo "umount #{image.expand_path('/proc')}"
+    end
+
+  end
+  
+end

data/lib/system_builder/cli.rb

@@ -0,0 +1,35 @@
+require 'optparse'
+
+module SystemBuilder
+  class CLI
+    def self.execute(stdout, arguments=[])
+      options = {}
+      mandatory_options = %w(config)
+
+      OptionParser.new do |opts|
+        opts.banner = <<-BANNER.gsub(/^          /,'')
+          Create and configure bootable systems
+
+          Usage: #{File.basename($0)} [options] image command
+
+          Options are:
+        BANNER
+        opts.separator ""
+        opts.on("-c", "--config=FILE", String,
+                "The file containing image and boot definitions") { |arg| options[:config] = arg }
+        opts.on("-h", "--help", "Show this help message.") { stdout.puts opts; exit }
+
+        opts.parse!(arguments)
+
+        if mandatory_options && mandatory_options.find { |option| options[option.to_sym].nil? }
+          stdout.puts opts; exit
+        end
+      end
+
+      load options[:config]
+
+      image = SystemBuilder.configuration(arguments.unshift)
+      image.create
+    end
+  end
+end

data/lib/system_builder/configurator.rb

@@ -0,0 +1,42 @@
+module SystemBuilder
+
+  class ProcConfigurator
+
+    def initialize(proc = nil, &block)
+      @proc = (proc or block)
+    end
+
+    def configure(chroot)
+      @proc.call chroot
+    end
+
+  end
+
+  class PuppetConfigurator
+
+    attr_reader :manifest
+
+    def initialize(manifest = ".")
+      @manifest = manifest
+    end
+
+    def configure(chroot)
+      chroot.apt_install :puppet
+      chroot.image.open("/etc/default/puppet") do |f|
+        f.puts "START=no"
+      end
+
+      unless File.directory?(manifest)
+        chroot.image.install "/tmp/puppet.pp", manifest
+        chroot.sudo "puppet tmp/puppet.pp"
+      else
+        chroot.image.mkdir "/tmp/puppet"
+
+        chroot.image.rsync "/tmp/puppet", "#{manifest}/manifests", "#{manifest}/files", :exclude => "*~"
+        chroot.sudo "puppet tmp/puppet/manifests/site.pp"
+      end
+    end
+
+  end
+
+end

data/lib/system_builder/core_ext.rb

@@ -0,0 +1,53 @@
+class Numeric
+
+  def megabytes
+    self * 1024 * 1024
+  end
+  alias_method :megabyte, :megabytes
+
+  def in_megabytes
+    self.to_f / 1.megabyte
+  end
+
+end
+
+class Object
+
+  def blank?
+    false
+  end
+
+end
+
+class NilClass 
+
+  def blank?
+    true
+  end
+
+end
+
+class Array 
+
+  alias_method :blank?, :empty?
+
+end
+
+class String
+
+  alias_method :blank?, :empty?
+
+end
+
+module FileUtils
+
+  def self.sh(*cmd)
+    puts "* #{cmd.join(' ')}"
+    raise "Command failed: #{$?}" unless system(cmd.join(' '))
+  end
+
+  def self.sudo(*cmd)
+    sh ["sudo", *cmd]
+  end
+
+end

data/lib/system_builder/image.rb

@@ -0,0 +1,118 @@
+require 'tempfile'
+
+class SystemBuilder::DiskImage
+
+  attr_accessor :boot, :size
+  attr_reader :file
+
+  def initialize(file)
+    @file = file
+    @size = 512.megabytes
+  end
+
+  def create
+    boot.create
+    # TODO
+
+    file_creation = (not File.exists?(file))
+    if file_creation
+      create_file
+      create_partition_table
+      format_root_fs
+    end
+
+    install_grub_files :stage_files => %w{e2fs_stage1_5 stage?}
+
+    sync_root_fs
+    install_grub if file_creation
+
+    self
+  end
+
+  def create_file
+    FileUtils::sh "dd if=/dev/zero of=#{file} count=#{size.in_megabytes.to_i} bs=1M"
+  end
+
+  def create_partition_table
+    FileUtils::sh "echo '63,' | /sbin/sfdisk --no-reread -uS -H16 -S63 #{file}"
+  end
+
+  def format_root_fs
+    loop_device = "/dev/loop0"
+    begin
+      FileUtils::sudo "losetup -o #{fs_offset} #{loop_device} #{file}"
+      FileUtils::sudo "mke2fs -L #{fs_label} -jqF #{loop_device} #{fs_block_size}"
+    ensure
+      FileUtils::sudo "losetup -d #{loop_device}"
+    end
+  end
+
+  def sync_root_fs
+    mount_dir = "/tmp/mount_root_fs"
+    FileUtils::mkdir_p mount_dir
+    
+    begin
+      FileUtils::sudo "mount -o loop,offset=#{fs_offset} #{file} #{mount_dir}"
+      FileUtils::sudo "rsync -av #{boot.root}/ #{mount_dir}"
+    ensure
+      FileUtils::sudo "umount #{mount_dir}"
+    end
+
+    FileUtils.touch file
+  end
+
+  def install_grub_files(options = {})
+    stage_files = Array(options[:stage_files]).flatten
+
+    boot.image do |image|
+      image.mkdir "/boot/grub"
+
+      install_grub_menu options  
+      image.install "boot/grub", stage_files.collect { |f| '/usr/lib/grub/**/' + f }
+    end
+  end
+
+  def install_grub
+    IO.popen("sudo grub --device-map=/dev/null","w") { |grub| 
+      grub.puts "device (hd0) #{file}"
+      grub.puts "root (hd0,0)"
+      grub.puts "setup (hd0)"
+      grub.puts "quit"
+    }
+  end
+
+  def install_grub_menu(options = {})
+    root = (options[:root] or "LABEL=#{fs_label}")
+    version = (options[:version] or Time.now.strftime("%Y%m%d%H%M"))
+
+    boot.image.open("/boot/grub/menu.lst") do |f|
+      f.puts "default 0"
+      f.puts "timeout 2"
+      f.puts "title #{version} Debian GNU/Linux"
+      f.puts "kernel /vmlinuz root=#{root} ro"
+      f.puts "initrd /initrd.img"
+    end
+  end
+
+  def convert(export_file, options = {})
+    unless FileUtils.uptodate? export_file, file
+      arguments = []
+      arguments << "-O #{options[:format]}" if options[:format]
+      FileUtils::sh "qemu-img convert -f raw #{file} #{arguments.join(' ')} #{export_file}"
+    end
+  end
+
+  def fs_block_size
+    linux_partition_info = `/sbin/sfdisk -l #{file}`.scan(%r{#{file}.*Linux}).first
+    linux_partition_info.split[4].to_i
+  end
+
+  def fs_offset
+    32256
+  end
+
+  def fs_label
+    "root"
+  end
+
+end

data/lib/system_builder/task.rb

@@ -0,0 +1,36 @@
+require 'rake/tasklib'
+
+class SystemBuilder::Task < Rake::TaskLib
+
+  attr_reader :name
+
+  def initialize(name, &block)
+    @name = name
+
+    @image =
+      if block_given?
+        block.call
+      else
+        SystemBuilder.config(name)
+      end
+
+    define
+  end
+
+  def define
+    namespace name do
+      desc "Create image #{name} in #{@image.file}"
+      task :dist do
+        @image.create
+      end
+      namespace :dist do
+        desc "Create vmwaire image in #{@image.file}.vdmk"
+        task :vmware do
+          @image.convert "#{@image.file}.vmdk", :format => "vmdk"
+        end
+      end
+      task "dist:vmware" => "dist"
+    end
+  end
+
+end

data/script/console

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+# File: script/console
+irb = RUBY_PLATFORM =~ /(:?mswin|mingw)/ ? 'irb.bat' : 'irb'
+
+libs =  " -r irb/completion"
+# Perhaps use a console_lib to store any extra methods I may want available in the cosole
+# libs << " -r #{File.dirname(__FILE__) + '/../lib/console_lib/console_logger.rb'}"
+libs <<  " -r #{File.dirname(__FILE__) + '/../lib/system-builder.rb'}"
+puts "Loading system-builder gem"
+exec "#{irb} #{libs} --simple-prompt"

data/script/destroy

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/destroy'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Destroy.new.run(ARGV)

data/script/generate

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/generate'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Generate.new.run(ARGV)

data/spec/spec.opts

@@ -0,0 +1 @@
+--colour

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+begin
+  require 'spec'
+rescue LoadError
+  require 'rubygems' unless ENV['NO_RUBYGEMS']
+  gem 'rspec'
+  require 'spec'
+end
+
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+require 'system-builder'

data/spec/system-builder_cli_spec.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+require 'system-builder/cli'
+
+describe SystemBuilder::CLI, "execute" do
+  before(:each) do
+    @stdout_io = StringIO.new
+    SystemBuilder::CLI.execute(@stdout_io, [])
+    @stdout_io.rewind
+    @stdout = @stdout_io.read
+  end
+  
+  it "should print default output" do
+    @stdout.should =~ /To update this executable/
+  end
+end

data/spec/system-builder_spec.rb

@@ -0,0 +1,11 @@
+require File.dirname(__FILE__) + '/spec_helper.rb'
+
+# Time to add your specs!
+# http://rspec.info/
+describe "Place your specs here" do
+  
+  it "find this spec in spec directory" do
+    # violated "Be sure to write your specs"
+  end
+  
+end

data/system-builder.gemspec

@@ -0,0 +1,35 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{system-builder}
+  s.version = "0.0.2"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Alban Peignier"]
+  s.date = %q{2009-09-02}
+  s.default_executable = %q{system-builder}
+  s.description = %q{FIX (describe your package)}
+  s.email = ["alban.peignier@free.fr"]
+  s.executables = ["system-builder"]
+  s.extra_rdoc_files = ["History.txt", "Manifest.txt", "PostInstall.txt"]
+  s.files = ["History.txt", "Manifest.txt", "PostInstall.txt", "README.rdoc", "Rakefile", "bin/system-builder", "examples/Rakefile", "examples/config.rb", "examples/manifests/classes/test.pp", "examples/manifests/site.pp", "features/development.feature", "features/step_definitions/common_steps.rb", "features/support/common.rb", "features/support/env.rb", "features/support/matchers.rb", "lib/system_builder.rb", "lib/system_builder/boot.rb", "lib/system_builder/cli.rb", "lib/system_builder/configurator.rb", "lib/system_builder/core_ext.rb", "lib/system_builder/image.rb", "lib/system_builder/task.rb", "script/console", "script/destroy", "script/generate", "spec/spec.opts", "spec/spec_helper.rb", "spec/system-builder_cli_spec.rb", "spec/system-builder_spec.rb", "system-builder.gemspec", "tasks/rspec.rake"]
+  s.homepage = %q{http://github.com/#{github_username}/#{project_name}}
+  s.rdoc_options = ["--main", "README.rdoc"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{system-builder}
+  s.rubygems_version = %q{1.3.4}
+  s.summary = %q{FIX (describe your package)}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<hoe>, [">= 2.3.3"])
+    else
+      s.add_dependency(%q<hoe>, [">= 2.3.3"])
+    end
+  else
+    s.add_dependency(%q<hoe>, [">= 2.3.3"])
+  end
+end

data/tasks/rspec.rake

@@ -0,0 +1,21 @@
+begin
+  require 'spec'
+rescue LoadError
+  require 'rubygems' unless ENV['NO_RUBYGEMS']
+  require 'spec'
+end
+begin
+  require 'spec/rake/spectask'
+rescue LoadError
+  puts <<-EOS
+To use rspec for testing you must install rspec gem:
+    gem install rspec
+EOS
+  exit(0)
+end
+
+desc "Run the specs under spec/models"
+Spec::Rake::SpecTask.new do |t|
+  t.spec_opts = ['--options', "spec/spec.opts"]
+  t.spec_files = FileList['spec/**/*_spec.rb']
+end

metadata

@@ -0,0 +1,96 @@
+--- !ruby/object:Gem::Specification 
+name: albanpeignier-system-builder
+version: !ruby/object:Gem::Version 
+  version: 0.0.2
+platform: ruby
+authors: 
+- Alban Peignier
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-09-02 00:00:00 -07:00
+default_executable: system-builder
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.3.3
+    version: 
+description: FIX (describe your package)
+email: 
+- alban.peignier@free.fr
+executables: 
+- system-builder
+extensions: []
+
+extra_rdoc_files: 
+- History.txt
+- Manifest.txt
+- PostInstall.txt
+files: 
+- History.txt
+- Manifest.txt
+- PostInstall.txt
+- README.rdoc
+- Rakefile
+- bin/system-builder
+- examples/Rakefile
+- examples/config.rb
+- examples/manifests/classes/test.pp
+- examples/manifests/site.pp
+- features/development.feature
+- features/step_definitions/common_steps.rb
+- features/support/common.rb
+- features/support/env.rb
+- features/support/matchers.rb
+- lib/system_builder.rb
+- lib/system_builder/boot.rb
+- lib/system_builder/cli.rb
+- lib/system_builder/configurator.rb
+- lib/system_builder/core_ext.rb
+- lib/system_builder/image.rb
+- lib/system_builder/task.rb
+- script/console
+- script/destroy
+- script/generate
+- spec/spec.opts
+- spec/spec_helper.rb
+- spec/system-builder_cli_spec.rb
+- spec/system-builder_spec.rb
+- system-builder.gemspec
+- tasks/rspec.rake
+has_rdoc: false
+homepage: http://github.com/#{github_username}/#{project_name}
+post_install_message: 
+rdoc_options: 
+- --main
+- README.rdoc
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: system-builder
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 3
+summary: FIX (describe your package)
+test_files: []
+