From 3df8cc53e367b39cf4bafe7178f26c34ac96e229 Mon Sep 17 00:00:00 2001
From: Abdullah Atta <abdullahatta@streetwriters.co>
Date: Wed, 7 Feb 2024 13:27:17 +0500
Subject: [PATCH] desktop: expose electron's safeStorage api via trpc

---
 apps/desktop/src/api/index.ts        |  4 ++-
 apps/desktop/src/api/safe-storage.ts | 38 ++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 1 deletion(-)
 create mode 100644 apps/desktop/src/api/safe-storage.ts

diff --git a/apps/desktop/src/api/index.ts b/apps/desktop/src/api/index.ts
index 9c20fdb86..5b45d9164 100644
--- a/apps/desktop/src/api/index.ts
+++ b/apps/desktop/src/api/index.ts
@@ -23,6 +23,7 @@ import { osIntegrationRouter } from "./os-integration";
 import { spellCheckerRouter } from "./spell-checker";
 import { updaterRouter } from "./updater";
 import { bridgeRouter } from "./bridge";
+import { safeStorageRouter } from "./safe-storage";
 
 const t = initTRPC.create();
 
@@ -31,7 +32,8 @@ export const router = t.router({
   integration: osIntegrationRouter,
   spellChecker: spellCheckerRouter,
   updater: updaterRouter,
-  bridge: bridgeRouter
+  bridge: bridgeRouter,
+  safeStorage: safeStorageRouter
 });
 
 export const api = router.createCaller({});
diff --git a/apps/desktop/src/api/safe-storage.ts b/apps/desktop/src/api/safe-storage.ts
new file mode 100644
index 000000000..59692b89a
--- /dev/null
+++ b/apps/desktop/src/api/safe-storage.ts
@@ -0,0 +1,38 @@
+/*
+This file is part of the Notesnook project (https://notesnook.com/)
+
+Copyright (C) 2023 Streetwriters (Private) Limited
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+import { safeStorage } from "electron";
+import { initTRPC } from "@trpc/server";
+import { z } from "zod";
+
+const t = initTRPC.create();
+
+export const safeStorageRouter = t.router({
+  /**
+   * Takes a string and returns an encrypted base64 string
+   */
+  encryptString: t.procedure.input(z.string()).query(async ({ input }) => {
+    return safeStorage.encryptString(input).toString("base64");
+  }),
+  /**
+   * Takes an encrypted base64 string and returns a string
+   */
+  decryptString: t.procedure.input(z.string()).query(async ({ input }) => {
+    return safeStorage.decryptString(Buffer.from(input, "base64"));
+  })
+});