def tasksel_steps
  [
	'I create a 6 GiB disk named "'+JOB_NAME+'"',
	'I plug ide drive "'+JOB_NAME+'"',
	'I start the computer',
	'I select the install mode',
	'I select British English',
	'I accept the hostname, using "example.com" as the domain',
	'I set the root password to "rootme"',
	'I set the password for "Philip Hands" to be "verysecret"',
	'I select full-disk, single-filesystem partitioning',
	'I note that the Base system is being installed',
	'I accept the default mirror',
	'I ignore Popcon',
	'we reach the Tasksel prompt',
  ]
end

def firstboot_steps
  [
	'I wait while the bulk of the packages are installed',
	'I install GRUB',
	'I allow reboot after the install is complete',
	'I wait for the reboot',
	'I power off the computer',
	'the computer is set to boot from ide drive "'+JOB_NAME+'"',
  ]
end

def checkpoints
  {
    'boot-d-i-to-tasksel' => {
      :description => "I have started Debian Installer and stopped at the Tasksel prompt",
      :parent_checkpoint => nil,
      :steps => [
    'I intend to use text mode',
      ] + tasksel_steps
    },

    'boot-g-i-to-tasksel' => {
      :description => "I have started GUI Debian Installer and stopped at the Tasksel prompt",
      :parent_checkpoint => nil,
      :steps => [
    'I intend to use gui mode',
      ] + tasksel_steps
    },

    'debian-console-install' => {
      :description => "I install a non-GUI Debian system, in text mode",
      :parent_checkpoint => 'boot-d-i-to-tasksel',
      :steps => [
    'I intend to use text mode',
	'I unset the Desktop task',
      ] + firstboot_steps,
    },

    'debian-gui-console-install' => {
      :description => "I install a non-GUI Debian system, in gui mode",
      :parent_checkpoint => 'boot-g-i-to-tasksel',
      :steps => [
    'I intend to use gui mode',
	'I unset the Desktop task',
      ] + firstboot_steps,
    },

    'debian-minimal-install' => {
      :description => "I install a Minimal Debian system, in text mode",
      :parent_checkpoint => 'boot-d-i-to-tasksel',
      :steps => [
    'I intend to use text mode',
	'I unset the Desktop and Print tasks',
      ] + firstboot_steps,
    },

    'debian-gui-minimal-install' => {
      :description => "I install a Minimal Debian system, in gui mode",
      :parent_checkpoint => 'boot-g-i-to-tasksel',
      :steps => [
    'I intend to use gui mode',
	'I unset the Desktop and Print tasks',
      ] + firstboot_steps,
    },

    'debian-gnome-install' => {
      :description => "I install a Gnome Desktop Debian system, in text mode",
      :parent_checkpoint => 'boot-d-i-to-tasksel',
      :steps => [
    'I intend to use text mode',
	'I select the Gnome Desktop task',
      ] + firstboot_steps,
    },

    'debian-gui-gnome-install' => {
      :description => "I install a Gnome Desktop Debian system, in gui mode",
      :parent_checkpoint => 'boot-g-i-to-tasksel',
      :steps => [
    'I intend to use gui mode',
	'I select the Gnome Desktop task',
      ] + firstboot_steps,
    },

    'debian-xfce-install' => {
      :description => "I install a XFCE Desktop Debian system, in text mode",
      :parent_checkpoint => 'boot-d-i-to-tasksel',
      :steps => [
    'I intend to use text mode',
	'I select the XFCE Desktop task',
      ] + firstboot_steps,
    },

    'debian-gui-xfce-install' => {
      :description => "I install a XFCE Desktop Debian system, in gui mode",
      :parent_checkpoint => 'boot-g-i-to-tasksel',
      :steps => [
    'I intend to use gui mode',
	'I select the XFCE Desktop task',
      ] + firstboot_steps,
    },

    'debian-gui-lxde-install' => {
      :description => "I install a LXDE Desktop Debian system, in gui mode",
      :parent_checkpoint => 'boot-g-i-to-tasksel',
      :steps => [
    'I intend to use gui mode',
	'I select the LXDE Desktop task',
      ] + firstboot_steps,
    },

    'debian-gui-kde-install' => {
      :description => "I install a KDE Desktop Debian system, in gui mode",
      :parent_checkpoint => 'boot-g-i-to-tasksel',
      :steps => [
    'I intend to use gui mode',
	'I select the KDE Desktop task',
      ] + firstboot_steps,
    },

  }
end

def live_screenshot()
  debug_log("debug: publishing live screenshot.")
  screen_capture = @screen.capture
  p = screen_capture.getFilename
  if File.exist?(p)
    s = ENV['WORKSPACE']
    s_path = "#{s}/screenshot.png"
    FileUtils.mv(p, s_path)
    convert = IO.popen(['convert',
                      s_path, '-adaptive-resize', '128x96', "#{s}/screenshot-thumb.png",
                      :err => ['/dev/null', 'w'],
                     ])
  end
end

def reach_checkpoint(name)
  scenario_indent = " "*4
  step_indent = " "*6

  step "a computer"
  live_screenshot
  if VM.snapshot_exists?(name)
    $vm.restore_snapshot(name)
    post_snapshot_restore_hook
  else
    checkpoint = checkpoints[name]
    checkpoint_description = checkpoint[:description]
    parent_checkpoint = checkpoint[:parent_checkpoint]
    steps = checkpoint[:steps]
    if parent_checkpoint
      if VM.snapshot_exists?(parent_checkpoint)
        $vm.restore_snapshot(parent_checkpoint)
      else
        reach_checkpoint(parent_checkpoint)
      end
      post_snapshot_restore_hook
    end
    debug_log(scenario_indent + "Checkpoint: #{checkpoint_description}",
              :color => :white)
    step_action = "Given"
    if parent_checkpoint
      parent_description = checkpoints[parent_checkpoint][:description]
      debug_log(step_indent + "#{step_action} #{parent_description}",
                :color => :green)
      step_action = "And"
    end
    steps.each do |s|
      begin
        step(s)
      rescue Exception => e
        debug_log(scenario_indent +
                  "Step failed while creating checkpoint: #{s}",
                  :color => :red)
        live_screenshot
        raise e
      end
      live_screenshot
      debug_log(step_indent + "#{step_action} #{s}", :color => :green)
      step_action = "And"
    end
    $vm.save_snapshot(name)
  end
  live_screenshot
end

# For each checkpoint we generate a step to reach it.
checkpoints.each do |name, desc|
  step_regex = Regexp.new("^#{Regexp.escape(desc[:description])}$")
  Given step_regex do
    reach_checkpoint(name)
  end
end