# give super-dwarven speed to an unit

$superdwarf_onupdate ||= nil
$superdwarf_ids ||= []

def unregister
    unless $superdwarf_onupdate.nil?
        df.onupdate_unregister($superdwarf_onupdate)
        $superdwarf_onupdate = nil
    end
end

case $script_args[0]
when 'add'
	if u = df.unit_find
		$superdwarf_ids |= [u.id]

		$superdwarf_onupdate ||= df.onupdate_register('superdwarf', 1) {
			if $superdwarf_ids.empty?
				unregister()
			else
				$superdwarf_ids.each { |id|
					if u = df.unit_find(id) and not u.flags1.dead
						u.actions.each { |a|
							case a.type
							when :Move
								a.data.move.timer = 1
							when :Climb
								a.data.climb.timer = 1
							when :Job
								a.data.job.timer = 1
							when :Job2
								a.data.job2.timer = 1
							when :Attack
                                # Attack execution timer; fires when reaches zero.
								a.data.attack.timer1 = 1
                                # Attack completion timer: finishes action at zero.
                                # An action must complete before target re-seleciton
                                # occurs.
								a.data.attack.timer2 = 0
							end
						}

						# no sleep
						if u.counters2.sleepiness_timer > 10000
							u.counters2.sleepiness_timer = 1
						end

						# no break
						if b = u.status.misc_traits.find { |t| t.id == :OnBreak }
							b.value = 500_000
						end
					else
						$superdwarf_ids.delete id
					end
				}
			end
		}
	else
		puts "Select a creature using 'v'"
	end

when 'del'
	if u = df.unit_find
		$superdwarf_ids.delete u.id
	else
		puts "Select a creature using 'v'"
	end

    if $superdwarf_ids.empty?
        unregister()
    end

when 'clear'
	$superdwarf_ids.clear
    unregister()

when 'list'
	puts "current superdwarves:", $superdwarf_ids.map { |id| df.unit_find(id).name }

else
	puts "Usage:",
		" - superdwarf add: give superspeed to currently selected creature",
		" - superdwarf del: remove superspeed to current creature",
		" - superdwarf clear: remove all superpowers",
		" - superdwarf list: list super-dwarves"
end