module Celluloid

# File lib/celluloid.rb, line 24
def actor?
  !!Thread.current[:celluloid_actor]
end

# File lib/celluloid.rb, line 68
def boot
  internal_pool.reset
  Celluloid::Notifications::Fanout.supervise_as :notifications_fanout
  Celluloid::IncidentReporter.supervise_as :default_incident_reporter, STDERR
end

# File lib/celluloid.rb, line 39
def cores
 CPUCounter.cores
end

# File lib/celluloid.rb, line 52
def exception_handler(&block)
  Logger.exception_handler(&block)
end

# File lib/celluloid.rb, line 18
def included(klass)
  klass.send :extend,  ClassMethods
  klass.send :include, InstanceMethods
end

# File lib/celluloid.rb, line 29
def mailbox
  Thread.current[:celluloid_mailbox] ||= Celluloid::Mailbox.new
end

# File lib/celluloid.rb, line 74
def register_shutdown
  return if @shutdown_registered
  # Terminate all actors at exit
  at_exit do
    if defined?(RUBY_ENGINE) && RUBY_ENGINE == "ruby" && RUBY_VERSION >= "1.9"
      # workaround for MRI bug losing exit status in at_exit block
      # http://bugs.ruby-lang.org/issues/5218
      exit_status = $!.status if $!.is_a?(SystemExit)
      Celluloid.shutdown
      exit exit_status if exit_status
    else
      Celluloid.shutdown
    end
  end
  @shutdown_registered = true
end

# File lib/celluloid.rb, line 92
def shutdown
  Timeout.timeout(shutdown_timeout) do
    internal_pool.shutdown

    actors = Actor.all
    Logger.debug "Terminating #{actors.size} actors..." if actors.size > 0

    # Attempt to shut down the supervision tree, if available
    Supervisor.root.terminate if Supervisor.root

    # Actors cannot self-terminate, you must do it for them
    actors.each do |actor|
      begin
        actor.terminate!
      rescue DeadActorError, MailboxError
      end
    end

    actors.each do |actor|
      begin
        Actor.join(actor)
      rescue DeadActorError, MailboxError
      end
    end

    Logger.debug "Shutdown completed cleanly"
  end
rescue Timeout::Error
  Logger.error("Couldn't cleanly terminate all actors in #{shutdown_timeout} seconds!")
end

# File lib/celluloid.rb, line 46
def stack_dump(output = STDERR)
  Celluloid::StackDump.new.dump(output)
end

# File lib/celluloid.rb, line 56
def suspend(status, waiter)
  task = Thread.current[:celluloid_task]
  if task && !Celluloid.exclusive?
    waiter.before_suspend(task) if waiter.respond_to?(:before_suspend)
    Task.suspend(status)
  else
    waiter.wait
  end
end

# File lib/celluloid.rb, line 34
def uuid
  UUID.generate
end

# File lib/celluloid/version.rb, line 3
def self.version; VERSION; end

# File lib/celluloid.rb, line 355
def abort(cause)
  cause = case cause
    when String then RuntimeError.new(cause)
    when Exception then cause
    else raise TypeError, "Exception object/String expected, but #{cause.class} received"
  end
  raise AbortError.new(cause)
end

# File lib/celluloid.rb, line 462
def after(interval, &block)
  Thread.current[:celluloid_actor].after(interval, &block)
end

# File lib/celluloid.rb, line 481
def async(meth = nil, *args, &block)
  Thread.current[:celluloid_actor].proxy.async meth, *args, &block
end

# File lib/celluloid.rb, line 385
def call_chain_id
  Thread.current[:celluloid_chain_id]
end

# File lib/celluloid.rb, line 380
def current_actor
  Actor.current
end

# File lib/celluloid.rb, line 474
def defer(&block)
  # This implementation relies on the present implementation of
  # Celluloid::Future, which uses a thread from InternalPool to run the block
  Future.new(&block).value
end

# File lib/celluloid.rb, line 467
def every(interval, &block)
  Thread.current[:celluloid_actor].every(interval, &block)
end

# File lib/celluloid.rb, line 451
def exclusive(&block)
  Thread.current[:celluloid_actor].exclusive(&block)
end

# File lib/celluloid.rb, line 456
def exclusive?
  actor = Thread.current[:celluloid_actor]
  actor && actor.exclusive?
end

# File lib/celluloid.rb, line 486
def future(meth = nil, *args, &block)
  Thread.current[:celluloid_actor].proxy.future meth, *args, &block
end

# File lib/celluloid.rb, line 410
def link(actor)
  Actor.link(actor)
end

# File lib/celluloid.rb, line 425
def linked_to?(actor)
  Actor.linked_to?(actor)
end

# File lib/celluloid.rb, line 395
def links
  Thread.current[:celluloid_actor].links
end

# File lib/celluloid.rb, line 400
def monitor(actor)
  Actor.monitor(actor)
end

# File lib/celluloid.rb, line 420
def monitoring?(actor)
  Actor.monitoring?(actor)
end

# File lib/celluloid.rb, line 430
def receive(timeout = nil, &block)
  actor = Thread.current[:celluloid_actor]
  if actor
    actor.receive(timeout, &block)
  else
    Celluloid.mailbox.receive(timeout, &block)
  end
end

# File lib/celluloid.rb, line 370
def signal(name, value = nil)
  Thread.current[:celluloid_actor].signal name, value
end

# File lib/celluloid.rb, line 440
def sleep(interval)
  actor = Thread.current[:celluloid_actor]
  if actor
    actor.sleep(interval)
  else
    Kernel.sleep interval
  end
end

# File lib/celluloid.rb, line 390
def tasks
  Thread.current[:celluloid_actor].tasks.to_a
end

# File lib/celluloid.rb, line 365
def terminate
  Thread.current[:celluloid_actor].terminate
end

# File lib/celluloid.rb, line 415
def unlink(actor)
  Actor.unlink(actor)
end

# File lib/celluloid.rb, line 405
def unmonitor(actor)
  Actor.unmonitor(actor)
end

# File lib/celluloid.rb, line 375
def wait(name)
  Thread.current[:celluloid_actor].wait name
end