@lobehub/chat 1.36.46 → 1.37.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +25 -0
- package/README.ja-JP.md +8 -8
- package/README.md +8 -8
- package/README.zh-CN.md +8 -8
- package/changelog/v1.json +9 -0
- package/next.config.mjs +4 -1
- package/package.json +5 -3
- package/scripts/migrateClientDB/compile-migrations.ts +14 -0
- package/src/app/(main)/(mobile)/me/(home)/layout.tsx +2 -0
- package/src/app/(main)/chat/_layout/Desktop/index.tsx +3 -2
- package/src/app/(main)/chat/_layout/Mobile.tsx +5 -3
- package/src/app/(main)/chat/features/Migration/DBReader.ts +290 -0
- package/src/app/(main)/chat/features/Migration/UpgradeButton.tsx +4 -8
- package/src/app/(main)/chat/features/Migration/index.tsx +26 -15
- package/src/app/(main)/settings/_layout/Desktop/index.tsx +2 -0
- package/src/app/loading/Client/Content.tsx +11 -1
- package/src/app/loading/Client/Error.tsx +27 -0
- package/src/app/loading/stage.ts +8 -0
- package/src/components/FullscreenLoading/index.tsx +4 -3
- package/src/const/version.ts +1 -0
- package/src/database/client/db.test.ts +172 -0
- package/src/database/client/db.ts +246 -0
- package/src/database/client/migrations.json +289 -0
- package/src/features/InitClientDB/EnableModal.tsx +111 -0
- package/src/features/InitClientDB/ErrorResult.tsx +125 -0
- package/src/features/InitClientDB/InitIndicator.tsx +124 -0
- package/src/features/InitClientDB/PGliteSVG.tsx +22 -0
- package/src/features/InitClientDB/index.tsx +37 -0
- package/src/hooks/useCheckPluginsIsInstalled.ts +2 -2
- package/src/hooks/useFetchInstalledPlugins.ts +2 -2
- package/src/hooks/useFetchMessages.ts +2 -2
- package/src/hooks/useFetchSessions.ts +2 -2
- package/src/hooks/useFetchThreads.ts +2 -2
- package/src/hooks/useFetchTopics.ts +2 -2
- package/src/layout/GlobalProvider/StoreInitialization.tsx +2 -2
- package/src/services/baseClientService/index.ts +9 -0
- package/src/services/debug.ts +32 -34
- package/src/services/file/index.ts +6 -2
- package/src/services/file/pglite.test.ts +198 -0
- package/src/services/file/pglite.ts +84 -0
- package/src/services/file/type.ts +4 -3
- package/src/services/github.ts +17 -0
- package/src/services/import/index.ts +6 -2
- package/src/services/import/pglite.test.ts +997 -0
- package/src/services/import/pglite.ts +34 -0
- package/src/services/message/client.ts +2 -0
- package/src/services/message/index.ts +6 -2
- package/src/services/message/pglite.test.ts +430 -0
- package/src/services/message/pglite.ts +118 -0
- package/src/services/message/server.ts +9 -9
- package/src/services/message/type.ts +3 -4
- package/src/services/plugin/index.ts +6 -2
- package/src/services/plugin/pglite.test.ts +175 -0
- package/src/services/plugin/pglite.ts +51 -0
- package/src/services/session/client.ts +1 -1
- package/src/services/session/index.ts +6 -2
- package/src/services/session/pglite.test.ts +411 -0
- package/src/services/session/pglite.ts +184 -0
- package/src/services/session/type.ts +14 -1
- package/src/services/topic/index.ts +6 -3
- package/src/services/topic/pglite.test.ts +212 -0
- package/src/services/topic/pglite.ts +85 -0
- package/src/services/user/client.test.ts +0 -1
- package/src/services/user/index.ts +8 -2
- package/src/services/user/pglite.test.ts +98 -0
- package/src/services/user/pglite.ts +92 -0
- package/src/store/chat/slices/builtinTool/action.test.ts +3 -4
- package/src/store/global/actions/clientDb.ts +51 -0
- package/src/store/global/initialState.ts +13 -0
- package/src/store/global/selectors.ts +24 -3
- package/src/store/global/store.ts +3 -1
- package/src/store/session/slices/sessionGroup/reducer.test.ts +6 -6
- package/src/store/user/slices/common/action.ts +2 -4
- package/src/types/clientDB.ts +29 -0
- package/src/types/importer.ts +17 -5
- package/src/types/meta.ts +0 -9
- package/src/types/session/sessionGroup.ts +3 -3
- package/src/services/message/index.test.ts +0 -48
@@ -0,0 +1,92 @@
|
|
1
|
+
import { DeepPartial } from 'utility-types';
|
2
|
+
|
3
|
+
import { clientDB } from '@/database/client/db';
|
4
|
+
import { users } from '@/database/schemas';
|
5
|
+
import { MessageModel } from '@/database/server/models/message';
|
6
|
+
import { SessionModel } from '@/database/server/models/session';
|
7
|
+
import { UserModel } from '@/database/server/models/user';
|
8
|
+
import { BaseClientService } from '@/services/baseClientService';
|
9
|
+
import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
|
10
|
+
import { UserSettings } from '@/types/user/settings';
|
11
|
+
import { AsyncLocalStorage } from '@/utils/localStorage';
|
12
|
+
|
13
|
+
import { IUserService } from './type';
|
14
|
+
|
15
|
+
export class ClientService extends BaseClientService implements IUserService {
|
16
|
+
private preferenceStorage: AsyncLocalStorage<UserPreference>;
|
17
|
+
|
18
|
+
private get userModel(): UserModel {
|
19
|
+
return new UserModel(clientDB as any, this.userId);
|
20
|
+
}
|
21
|
+
private get messageModel(): MessageModel {
|
22
|
+
return new MessageModel(clientDB as any, this.userId);
|
23
|
+
}
|
24
|
+
private get sessionModel(): SessionModel {
|
25
|
+
return new SessionModel(clientDB as any, this.userId);
|
26
|
+
}
|
27
|
+
|
28
|
+
constructor(userId?: string) {
|
29
|
+
super(userId);
|
30
|
+
this.preferenceStorage = new AsyncLocalStorage('LOBE_PREFERENCE');
|
31
|
+
}
|
32
|
+
|
33
|
+
async getUserState(): Promise<UserInitializationState> {
|
34
|
+
// if user not exist in the db, create one to make sure the user exist
|
35
|
+
await this.makeSureUserExist();
|
36
|
+
|
37
|
+
const state = await this.userModel.getUserState((encryptKeyVaultsStr) =>
|
38
|
+
encryptKeyVaultsStr ? JSON.parse(encryptKeyVaultsStr) : {},
|
39
|
+
);
|
40
|
+
|
41
|
+
const user = await UserModel.findById(clientDB as any, this.userId);
|
42
|
+
const messageCount = await this.messageModel.count();
|
43
|
+
const sessionCount = await this.sessionModel.count();
|
44
|
+
|
45
|
+
return {
|
46
|
+
...state,
|
47
|
+
avatar: user?.avatar as string,
|
48
|
+
canEnablePWAGuide: messageCount >= 4,
|
49
|
+
canEnableTrace: messageCount >= 4,
|
50
|
+
hasConversation: messageCount > 0 || sessionCount > 0,
|
51
|
+
isOnboard: true,
|
52
|
+
preference: await this.preferenceStorage.getFromLocalStorage(),
|
53
|
+
};
|
54
|
+
}
|
55
|
+
|
56
|
+
updateUserSettings = async (value: DeepPartial<UserSettings>) => {
|
57
|
+
const { keyVaults, ...res } = value;
|
58
|
+
|
59
|
+
return this.userModel.updateSetting({ ...res, keyVaults: JSON.stringify(keyVaults) });
|
60
|
+
};
|
61
|
+
|
62
|
+
resetUserSettings = async () => {
|
63
|
+
return this.userModel.deleteSetting();
|
64
|
+
};
|
65
|
+
|
66
|
+
async updateAvatar(avatar: string) {
|
67
|
+
await this.userModel.updateUser({ avatar });
|
68
|
+
}
|
69
|
+
|
70
|
+
async updatePreference(preference: Partial<UserPreference>) {
|
71
|
+
await this.preferenceStorage.saveToLocalStorage(preference);
|
72
|
+
}
|
73
|
+
|
74
|
+
// eslint-disable-next-line @typescript-eslint/no-unused-vars,unused-imports/no-unused-vars
|
75
|
+
async updateGuide(guide: Partial<UserGuide>) {
|
76
|
+
throw new Error('Method not implemented.');
|
77
|
+
}
|
78
|
+
|
79
|
+
async makeSureUserExist() {
|
80
|
+
const existUsers = await clientDB.query.users.findMany();
|
81
|
+
|
82
|
+
let user: { id: string };
|
83
|
+
if (existUsers.length === 0) {
|
84
|
+
const result = await clientDB.insert(users).values({ id: this.userId }).returning();
|
85
|
+
user = result[0];
|
86
|
+
} else {
|
87
|
+
user = existUsers[0];
|
88
|
+
}
|
89
|
+
|
90
|
+
return user;
|
91
|
+
}
|
92
|
+
}
|
@@ -47,10 +47,9 @@ describe('chatToolSlice', () => {
|
|
47
47
|
url: '',
|
48
48
|
});
|
49
49
|
vi.spyOn(result.current, 'toggleDallEImageLoading');
|
50
|
-
vi.spyOn(ClientService.prototype, 'checkFileHash').mockImplementation(
|
51
|
-
isExist: false,
|
52
|
-
|
53
|
-
}));
|
50
|
+
vi.spyOn(ClientService.prototype, 'checkFileHash').mockImplementation(
|
51
|
+
async () => ({ isExist: false }) as any,
|
52
|
+
);
|
54
53
|
|
55
54
|
await act(async () => {
|
56
55
|
await result.current.generateImageFromPrompts(prompts, messageId);
|
@@ -0,0 +1,51 @@
|
|
1
|
+
import { SWRResponse } from 'swr';
|
2
|
+
import type { StateCreator } from 'zustand/vanilla';
|
3
|
+
|
4
|
+
import { useOnlyFetchOnceSWR } from '@/libs/swr';
|
5
|
+
import type { GlobalStore } from '@/store/global';
|
6
|
+
import { DatabaseLoadingState, OnStageChange } from '@/types/clientDB';
|
7
|
+
|
8
|
+
type InitClientDBParams = { onStateChange: OnStageChange };
|
9
|
+
/**
|
10
|
+
* 设置操作
|
11
|
+
*/
|
12
|
+
export interface GlobalClientDBAction {
|
13
|
+
initializeClientDB: (params?: InitClientDBParams) => Promise<void>;
|
14
|
+
markPgliteEnabled: () => void;
|
15
|
+
useInitClientDB: (params?: InitClientDBParams) => SWRResponse;
|
16
|
+
}
|
17
|
+
|
18
|
+
export const clientDBSlice: StateCreator<
|
19
|
+
GlobalStore,
|
20
|
+
[['zustand/devtools', never]],
|
21
|
+
[],
|
22
|
+
GlobalClientDBAction
|
23
|
+
> = (set, get) => ({
|
24
|
+
initializeClientDB: async (params) => {
|
25
|
+
// if the db has started initialized or not error, just skip.
|
26
|
+
if (
|
27
|
+
get().initClientDBStage !== DatabaseLoadingState.Idle &&
|
28
|
+
get().initClientDBStage !== DatabaseLoadingState.Error
|
29
|
+
)
|
30
|
+
return;
|
31
|
+
|
32
|
+
const { initializeDB } = await import('@/database/client/db');
|
33
|
+
await initializeDB({
|
34
|
+
onError: (error) => {
|
35
|
+
set({ initClientDBError: error });
|
36
|
+
},
|
37
|
+
onProgress: (data) => {
|
38
|
+
set({ initClientDBProcess: data });
|
39
|
+
},
|
40
|
+
onStateChange: (state) => {
|
41
|
+
set({ initClientDBStage: state });
|
42
|
+
params?.onStateChange?.(state);
|
43
|
+
},
|
44
|
+
});
|
45
|
+
},
|
46
|
+
markPgliteEnabled: () => {
|
47
|
+
get().updateSystemStatus({ isEnablePglite: true });
|
48
|
+
},
|
49
|
+
useInitClientDB: (params) =>
|
50
|
+
useOnlyFetchOnceSWR('initClientDB', () => get().initializeClientDB(params)),
|
51
|
+
});
|
@@ -1,5 +1,6 @@
|
|
1
1
|
import { AppRouterInstance } from 'next/dist/shared/lib/app-router-context.shared-runtime';
|
2
2
|
|
3
|
+
import { DatabaseLoadingState } from '@/types/clientDB';
|
3
4
|
import { SessionDefaultGroup } from '@/types/session';
|
4
5
|
import { AsyncLocalStorage } from '@/utils/localStorage';
|
5
6
|
|
@@ -37,6 +38,10 @@ export interface SystemStatus {
|
|
37
38
|
hidePWAInstaller?: boolean;
|
38
39
|
hideThreadLimitAlert?: boolean;
|
39
40
|
inputHeight: number;
|
41
|
+
/**
|
42
|
+
* 应用初始化时不启用 PGLite,只有当用户手动开启时才启用
|
43
|
+
*/
|
44
|
+
isEnablePglite?: boolean;
|
40
45
|
mobileShowPortal?: boolean;
|
41
46
|
mobileShowTopic?: boolean;
|
42
47
|
sessionsWidth: number;
|
@@ -50,6 +55,13 @@ export interface SystemStatus {
|
|
50
55
|
|
51
56
|
export interface GlobalState {
|
52
57
|
hasNewVersion?: boolean;
|
58
|
+
initClientDBError?: Error;
|
59
|
+
initClientDBProcess?: { costTime?: number; phase: 'wasm' | 'dependencies'; progress: number };
|
60
|
+
/**
|
61
|
+
* 客户端数据库初始化状态
|
62
|
+
* 启动时为 Idle,完成为 Ready,报错为 Error
|
63
|
+
*/
|
64
|
+
initClientDBStage: DatabaseLoadingState;
|
53
65
|
isMobile?: boolean;
|
54
66
|
isStatusInit?: boolean;
|
55
67
|
latestVersion?: string;
|
@@ -76,6 +88,7 @@ export const INITIAL_STATUS = {
|
|
76
88
|
} satisfies SystemStatus;
|
77
89
|
|
78
90
|
export const initialState: GlobalState = {
|
91
|
+
initClientDBStage: DatabaseLoadingState.Idle,
|
79
92
|
isMobile: false,
|
80
93
|
isStatusInit: false,
|
81
94
|
sidebarKey: SidebarTabKey.Chat,
|
@@ -1,4 +1,6 @@
|
|
1
|
+
import { isServerMode, isUsePgliteDB } from '@/const/version';
|
1
2
|
import { GlobalStore } from '@/store/global';
|
3
|
+
import { DatabaseLoadingState } from '@/types/clientDB';
|
2
4
|
|
3
5
|
import { INITIAL_STATUS } from './initialState';
|
4
6
|
|
@@ -22,17 +24,36 @@ const filePanelWidth = (s: GlobalStore) => s.status.filePanelWidth;
|
|
22
24
|
const inputHeight = (s: GlobalStore) => s.status.inputHeight;
|
23
25
|
const threadInputHeight = (s: GlobalStore) => s.status.threadInputHeight;
|
24
26
|
|
25
|
-
const isPgliteNotEnabled = () =>
|
27
|
+
const isPgliteNotEnabled = (s: GlobalStore) =>
|
28
|
+
isUsePgliteDB && !isServerMode && s.isStatusInit && !s.status.isEnablePglite;
|
26
29
|
|
27
|
-
|
30
|
+
/**
|
31
|
+
* 当且仅当 client db 模式,且 pglite 未初始化完成时返回 true
|
32
|
+
*/
|
33
|
+
const isPgliteNotInited = (s: GlobalStore) =>
|
34
|
+
isUsePgliteDB &&
|
35
|
+
s.isStatusInit &&
|
36
|
+
s.status.isEnablePglite &&
|
37
|
+
s.initClientDBStage !== DatabaseLoadingState.Ready;
|
28
38
|
|
29
|
-
|
39
|
+
/**
|
40
|
+
* 当且仅当 client db 模式,且 pglite 初始化完成时返回 true
|
41
|
+
*/
|
42
|
+
const isPgliteInited = (s: GlobalStore): boolean =>
|
43
|
+
(s.isStatusInit &&
|
44
|
+
s.status.isEnablePglite &&
|
45
|
+
s.initClientDBStage === DatabaseLoadingState.Ready) ||
|
46
|
+
false;
|
47
|
+
|
48
|
+
// 这个变量控制 clientdb 是否完成初始化,正常来说,只有 pgliteDB 模式下,才会存在变化,其他时候都是 true
|
49
|
+
const isDBInited = (s: GlobalStore): boolean => (isUsePgliteDB ? isPgliteInited(s) : true);
|
30
50
|
|
31
51
|
export const systemStatusSelectors = {
|
32
52
|
filePanelWidth,
|
33
53
|
hidePWAInstaller,
|
34
54
|
inZenMode,
|
35
55
|
inputHeight,
|
56
|
+
isDBInited,
|
36
57
|
isPgliteInited,
|
37
58
|
isPgliteNotEnabled,
|
38
59
|
isPgliteNotInited,
|
@@ -5,15 +5,17 @@ import { StateCreator } from 'zustand/vanilla';
|
|
5
5
|
|
6
6
|
import { createDevtools } from '../middleware/createDevtools';
|
7
7
|
import { type GlobalStoreAction, globalActionSlice } from './action';
|
8
|
+
import { type GlobalClientDBAction, clientDBSlice } from './actions/clientDb';
|
8
9
|
import { type GlobalState, initialState } from './initialState';
|
9
10
|
|
10
11
|
// =============== 聚合 createStoreFn ============ //
|
11
12
|
|
12
|
-
export type GlobalStore = GlobalState & GlobalStoreAction;
|
13
|
+
export type GlobalStore = GlobalState & GlobalStoreAction & GlobalClientDBAction;
|
13
14
|
|
14
15
|
const createStore: StateCreator<GlobalStore, [['zustand/devtools', never]]> = (...parameters) => ({
|
15
16
|
...initialState,
|
16
17
|
...globalActionSlice(...parameters),
|
18
|
+
...clientDBSlice(...parameters),
|
17
19
|
});
|
18
20
|
|
19
21
|
// =============== 实装 useStore ============ //
|
@@ -10,14 +10,14 @@ describe('sessionGroupsReducer', () => {
|
|
10
10
|
{
|
11
11
|
id: nanoid(),
|
12
12
|
name: 'Group 1',
|
13
|
-
createdAt: Date
|
14
|
-
updatedAt: Date
|
13
|
+
createdAt: new Date(),
|
14
|
+
updatedAt: new Date(),
|
15
15
|
},
|
16
16
|
{
|
17
17
|
id: nanoid(),
|
18
18
|
name: 'Group 2',
|
19
|
-
createdAt: Date
|
20
|
-
updatedAt: Date
|
19
|
+
createdAt: new Date(),
|
20
|
+
updatedAt: new Date(),
|
21
21
|
sort: 1,
|
22
22
|
},
|
23
23
|
];
|
@@ -26,8 +26,8 @@ describe('sessionGroupsReducer', () => {
|
|
26
26
|
const newItem: SessionGroupItem = {
|
27
27
|
id: nanoid(),
|
28
28
|
name: 'New Group',
|
29
|
-
createdAt: Date
|
30
|
-
updatedAt: Date
|
29
|
+
createdAt: new Date(),
|
30
|
+
updatedAt: new Date(),
|
31
31
|
};
|
32
32
|
|
33
33
|
const result = sessionGroupsReducer(initialState, {
|
@@ -46,11 +46,9 @@ export const createCommonSlice: StateCreator<
|
|
46
46
|
await mutate(GET_USER_STATE_KEY);
|
47
47
|
},
|
48
48
|
updateAvatar: async (avatar) => {
|
49
|
-
const {
|
49
|
+
const { userClientService } = await import('@/services/user');
|
50
50
|
|
51
|
-
|
52
|
-
|
53
|
-
await clientService.updateAvatar(avatar);
|
51
|
+
await userClientService.updateAvatar(avatar);
|
54
52
|
await get().refreshUserState();
|
55
53
|
},
|
56
54
|
|
@@ -0,0 +1,29 @@
|
|
1
|
+
// 定义加载状态类型
|
2
|
+
export enum DatabaseLoadingState {
|
3
|
+
Error = 'error',
|
4
|
+
Finished = 'finished',
|
5
|
+
Idle = 'idle',
|
6
|
+
Initializing = 'initializing',
|
7
|
+
LoadingDependencies = 'loadingDependencies',
|
8
|
+
LoadingWasm = 'loadingWasm',
|
9
|
+
Migrating = 'migrating',
|
10
|
+
Ready = 'ready',
|
11
|
+
}
|
12
|
+
|
13
|
+
export const ClientDatabaseInitStages = [
|
14
|
+
DatabaseLoadingState.Idle,
|
15
|
+
DatabaseLoadingState.Initializing,
|
16
|
+
DatabaseLoadingState.LoadingDependencies,
|
17
|
+
DatabaseLoadingState.LoadingWasm,
|
18
|
+
DatabaseLoadingState.Migrating,
|
19
|
+
DatabaseLoadingState.Finished,
|
20
|
+
];
|
21
|
+
|
22
|
+
// 定义进度回调接口
|
23
|
+
export interface ClientDBLoadingProgress {
|
24
|
+
costTime?: number;
|
25
|
+
phase: 'wasm' | 'dependencies';
|
26
|
+
progress: number;
|
27
|
+
}
|
28
|
+
|
29
|
+
export type OnStageChange = (state: DatabaseLoadingState) => void;
|
package/src/types/importer.ts
CHANGED
@@ -9,9 +9,8 @@ import {
|
|
9
9
|
} from '@/types/message';
|
10
10
|
import { MetaData } from '@/types/meta';
|
11
11
|
import { SessionGroupId } from '@/types/session';
|
12
|
-
import { ChatTopic } from '@/types/topic';
|
13
12
|
|
14
|
-
interface ImportSession {
|
13
|
+
export interface ImportSession {
|
15
14
|
config: LobeAgentConfig;
|
16
15
|
createdAt: string;
|
17
16
|
group?: SessionGroupId;
|
@@ -22,7 +21,7 @@ interface ImportSession {
|
|
22
21
|
updatedAt: string;
|
23
22
|
}
|
24
23
|
|
25
|
-
interface ImportMessage {
|
24
|
+
export interface ImportMessage {
|
26
25
|
content: string;
|
27
26
|
createdAt: number;
|
28
27
|
error?: ChatMessageError;
|
@@ -64,19 +63,32 @@ interface ImportMessage {
|
|
64
63
|
updatedAt: number;
|
65
64
|
}
|
66
65
|
|
67
|
-
interface ImportSessionGroup {
|
66
|
+
export interface ImportSessionGroup {
|
68
67
|
createdAt: number;
|
69
68
|
id: string;
|
70
69
|
name: string;
|
71
70
|
sort?: number | null;
|
72
71
|
updatedAt: number;
|
73
72
|
}
|
73
|
+
export interface ImportTopic {
|
74
|
+
createdAt: number;
|
75
|
+
favorite?: boolean;
|
76
|
+
historySummary?: string;
|
77
|
+
id: string;
|
78
|
+
metadata?: {
|
79
|
+
model?: string;
|
80
|
+
provider?: string;
|
81
|
+
};
|
82
|
+
sessionId?: string;
|
83
|
+
title: string;
|
84
|
+
updatedAt: number;
|
85
|
+
}
|
74
86
|
|
75
87
|
export interface ImporterEntryData {
|
76
88
|
messages?: ImportMessage[];
|
77
89
|
sessionGroups?: ImportSessionGroup[];
|
78
90
|
sessions?: ImportSession[];
|
79
|
-
topics?:
|
91
|
+
topics?: ImportTopic[];
|
80
92
|
version: number;
|
81
93
|
}
|
82
94
|
|
package/src/types/meta.ts
CHANGED
@@ -21,19 +21,10 @@ export const LobeMetaDataSchema = z.object({
|
|
21
21
|
export type MetaData = z.infer<typeof LobeMetaDataSchema>;
|
22
22
|
|
23
23
|
export interface BaseDataModel {
|
24
|
-
/**
|
25
|
-
* @deprecated
|
26
|
-
*/
|
27
|
-
createAt?: number;
|
28
|
-
|
29
24
|
createdAt: number;
|
30
25
|
|
31
26
|
id: string;
|
32
27
|
meta: MetaData;
|
33
28
|
|
34
|
-
/**
|
35
|
-
* @deprecated
|
36
|
-
*/
|
37
|
-
updateAt?: number;
|
38
29
|
updatedAt: number;
|
39
30
|
}
|
@@ -8,11 +8,11 @@ export enum SessionDefaultGroup {
|
|
8
8
|
export type SessionGroupId = SessionDefaultGroup | string;
|
9
9
|
|
10
10
|
export interface SessionGroupItem {
|
11
|
-
createdAt:
|
11
|
+
createdAt: Date;
|
12
12
|
id: string;
|
13
13
|
name: string;
|
14
|
-
sort?: number;
|
15
|
-
updatedAt:
|
14
|
+
sort?: number | null;
|
15
|
+
updatedAt: Date;
|
16
16
|
}
|
17
17
|
|
18
18
|
export type SessionGroups = SessionGroupItem[];
|
@@ -1,48 +0,0 @@
|
|
1
|
-
import { Mock, describe, expect, it, vi } from 'vitest';
|
2
|
-
|
3
|
-
import { CreateMessageParams, MessageModel } from '@/database/_deprecated/models/message';
|
4
|
-
import { ChatMessage, ChatMessageError, ChatPluginPayload } from '@/types/message';
|
5
|
-
|
6
|
-
import { messageService } from './index';
|
7
|
-
|
8
|
-
// Mock the MessageModel
|
9
|
-
vi.mock('@/database/_deprecated/models/message', () => {
|
10
|
-
return {
|
11
|
-
MessageModel: {
|
12
|
-
count: vi.fn(),
|
13
|
-
},
|
14
|
-
};
|
15
|
-
});
|
16
|
-
|
17
|
-
describe('MessageService', () => {
|
18
|
-
beforeEach(() => {
|
19
|
-
// Reset all mocks before running each test case
|
20
|
-
vi.resetAllMocks();
|
21
|
-
});
|
22
|
-
|
23
|
-
describe('hasMessages', () => {
|
24
|
-
it('should return true if there are messages', async () => {
|
25
|
-
// Setup
|
26
|
-
(MessageModel.count as Mock).mockResolvedValue(1);
|
27
|
-
|
28
|
-
// Execute
|
29
|
-
const hasMessages = await messageService.hasMessages();
|
30
|
-
|
31
|
-
// Assert
|
32
|
-
expect(MessageModel.count).toHaveBeenCalled();
|
33
|
-
expect(hasMessages).toBe(true);
|
34
|
-
});
|
35
|
-
|
36
|
-
it('should return false if there are no messages', async () => {
|
37
|
-
// Setup
|
38
|
-
(MessageModel.count as Mock).mockResolvedValue(0);
|
39
|
-
|
40
|
-
// Execute
|
41
|
-
const hasMessages = await messageService.hasMessages();
|
42
|
-
|
43
|
-
// Assert
|
44
|
-
expect(MessageModel.count).toHaveBeenCalled();
|
45
|
-
expect(hasMessages).toBe(false);
|
46
|
-
});
|
47
|
-
});
|
48
|
-
});
|