feat(knowledge): add FAQ Excel import export

This commit is contained in:
mlogclub
2026-06-01 13:57:49 +08:00
parent d8692b45cd
commit 8933e963f5
15 changed files with 1133 additions and 363 deletions
@@ -5,8 +5,11 @@ import (
"agent-desk/internal/pkg/constants"
"agent-desk/internal/pkg/dto/request"
"agent-desk/internal/pkg/dto/response"
"agent-desk/internal/pkg/errorsx"
"agent-desk/internal/pkg/httpx"
"agent-desk/internal/services"
"fmt"
"net/http"
"agent-desk/internal/pkg/httpx/params"
@@ -14,6 +17,79 @@ import (
"github.com/mlogclub/simple/web"
)
func KnowledgeFAQGetImport_template(ctx *gin.Context) {
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeFAQView); err != nil {
httpx.WriteJSON(ctx, err)
return
}
file, err := services.KnowledgeFAQService.BuildKnowledgeFAQImportTemplate()
if err != nil {
httpx.WriteJSON(ctx, err)
return
}
writeKnowledgeFAQExcelFile(ctx, file)
}
func KnowledgeFAQGetExport(ctx *gin.Context) {
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeFAQView); err != nil {
httpx.WriteJSON(ctx, err)
return
}
knowledgeBaseID, ok := params.GetInt64(ctx, "knowledgeBaseId")
if !ok || knowledgeBaseID <= 0 {
httpx.WriteJSON(ctx, errorsx.InvalidParam("知识库不存在"))
return
}
file, err := services.KnowledgeFAQService.ExportKnowledgeFAQs(knowledgeBaseID)
if err != nil {
httpx.WriteJSON(ctx, err)
return
}
writeKnowledgeFAQExcelFile(ctx, file)
}
func KnowledgeFAQPostImport(ctx *gin.Context) {
operator, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeFAQCreate)
if err != nil {
httpx.WriteJSON(ctx, err)
return
}
mode := request.KnowledgeFAQImportMode(ctx.PostForm("mode"))
if mode == request.KnowledgeFAQImportModeOverwrite {
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeFAQUpdate); err != nil {
httpx.WriteJSON(ctx, err)
return
}
}
knowledgeBaseID, ok := params.GetInt64(ctx, "knowledgeBaseId")
if !ok || knowledgeBaseID <= 0 {
httpx.WriteJSON(ctx, errorsx.InvalidParam("知识库不存在"))
return
}
header, err := ctx.FormFile("file")
if err != nil {
httpx.WriteJSON(ctx, errorsx.InvalidParam("请选择导入文件"))
return
}
file, err := header.Open()
if err != nil {
httpx.WriteJSON(ctx, errorsx.InvalidParam("导入文件读取失败"))
return
}
defer file.Close()
result, err := services.KnowledgeFAQService.ImportKnowledgeFAQs(request.ImportKnowledgeFAQRequest{
KnowledgeBaseID: knowledgeBaseID,
Mode: mode,
Filename: header.Filename,
Reader: file,
}, operator)
if err != nil {
httpx.WriteJSON(ctx, err)
return
}
httpx.WriteJSON(ctx, result)
}
func KnowledgeFAQAnyList(ctx *gin.Context) {
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeFAQView); err != nil {
httpx.WriteJSON(ctx, err)
@@ -106,3 +182,9 @@ func KnowledgeFAQPostDelete(ctx *gin.Context) {
}
httpx.WriteJSON(ctx, nil)
}
func writeKnowledgeFAQExcelFile(ctx *gin.Context, file *response.KnowledgeFAQExportedFile) {
ctx.Header("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, file.Filename))
ctx.Header("Cache-Control", "no-store")
ctx.Data(http.StatusOK, file.ContentType, file.Data)
}