feat: add knowledge directory management for documents and FAQs

- Implemented directory validation in Create and Update knowledge document services.
- Added directory selection in document and FAQ edit dialogs with a new KnowledgeDirectoryPanel component.
- Enhanced API to support fetching, creating, updating, and deleting knowledge directories.
- Updated data models to include directory information for knowledge documents and FAQs.
- Added translations for directory-related messages in English and Chinese.

internal/services/knowledge_directory_service.go

@@ -0,0 +1,212 @@
+package services
+
+import (
+	"strings"
+	"time"
+
+	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/dto"
+	"agent-desk/internal/pkg/dto/request"
+	"agent-desk/internal/pkg/enums"
+	"agent-desk/internal/pkg/errorsx"
+	"agent-desk/internal/pkg/utils"
+	"agent-desk/internal/repositories"
+
+	"github.com/mlogclub/simple/sqls"
+)
+
+var KnowledgeDirectoryService = newKnowledgeDirectoryService()
+
+func newKnowledgeDirectoryService() *knowledgeDirectoryService {
+	return &knowledgeDirectoryService{}
+}
+
+type knowledgeDirectoryService struct {
+}
+
+func (s *knowledgeDirectoryService) Get(id int64) *models.KnowledgeDirectory {
+	return repositories.KnowledgeDirectoryRepository.Get(sqls.DB(), id)
+}
+
+func (s *knowledgeDirectoryService) Find(cnd *sqls.Cnd) []models.KnowledgeDirectory {
+	return repositories.KnowledgeDirectoryRepository.Find(sqls.DB(), cnd)
+}
+
+func (s *knowledgeDirectoryService) FindAllByKnowledgeBaseID(knowledgeBaseID int64) []models.KnowledgeDirectory {
+	return s.Find(sqls.NewCnd().Eq("knowledge_base_id", knowledgeBaseID).Asc("parent_id").Asc("sort_no").Asc("id"))
+}
+
+func (s *knowledgeDirectoryService) Count(cnd *sqls.Cnd) int64 {
+	return repositories.KnowledgeDirectoryRepository.Count(sqls.DB(), cnd)
+}
+
+func (s *knowledgeDirectoryService) CreateDirectory(req request.CreateKnowledgeDirectoryRequest, operator *dto.AuthPrincipal) (*models.KnowledgeDirectory, error) {
+	if operator == nil {
+		return nil, errorsx.Unauthorized("未登录或登录已过期")
+	}
+	name := strings.TrimSpace(req.Name)
+	if name == "" {
+		return nil, errorsx.InvalidParam("目录名称不能为空")
+	}
+	if req.KnowledgeBaseID <= 0 || KnowledgeBaseService.Get(req.KnowledgeBaseID) == nil {
+		return nil, errorsx.InvalidParam("知识库不存在")
+	}
+	if err := s.validateParent(req.KnowledgeBaseID, req.ParentID, 0); err != nil {
+		return nil, err
+	}
+	if existing := s.findByName(req.KnowledgeBaseID, req.ParentID, name); existing != nil {
+		return nil, errorsx.InvalidParam("同级下已存在相同名称的目录")
+	}
+	item := &models.KnowledgeDirectory{
+		KnowledgeBaseID: req.KnowledgeBaseID,
+		ParentID:        req.ParentID,
+		Name:            name,
+		SortNo:          s.NextSortNo(req.KnowledgeBaseID, req.ParentID),
+		Status:          enums.StatusOk,
+		Remark:          strings.TrimSpace(req.Remark),
+		AuditFields:     utils.BuildAuditFields(operator),
+	}
+	if err := repositories.KnowledgeDirectoryRepository.Create(sqls.DB(), item); err != nil {
+		return nil, err
+	}
+	return item, nil
+}
+
+func (s *knowledgeDirectoryService) UpdateDirectory(req request.UpdateKnowledgeDirectoryRequest, operator *dto.AuthPrincipal) error {
+	if operator == nil {
+		return errorsx.Unauthorized("未登录或登录已过期")
+	}
+	item := s.Get(req.ID)
+	if item == nil {
+		return errorsx.InvalidParam("目录不存在")
+	}
+	name := strings.TrimSpace(req.Name)
+	if name == "" {
+		return errorsx.InvalidParam("目录名称不能为空")
+	}
+	if req.KnowledgeBaseID <= 0 {
+		req.KnowledgeBaseID = item.KnowledgeBaseID
+	}
+	if req.KnowledgeBaseID != item.KnowledgeBaseID {
+		return errorsx.InvalidParam("目录不能移动到其他知识库")
+	}
+	if err := s.validateParent(item.KnowledgeBaseID, req.ParentID, req.ID); err != nil {
+		return err
+	}
+	if req.ParentID > 0 && s.Count(sqls.NewCnd().Eq("parent_id", req.ID)) > 0 {
+		return errorsx.InvalidParam("存在子目录的目录不能移动到二级目录")
+	}
+	if existing := s.findByName(item.KnowledgeBaseID, req.ParentID, name); existing != nil && existing.ID != req.ID {
+		return errorsx.InvalidParam("同级下已存在相同名称的目录")
+	}
+	return repositories.KnowledgeDirectoryRepository.Updates(sqls.DB(), req.ID, map[string]any{
+		"parent_id":        req.ParentID,
+		"name":             name,
+		"remark":           strings.TrimSpace(req.Remark),
+		"update_user_id":   operator.UserID,
+		"update_user_name": operator.Username,
+		"updated_at":       time.Now(),
+	})
+}
+
+func (s *knowledgeDirectoryService) DeleteDirectory(id int64) error {
+	item := s.Get(id)
+	if item == nil {
+		return errorsx.InvalidParam("目录不存在")
+	}
+	if s.Count(sqls.NewCnd().Eq("parent_id", id)) > 0 {
+		return errorsx.InvalidParam("该目录下存在子目录，无法删除")
+	}
+	if KnowledgeDocumentService.Count(sqls.NewCnd().Eq("directory_id", id)) > 0 {
+		return errorsx.InvalidParam("该目录下存在文档，无法删除")
+	}
+	if KnowledgeFAQService.Count(sqls.NewCnd().Eq("directory_id", id)) > 0 {
+		return errorsx.InvalidParam("该目录下存在FAQ，无法删除")
+	}
+	return repositories.KnowledgeDirectoryRepository.Delete(sqls.DB(), id)
+}
+
+func (s *knowledgeDirectoryService) UpdateSort(knowledgeBaseID int64, parentID int64, ids []int64) error {
+	return sqls.WithTransaction(func(ctx *sqls.TxContext) error {
+		for i, id := range ids {
+			item := repositories.KnowledgeDirectoryRepository.Get(ctx.Tx, id)
+			if item == nil {
+				return errorsx.InvalidParam("目录不存在")
+			}
+			if item.KnowledgeBaseID != knowledgeBaseID || item.ParentID != parentID {
+				return errorsx.InvalidParam("只能调整同知识库同级目录排序")
+			}
+			if err := repositories.KnowledgeDirectoryRepository.UpdateColumn(ctx.Tx, id, "sort_no", i+1); err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
+func (s *knowledgeDirectoryService) RequireUsableDirectory(knowledgeBaseID int64, directoryID int64) (*models.KnowledgeDirectory, error) {
+	if directoryID <= 0 {
+		return nil, nil
+	}
+	item := s.Get(directoryID)
+	if item == nil {
+		return nil, errorsx.InvalidParam("知识库目录不存在")
+	}
+	if item.KnowledgeBaseID != knowledgeBaseID {
+		return nil, errorsx.InvalidParam("知识库目录不属于当前知识库")
+	}
+	if item.Status != enums.StatusOk {
+		return nil, errorsx.InvalidParam("知识库目录不可用")
+	}
+	return item, nil
+}
+
+func (s *knowledgeDirectoryService) NextSortNo(knowledgeBaseID int64, parentID int64) int {
+	if temp := repositories.KnowledgeDirectoryRepository.FindOne(sqls.DB(), sqls.NewCnd().Eq("knowledge_base_id", knowledgeBaseID).Eq("parent_id", parentID).Desc("sort_no").Desc("id")); temp != nil {
+		return temp.SortNo + 1
+	}
+	return 1
+}
+
+func (s *knowledgeDirectoryService) PathMap(knowledgeBaseID int64) map[int64]string {
+	items := s.FindAllByKnowledgeBaseID(knowledgeBaseID)
+	byID := make(map[int64]models.KnowledgeDirectory, len(items))
+	for _, item := range items {
+		byID[item.ID] = item
+	}
+	ret := make(map[int64]string, len(items))
+	for _, item := range items {
+		if item.ParentID > 0 {
+			if parent, ok := byID[item.ParentID]; ok {
+				ret[item.ID] = parent.Name + " / " + item.Name
+				continue
+			}
+		}
+		ret[item.ID] = item.Name
+	}
+	return ret
+}
+
+func (s *knowledgeDirectoryService) validateParent(knowledgeBaseID int64, parentID int64, selfID int64) error {
+	if parentID <= 0 {
+		return nil
+	}
+	if parentID == selfID {
+		return errorsx.InvalidParam("不能将目录设为自己的子目录")
+	}
+	parent := s.Get(parentID)
+	if parent == nil {
+		return errorsx.InvalidParam("父目录不存在")
+	}
+	if parent.KnowledgeBaseID != knowledgeBaseID {
+		return errorsx.InvalidParam("父目录不属于当前知识库")
+	}
+	if parent.ParentID > 0 {
+		return errorsx.InvalidParam("知识库目录最多支持二级")
+	}
+	return nil
+}
+
+func (s *knowledgeDirectoryService) findByName(knowledgeBaseID int64, parentID int64, name string) *models.KnowledgeDirectory {
+	return repositories.KnowledgeDirectoryRepository.FindOne(sqls.DB(), sqls.NewCnd().Eq("knowledge_base_id", knowledgeBaseID).Eq("parent_id", parentID).Eq("name", name))
+}

internal/services/knowledge_directory_service_test.go

@@ -0,0 +1,126 @@
+package services
+
+import (
+	"testing"
+
+	"agent-desk/internal/models"
+	"agent-desk/internal/pkg/dto"
+	"agent-desk/internal/pkg/dto/request"
+	"agent-desk/internal/pkg/enums"
+	"agent-desk/internal/repositories"
+
+	"github.com/glebarez/sqlite"
+	"github.com/mlogclub/simple/sqls"
+	"gorm.io/gorm"
+)
+
+func TestKnowledgeDirectoryRejectsThirdLevel(t *testing.T) {
+	setupKnowledgeDirectoryTestDB(t)
+	operator := knowledgeDirectoryTestOperator()
+	kb := createKnowledgeDirectoryTestBase(t, "Document KB", string(enums.KnowledgeBaseTypeDocument))
+	parent, err := KnowledgeDirectoryService.CreateDirectory(request.CreateKnowledgeDirectoryRequest{
+		KnowledgeBaseID: kb.ID,
+		Name:            "一级目录",
+	}, operator)
+	if err != nil {
+		t.Fatalf("create parent: %v", err)
+	}
+	child, err := KnowledgeDirectoryService.CreateDirectory(request.CreateKnowledgeDirectoryRequest{
+		KnowledgeBaseID: kb.ID,
+		ParentID:        parent.ID,
+		Name:            "二级目录",
+	}, operator)
+	if err != nil {
+		t.Fatalf("create child: %v", err)
+	}
+
+	if _, err := KnowledgeDirectoryService.CreateDirectory(request.CreateKnowledgeDirectoryRequest{
+		KnowledgeBaseID: kb.ID,
+		ParentID:        child.ID,
+		Name:            "三级目录",
+	}, operator); err == nil {
+		t.Fatal("CreateDirectory() error is nil, want third-level rejection")
+	}
+}
+
+func TestKnowledgeDocumentRejectsDirectoryFromOtherKnowledgeBase(t *testing.T) {
+	setupKnowledgeDirectoryTestDB(t)
+	operator := knowledgeDirectoryTestOperator()
+	kb := createKnowledgeDirectoryTestBase(t, "Document KB", string(enums.KnowledgeBaseTypeDocument))
+	otherKB := createKnowledgeDirectoryTestBase(t, "Other KB", string(enums.KnowledgeBaseTypeDocument))
+	directory, err := KnowledgeDirectoryService.CreateDirectory(request.CreateKnowledgeDirectoryRequest{
+		KnowledgeBaseID: otherKB.ID,
+		Name:            "其他目录",
+	}, operator)
+	if err != nil {
+		t.Fatalf("create directory: %v", err)
+	}
+
+	_, err = KnowledgeDocumentService.CreateKnowledgeDocument(request.CreateKnowledgeDocumentRequest{
+		KnowledgeBaseID: kb.ID,
+		DirectoryID:     directory.ID,
+		Title:           "Doc",
+		Content:         "content",
+	}, operator)
+	if err == nil {
+		t.Fatal("CreateKnowledgeDocument() error is nil, want cross-knowledge-base directory rejection")
+	}
+}
+
+func TestKnowledgeDirectoryDeleteRejectsAttachedContent(t *testing.T) {
+	setupKnowledgeDirectoryTestDB(t)
+	operator := knowledgeDirectoryTestOperator()
+	kb := createKnowledgeDirectoryTestBase(t, "FAQ KB", string(enums.KnowledgeBaseTypeFAQ))
+	directory, err := KnowledgeDirectoryService.CreateDirectory(request.CreateKnowledgeDirectoryRequest{
+		KnowledgeBaseID: kb.ID,
+		Name:            "FAQ目录",
+	}, operator)
+	if err != nil {
+		t.Fatalf("create directory: %v", err)
+	}
+	if err := repositories.KnowledgeFAQRepository.Create(sqls.DB(), &models.KnowledgeFAQ{
+		KnowledgeBaseID: kb.ID,
+		DirectoryID:     directory.ID,
+		Question:        "Question",
+		Answer:          "Answer",
+		Status:          enums.StatusOk,
+	}); err != nil {
+		t.Fatalf("create faq: %v", err)
+	}
+
+	if err := KnowledgeDirectoryService.DeleteDirectory(directory.ID); err == nil {
+		t.Fatal("DeleteDirectory() error is nil, want attached content rejection")
+	}
+}
+
+func setupKnowledgeDirectoryTestDB(t *testing.T) {
+	t.Helper()
+	db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{})
+	if err != nil {
+		t.Fatalf("open sqlite db: %v", err)
+	}
+	if err := db.AutoMigrate(&models.KnowledgeBase{}, &models.KnowledgeDirectory{}, &models.KnowledgeDocument{}, &models.KnowledgeFAQ{}); err != nil {
+		t.Fatalf("auto migrate: %v", err)
+	}
+	sqls.SetDB(db)
+}
+
+func createKnowledgeDirectoryTestBase(t *testing.T, name string, knowledgeType string) *models.KnowledgeBase {
+	t.Helper()
+	item := &models.KnowledgeBase{
+		Name:          name,
+		KnowledgeType: knowledgeType,
+		Status:        enums.StatusOk,
+	}
+	if err := repositories.KnowledgeBaseRepository.Create(sqls.DB(), item); err != nil {
+		t.Fatalf("create knowledge base: %v", err)
+	}
+	return item
+}
+
+func knowledgeDirectoryTestOperator() *dto.AuthPrincipal {
+	return &dto.AuthPrincipal{
+		UserID:   1,
+		Username: "tester",
+	}
+}

internal/services/knowledge_document_service.go

@@ -94,6 +94,9 @@ func (s *knowledgeDocumentService) CreateKnowledgeDocument(req request.CreateKno
 	if kb.KnowledgeType == string(enums.KnowledgeBaseTypeFAQ) {
 		return nil, errorsx.InvalidParam("FAQ知识库不支持文档")
 	}
+	if _, err := KnowledgeDirectoryService.RequireUsableDirectory(req.KnowledgeBaseID, req.DirectoryID); err != nil {
+		return nil, err
+	}
 	item, err := s.buildKnowledgeDocumentModel(req)
 	if err != nil {
 		return nil, err
@@ -133,6 +136,9 @@ func (s *knowledgeDocumentService) UpdateKnowledgeDocument(req request.UpdateKno
 	if kb.KnowledgeType == string(enums.KnowledgeBaseTypeFAQ) {
 		return errorsx.InvalidParam("FAQ知识库不支持文档")
 	}
+	if _, err := KnowledgeDirectoryService.RequireUsableDirectory(req.KnowledgeBaseID, req.DirectoryID); err != nil {
+		return err
+	}
 	item, err := s.buildKnowledgeDocumentModel(req.CreateKnowledgeDocumentRequest)
 	if err != nil {
 		return err
@@ -140,6 +146,7 @@ func (s *knowledgeDocumentService) UpdateKnowledgeDocument(req request.UpdateKno
 	oldKnowledgeBaseID := current.KnowledgeBaseID
 	if err := repositories.KnowledgeDocumentRepository.Updates(sqls.DB(), req.ID, map[string]any{
 		"knowledge_base_id": item.KnowledgeBaseID,
+		"directory_id":      item.DirectoryID,
 		"title":             item.Title,
 		"content_type":      item.ContentType,
 		"content_hash":      item.ContentHash,
@@ -187,6 +194,7 @@ func (s *knowledgeDocumentService) buildKnowledgeDocumentModel(req request.Creat
 	plainText := rag.ExtractPlainText(req.Content, req.ContentType)
 	item := &models.KnowledgeDocument{
 		KnowledgeBaseID: req.KnowledgeBaseID,
+		DirectoryID:     req.DirectoryID,
 		Title:           req.Title,
 		ContentType:     req.ContentType,
 		Content:         req.Content,

internal/services/knowledge_faq_service.go

@@ -40,6 +40,10 @@ func (s *knowledgeFAQService) FindPageByParams(queryParams *params.QueryParams)
 	return repositories.KnowledgeFAQRepository.FindPageByParams(sqls.DB(), queryParams)
 }
 
+func (s *knowledgeFAQService) Count(cnd *sqls.Cnd) int64 {
+	return repositories.KnowledgeFAQRepository.Count(sqls.DB(), cnd)
+}
+
 func (s *knowledgeFAQService) CreateKnowledgeFAQ(req request.CreateKnowledgeFAQRequest, operator *dto.AuthPrincipal) (*models.KnowledgeFAQ, error) {
 	if operator == nil {
 		return nil, errorsx.Unauthorized("未登录或登录已过期")
@@ -48,6 +52,9 @@ func (s *knowledgeFAQService) CreateKnowledgeFAQ(req request.CreateKnowledgeFAQR
 	if err != nil {
 		return nil, err
 	}
+	if _, err := KnowledgeDirectoryService.RequireUsableDirectory(req.KnowledgeBaseID, req.DirectoryID); err != nil {
+		return nil, err
+	}
 	item, err := s.buildKnowledgeFAQModel(req)
 	if err != nil {
 		return nil, err
@@ -78,12 +85,16 @@ func (s *knowledgeFAQService) UpdateKnowledgeFAQ(req request.UpdateKnowledgeFAQR
 	if _, err := s.requireFAQKnowledgeBase(req.KnowledgeBaseID); err != nil {
 		return err
 	}
+	if _, err := KnowledgeDirectoryService.RequireUsableDirectory(req.KnowledgeBaseID, req.DirectoryID); err != nil {
+		return err
+	}
 	item, err := s.buildKnowledgeFAQModel(req.CreateKnowledgeFAQRequest)
 	if err != nil {
 		return err
 	}
 	if err := repositories.KnowledgeFAQRepository.Updates(sqls.DB(), req.ID, map[string]any{
 		"knowledge_base_id": item.KnowledgeBaseID,
+		"directory_id":      item.DirectoryID,
 		"question":          item.Question,
 		"answer":            item.Answer,
 		"similar_questions": item.SimilarQuestions,
@@ -127,6 +138,7 @@ func (s *knowledgeFAQService) buildKnowledgeFAQModel(req request.CreateKnowledge
 	}
 	return &models.KnowledgeFAQ{
 		KnowledgeBaseID:  req.KnowledgeBaseID,
+		DirectoryID:      req.DirectoryID,
 		Question:         req.Question,
 		Answer:           req.Answer,
 		SimilarQuestions: string(similarQuestions),