eta_api/controllers/ai/ai_pormpt.go

package ai

import (
	"eta/eta_api/models"
	"eta/eta_api/models/ai_summary"
	"eta/eta_api/models/system"
)

// AiPromptList
// @Title 获取所有ai纪要分类接口-包含沙盘
// @Description 获取所有ai纪要分类接口-包含沙盘
// @Success 200 {object} data_manage.ChartClassifyListResp
// @router /prompt/list [get]
func (this *AiController) AiPromptList() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	list, err := ai_summary.GetAiPromptList(this.SysUser.AdminId, 0)
	if err != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "获取数据失败，Err:" + err.Error()
		return
	}


	br.Ret = 200
	br.Success = true
	br.Msg = "获取成功"
	br.Data = list
}

// PromptGroupList
// @Title 获取可见的目录列表
// @Description 获取可见的目录列表接口
// @Success 200 {object} ppt_english.RespGroupList
// @router /prompt/groups [get]
func (this *AiController) PromptGroupList() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	list, err := ai_summary.GetAiPromptShareList()
	if err != nil {
		br.Msg = err.Error()
		return
	}
	resp := make([]*ai_summary.RespGroupListItem, 0)
	publicAdminIdList := make([]int, 0)
	publicPromptListMap := make(map[int][]*ai_summary.AiPromptItem)
	systemAdminMap := make(map[int]*system.Admin)

	for _, v := range list {
		publicPromptList, ok := publicPromptListMap[v.SysAdminId]
		if !ok {
			publicPromptList = make([]*ai_summary.AiPromptItem, 0)
			publicAdminIdList = append(publicAdminIdList, v.SysAdminId)

		}
		systemAdminMap[v.SysAdminId] = &system.Admin{
			AdminId: v.SysAdminId,
			RealName:v.SysAdminName,
		}
		publicPromptList = append(publicPromptList, v)
		publicPromptListMap[v.SysAdminId] = publicPromptList
	}
	for _, v := range publicAdminIdList {
		systemAdmin, ok := systemAdminMap[v]
		if !ok {
			continue
		}
		respGroupPptNameListItemList, ok := publicPromptListMap[v]
		if !ok {
			respGroupPptNameListItemList = make([]*ai_summary.AiPromptItem, 0)
		}

		//  分组信息
		tmpRespGroupListItem := &ai_summary.RespGroupListItem{
			GroupId:   int64(systemAdmin.AdminId),
			GroupName: systemAdmin.RealName + "的PPT",
			AdminId:   systemAdmin.AdminId,
			IsShare:   1,
			PromptList:   respGroupPptNameListItemList,
		}
		resp = append(resp, tmpRespGroupListItem)
	}

	br.Ret = 200
	br.Success = true
	br.Msg = "查询成功"
	br.Data = resp
	return
}