feat: expose tasks.current_task

See #4
nvim-neotest · Jan 21, 2024 · f12c3b1 · f12c3b1
1 parent b789a35
commit f12c3b1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/doc/nio.txt b/doc/nio.txt
@@ -96,6 +96,13 @@ Parameters~
 Return~
 `(any[])` Results of all functions
 
+                                                            *nio.current_task()*
+`current_task`()
+
+Get the current running task, if any.
+Return~
+`(nio.tasks.Task|nil)`
+
                                                                    *nio.first()*
 `first`({functions})
 

diff --git a/lua/nio/init.lua b/lua/nio/init.lua
@@ -121,6 +121,12 @@ function nio.gather(functions)
   return sorted
 end
 
+--- Get the current running task, if any.
+---@return nio.tasks.Task|nil
+function nio.current_task()
+  return tasks.current_task()
+end
+
 --- Run a collection of async functions concurrently and return the result of
 --- the first to finish.
 ---@async