✨ feat: add sso fetcher in services

Author: cy948

src/database/server/models/user.ts

@@ -1,6 +1,7 @@
 import { TRPCError } from '@trpc/server';
 import dayjs from 'dayjs';
 import { eq } from 'drizzle-orm/expressions';
+import type { AdapterAccount } from 'next-auth/adapters';
 import { DeepPartial } from 'utility-types';
 
 import { LobeChatDatabase } from '@/database/type';
@@ -9,7 +10,14 @@ import { UserKeyVaults, UserSettings } from '@/types/user/settings';
 import { merge } from '@/utils/merge';
 import { today } from '@/utils/time';
 
-import { NewUser, UserItem, UserSettingsItem, userSettings, users } from '../../schemas';
+import {
+  NewUser,
+  UserItem,
+  UserSettingsItem,
+  nextauthAccounts,
+  userSettings,
+  users,
+} from '../../schemas';
 import { SessionModel } from './session';
 
 type DecryptUserKeyVaults = (
@@ -97,6 +105,25 @@ export class UserModel {
     };
   };
 
+  getUserSSOProviders = async () => {
+    return this.db.query.nextauthAccounts
+      .findMany({
+        where: eq(nextauthAccounts.userId, this.userId),
+      })
+      .then((accounts) =>
+        accounts.map((account) => {
+          // Pick necessary fields, don't expose the sensitive information
+          return {
+            expires_at: account?.expires_at,
+            provider: account?.provider,
+            providerAccountId: account?.providerAccountId,
+            scope: account?.scope,
+            type: account?.type,
+          } as unknown as AdapterAccount;
+        }),
+      ) as Promise<AdapterAccount[]>;
+  };
+
   getUserSettings = async () => {
     return this.db.query.userSettings.findFirst({ where: eq(userSettings.id, this.userId) });
   };

src/server/routers/lambda/user.ts

@@ -24,6 +24,10 @@ export const userRouter = router({
     return ctx.userModel.getUserRegistrationDuration();
   }),
 
+  getUserSSOProviders: userProcedure.query(async ({ ctx }) => {
+    return ctx.userModel.getUserSSOProviders();
+  }),
+
   getUserState: userProcedure.query(async ({ ctx }): Promise<UserInitializationState> => {
     let state: Awaited<ReturnType<UserModel['getUserState']>> | undefined;
 

src/services/user/_deprecated.ts

@@ -37,6 +37,10 @@ export class ClientService implements IUserService {
     };
   }
 
+  getUserSSOProviders = async () => {
+    return [];
+  };
+
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
   updateUserSettings = async (patch: DeepPartial<UserSettings>, _?: any) => {
     return UserModel.updateSettings(patch);

src/services/user/client.ts

@@ -54,6 +54,11 @@ export class ClientService extends BaseClientService implements IUserService {
     };
   };
 
+  getUserSSOProviders: IUserService['getUserSSOProviders'] = async () => {
+    // Account not exist on next-auth in client mode
+    return [];
+  };
+
   updateUserSettings: IUserService['updateUserSettings'] = async (value) => {
     const { keyVaults, ...res } = value;
 

src/services/user/server.ts

@@ -10,6 +10,10 @@ export class ServerService implements IUserService {
     return lambdaClient.user.getUserState.query();
   };
 
+  getUserSSOProviders: IUserService['getUserSSOProviders'] = async () => {
+    return lambdaClient.user.getUserSSOProviders.query();
+  };
+
   makeUserOnboarded = async () => {
     return lambdaClient.user.makeUserOnboarded.mutate();
   };

src/services/user/type.ts

@@ -1,3 +1,4 @@
+import type { AdapterAccount } from 'next-auth/adapters';
 import { DeepPartial } from 'utility-types';
 
 import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
@@ -9,6 +10,7 @@ export interface IUserService {
     duration: number;
     updatedAt: string;
   }>;
+  getUserSSOProviders: () => Promise<AdapterAccount[]>;
   getUserState: () => Promise<UserInitializationState>;
   resetUserSettings: () => Promise<any>;
   updateGuide: (guide: Partial<UserGuide>) => Promise<any>;