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

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

@@ -1,245 +0,0 @@
-import { Mock, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { SessionModel } from '@/database/_deprecated/models/session';
-import { CreateTopicParams, TopicModel } from '@/database/_deprecated/models/topic';
-import { ChatTopic } from '@/types/topic';
-
-import { ClientService } from './_deprecated';
-
-const topicService = new ClientService();
-// Mock the TopicModel
-vi.mock('@/database/_deprecated/models/topic', () => {
-  return {
-    TopicModel: {
-      create: vi.fn(),
-      query: vi.fn(),
-      delete: vi.fn(),
-      count: vi.fn(),
-      batchDeleteBySessionId: vi.fn(),
-      batchDelete: vi.fn(),
-      clearTable: vi.fn(),
-      toggleFavorite: vi.fn(),
-      batchCreate: vi.fn(),
-      update: vi.fn(),
-      queryAll: vi.fn(),
-      queryByKeyword: vi.fn(),
-    },
-  };
-});
-
-describe('TopicService', () => {
-  // Mock data
-  const mockTopicId = 'mock-topic-id';
-  const mockTopic: ChatTopic = {
-    createdAt: 100,
-    updatedAt: 100,
-    id: mockTopicId,
-    title: 'Mock Topic',
-  };
-  const mockTopics = [mockTopic];
-
-  beforeEach(() => {
-    // Reset all mocks before running each test case
-    vi.resetAllMocks();
-  });
-
-  describe('createTopic', () => {
-    it('should create a topic and return its id', async () => {
-      // Setup
-      const createParams: CreateTopicParams = {
-        title: 'New Topic',
-        sessionId: '1',
-      };
-      (TopicModel.create as Mock).mockResolvedValue(mockTopic);
-
-      // Execute
-      const topicId = await topicService.createTopic(createParams);
-
-      // Assert
-      expect(TopicModel.create).toHaveBeenCalledWith(createParams);
-      expect(topicId).toBe(mockTopicId);
-    });
-    it('should throw an error if topic creation fails', async () => {
-      // Setup
-      const createParams: CreateTopicParams = {
-        title: 'New Topic',
-        sessionId: '1',
-      };
-
-      (TopicModel.create as Mock).mockResolvedValue(null);
-
-      // Execute & Assert
-      await expect(topicService.createTopic(createParams)).rejects.toThrow('topic create Error');
-    });
-  });
-
-  describe('getTopics', () => {
-    // Example for getTopics
-    it('should query topics with given parameters', async () => {
-      // Setup
-      const queryParams = { containerId: 'session-id' };
-      (TopicModel.query as Mock).mockResolvedValue(mockTopics);
-
-      // Execute
-      const topics = await topicService.getTopics(queryParams);
-
-      // Assert
-      expect(TopicModel.query).toHaveBeenCalledWith(queryParams);
-      expect(topics).toBe(mockTopics);
-    });
-  });
-
-  describe('updateTopic', () => {
-    // Example for updateFavorite
-    it('should toggle favorite status of a topic', async () => {
-      // Setup
-      const newState = true;
-
-      // Execute
-      await topicService.updateTopic(mockTopicId, { favorite: newState });
-
-      // Assert
-      expect(TopicModel.update).toHaveBeenCalledWith(mockTopicId, { favorite: 1 });
-    });
-
-    it('should update the title of a topic', async () => {
-      // Setup
-      const newTitle = 'Updated Topic Title';
-      (TopicModel.update as Mock).mockResolvedValue({ ...mockTopic, title: newTitle });
-
-      // Execute
-      const result = await topicService.updateTopic(mockTopicId, { title: newTitle });
-
-      // Assert
-      expect(TopicModel.update).toHaveBeenCalledWith(mockTopicId, { title: newTitle });
-      expect(result).toEqual({ ...mockTopic, title: newTitle });
-    });
-  });
-
-  describe('removeTopic', () => {
-    it('should remove a topic by id', async () => {
-      // Setup
-      (TopicModel.delete as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await topicService.removeTopic(mockTopicId);
-
-      // Assert
-      expect(TopicModel.delete).toHaveBeenCalledWith(mockTopicId);
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('removeTopics', () => {
-    it('should remove all topics with a given session id', async () => {
-      // Setup
-      const sessionId = 'session-id';
-      (TopicModel.batchDeleteBySessionId as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await topicService.removeTopics(sessionId);
-
-      // Assert
-      expect(TopicModel.batchDeleteBySessionId).toHaveBeenCalledWith(sessionId);
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('batchRemoveTopics', () => {
-    it('should batch remove topics', async () => {
-      // Setup
-      const topicIds = [mockTopicId, 'another-topic-id'];
-      (TopicModel.batchDelete as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await topicService.batchRemoveTopics(topicIds);
-
-      // Assert
-      expect(TopicModel.batchDelete).toHaveBeenCalledWith(topicIds);
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('removeAllTopic', () => {
-    it('should clear all topics from the table', async () => {
-      // Setup
-      (TopicModel.clearTable as Mock).mockResolvedValue(true);
-
-      // Execute
-      const result = await topicService.removeAllTopic();
-
-      // Assert
-      expect(TopicModel.clearTable).toHaveBeenCalled();
-      expect(result).toBe(true);
-    });
-  });
-
-  describe('batchCreateTopics', () => {
-    it('should batch create topics', async () => {
-      // Setup
-      (TopicModel.batchCreate as Mock).mockResolvedValue(mockTopics);
-
-      // Execute
-      const result = await topicService.batchCreateTopics(mockTopics);
-
-      // Assert
-      expect(TopicModel.batchCreate).toHaveBeenCalledWith(mockTopics);
-      expect(result).toBe(mockTopics);
-    });
-  });
-
-  describe('getAllTopics', () => {
-    it('should retrieve all topics', async () => {
-      // Setup
-      (TopicModel.queryAll as Mock).mockResolvedValue(mockTopics);
-
-      // Execute
-      const result = await topicService.getAllTopics();
-
-      // Assert
-      expect(TopicModel.queryAll).toHaveBeenCalled();
-      expect(result).toBe(mockTopics);
-    });
-  });
-
-  describe('searchTopics', () => {
-    it('should return all topics that match the keyword', async () => {
-      // Setup
-      const keyword = 'search';
-      (TopicModel.queryByKeyword as Mock).mockResolvedValue(mockTopics);
-
-      // Execute
-      const result = await topicService.searchTopics(keyword, undefined);
-
-      // Assert
-      expect(TopicModel.queryByKeyword).toHaveBeenCalledWith(keyword, undefined);
-      expect(result).toBe(mockTopics);
-    });
-  });
-
-  describe('countTopics', () => {
-    it('should return false if no topics exist', async () => {
-      // Setup
-      (TopicModel.count as Mock).mockResolvedValue(0);
-
-      // Execute
-      const result = await topicService.countTopics();
-
-      // Assert
-      expect(TopicModel.count).toHaveBeenCalled();
-      expect(result).toBe(0);
-    });
-
-    it('should return true if topics exist', async () => {
-      // Setup
-      (TopicModel.count as Mock).mockResolvedValue(1);
-
-      // Execute
-      const result = await topicService.countTopics();
-
-      // Assert
-      expect(TopicModel.count).toHaveBeenCalled();
-      expect(result).toBe(1);
-    });
-  });
-});

package/src/services/topic/_deprecated.ts

@@ -1,75 +0,0 @@
-import { TopicModel } from '@/database/_deprecated/models/topic';
-import { ChatTopic } from '@/types/topic';
-
-import { CreateTopicParams, ITopicService, QueryTopicParams } from './type';
-
-export class ClientService implements ITopicService {
-  async createTopic(params: CreateTopicParams): Promise<string> {
-    const item = await TopicModel.create(params as any);
-
-    if (!item) {
-      throw new Error('topic create Error');
-    }
-
-    return item.id;
-  }
-
-  async batchCreateTopics(importTopics: ChatTopic[]) {
-    return TopicModel.batchCreate(importTopics as any);
-  }
-
-  async cloneTopic(id: string, newTitle?: string) {
-    return TopicModel.duplicateTopic(id, newTitle);
-  }
-
-  async getTopics(params: QueryTopicParams): Promise<ChatTopic[]> {
-    return TopicModel.query(params);
-  }
-
-  async searchTopics(keyword: string, sessionId?: string) {
-    return TopicModel.queryByKeyword(keyword, sessionId);
-  }
-
-  async getAllTopics() {
-    return TopicModel.queryAll();
-  }
-
-  async countTopics() {
-    return TopicModel.count();
-  }
-
-  // @ts-ignore
-  async rankTopics() {
-    throw new Error('Method not implemented.');
-  }
-
-  async updateTopicFavorite(id: string, favorite?: boolean) {
-    return this.updateTopic(id, { favorite });
-  }
-
-  async updateTopicTitle(id: string, text: string) {
-    return this.updateTopic(id, { title: text });
-  }
-
-  async updateTopic(id: string, data: Partial<ChatTopic>) {
-    const favorite = typeof data.favorite !== 'undefined' ? (data.favorite ? 1 : 0) : undefined;
-
-    return TopicModel.update(id, { ...data, favorite });
-  }
-
-  async removeTopic(id: string) {
-    return TopicModel.delete(id);
-  }
-
-  async removeTopics(sessionId: string) {
-    return TopicModel.batchDeleteBySessionId(sessionId);
-  }
-
-  async batchRemoveTopics(topics: string[]) {
-    return TopicModel.batchDelete(topics);
-  }
-
-  async removeAllTopic() {
-    return TopicModel.clearTable();
-  }
-}

package/src/services/topic/client.ts

@@ -1,89 +0,0 @@
-import { INBOX_SESSION_ID } from '@/const/session';
-import { clientDB } from '@/database/client/db';
-import { TopicModel } from '@/database/models/topic';
-import { BaseClientService } from '@/services/baseClientService';
-import { ChatTopic } from '@/types/topic';
-
-import { ITopicService } from './type';
-
-export class ClientService extends BaseClientService implements ITopicService {
-  private get topicModel(): TopicModel {
-    return new TopicModel(clientDB as any, this.userId);
-  }
-
-  createTopic: ITopicService['createTopic'] = async (params) => {
-    const item = await this.topicModel.create({
-      ...params,
-      sessionId: this.toDbSessionId(params.sessionId),
-    } as any);
-
-    if (!item) {
-      throw new Error('topic create Error');
-    }
-
-    return item.id;
-  };
-
-  batchCreateTopics: ITopicService['batchCreateTopics'] = async (importTopics) => {
-    const data = await this.topicModel.batchCreate(importTopics as any);
-
-    return { added: data.length, ids: [], skips: [], success: true };
-  };
-
-  cloneTopic: ITopicService['cloneTopic'] = async (id, newTitle) => {
-    const data = await this.topicModel.duplicate(id, newTitle);
-    return data.topic.id;
-  };
-
-  getTopics: ITopicService['getTopics'] = async (params) => {
-    const data = await this.topicModel.query({
-      ...params,
-      containerId: this.toDbSessionId(params.containerId),
-    });
-    return data as unknown as Promise<ChatTopic[]>;
-  };
-
-  searchTopics: ITopicService['searchTopics'] = async (keyword, sessionId) => {
-    const data = await this.topicModel.queryByKeyword(keyword, this.toDbSessionId(sessionId));
-
-    return data as unknown as Promise<ChatTopic[]>;
-  };
-
-  getAllTopics: ITopicService['getAllTopics'] = async () => {
-    const data = await this.topicModel.queryAll();
-
-    return data as unknown as Promise<ChatTopic[]>;
-  };
-
-  countTopics: ITopicService['countTopics'] = async (params) => {
-    return this.topicModel.count(params);
-  };
-
-  rankTopics: ITopicService['rankTopics'] = async (limit) => {
-    return this.topicModel.rank(limit);
-  };
-
-  updateTopic: ITopicService['updateTopic'] = async (id, data) => {
-    return this.topicModel.update(id, data as any);
-  };
-
-  removeTopic: ITopicService['removeTopic'] = async (id) => {
-    return this.topicModel.delete(id);
-  };
-
-  removeTopics: ITopicService['removeTopics'] = async (sessionId) => {
-    return this.topicModel.batchDeleteBySessionId(this.toDbSessionId(sessionId));
-  };
-
-  batchRemoveTopics: ITopicService['batchRemoveTopics'] = async (topics) => {
-    return this.topicModel.batchDelete(topics);
-  };
-
-  removeAllTopic: ITopicService['removeAllTopic'] = async () => {
-    return this.topicModel.deleteAll();
-  };
-
-  private toDbSessionId(sessionId?: string | null) {
-    return sessionId === INBOX_SESSION_ID ? null : sessionId;
-  }
-}

package/src/services/topic/pglite.test.ts

@@ -1,212 +0,0 @@
-import { eq } from 'drizzle-orm';
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { clientDB, initializeDB } from '@/database/client/db';
-import { sessions, topics, users } from '@/database/schemas';
-import { ChatTopic } from '@/types/topic';
-
-import { ClientService } from './client';
-
-// Mock data
-const userId = 'topic-user-test';
-const sessionId = 'topic-session';
-const mockTopicId = 'mock-topic-id';
-
-const mockTopic = {
-  id: mockTopicId,
-  title: 'Mock Topic',
-};
-
-const topicService = new ClientService(userId);
-
-beforeEach(async () => {
-  await initializeDB();
-
-  await clientDB.delete(users);
-
-  // 创建测试数据
-  await clientDB.transaction(async (tx) => {
-    await tx.insert(users).values({ id: userId }).onConflictDoNothing();
-    await tx.insert(sessions).values({ id: sessionId, userId });
-    await tx.insert(topics).values({ ...mockTopic, sessionId, userId });
-  });
-});
-
-describe('TopicService', () => {
-  describe('createTopic', () => {
-    it('should create a topic and return its id', async () => {
-      // Setup
-      const createParams = {
-        title: 'New Topic',
-        sessionId: sessionId,
-      };
-
-      // Execute
-      const topicId = await topicService.createTopic(createParams);
-
-      // Assert
-      expect(topicId).toBeDefined();
-    });
-
-    it('should throw an error if topic creation fails', async () => {
-      // Setup
-      const createParams = {
-        title: 'New Topic',
-        sessionId: 123 as any, // sessionId should be string
-      };
-
-      // Execute & Assert
-      await expect(topicService.createTopic(createParams)).rejects.toThrowError();
-    });
-  });
-
-  describe('getTopics', () => {
-    // Example for getTopics
-    it('should query topics with given parameters', async () => {
-      // Setup
-      const queryParams = { containerId: sessionId };
-
-      // Execute
-      const data = await topicService.getTopics(queryParams);
-
-      // Assert
-      expect(data[0]).toMatchObject(mockTopic);
-    });
-  });
-
-  describe('updateTopic', () => {
-    // Example for updateFavorite
-    it('should toggle favorite status of a topic', async () => {
-      // Execute
-      const result = await topicService.updateTopic(mockTopicId, { favorite: true });
-
-      // Assert
-      expect(result[0].favorite).toBeTruthy();
-    });
-
-    it('should update the title of a topic', async () => {
-      // Setup
-      const newTitle = 'Updated Topic Title';
-
-      // Execute
-      const result = await topicService.updateTopic(mockTopicId, { title: newTitle });
-
-      // Assert
-      expect(result[0].title).toEqual(newTitle);
-    });
-  });
-
-  describe('removeTopic', () => {
-    it('should remove a topic by id', async () => {
-      // Execute
-      await topicService.removeTopic(mockTopicId);
-      const result = await clientDB.query.topics.findFirst({ where: eq(topics.id, mockTopicId) });
-
-      // Assert
-      expect(result).toBeUndefined();
-    });
-  });
-
-  describe('removeTopics', () => {
-    it('should remove all topics with a given session id', async () => {
-      // Setup
-      const sessionId = 'session-id';
-
-      // Execute
-      await topicService.removeTopics(sessionId);
-      const result = await clientDB.query.topics.findMany({
-        where: eq(topics.sessionId, sessionId),
-      });
-
-      expect(result.length).toEqual(0);
-    });
-  });
-
-  describe('batchRemoveTopics', () => {
-    it('should batch remove topics', async () => {
-      await clientDB.insert(topics).values([{ id: 'topic-id-1', title: 'topic-title', userId }]);
-      // Setup
-      const topicIds = [mockTopicId, 'another-topic-id'];
-
-      // Execute
-      await topicService.batchRemoveTopics(topicIds);
-
-      const count = await clientDB.$count(topics);
-
-      // Assert
-      expect(count).toBe(1);
-    });
-  });
-
-  describe('removeAllTopic', () => {
-    it('should clear all topics from the table', async () => {
-      // Execute
-      await topicService.removeAllTopic();
-
-      const count = await clientDB.$count(topics);
-      // Assert
-      expect(count).toBe(0);
-    });
-  });
-
-  describe('batchCreateTopics', () => {
-    it('should batch create topics', async () => {
-      // Execute
-      const result = await topicService.batchCreateTopics([
-        { id: 'topic-id-1', title: 'topic-title' },
-        { id: 'topic-id-2', title: 'topic-title' },
-      ] as ChatTopic[]);
-
-      // Assert
-      expect(result.success).toBeTruthy();
-      expect(result.added).toBe(2);
-    });
-  });
-
-  describe('getAllTopics', () => {
-    it('should retrieve all topics', async () => {
-      await clientDB.insert(topics).values([
-        { id: 'topic-id-1', title: 'topic-title', userId },
-        { id: 'topic-id-2', title: 'topic-title', userId },
-      ]);
-      // Execute
-      const result = await topicService.getAllTopics();
-
-      // Assert
-      expect(result.length).toEqual(3);
-    });
-  });
-
-  describe('searchTopics', () => {
-    it('should return all topics that match the keyword', async () => {
-      // Setup
-      const keyword = 'Topic';
-
-      // Execute
-      const result = await topicService.searchTopics(keyword, sessionId);
-
-      // Assert
-      expect(result.length).toEqual(1);
-    });
-    it('should return empty topic if not match the keyword', async () => {
-      // Setup
-      const keyword = 'search';
-
-      // Execute
-      const result = await topicService.searchTopics(keyword, sessionId);
-
-      // Assert
-      expect(result.length).toEqual(0);
-    });
-  });
-
-  describe('countTopics', () => {
-    it('should return topic counts', async () => {
-      // Execute
-      const result = await topicService.countTopics();
-
-      // Assert
-      expect(result).toBe(1);
-    });
-  });
-});