Merge pull request #31 from hexedtech/feat/lua-promise-abort

feat(lua): cancel promise

dist/lua/annotations.lua

@@ -18,6 +18,9 @@ local NilPromise = {}
 --- block until promise is ready
 function NilPromise:await() end
 
+--- cancel promise execution
+function NilPromise:cancel() end
+
 ---@param cb fun() callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function NilPromise:and_then(cb) end
@@ -30,6 +33,9 @@ local StringPromise = {}
 --- @return string
 function StringPromise:await() end
 
+--- cancel promise execution
+function StringPromise:cancel() end
+
 ---@param cb fun(x: string) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function StringPromise:and_then(cb) end
@@ -40,6 +46,8 @@ local StringArrayPromise = {}
 --- block until promise is ready and return value
 --- @return string[]
 function StringArrayPromise:await() end
+--- cancel promise execution
+function StringArrayPromise:cancel() end
 ---@param cb fun(x: string[]) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function StringArrayPromise:and_then(cb) end
@@ -50,6 +58,8 @@ local ClientPromise = {}
 --- block until promise is ready and return value
 --- @return Client
 function ClientPromise:await() end
+--- cancel promise execution
+function ClientPromise:cancel() end
 ---@param cb fun(x: Client) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function ClientPromise:and_then(cb) end
@@ -60,6 +70,8 @@ local WorkspacePromise = {}
 --- block until promise is ready and return value
 --- @return Workspace
 function WorkspacePromise:await() end
+--- cancel promise execution
+function WorkspacePromise:cancel() end
 ---@param cb fun(x: Workspace) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function WorkspacePromise:and_then(cb) end
@@ -70,6 +82,8 @@ local WorkspaceEventPromise = {}
 --- block until promise is ready and return value
 --- @return WorkspaceEvent
 function WorkspaceEventPromise:await() end
+--- cancel promise execution
+function WorkspaceEventPromise:cancel() end
 ---@param cb fun(x: WorkspaceEvent) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function WorkspaceEventPromise:and_then(cb) end
@@ -80,6 +94,8 @@ local BufferControllerPromise = {}
 --- block until promise is ready and return value
 --- @return BufferController
 function BufferControllerPromise:await() end
+--- cancel promise execution
+function BufferControllerPromise:cancel() end
 ---@param cb fun(x: BufferController) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function BufferControllerPromise:and_then(cb) end
@@ -90,6 +106,8 @@ local CursorPromise = {}
 --- block until promise is ready and return value
 --- @return Cursor
 function CursorPromise:await() end
+--- cancel promise execution
+function CursorPromise:cancel() end
 ---@param cb fun(x: Cursor) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function CursorPromise:and_then(cb) end
@@ -100,6 +118,8 @@ local MaybeCursorPromise = {}
 --- block until promise is ready and return value
 --- @return Cursor | nil
 function MaybeCursorPromise:await() end
+--- cancel promise execution
+function MaybeCursorPromise:cancel() end
 ---@param cb fun(x: Cursor | nil) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function MaybeCursorPromise:and_then(cb) end
@@ -110,6 +130,8 @@ local TextChangePromise = {}
 --- block until promise is ready and return value
 --- @return TextChange
 function TextChangePromise:await() end
+--- cancel promise execution
+function TextChangePromise:cancel() end
 ---@param cb fun(x: TextChange) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function TextChangePromise:and_then(cb) end
@@ -120,6 +142,8 @@ local MaybeTextChangePromise = {}
 --- block until promise is ready and return value
 --- @return TextChange | nil
 function MaybeTextChangePromise:await() end
+--- cancel promise execution
+function MaybeTextChangePromise:cancel() end
 ---@param cb fun(x: TextChange | nil) callback to invoke
 ---invoke callback asynchronously as soon as promise is ready
 function MaybeTextChangePromise:and_then(cb) end

src/ffi/lua/ext/a_sync.rs

@@ -49,6 +49,13 @@ impl LuaUserData for Promise {
 			None => Err(LuaError::runtime("Promise already awaited")),
 			Some(x) => tokio().block_on(x).map_err(LuaError::runtime)?,
 		});
+		methods.add_method_mut("cancel", |_, this, ()| match this.0.take() {
+			None => Err(LuaError::runtime("Promise already awaited")),
+			Some(x) => {
+				x.abort();
+				Ok(())
+			},
+		});
 		methods.add_method_mut("and_then", |_, this, (cb,): (LuaFunction,)| {
 			match this.0.take() {
 				None => Err(LuaError::runtime("Promise already awaited")),