feat(knowledge): add FAQ Excel import export
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user