require 'ztk'

LAB_ACTION_ORDER = %W(build demolish recycle bounce create destroy up down provision deprovision).map(&:to_sym)

LAB_ACTIONS = {
  :create      => ["Initialize %s",   "Attempts to create the <%= @component %>."],
  :destroy     => ["Terminate %s",    "Attempts to destroy the <%= @component %>."],
  :up          => ["On-line %s",      "Attempts to online the <%= @component %>."],
  :down        => ["Off-line %s",     "Attempts to offline the <%= @component %>."],
  :provision   => ["Provision %s",    "Attempts to provision the <%= @component %>."],
  :deprovision => ["De-provision %s", "Attempts to deprovision the <%= @component %>."],
  :bounce      => ["Bounce %s (down->up)", <<-EOF],
    Attempts to bounce the <%= @component %>.  TestLab will attempt to offline, then online the <%= @component %>.

    The <%= @component %> are taken through the following states:

    Current -> Down -> Up
    EOF
  :recycle     => ["Recycle %s (demolish->build)", <<-EOF],
    Attempts to recycle the <%= @component %>.  TestLab will attempt to demolish, then build the <%= @component %>.

    The <%= @component %> are taken through the following states:

    Current -> Demolish -> Build
    EOF
  :build       => ["Build %s (create->up->provision)", <<-EOF],
    Attempts to build the <%= @component %>.  TestLab will attempt to create, online and provision the <%= @component %>.

    The <%= @component %> are taken through the following states:

    Current -> Create -> Up -> Provision
    EOF
  :demolish    => ["Demolish %s (deprovision->down->destroy)", <<-EOF]
    Attempts to demolish the <%= @component %>.  TestLab will attempt to deprovision, offline and destroy the <%= @component %>.

    The <%= @component %> are taken through the following states:

    Current -> Deprovision -> Down -> Destroy
    EOF
}

def build_lab_commands(component, klass, &block)
  desc %(Manage lab #{component}s)
  command component do |c|
    c.desc %(Optional #{component} ID or comma separated list of #{component} IDs)
    c.arg_name %(#{component}[,#{component},...])
    c.flag [:n, :name]

    c.desc %(Force the actions verbatium, do not attempt to infer shortcuts; this has no effect for most operations)
    c.switch [:f, :force]

    LAB_ACTION_ORDER.each do |lab_action|
      action_desc = LAB_ACTIONS[lab_action]
      c.desc(action_desc.first % "#{component}s")
      c.long_desc(ZTK::Template.string(action_desc.last, {:component => "#{component}s"}))

      c.command lab_action do |la|

        la.action do |global_options, options, args|
          objects  = iterate_objects_by_name(options[:name], klass)

          @testlab.do_parallel_actions(klass, objects, lab_action) do |object, action, klass|
            send_lab_action(object, options, action)
          end
        end

      end
    end

    !block.nil? and block.call(c)
  end
end

def send_lab_action(object, options, lab_action)
  if %w( build recycle ).map(&:to_sym).include?(lab_action)
    object.send(lab_action, options[:force])
  else
    object.send(lab_action)
  end
end

def iterate_objects_by_name(name, klass, &block)
  objects = Array.new
  klass_name = klass.to_s.split('::').last.downcase

  objects = klass.all.select{ |o| (!o.template rescue true) }
  if !name.nil?
    names = name.split(',')

    not_names = names.collect { |n| ((n[0] == '!') and n[1..-1]) }.delete_if {|n| n == false }.compact
    names.delete_if { |n| n[0] == '!' }

    if names.size > 0
      objects = klass.find(names).select{ |o| (!o.template rescue true) }
    end
    objects.delete_if{ |o| not_names.include?(o.id.to_s.downcase) }
  end

  (objects.nil? || (objects.count == 0)) and raise TestLab::TestLabError, "We could not find any of the #{klass_name}s you supplied!"

  objects.each do |object|
    !block.nil? and block.call(object)
  end

  objects
end