class ThreadOrder

# File lib/thread_order.rb, line 9
def initialize
  @mutex   = Mutex.new
  @bodies  = {}
  @threads = []
  @queue   = [] # Queue is in stdlib, but half the purpose of this lib is to avoid such deps, so using an array in a Mutex
  @worker  = Thread.new do
    Thread.current.abort_on_exception = true
    Thread.current[:thread_order_name] = :internal_worker
    loop { break if :shutdown == work() }
  end
end

# File lib/thread_order.rb, line 67
def apocalypse!(thread_method=:kill)
  enqueue do
    @threads.each(&thread_method)
    @queue.clear
    :shutdown
  end
  @worker.join
end

# File lib/thread_order.rb, line 25
def current
  Thread.current[:thread_order_name]
end

# File lib/thread_order.rb, line 21
def declare(name, &block)
  sync { @bodies[name] = block }
end

# File lib/thread_order.rb, line 76
def enqueue(&block)
  sync { @queue << block if @worker.alive? }
end

# File lib/thread_order.rb, line 63
def join_all
  sync { @threads }.each { |th| th.join }
end

# File lib/thread_order.rb, line 29
def pass_to(name, options={})
  child        = nil
  parent       = Thread.current
  resume_event = extract_resume_event!(options)
  enqueue do
    sync do
      @threads << Thread.new {
        child = Thread.current
        child[:thread_order_name] = name
        body = sync { @bodies.fetch(name) }
        wait_until { parent.stop? }
        :run == resume_event && parent.wakeup
        wake_on_sleep = lambda do
          child.status == 'sleep' ? parent.wakeup :
          child.status == nil     ? :noop         :
          child.status == false   ? parent.raise(CannotResume.new "#{name} exited instead of sleeping") :
                                    enqueue(&wake_on_sleep)
        end
        :sleep == resume_event && enqueue(&wake_on_sleep)
        begin
          body.call parent
        rescue Exception => e
          enqueue { parent.raise e }
          raise
        ensure
          :exit == resume_event && enqueue { parent.wakeup }
        end
      }
    end
  end
  sleep
  child
end

# File lib/thread_order.rb, line 80
def wait_until(&condition)
  return if condition.call
  thread = Thread.current
  wake_when_true = lambda do
    if thread.stop? && condition.call
      thread.wakeup
    else
      enqueue(&wake_when_true)
    end
  end
  enqueue(&wake_when_true)
  sleep
end

# File lib/thread_order.rb, line 106
def extract_resume_event!(options)
  resume_on = options.delete :resume_on
  options.any? &&
    raise(ArgumentError, "Unknown options: #{options.inspect}")
  resume_on && ![:run, :exit, :sleep, nil].include?(resume_on) and
    raise(ArgumentError, "Unknown status: #{resume_on.inspect}")
  resume_on || :none
end

# File lib/thread_order.rb, line 96
def sync(&block)
  @mutex.synchronize(&block)
end

# File lib/thread_order.rb, line 100
def work
  task = sync { @queue.shift }
  task ||= lambda { Thread.pass }
  task.call
end