@mindbase/express-knowledge 1.0.8 → 1.0.12

package/service/TagService.ts

@@ -1,271 +1,271 @@
-import { knowledgeTag, knowledgeTagRelation } from "../orm/Tag.schema";
-import { knowledge } from "../orm/Knowledge.schema";
-import { eq, and, sql, desc, or, inArray } from "drizzle-orm";
-import type { InsertTag } from "../types";
-
-let _db: any;
-
-/**
- * 初始化 Service 的数据库实例
- */
-export function initTagService(db: any) {
-  _db = db;
-}
-
-function getDB() {
-  if (!_db) throw new Error("TagService not initialized with DB");
-  return _db;
-}
-
-// ==================== 标签管理 ====================
-
-/**
- * 创建标签
- */
-export async function createTag(data: InsertTag & { userId?: number }) {
-  const db = getDB();
-  const now = new Date();
-
-  const result = await db
-    .insert(knowledgeTag)
-    .values({
-      name: data.name,
-      color: data.color,
-      userId: data.userId,
-      created: now,
-    })
-    .execute();
-
-  const lastId = result.lastInsertRowid as number;
-  return getTagById(lastId);
-}
-
-/**
- * 获取标签详情
- */
-export async function getTagById(id: number) {
-  const db = getDB();
-  const [tag] = await db.select().from(knowledgeTag).where(eq(knowledgeTag.id, id)).limit(1);
-
-  return tag || null;
-}
-
-/**
- * 获取标签列表
- */
-export async function listTags(userId?: number) {
-  const db = getDB();
-
-  let conditions = [];
-  if (userId !== undefined) {
-    conditions.push(eq(knowledgeTag.userId, userId));
-  }
-
-  return await db
-    .select()
-    .from(knowledgeTag)
-    .where(conditions.length > 0 ? and(...conditions) : undefined)
-    .orderBy(knowledgeTag.name)
-    .execute();
-}
-
-/**
- * 更新标签
- */
-export async function updateTag(id: number, data: Partial<InsertTag>, userId: number) {
-  const db = getDB();
-
-  // 检查权限
-  const [tag] = await db
-    .select()
-    .from(knowledgeTag)
-    .where(and(eq(knowledgeTag.id, id), eq(knowledgeTag.userId, userId)))
-    .limit(1);
-
-  if (!tag) {
-    throw new Error("标签不存在或无权操作");
-  }
-
-  await db
-    .update(knowledgeTag)
-    .set({
-      ...(data.name !== undefined && { name: data.name }),
-      ...(data.color !== undefined && { color: data.color }),
-    })
-    .where(eq(knowledgeTag.id, id))
-    .execute();
-
-  return getTagById(id);
-}
-
-/**
- * 删除标签
- */
-export async function deleteTag(id: number, userId: number) {
-  const db = getDB();
-
-  // 检查权限
-  const [tag] = await db
-    .select()
-    .from(knowledgeTag)
-    .where(and(eq(knowledgeTag.id, id), eq(knowledgeTag.userId, userId)))
-    .limit(1);
-
-  if (!tag) {
-    throw new Error("标签不存在或无权操作");
-  }
-
-  // 删除标签关联
-  await db.delete(knowledgeTagRelation).where(eq(knowledgeTagRelation.tagId, id)).execute();
-
-  // 删除标签
-  await db.delete(knowledgeTag).where(eq(knowledgeTag.id, id)).execute();
-
-  return { success: true };
-}
-
-// ==================== 标签关联 ====================
-
-/**
- * 为知识库内容添加标签
- */
-export async function addTagToKnowledge(knowledgeId: number, tagId: number, userId: number) {
-  const db = getDB();
-
-  // 检查标签是否存在
-  const [tag] = await db.select().from(knowledgeTag).where(eq(knowledgeTag.id, tagId)).limit(1);
-
-  if (!tag) {
-    throw new Error("标签不存在");
-  }
-
-  // 检查是否已存在关联
-  const [existing] = await db
-    .select()
-    .from(knowledgeTagRelation)
-    .where(and(eq(knowledgeTagRelation.knowledgeId, knowledgeId), eq(knowledgeTagRelation.tagId, tagId)))
-    .limit(1);
-
-  if (existing) {
-    return { success: true, message: "标签已存在" };
-  }
-
-  // 创建关联
-  await db
-    .insert(knowledgeTagRelation)
-    .values({
-      knowledgeId,
-      tagId,
-    })
-    .execute();
-
-  return { success: true };
-}
-
-/**
- * 移除知识库内容的标签
- */
-export async function removeTagFromKnowledge(knowledgeId: number, tagId: number) {
-  const db = getDB();
-
-  await db
-    .delete(knowledgeTagRelation)
-    .where(and(eq(knowledgeTagRelation.knowledgeId, knowledgeId), eq(knowledgeTagRelation.tagId, tagId)))
-    .execute();
-
-  return { success: true };
-}
-
-/**
- * 获取知识库内容的标签列表
- */
-export async function getKnowledgeTags(knowledgeId: number) {
-  const db = getDB();
-
-  return await db
-    .select({
-      id: knowledgeTag.id,
-      name: knowledgeTag.name,
-      color: knowledgeTag.color,
-    })
-    .from(knowledgeTagRelation)
-    .innerJoin(knowledgeTag, eq(knowledgeTagRelation.tagId, knowledgeTag.id))
-    .where(eq(knowledgeTagRelation.knowledgeId, knowledgeId))
-    .execute();
-}
-
-/**
- * 按标签查询知识库内容
- */
-export async function getKnowledgeByTag(
-  tagId: number,
-  options: {
-    pageIndex?: number;
-    pageSize?: number;
-    safe?: number[];
-    userId?: number;
-    isAdmin?: boolean;
-  }
-) {
-  const db = getDB();
-  const { pageIndex = 1, pageSize = 10, safe = [0], userId, isAdmin } = options;
-
-  // 获取带有该标签的知识库 ID 列表
-  const relations = await db.select().from(knowledgeTagRelation).where(eq(knowledgeTagRelation.tagId, tagId)).execute();
-
-  if (relations.length === 0) {
-    return { list: [], total: 0, pageIndex, pageSize };
-  }
-
-  const knowledgeIds = relations.map((r) => r.knowledgeId);
-
-  let conditions = [eq(knowledge.isDelete, 0), inArray(knowledge.id, knowledgeIds)];
-
-  // 权限筛选
-  const safeConditions = [];
-  for (const s of safe) {
-    switch (s) {
-      case 0: // 公开
-        safeConditions.push(eq(knowledge.safe, s));
-        break;
-      case 1: // 登录共享
-        if (userId) {
-          safeConditions.push(eq(knowledge.safe, s));
-        }
-        break;
-      case 2: // 个人私密
-        if (userId) {
-          safeConditions.push(and(eq(knowledge.safe, s), isAdmin ? sql`1=1` : eq(knowledge.userId, userId)));
-        }
-        break;
-    }
-  }
-
-  if (safeConditions.length > 0) {
-    conditions.push(or(...safeConditions));
-  }
-
-  const offset = (pageIndex - 1) * pageSize;
-
-  const list = await db
-    .select()
-    .from(knowledge)
-    .where(and(...conditions))
-    .limit(pageSize)
-    .offset(offset)
-    .orderBy(desc(knowledge.created))
-    .execute();
-
-  const [totalResult] = await db
-    .select({ count: sql<number>`count(*)` })
-    .from(knowledge)
-    .where(and(...conditions))
-    .execute();
-
-  return {
-    list,
-    total: totalResult.count,
-    pageIndex,
-    pageSize,
-  };
-}
+import { knowledgeTag, knowledgeTagRelation } from "../orm/Tag.schema";
+import { knowledge } from "../orm/Knowledge.schema";
+import { eq, and, sql, desc, or, inArray } from "drizzle-orm";
+import type { InsertTag } from "../types";
+
+let _db: any;
+
+/**
+ * 初始化 Service 的数据库实例
+ */
+export function initTagService(db: any) {
+  _db = db;
+}
+
+function getDB() {
+  if (!_db) throw new Error("TagService not initialized with DB");
+  return _db;
+}
+
+// ==================== 标签管理 ====================
+
+/**
+ * 创建标签
+ */
+export async function createTag(data: InsertTag & { userId?: number }) {
+  const db = getDB();
+  const now = new Date();
+
+  const result = await db
+    .insert(knowledgeTag)
+    .values({
+      name: data.name,
+      color: data.color,
+      userId: data.userId,
+      created: now,
+    })
+    .execute();
+
+  const lastId = result.lastInsertRowid as number;
+  return getTagById(lastId);
+}
+
+/**
+ * 获取标签详情
+ */
+export async function getTagById(id: number) {
+  const db = getDB();
+  const [tag] = await db.select().from(knowledgeTag).where(eq(knowledgeTag.id, id)).limit(1);
+
+  return tag || null;
+}
+
+/**
+ * 获取标签列表
+ */
+export async function listTags(userId?: number) {
+  const db = getDB();
+
+  let conditions = [];
+  if (userId !== undefined) {
+    conditions.push(eq(knowledgeTag.userId, userId));
+  }
+
+  return await db
+    .select()
+    .from(knowledgeTag)
+    .where(conditions.length > 0 ? and(...conditions) : undefined)
+    .orderBy(knowledgeTag.name)
+    .execute();
+}
+
+/**
+ * 更新标签
+ */
+export async function updateTag(id: number, data: Partial<InsertTag>, userId: number) {
+  const db = getDB();
+
+  // 检查权限
+  const [tag] = await db
+    .select()
+    .from(knowledgeTag)
+    .where(and(eq(knowledgeTag.id, id), eq(knowledgeTag.userId, userId)))
+    .limit(1);
+
+  if (!tag) {
+    throw new Error("标签不存在或无权操作");
+  }
+
+  await db
+    .update(knowledgeTag)
+    .set({
+      ...(data.name !== undefined && { name: data.name }),
+      ...(data.color !== undefined && { color: data.color }),
+    })
+    .where(eq(knowledgeTag.id, id))
+    .execute();
+
+  return getTagById(id);
+}
+
+/**
+ * 删除标签
+ */
+export async function deleteTag(id: number, userId: number) {
+  const db = getDB();
+
+  // 检查权限
+  const [tag] = await db
+    .select()
+    .from(knowledgeTag)
+    .where(and(eq(knowledgeTag.id, id), eq(knowledgeTag.userId, userId)))
+    .limit(1);
+
+  if (!tag) {
+    throw new Error("标签不存在或无权操作");
+  }
+
+  // 删除标签关联
+  await db.delete(knowledgeTagRelation).where(eq(knowledgeTagRelation.tagId, id)).execute();
+
+  // 删除标签
+  await db.delete(knowledgeTag).where(eq(knowledgeTag.id, id)).execute();
+
+  return { success: true };
+}
+
+// ==================== 标签关联 ====================
+
+/**
+ * 为知识库内容添加标签
+ */
+export async function addTagToKnowledge(knowledgeId: number, tagId: number, userId: number) {
+  const db = getDB();
+
+  // 检查标签是否存在
+  const [tag] = await db.select().from(knowledgeTag).where(eq(knowledgeTag.id, tagId)).limit(1);
+
+  if (!tag) {
+    throw new Error("标签不存在");
+  }
+
+  // 检查是否已存在关联
+  const [existing] = await db
+    .select()
+    .from(knowledgeTagRelation)
+    .where(and(eq(knowledgeTagRelation.knowledgeId, knowledgeId), eq(knowledgeTagRelation.tagId, tagId)))
+    .limit(1);
+
+  if (existing) {
+    return { success: true, message: "标签已存在" };
+  }
+
+  // 创建关联
+  await db
+    .insert(knowledgeTagRelation)
+    .values({
+      knowledgeId,
+      tagId,
+    })
+    .execute();
+
+  return { success: true };
+}
+
+/**
+ * 移除知识库内容的标签
+ */
+export async function removeTagFromKnowledge(knowledgeId: number, tagId: number) {
+  const db = getDB();
+
+  await db
+    .delete(knowledgeTagRelation)
+    .where(and(eq(knowledgeTagRelation.knowledgeId, knowledgeId), eq(knowledgeTagRelation.tagId, tagId)))
+    .execute();
+
+  return { success: true };
+}
+
+/**
+ * 获取知识库内容的标签列表
+ */
+export async function getKnowledgeTags(knowledgeId: number) {
+  const db = getDB();
+
+  return await db
+    .select({
+      id: knowledgeTag.id,
+      name: knowledgeTag.name,
+      color: knowledgeTag.color,
+    })
+    .from(knowledgeTagRelation)
+    .innerJoin(knowledgeTag, eq(knowledgeTagRelation.tagId, knowledgeTag.id))
+    .where(eq(knowledgeTagRelation.knowledgeId, knowledgeId))
+    .execute();
+}
+
+/**
+ * 按标签查询知识库内容
+ */
+export async function getKnowledgeByTag(
+  tagId: number,
+  options: {
+    pageIndex?: number;
+    pageSize?: number;
+    safe?: number[];
+    userId?: number;
+    isAdmin?: boolean;
+  }
+) {
+  const db = getDB();
+  const { pageIndex = 1, pageSize = 10, safe = [0], userId, isAdmin } = options;
+
+  // 获取带有该标签的知识库 ID 列表
+  const relations = await db.select().from(knowledgeTagRelation).where(eq(knowledgeTagRelation.tagId, tagId)).execute();
+
+  if (relations.length === 0) {
+    return { list: [], total: 0, pageIndex, pageSize };
+  }
+
+  const knowledgeIds = relations.map((r) => r.knowledgeId);
+
+  let conditions = [eq(knowledge.isDelete, 0), inArray(knowledge.id, knowledgeIds)];
+
+  // 权限筛选
+  const safeConditions = [];
+  for (const s of safe) {
+    switch (s) {
+      case 0: // 公开
+        safeConditions.push(eq(knowledge.safe, s));
+        break;
+      case 1: // 登录共享
+        if (userId) {
+          safeConditions.push(eq(knowledge.safe, s));
+        }
+        break;
+      case 2: // 个人私密
+        if (userId) {
+          safeConditions.push(and(eq(knowledge.safe, s), isAdmin ? sql`1=1` : eq(knowledge.userId, userId)));
+        }
+        break;
+    }
+  }
+
+  if (safeConditions.length > 0) {
+    conditions.push(or(...safeConditions));
+  }
+
+  const offset = (pageIndex - 1) * pageSize;
+
+  const list = await db
+    .select()
+    .from(knowledge)
+    .where(and(...conditions))
+    .limit(pageSize)
+    .offset(offset)
+    .orderBy(desc(knowledge.created))
+    .execute();
+
+  const [totalResult] = await db
+    .select({ count: sql<number>`count(*)` })
+    .from(knowledge)
+    .where(and(...conditions))
+    .execute();
+
+  return {
+    list,
+    total: totalResult.count,
+    pageIndex,
+    pageSize,
+  };
+}