try the desktop db

Author: arvinxx

.gitignore

@@ -68,4 +68,5 @@ public/swe-worker*
 *.patch
 *.pdf
 vertex-ai-key.json
-.pnpm-store
+.pnpm-store
+lobechat-db

src/database/server/core/db.ts

@@ -1,14 +1,21 @@
+import { PGlite } from '@electric-sql/pglite';
 import { Pool as NeonPool, neonConfig } from '@neondatabase/serverless';
 import { NeonDatabase, drizzle as neonDrizzle } from 'drizzle-orm/neon-serverless';
 import { drizzle as nodeDrizzle } from 'drizzle-orm/node-postgres';
+import { drizzle as pgliteDrizzle } from 'drizzle-orm/pglite';
 import { Pool as NodePool } from 'pg';
 import ws from 'ws';
 
 import { serverDBEnv } from '@/config/db';
-import { isServerMode } from '@/const/version';
+import { isDesktop, isServerMode } from '@/const/version';
 
 import * as schema from '../../schemas';
 
+export const getPgliteInstance = () => {
+  const client = new PGlite('lobechat-db');
+  return pgliteDrizzle({ client, schema });
+};
+
 export const getDBInstance = (): NeonDatabase<typeof schema> => {
   if (!isServerMode) return {} as any;
 
@@ -41,4 +48,4 @@ If you don't have it, please run \`openssl rand -base64 32\` to create one.
   return neonDrizzle(client, { schema });
 };
 
-export const serverDB = getDBInstance();
+export const serverDB = isDesktop ? getPgliteInstance() : getDBInstance();

src/libs/trpc/middleware/userAuth.ts

@@ -1,11 +1,19 @@
 import { TRPCError } from '@trpc/server';
 
 import { enableClerk } from '@/const/auth';
+import { isDesktop } from '@/const/version';
 
 import { trpc } from '../init';
 
 export const userAuth = trpc.middleware(async (opts) => {
   const { ctx } = opts;
+  if (isDesktop) {
+    return opts.next({
+      ctx: {
+        userId: 'desktop_user',
+      },
+    });
+  }
   // `ctx.user` is nullable
   if (!ctx.userId) {
     if (enableClerk) {

src/services/aiModel/index.ts

@@ -1,5 +1,9 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService } from './client';
 import { ServerService } from './server';
 
 export const aiModelService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : new ClientService();
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : new ClientService();

src/services/aiProvider/index.ts

@@ -1,5 +1,9 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService } from './client';
 import { ServerService } from './server';
 
 export const aiProviderService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : new ClientService();
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : new ClientService();

src/services/file/index.ts

@@ -1,3 +1,5 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService as DeprecatedService } from './_deprecated';
 import { ClientService } from './client';
 import { ServerService } from './server';
@@ -6,4 +8,6 @@ const clientService =
   process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
 
 export const fileService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : clientService;

src/services/message/index.ts

@@ -1,3 +1,5 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService as DeprecatedService } from './_deprecated';
 import { ClientService } from './client';
 import { ServerService } from './server';
@@ -6,4 +8,6 @@ const clientService =
   process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
 
 export const messageService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : clientService;

src/services/plugin/index.ts

@@ -1,3 +1,5 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService as DeprecatedService } from './_deprecated';
 import { ClientService } from './client';
 import { ServerService } from './server';
@@ -6,4 +8,6 @@ const clientService =
   process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
 
 export const pluginService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : clientService;

src/services/session/index.ts

@@ -1,3 +1,5 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService as DeprecatedService } from './_deprecated';
 import { ClientService } from './client';
 import { ServerService } from './server';
@@ -6,4 +8,6 @@ const clientService =
   process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
 
 export const sessionService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : clientService;

src/services/thread/index.ts

@@ -1,5 +1,9 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService } from './client';
 import { ServerService } from './server';
 
 export const threadService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : new ClientService();
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : new ClientService();

src/services/topic/index.ts

@@ -1,3 +1,5 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService as DeprecatedService } from './_deprecated';
 import { ClientService } from './client';
 import { ServerService } from './server';
@@ -6,4 +8,6 @@ const clientService =
   process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
 
 export const topicService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : clientService;

src/services/user/index.ts

@@ -1,3 +1,5 @@
+import { isDesktop } from '@/const/version';
+
 import { ClientService as DeprecatedService } from './_deprecated';
 import { ClientService } from './client';
 import { ServerService } from './server';
@@ -6,6 +8,8 @@ const clientService =
   process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
 
 export const userService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
+    ? new ServerService()
+    : clientService;
 
 export const userClientService = clientService;