@lobehub/lobehub 2.0.0-next.13 → 2.0.0-next.14

package/src/services/user/_deprecated.test.ts

@@ -1,101 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { UserModel } from '@/database/_deprecated/models/user';
-import { UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-
-import { ClientService } from './_deprecated';
-
-vi.mock('@/database/_deprecated/models/user', () => ({
-  UserModel: {
-    getUser: vi.fn(),
-    updateSettings: vi.fn(),
-    resetSettings: vi.fn(),
-    updateAvatar: vi.fn(),
-  },
-}));
-
-const mockUser = {
-  avatar: 'avatar.png',
-  settings: { themeMode: 'light' } as unknown as UserSettings,
-  uuid: 'user-id',
-};
-
-const mockPreference = {
-  useCmdEnterToSend: true,
-} as UserPreference;
-
-describe('ClientService', () => {
-  let clientService: ClientService;
-
-  beforeEach(() => {
-    vi.clearAllMocks();
-    clientService = new ClientService();
-  });
-
-  it('should get user state correctly', async () => {
-    (UserModel.getUser as Mock).mockResolvedValue(mockUser);
-    const spyOn = vi
-      .spyOn(clientService['preferenceStorage'], 'getFromLocalStorage')
-      .mockResolvedValue(mockPreference);
-
-    const userState = await clientService.getUserState();
-
-    expect(userState).toEqual({
-      avatar: mockUser.avatar,
-      isOnboard: true,
-      canEnablePWAGuide: false,
-      hasConversation: false,
-      canEnableTrace: false,
-      preference: mockPreference,
-      settings: mockUser.settings,
-      userId: mockUser.uuid,
-    });
-    expect(UserModel.getUser).toHaveBeenCalledTimes(1);
-    expect(spyOn).toHaveBeenCalledTimes(1);
-  });
-
-  it('should update user settings correctly', async () => {
-    const settingsPatch: PartialDeep<UserSettings> = { general: { fontSize: 12 } };
-    (UserModel.updateSettings as Mock).mockResolvedValue(undefined);
-
-    await clientService.updateUserSettings(settingsPatch);
-
-    expect(UserModel.updateSettings).toHaveBeenCalledWith(settingsPatch);
-    expect(UserModel.updateSettings).toHaveBeenCalledTimes(1);
-  });
-
-  it('should reset user settings correctly', async () => {
-    (UserModel.resetSettings as Mock).mockResolvedValue(undefined);
-
-    await clientService.resetUserSettings();
-
-    expect(UserModel.resetSettings).toHaveBeenCalledTimes(1);
-  });
-
-  it('should update user avatar correctly', async () => {
-    const newAvatar = 'new-avatar.png';
-    (UserModel.updateAvatar as Mock).mockResolvedValue(undefined);
-
-    await clientService.updateAvatar(newAvatar);
-
-    expect(UserModel.updateAvatar).toHaveBeenCalledWith(newAvatar);
-    expect(UserModel.updateAvatar).toHaveBeenCalledTimes(1);
-  });
-
-  it('should update user preference correctly', async () => {
-    const newPreference = {
-      useCmdEnterToSend: false,
-    } as UserPreference;
-
-    const spyOn = vi
-      .spyOn(clientService['preferenceStorage'], 'saveToLocalStorage')
-      .mockResolvedValue(undefined);
-
-    await clientService.updatePreference(newPreference);
-
-    expect(spyOn).toHaveBeenCalledWith(newPreference);
-    expect(spyOn).toHaveBeenCalledTimes(1);
-  });
-});

package/src/services/user/_deprecated.ts

@@ -1,70 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-
-import { MessageModel } from '@/database/_deprecated/models/message';
-import { SessionModel } from '@/database/_deprecated/models/session';
-import { UserModel } from '@/database/_deprecated/models/user';
-import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-import { AsyncLocalStorage } from '@/utils/localStorage';
-
-import { IUserService } from './type';
-
-export class ClientService implements IUserService {
-  private preferenceStorage: AsyncLocalStorage<UserPreference>;
-
-  constructor() {
-    this.preferenceStorage = new AsyncLocalStorage('LOBE_PREFERENCE');
-  }
-
-  getUserRegistrationDuration = async () => {
-    throw new Error('Method not implemented.');
-  };
-
-  async getUserState(): Promise<UserInitializationState> {
-    const user = await UserModel.getUser();
-    const messageCount = await MessageModel.count();
-    const sessionCount = await SessionModel.count();
-
-    return {
-      avatar: user.avatar,
-      canEnablePWAGuide: messageCount >= 4,
-      canEnableTrace: messageCount >= 4,
-      hasConversation: messageCount > 0 || sessionCount > 0,
-      isOnboard: true,
-      preference: await this.preferenceStorage.getFromLocalStorage(),
-      settings: user.settings as UserSettings,
-      userId: user.uuid,
-    };
-  }
-
-  getUserSSOProviders = async () => {
-    // Account not exist on next-auth in client mode, no need to implement this method
-    return [];
-  };
-
-  unlinkSSOProvider = async () => {
-    // Account not exist on next-auth in client mode, no need to implement this method
-  };
-
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  updateUserSettings = async (patch: PartialDeep<UserSettings>, _?: any) => {
-    return UserModel.updateSettings(patch);
-  };
-
-  resetUserSettings = async () => {
-    return UserModel.resetSettings();
-  };
-
-  async updateAvatar(avatar: string) {
-    await UserModel.updateAvatar(avatar);
-  }
-
-  async updatePreference(preference: Partial<UserPreference>) {
-    await this.preferenceStorage.saveToLocalStorage(preference);
-  }
-
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars,unused-imports/no-unused-vars
-  async updateGuide(guide: Partial<UserGuide>) {
-    throw new Error('Method not implemented.');
-  }
-}

package/src/services/user/client.test.ts

@@ -1,111 +0,0 @@
-import { eq } from 'drizzle-orm';
-import type { PartialDeep } from 'type-fest';
-import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { clientDB, initializeDB } from '@/database/client/db';
-import { userSettings, users } from '@/database/schemas';
-import { UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-
-import { ClientService } from './client';
-
-const mockUser = {
-  avatar: 'avatar.png',
-  settings: { themeMode: 'light' } as unknown as UserSettings,
-  uuid: 'user-id',
-};
-
-const mockPreference = {
-  useCmdEnterToSend: true,
-} as UserPreference;
-const clientService = new ClientService(mockUser.uuid);
-
-beforeEach(async () => {
-  vi.clearAllMocks();
-
-  await initializeDB();
-  await clientDB.delete(users);
-
-  await clientDB
-    .insert(users)
-    .values({ id: mockUser.uuid, avatar: 'avatar.png' })
-    .onConflictDoNothing();
-  await clientDB
-    .insert(userSettings)
-    .values({ id: mockUser.uuid, general: { themeMode: 'light' } });
-});
-
-describe('ClientService', () => {
-  it('should get user state correctly', async () => {
-    const spyOn = vi
-      .spyOn(clientService['preferenceStorage'], 'getFromLocalStorage')
-      .mockResolvedValue(mockPreference);
-
-    const userState = await clientService.getUserState();
-
-    expect(userState).toMatchObject({
-      avatar: mockUser.avatar,
-      isOnboard: true,
-      canEnablePWAGuide: false,
-      hasConversation: false,
-      canEnableTrace: false,
-      preference: mockPreference,
-      settings: { general: { themeMode: 'light' } },
-      userId: mockUser.uuid,
-    });
-    expect(spyOn).toHaveBeenCalledTimes(1);
-  });
-
-  it('should update user settings correctly', async () => {
-    const settingsPatch: PartialDeep<UserSettings> = { general: { fontSize: 12 } };
-
-    await clientService.updateUserSettings(settingsPatch);
-
-    const result = await clientDB.query.userSettings.findFirst({
-      where: eq(userSettings.id, mockUser.uuid),
-    });
-
-    expect(result).toMatchObject(settingsPatch);
-  });
-
-  it('should reset user settings correctly', async () => {
-    await clientService.resetUserSettings();
-
-    const result = await clientDB.query.userSettings.findFirst({
-      where: eq(userSettings.id, mockUser.uuid),
-    });
-
-    expect(result).toBeUndefined();
-  });
-
-  it('should update user avatar correctly', async () => {
-    const newAvatar = 'new-avatar.png';
-
-    await clientService.updateAvatar(newAvatar);
-  });
-
-  it('should update user preference correctly', async () => {
-    const newPreference = {
-      useCmdEnterToSend: false,
-    } as UserPreference;
-
-    const spyOn = vi
-      .spyOn(clientService['preferenceStorage'], 'saveToLocalStorage')
-      .mockResolvedValue(undefined);
-
-    await clientService.updatePreference(newPreference);
-
-    expect(spyOn).toHaveBeenCalledWith(newPreference);
-    expect(spyOn).toHaveBeenCalledTimes(1);
-  });
-
-  it('should return empty array for getUserSSOProviders', async () => {
-    const providers = await clientService.getUserSSOProviders();
-    expect(providers).toEqual([]);
-  });
-
-  it('should do nothing when unlinkSSOProvider is called', async () => {
-    const result = await clientService.unlinkSSOProvider('google', '123');
-    expect(result).toBeUndefined();
-  });
-});

package/src/services/user/client.ts

@@ -1,104 +0,0 @@
-import { clientDB } from '@/database/client/db';
-import { MessageModel } from '@/database/models/message';
-import { SessionModel } from '@/database/models/session';
-import { UserModel } from '@/database/models/user';
-import { users } from '@/database/schemas';
-import { BaseClientService } from '@/services/baseClientService';
-import { UserPreference } from '@/types/user';
-import { AsyncLocalStorage } from '@/utils/localStorage';
-
-import { IUserService } from './type';
-
-export class ClientService extends BaseClientService implements IUserService {
-  private preferenceStorage: AsyncLocalStorage<UserPreference>;
-
-  private get userModel(): UserModel {
-    return new UserModel(clientDB as any, this.userId);
-  }
-  private get messageModel(): MessageModel {
-    return new MessageModel(clientDB as any, this.userId);
-  }
-  private get sessionModel(): SessionModel {
-    return new SessionModel(clientDB as any, this.userId);
-  }
-
-  constructor(userId?: string) {
-    super(userId);
-    this.preferenceStorage = new AsyncLocalStorage('LOBE_PREFERENCE');
-  }
-
-  getUserRegistrationDuration: IUserService['getUserRegistrationDuration'] = async () => {
-    return this.userModel.getUserRegistrationDuration();
-  };
-
-  getUserState: IUserService['getUserState'] = async () => {
-    // if user not exist in the db, create one to make sure the user exist
-    await this.makeSureUserExist();
-
-    const state = await this.userModel.getUserState((encryptKeyVaultsStr) =>
-      encryptKeyVaultsStr ? JSON.parse(encryptKeyVaultsStr) : {},
-    );
-
-    const messageCount = await this.messageModel.count();
-    const sessionCount = await this.sessionModel.count();
-
-    return {
-      ...state,
-      avatar: state.avatar ?? '',
-      canEnablePWAGuide: messageCount >= 4,
-      canEnableTrace: messageCount >= 4,
-      firstName: state.firstName,
-      fullName: state.fullName,
-      hasConversation: messageCount > 0 || sessionCount > 0,
-      isOnboard: true,
-      lastName: state.lastName,
-      preference: await this.preferenceStorage.getFromLocalStorage(),
-      username: state.username,
-    };
-  };
-
-  getUserSSOProviders: IUserService['getUserSSOProviders'] = async () => {
-    // Account not exist on next-auth in client mode, no need to implement this method
-    return [];
-  };
-
-  unlinkSSOProvider: IUserService['unlinkSSOProvider'] = async () => {
-    // Account not exist on next-auth in client mode, no need to implement this method
-  };
-
-  updateUserSettings: IUserService['updateUserSettings'] = async (value) => {
-    const { keyVaults, ...res } = value;
-
-    return this.userModel.updateSetting({ ...res, keyVaults: JSON.stringify(keyVaults) });
-  };
-
-  resetUserSettings: IUserService['resetUserSettings'] = async () => {
-    return this.userModel.deleteSetting();
-  };
-
-  updateAvatar = async (avatar: string) => {
-    await this.userModel.updateUser({ avatar });
-  };
-
-  updatePreference: IUserService['updatePreference'] = async (preference) => {
-    await this.preferenceStorage.saveToLocalStorage(preference);
-  };
-
-  updateGuide: IUserService['updateGuide'] = async () => {
-    throw new Error('Method not implemented.');
-  };
-
-  makeSureUserExist = async () => {
-    const existUsers = await clientDB.query.users.findMany();
-
-    let user: { id: string };
-    if (existUsers.length === 0) {
-      const result = await clientDB.insert(users).values({ id: this.userId }).returning();
-      user = result[0];
-    } else {
-      user = existUsers[0];
-    }
-
-    return user;
-  };
-}