WIP

Author: ioquatix

Steepfile

@@ -0,0 +1,6 @@
+D = Steep::Diagnostic
+
+target :lib do
+  signature "sig"
+  check "lib"
+end

gems.rb

@@ -34,6 +34,8 @@
 	gem "rubocop"
 	gem "rubocop-socketry"
 
+	gem "steep", git: "https://github.com/soutaro/steep"
+	
 	gem "sus-fixtures-async"
 	gem "sus-fixtures-console"
 	gem "sus-fixtures-time"

lib/async/barrier.rb

@@ -63,9 +63,9 @@ def empty?
 		# Wait for all tasks to complete by invoking {Task#wait} on each waiting task, which may raise an error. As long as the task has completed, it will be removed from the barrier.
 		#
 		# @yields {|task| ...} If a block is given, the unwaited task is yielded. You must invoke {Task#wait} yourself. In addition, you may `break` if you have captured enough results.
+		# 	@parameter task [Task] The task which has completed.
 		#
 		# @asynchronous Will wait for tasks to finish executing.
-		# @rbs () ?{(Task) -> untyped} -> void
 		def wait
 			while !@tasks.empty?
 				# Wait for a task to finish (we get the task node):

lib/async/idler.rb

@@ -12,7 +12,7 @@ class Idler
 		#
 		# @parameter maximum_load [Numeric] The maximum load before we start shedding work.
 		# @parameter backoff [Numeric] The initial backoff time, used for delaying work.
-		# @parameter parent [Interface(Asyncable) | Nil] The parent task to use for async operations.
+		# @parameter parent [_Asyncable | Nil] The parent task to use for async operations.
 		def initialize(maximum_load = 0.8, backoff: 0.01, parent: nil)
 			@maximum_load = maximum_load
 			@backoff = backoff
@@ -24,7 +24,7 @@ def initialize(maximum_load = 0.8, backoff: 0.01, parent: nil)
 		# @asynchronous Executes the given block concurrently.
 		#
 		# @parameter arguments [Array] The arguments to pass to the block.
-		# @parameter parent [Interface(Asyncable) | Nil] The parent task to use for async operations.
+		# @parameter parent [_Asyncable] The parent task to use for async operations.
 		# @parameter options [Hash] The options to pass to the task.
 		# @yields {|task| ...} When the system is idle, the block will be executed in a new task.
 		def async(*arguments, parent: (@parent or Task.current), **options, &block)

lib/async/list.rb

@@ -83,6 +83,7 @@ def prepend(node)
 
 		# Add the node, yield, and the remove the node.
 		# @yields {|node| ...} Yields the node.
+		# 	@parameter node [Node] The node to add to the list.
 		# @returns [Object] Returns the result of the block.
 		def stack(node, &block)
 			append(node)

lib/async/notification.rb

@@ -11,7 +11,7 @@ module Async
 	class Notification < Condition
 		# Signal to a given task that it should resume operations.
 		#
-		# @returns [Boolean] if a task was signalled.
+		# @returns [bool] If a task was signalled.
 		def signal(value = nil, task: Task.current)
 			return false if @waiting.empty?
 
@@ -21,14 +21,19 @@ def signal(value = nil, task: Task.current)
 		end
 
 		Signal = Struct.new(:waiting, :value) do
+			# @returns [bool] Returns true if the signal is still alive.
 			def alive?
 				true
 			end
 
+			# Transfer the value to all waiting fibers.
+			# @returns [self]
 			def transfer
 				waiting.each do |fiber|
 					fiber.transfer(value) if fiber.alive?
 				end
+				
+				return self
 			end
 		end
 

lib/async/queue.rb

@@ -23,7 +23,7 @@ class ClosedError < RuntimeError
 
 		# Create a new queue.
 		#
-		# @parameter parent [Interface(Asyncable) | Nil] The parent task to use for async operations.
+		# @parameter parent [_Asyncable | Nil] The parent task to use for async operations.
 		# @parameter available [Notification] The notification to use for signaling when items are available.
 		def initialize(parent: nil, available: Notification.new)
 			@items = []
@@ -104,7 +104,7 @@ def pop
 		# @asynchronous Executes the given block concurrently for each item.
 		#
 		# @parameter arguments [Array] The arguments to pass to the block.
-		# @parameter parent [Interface(Asyncable) | Nil] The parent task to use for async operations.
+		# @parameter parent [_Asyncable | Nil] The parent task to use for async operations.
 		# @parameter options [Hash] The options to pass to the task.
 		# @yields {|task| ...} When the system is idle, the block will be executed in a new task.
 		def async(parent: (@parent or Task.current), **options, &block)

lib/async/task.rb

@@ -225,7 +225,7 @@ def run(*arguments)
 		# @yields {|task| ...} in the context of the new task.
 		# @raises [FinishedError] If the task has already finished.
 		# @returns [Task] The child task.
-		# @rbs [ResultType] (*untyped, **untyped) { () -> ResultType } -> Task[ResultType]
+		# @rbs [T] (*untyped, **untyped) { (Task[T], *untyped, **untyped) -> T } -> Task[T]
 		def async(*arguments, **options, &block)
 			raise FinishedError if self.finished?
 
@@ -370,7 +370,7 @@ def self.current
 		end
 
 		# Check if there is a task defined for the current fiber.
-		# @returns [Interface(Asyncable) | Nil]
+		# @returns [_Asyncable | Nil]
 		def self.current?
 			Fiber.current.async_task
 		end

lib/async/variable.rb

@@ -7,6 +7,7 @@
 
 module Async
 	# A synchronization primitive that allows one task to wait for another task to resolve a value.
+	# @rbs generic T
 	class Variable
 		# Create a new variable.
 		#
@@ -22,13 +23,14 @@ def initialize(condition = Condition.new)
 		#
 		# @parameter value [Object] The value to resolve.
 		def resolve(value = true)
-			@value = value
-			condition = @condition
-			@condition = nil
-			
-			self.freeze
-			
-			condition.signal(value)
+			if condition = @condition
+				@value = value
+				@condition = nil
+				
+				self.freeze
+				
+				condition.signal(value)
+			end
 		end
 
 		# Alias for {#resolve}.
@@ -45,7 +47,7 @@ def resolved?
 
 		# Wait for the value to be resolved.
 		#
-		# @returns [Object] The resolved value.
+		# @returns [T?] The resolved value.
 		def wait
 			@condition&.wait
 			return @value

lib/async/waiter.rb

@@ -10,7 +10,7 @@ module Async
 	class Waiter
 		# Create a waiter instance.
 		#
-		# @parameter parent [Interface(Asyncable) | Nil] The parent task to use for asynchronous operations.
+		# @parameter parent [_Asyncable | Nil] The parent task to use for asynchronous operations.
 		# @parameter finished [::Async::Condition] The condition to signal when a task completes.
 		def initialize(parent: nil, finished: Async::Condition.new)
 			warn("`Async::Waiter` is deprecated, use `Async::Barrier` instead.", uplevel: 1, category: :deprecated) if $VERBOSE