class Async::Task

def self.current
	Thread.current[:async_task] or raise RuntimeError, "No async task available!"
end

def self.current?
	Thread.current[:async_task]
end

def self.yield
	if block_given?
		result = yield
	else
		result = Fiber.yield
	end
	
	if result.is_a? Exception
		raise result
	else
		return result
	end
end

def alive?
	@fiber&.alive?
end

def async(*arguments, **options, &block)
	task = Task.new(@reactor, self, **options, &block)
	
	task.run(*arguments)
	
	return task
end

def complete?
	@status == :complete
end

def current?
	self.equal?(Thread.current[:async_task])
end

def fail!(exception = nil, propagate = true)
	@status = :failed
	@result = exception
	
	if propagate
		raise
	elsif @finished.nil?
		# If no one has called wait, we log this as an error:
		logger.error(self) {$!}
	else
		logger.debug(self) {$!}
	end
end

def failed?
	@status == :failed
end

def finish!
	# Allow the fiber to be recycled.
	@fiber = nil
	
	# Attempt to remove this node from the task tree.
	consume
	
	# If this task was being used as a future, signal completion here:
	if @finished
		@finished.signal(@result)
	end
end

def finished?
	super && @status != :running
end

def initialize(reactor, parent = Task.current?, logger: nil, &block)
	super(parent || reactor)
	
	@reactor = reactor
	
	@status = :initialized
	@result = nil
	@finished = nil
	
	@logger = logger
	
	@fiber = make_fiber(&block)
end

def logger
	@logger ||= @parent&.logger
end

def make_fiber(&block)
	Fiber.new do |*arguments|
		set!
		
		begin
			@result = yield(self, *arguments)
			@status = :complete
			# logger.debug(self) {"Task was completed with #{@children.size} children!"}
		rescue Stop
			stop!
		rescue StandardError => error
			fail!(error, false)
		rescue Exception => exception
			fail!(exception, true)
		ensure
			# logger.debug(self) {"Task ensure $!=#{$!} with #{@children.size} children!"}
			finish!
		end
	end
end

def run(*arguments)
	if @status == :initialized
		@status = :running
		
		@fiber.resume(*arguments)
	else
		raise RuntimeError, "Task already running!"
	end
end

def running?
	@status == :running
end

def set!
	# This is actually fiber-local:
	Thread.current[:async_task] = self
end

def stop(later = false)
	if self.stopped?
		# If we already stopped this task... don't try to stop it again:
		return
	end
	
	if self.running?
		if self.current?
			if later
				@reactor << Stop::Later.new(self)
			else
				raise Stop, "Stopping current task!"
			end
		elsif @fiber&.alive?
			begin
				@fiber.resume(Stop.new)
			rescue FiberError
				@reactor << Stop::Later.new(self)
			end
		end
	else
		# We are not running, but children might be, so transition directly into stopped state:
		stop!
	end
end

def stop!
	# logger.debug(self) {"Task was stopped with #{@children&.size.inspect} children!"}
	@status = :stopped
	
	@children&.each do |child|
		child.stop(true)
	end
end

def stopped?
	@status == :stopped
end

def stopping?
	@status == :stopping
end

def to_s
	"\#<#{self.description} (#{@status})>"
end

def wait
	raise RuntimeError, "Cannot wait on own fiber" if Fiber.current.equal?(@fiber)
	
	if running?
		@finished ||= Condition.new
		@finished.wait
	else
		Task.yield{@result}
	end
end

def yield
	Task.yield{reactor.yield}
end