@lobehub/lobehub 2.0.0-next.26 → 2.0.0-next.28

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

@@ -1,149 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-import { describe, expect, it, vi } from 'vitest';
-
-import { lambdaClient } from '@/libs/trpc/client';
-import { UserInitializationState, UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-
-import { ServerService } from './server';
-
-vi.mock('@/libs/trpc/client', () => ({
-  lambdaClient: {
-    user: {
-      getUserRegistrationDuration: {
-        query: vi.fn(),
-      },
-      getUserState: {
-        query: vi.fn(),
-      },
-      getUserSSOProviders: {
-        query: vi.fn(),
-      },
-      unlinkSSOProvider: {
-        mutate: vi.fn(),
-      },
-      makeUserOnboarded: {
-        mutate: vi.fn(),
-      },
-      updatePreference: {
-        mutate: vi.fn(),
-      },
-      updateGuide: {
-        mutate: vi.fn(),
-      },
-      updateSettings: {
-        mutate: vi.fn(),
-      },
-      resetSettings: {
-        mutate: vi.fn(),
-      },
-    },
-  },
-}));
-
-describe('ServerService', () => {
-  const service = new ServerService();
-
-  it('should get user registration duration', async () => {
-    const mockData = {
-      createdAt: '2023-01-01',
-      duration: 100,
-      updatedAt: '2023-01-02',
-    };
-    vi.mocked(lambdaClient.user.getUserRegistrationDuration.query).mockResolvedValue(mockData);
-
-    const result = await service.getUserRegistrationDuration();
-    expect(result).toEqual(mockData);
-  });
-
-  it('should get user state', async () => {
-    const mockState: UserInitializationState = {
-      isOnboard: true,
-      preference: {
-        telemetry: true,
-      },
-      settings: {},
-    };
-    vi.mocked(lambdaClient.user.getUserState.query).mockResolvedValue(mockState);
-
-    const result = await service.getUserState();
-    expect(result).toEqual(mockState);
-  });
-
-  it('should get user SSO providers', async () => {
-    const mockProviders = [
-      {
-        provider: 'google',
-        providerAccountId: '123',
-        userId: 'user1',
-        type: 'oauth' as const,
-        access_token: 'token',
-        token_type: 'bearer' as const,
-        expires_at: 123,
-        scope: 'email profile',
-      },
-    ];
-    vi.mocked(lambdaClient.user.getUserSSOProviders.query).mockResolvedValue(mockProviders);
-
-    const result = await service.getUserSSOProviders();
-    expect(result).toEqual(mockProviders);
-  });
-
-  it('should unlink SSO provider', async () => {
-    const provider = 'google';
-    const providerAccountId = '123';
-    await service.unlinkSSOProvider(provider, providerAccountId);
-
-    expect(lambdaClient.user.unlinkSSOProvider.mutate).toHaveBeenCalledWith({
-      provider,
-      providerAccountId,
-    });
-  });
-
-  it('should make user onboarded', async () => {
-    await service.makeUserOnboarded();
-    expect(lambdaClient.user.makeUserOnboarded.mutate).toHaveBeenCalled();
-  });
-
-  it('should update user preference', async () => {
-    const preference: Partial<UserPreference> = {
-      telemetry: true,
-      useCmdEnterToSend: true,
-    };
-    await service.updatePreference(preference);
-    expect(lambdaClient.user.updatePreference.mutate).toHaveBeenCalledWith(preference);
-  });
-
-  it('should update user guide', async () => {
-    const guide = {
-      moveSettingsToAvatar: true,
-      topic: false,
-      uploadFileInKnowledgeBase: true,
-    };
-    await service.updateGuide(guide);
-    expect(lambdaClient.user.updateGuide.mutate).toHaveBeenCalledWith(guide);
-  });
-
-  it('should update user settings', async () => {
-    const settings: PartialDeep<UserSettings> = {
-      defaultAgent: {
-        config: {
-          model: 'gpt-4',
-          provider: 'openai',
-        },
-        meta: {
-          avatar: 'avatar',
-          description: 'test agent',
-        },
-      },
-    };
-    const signal = new AbortController().signal;
-    await service.updateUserSettings(settings, signal);
-    expect(lambdaClient.user.updateSettings.mutate).toHaveBeenCalledWith(settings, { signal });
-  });
-
-  it('should reset user settings', async () => {
-    await service.resetUserSettings();
-    expect(lambdaClient.user.resetSettings.mutate).toHaveBeenCalled();
-  });
-});

package/src/services/user/server.ts

@@ -1,47 +0,0 @@
-import { lambdaClient } from '@/libs/trpc/client';
-import { IUserService } from '@/services/user/type';
-
-export class ServerService implements IUserService {
-  getUserRegistrationDuration: IUserService['getUserRegistrationDuration'] = async () => {
-    return lambdaClient.user.getUserRegistrationDuration.query();
-  };
-
-  getUserState: IUserService['getUserState'] = async () => {
-    return lambdaClient.user.getUserState.query();
-  };
-
-  getUserSSOProviders: IUserService['getUserSSOProviders'] = async () => {
-    return lambdaClient.user.getUserSSOProviders.query();
-  };
-
-  unlinkSSOProvider: IUserService['unlinkSSOProvider'] = async (
-    provider: string,
-    providerAccountId: string,
-  ) => {
-    return lambdaClient.user.unlinkSSOProvider.mutate({ provider, providerAccountId });
-  };
-
-  makeUserOnboarded = async () => {
-    return lambdaClient.user.makeUserOnboarded.mutate();
-  };
-
-  updateAvatar: IUserService['updateAvatar'] = async (avatar) => {
-    return lambdaClient.user.updateAvatar.mutate(avatar);
-  };
-
-  updatePreference: IUserService['updatePreference'] = async (preference) => {
-    return lambdaClient.user.updatePreference.mutate(preference);
-  };
-
-  updateGuide: IUserService['updateGuide'] = async (guide) => {
-    return lambdaClient.user.updateGuide.mutate(guide);
-  };
-
-  updateUserSettings: IUserService['updateUserSettings'] = async (value, signal) => {
-    return lambdaClient.user.updateSettings.mutate(value, { signal });
-  };
-
-  resetUserSettings: IUserService['resetUserSettings'] = async () => {
-    return lambdaClient.user.resetSettings.mutate();
-  };
-}

package/src/services/user/type.ts

@@ -1,21 +0,0 @@
-import type { AdapterAccount } from 'next-auth/adapters';
-import type { PartialDeep } from 'type-fest';
-
-import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
-import { UserSettings } from '@/types/user/settings';
-
-export interface IUserService {
-  getUserRegistrationDuration: () => Promise<{
-    createdAt: string;
-    duration: number;
-    updatedAt: string;
-  }>;
-  getUserSSOProviders: () => Promise<AdapterAccount[]>;
-  getUserState: () => Promise<UserInitializationState>;
-  resetUserSettings: () => Promise<any>;
-  unlinkSSOProvider: (provider: string, providerAccountId: string) => Promise<any>;
-  updateAvatar: (avatar: string) => Promise<any>;
-  updateGuide: (guide: Partial<UserGuide>) => Promise<any>;
-  updatePreference: (preference: Partial<UserPreference>) => Promise<any>;
-  updateUserSettings: (value: PartialDeep<UserSettings>, signal?: AbortSignal) => Promise<any>;
-}