Migrate address book on app start and delete old address book (#164)

Author: dguenther

main/api/contacts/index.ts

@@ -2,7 +2,6 @@ import Store, { Schema } from "electron-store";
 import { v4 as uuidv4 } from "uuid";
 import { z } from "zod";
 
-import { getNodeAppBetaContacts } from "./utils/getNodeAppBetaContacts";
 import { t } from "../trpc";
 
 const contactDefinition = z.object({
@@ -35,7 +34,7 @@ const schema: Schema<{
   },
 };
 
-const store = new Store({ schema, name: "contacts" });
+export const store = new Store({ schema, name: "contacts" });
 
 export const contactsRouter = t.router({
   getContacts: t.procedure.query(async () => {
@@ -54,27 +53,6 @@ export const contactsRouter = t.router({
         null
       );
     }),
-  migrateNodeAppBetaContacts: t.procedure.mutation(async () => {
-    const betaContacts = await getNodeAppBetaContacts();
-    const contacts = await store.get("contacts", []);
-    const addressLookup = new Set(contacts.map((contact) => contact.address));
-
-    let order = 0;
-
-    for (const contact of betaContacts) {
-      if (!addressLookup.has(contact.address)) {
-        contacts.push({
-          id: uuidv4(),
-          name: contact.name,
-          address: contact.address,
-          order: order++,
-        });
-      }
-    }
-
-    store.set("contacts", contacts);
-    return contacts;
-  }),
   addContact: t.procedure
     .input(
       z.object({

main/api/contacts/utils/getNodeAppBetaContacts.ts

@@ -0,0 +1,80 @@
+import fs from "fs";
+import os from "os";
+import path from "path";
+
+import NeDBStorage from "nedb";
+import { v4 as uuidv4 } from "uuid";
+
+import { store } from "..";
+
+type BetaContact = {
+  name: string;
+  address: string;
+};
+
+let nedb: NeDBStorage;
+
+function getNodeAppBetaContacts(): Promise<Array<BetaContact>> {
+  const betaContactsDbPath = path.join(
+    os.homedir(),
+    "/.ironfish/node-app",
+    "address_book.db",
+  );
+
+  if (!fs.existsSync(betaContactsDbPath)) {
+    return Promise.resolve([]);
+  }
+
+  if (!nedb) {
+    nedb = new NeDBStorage({
+      filename: betaContactsDbPath,
+      autoload: true,
+      timestampData: true,
+    });
+  }
+
+  return new Promise((resolve, reject) => {
+    nedb.find({}, (err: unknown, docs: Array<BetaContact>) => {
+      if (err) {
+        reject(err);
+      } else {
+        resolve(docs || []);
+      }
+    });
+  });
+}
+
+function deleteNodeAppBetaContacts(): void {
+  const betaContactsDbPath = path.join(
+    os.homedir(),
+    "/.ironfish/node-app",
+    "address_book.db",
+  );
+
+  if (fs.existsSync(betaContactsDbPath)) {
+    fs.rmSync(betaContactsDbPath);
+  }
+}
+
+export async function migrateNodeAppBetaContacts() {
+  const betaContacts = await getNodeAppBetaContacts();
+  const contacts = store.get("contacts", []);
+  const addressLookup = new Set(contacts.map((contact) => contact.address));
+
+  let order = 0;
+
+  for (const contact of betaContacts) {
+    if (!addressLookup.has(contact.address)) {
+      contacts.push({
+        id: uuidv4(),
+        name: contact.name,
+        address: contact.address,
+        order: order++,
+      });
+    }
+  }
+
+  store.set("contacts", contacts);
+  deleteNodeAppBetaContacts();
+  return contacts;
+}

main/api/contacts/utils/migrateNodeAppBetaContacts.ts

@@ -4,6 +4,7 @@ import serve from "electron-serve";
 import { createIPCHandler } from "electron-trpc/main";
 
 import { router } from "./api";
+import { migrateNodeAppBetaContacts } from "./api/contacts/utils/migrateNodeAppBetaContacts";
 import { manager } from "./api/manager";
 import { getUserSettings } from "./api/user-settings/userSettings";
 import { mainWindow } from "./main-window";
@@ -79,6 +80,7 @@ app.whenReady().then(() => {
 
   createThemeChangeHandler();
   setNativeThemeSource();
+  migrateNodeAppBetaContacts();
 
   const handler = createIPCHandler({ router });