From 617cf380695fe957798aee5c7f3a96e19f46712a Mon Sep 17 00:00:00 2001
From: Amr Bashir <amr.bashir2015@gmail.com>
Date: Tue, 13 Jun 2023 15:47:55 +0300
Subject: [PATCH] fix(window-state): propagate promise (#435)

closes #432
---
 .changes/window-state-promise.md       |  5 +++++
 plugins/window-state/guest-js/index.ts | 15 +++++++++------
 2 files changed, 14 insertions(+), 6 deletions(-)
 create mode 100644 .changes/window-state-promise.md

diff --git a/.changes/window-state-promise.md b/.changes/window-state-promise.md
new file mode 100644
index 00000000..f97c43df
--- /dev/null
+++ b/.changes/window-state-promise.md
@@ -0,0 +1,5 @@
+---
+"window-state-js": "patch"
+---
+
+Correctly propagate the promise inside `saveWindowState`, `restoreState` and `restoreStateCurrent` so callers can choose to `await` them.
diff --git a/plugins/window-state/guest-js/index.ts b/plugins/window-state/guest-js/index.ts
index 11ead0f4..0cc32649 100644
--- a/plugins/window-state/guest-js/index.ts
+++ b/plugins/window-state/guest-js/index.ts
@@ -14,22 +14,25 @@ export enum StateFlags {
 /**
  *  Save the state of all open windows to disk.
  */
-async function saveWindowState(flags: StateFlags) {
-  invoke("plugin:window-state|save_window_state", { flags });
+async function saveWindowState(flags: StateFlags): Promise<void> {
+  return invoke("plugin:window-state|save_window_state", { flags });
 }
 
 /**
  *  Restore the state for the specified window from disk.
  */
-async function restoreState(label: WindowLabel, flags: StateFlags) {
-  invoke("plugin:window-state|restore_state", { label, flags });
+async function restoreState(
+  label: WindowLabel,
+  flags: StateFlags
+): Promise<void> {
+  return invoke("plugin:window-state|restore_state", { label, flags });
 }
 
 /**
  *  Restore the state for the current window from disk.
  */
-async function restoreStateCurrent(flags: StateFlags) {
-  restoreState(getCurrent().label, flags);
+async function restoreStateCurrent(flags: StateFlags): Promise<void> {
+  return restoreState(getCurrent().label, flags);
 }
 
 export { restoreState, restoreStateCurrent, saveWindowState };