From 8a46ecb7373d6c5e7300861169ea51a7917cd2b4 Mon Sep 17 00:00:00 2001
From: MohamedBassem <me@mbassem.com>
Date: Tue, 5 Mar 2024 13:11:06 +0000
Subject: refactor: Extract trpc logic into its package

---
 packages/trpc/routers/admin.ts | 77 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 packages/trpc/routers/admin.ts

(limited to 'packages/trpc/routers/admin.ts')

diff --git a/packages/trpc/routers/admin.ts b/packages/trpc/routers/admin.ts
new file mode 100644
index 00000000..8a7b592d
--- /dev/null
+++ b/packages/trpc/routers/admin.ts
@@ -0,0 +1,77 @@
+import { adminProcedure, router } from "../index";
+import { z } from "zod";
+import { count } from "drizzle-orm";
+import { bookmarks, users } from "@hoarder/db/schema";
+import {
+  LinkCrawlerQueue,
+  OpenAIQueue,
+  SearchIndexingQueue,
+} from "@hoarder/shared/queues";
+
+export const adminAppRouter = router({
+  stats: adminProcedure
+    .output(
+      z.object({
+        numUsers: z.number(),
+        numBookmarks: z.number(),
+        pendingCrawls: z.number(),
+        pendingIndexing: z.number(),
+        pendingOpenai: z.number(),
+      }),
+    )
+    .query(async ({ ctx }) => {
+      const [
+        [{ value: numUsers }],
+        [{ value: numBookmarks }],
+        pendingCrawls,
+        pendingIndexing,
+        pendingOpenai,
+      ] = await Promise.all([
+        ctx.db.select({ value: count() }).from(users),
+        ctx.db.select({ value: count() }).from(bookmarks),
+        LinkCrawlerQueue.getWaitingCount(),
+        SearchIndexingQueue.getWaitingCount(),
+        OpenAIQueue.getWaitingCount(),
+      ]);
+
+      return {
+        numUsers,
+        numBookmarks,
+        pendingCrawls,
+        pendingIndexing,
+        pendingOpenai,
+      };
+    }),
+  recrawlAllLinks: adminProcedure.mutation(async ({ ctx }) => {
+    const bookmarkIds = await ctx.db.query.bookmarkLinks.findMany({
+      columns: {
+        id: true,
+      },
+    });
+
+    await Promise.all(
+      bookmarkIds.map((b) =>
+        LinkCrawlerQueue.add("crawl", {
+          bookmarkId: b.id,
+        }),
+      ),
+    );
+  }),
+
+  reindexAllBookmarks: adminProcedure.mutation(async ({ ctx }) => {
+    const bookmarkIds = await ctx.db.query.bookmarks.findMany({
+      columns: {
+        id: true,
+      },
+    });
+
+    await Promise.all(
+      bookmarkIds.map((b) =>
+        SearchIndexingQueue.add("search_indexing", {
+          bookmarkId: b.id,
+          type: "index",
+        }),
+      ),
+    );
+  }),
+});
-- 
cgit v1.2.3-70-g09d2