From 0e064d0613c9af951af499aadca67d00a337a408 Mon Sep 17 00:00:00 2001
From: BluJ <mogulslayer@gmail.com>
Date: Thu, 4 May 2023 16:10:14 -0600
Subject: [PATCH] chore: Add keys

---
 lib/util/index.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/lib/util/index.ts b/lib/util/index.ts
index 6fe40dd..296fb00 100644
--- a/lib/util/index.ts
+++ b/lib/util/index.ts
@@ -48,7 +48,7 @@ export type Utils<WD, WrapperOverWrite = { const: never }> = {
     key: string
     value: string | null | undefined
     generator: DefaultString
-  }) => Promise<void>
+  }) => Promise<null | string>
   readFile: <A>(fileHelper: FileHelper<A>) => ReturnType<FileHelper<A>["read"]>
   writeFile: <A>(
     fileHelper: FileHelper<A>,
@@ -102,12 +102,14 @@ export const utils = <WrapperData = never, WrapperOverWrite = { const: never }>(
   }) => {
     if (value) {
       await effects.vaultSet({ key, value })
-      return
+      return value
     }
     if (await effects.vaultList().then((x) => x.includes(key))) {
-      return
+      return null
     }
-    await effects.vaultSet({ key, value: getDefaultString(generator) })
+    const newValue = getDefaultString(generator)
+    await effects.vaultSet({ key, value: newValue })
+    return newValue
   },
   getSystemSmtp: () =>
     new GetSystemSmtp(effects) as GetSystemSmtp & WrapperOverWrite,