142 lines
2.9 KiB
Go
142 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/XingfenD/yoresee_doc/internal/model"
|
|
"github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type templateSeed struct {
|
|
Name string
|
|
Description string
|
|
Content string
|
|
Scope string
|
|
KnowledgeBaseID *int64
|
|
Tags []string
|
|
}
|
|
|
|
func initializeTemplatesInTx(tx *gorm.DB) error {
|
|
logrus.Println("Initializing default templates in transaction...")
|
|
|
|
var adminUser model.User
|
|
if err := tx.Where("username = ?", "admin").First(&adminUser).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
var defaultKB model.KnowledgeBase
|
|
if err := tx.Where("name = ?", "默认知识库").First(&defaultKB).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
seeds := []templateSeed{
|
|
{
|
|
Name: "个人周报模板",
|
|
Description: "用于个人每周工作复盘与计划",
|
|
Content: `# 个人周报
|
|
|
|
## 本周完成
|
|
-
|
|
|
|
## 问题与风险
|
|
-
|
|
|
|
## 下周计划
|
|
- `,
|
|
Scope: "private",
|
|
Tags: []string{"个人", "周报"},
|
|
},
|
|
{
|
|
Name: "会议纪要模板",
|
|
Description: "通用会议记录模板",
|
|
Content: `# 会议纪要
|
|
|
|
## 会议信息
|
|
- 时间：
|
|
- 参与人：
|
|
|
|
## 讨论内容
|
|
-
|
|
|
|
## 行动项
|
|
- [ ] `,
|
|
Scope: "private",
|
|
Tags: []string{"个人", "会议"},
|
|
},
|
|
{
|
|
Name: "需求评审模板",
|
|
Description: "知识库内需求评审记录模板",
|
|
KnowledgeBaseID: &defaultKB.ID,
|
|
Scope: "knowledge_base",
|
|
Content: `# 需求评审
|
|
|
|
## 背景
|
|
|
|
## 目标
|
|
|
|
## 方案评审
|
|
- 优点：
|
|
- 风险：
|
|
|
|
## 结论`,
|
|
Tags: []string{"知识库", "需求"},
|
|
},
|
|
{
|
|
Name: "技术方案模板",
|
|
Description: "知识库内技术方案沉淀模板",
|
|
KnowledgeBaseID: &defaultKB.ID,
|
|
Scope: "knowledge_base",
|
|
Content: `# 技术方案
|
|
|
|
## 问题定义
|
|
|
|
## 方案设计
|
|
|
|
## 数据模型
|
|
|
|
## 发布计划
|
|
`,
|
|
Tags: []string{"知识库", "技术"},
|
|
},
|
|
}
|
|
|
|
for _, seed := range seeds {
|
|
if err := createTemplateIfNotExists(tx, adminUser.ID, seed); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
logrus.Println("Default templates initialized successfully in transaction")
|
|
return nil
|
|
}
|
|
|
|
func createTemplateIfNotExists(tx *gorm.DB, userID int64, seed templateSeed) error {
|
|
query := tx.Model(&model.Template{}).
|
|
Where("user_id = ? AND name = ? AND scope = ?", userID, seed.Name, seed.Scope)
|
|
if seed.KnowledgeBaseID == nil {
|
|
query = query.Where("knowledge_base_id IS NULL")
|
|
} else {
|
|
query = query.Where("knowledge_base_id = ?", *seed.KnowledgeBaseID)
|
|
}
|
|
|
|
var count int64
|
|
if err := query.Count(&count).Error; err != nil {
|
|
return err
|
|
}
|
|
if count > 0 {
|
|
return nil
|
|
}
|
|
|
|
template := model.Template{
|
|
Name: seed.Name,
|
|
Description: seed.Description,
|
|
DocumentType: model.DocumentType_Markdown,
|
|
Content: seed.Content,
|
|
UserID: userID,
|
|
Scope: seed.Scope,
|
|
KnowledgeBaseID: seed.KnowledgeBaseID,
|
|
IsPublic: false,
|
|
Tags: seed.Tags,
|
|
}
|
|
return tx.Create(&template).Error
|
|
}
|