feat: add getClientIds and load accounts on startup

Author: mrepol742

src/components/services/account.ts

@@ -74,3 +74,25 @@ export async function getAccountCount(): Promise<number> {
     return 0;
   }
 }
+
+export async function getClientIds(): Promise<Map<string, boolean>> {
+  try {
+    const accounts = await prisma.account.findMany({
+      select: {
+        clientId: true,
+        isRoot: true,
+      },
+    });
+
+    const clientIds = new Map<string, boolean>();
+    accounts.forEach((account) => {
+      clientIds.set(account.clientId, account.isRoot);
+    });
+
+    return clientIds;
+  } catch (error) {
+    Sentry.captureException(error);
+    log.error("Database", "Failed to get client IDs.", error);
+    return new Map<string, boolean>();
+  }
+}

src/index.ts

@@ -16,7 +16,7 @@ import {
   PHISHTANK_AUTO_UPDATE,
   PROJECT_THRESHOLD_MEMORY,
 } from "./config";
-import { getRootAccount } from "./components/services/account";
+import { getClientIds, getRootAccount } from "./components/services/account";
 
 const monitor = new MemoryMonitor({
   interval: 60000,
@@ -39,8 +39,13 @@ async function main() {
   ]);
 
   phishingSet = phishtank.getPhishingSet();
-  const rootClientId = await getRootAccount();
-  await addAccount(rootClientId, undefined, true);
+
+  const accountsIds = await getClientIds();
+  await Promise.all(
+    Array.from(accountsIds.entries()).map(async ([clientId, isRoot]) => {
+      await addAccount(clientId, undefined, isRoot);
+    }),
+  );
 
   await mapCommands();
   // Watch for changes