package ai_summary

import (
	"github.com/beego/beego/v2/client/orm"
	"time"
)

type AiPrompt struct {
	AiPromptId    int       `orm:"column(ai_prompt_id);pk"` // ai纪要提示词id
	PromptContent string    // 原始内容
	SysAdminId    int       // 创建人ID
	SysAdminName  string    // 创建人姓名
	Title         string    // 文档标题
	CreateTime    time.Time // 创建时间
	ModifyTime    time.Time // 更新时间
	IsShare       int       // 是否分享，0：不分享，1：分享
}

type AiPromptItem struct {
	AiPromptId    int    `orm:"column(ai_prompt_id);pk"` // ai纪要提示词id
	PromptContent string // 原始内容
	SysAdminId    int    // 创建人ID
	SysAdminName  string // 创建人姓名
	Title         string // 文档标题
	CreateTime    string // 创建时间
	ModifyTime    string // 更新时间
	IsShare       int    // 是否分享，0：不分享，1：分享
}

func GetAiPromptList(sysUserId, isShare int) (items []*AiPromptItem, err error) {
	sql := ``
	if isShare == 1 {
		sql = ` SELECT * FROM ai_prompt WHERE sys_admin_id=? and is_share=1 ORDER BY create_time DESC `
	} else {
		sql = ` SELECT * FROM ai_prompt WHERE sys_admin_id=? ORDER BY create_time DESC `
	}
	o := orm.NewOrm()
	_, err = o.Raw(sql, sysUserId).QueryRows(&items)
	return
}

func GetAiPromptShareList() (items []*AiPromptItem, err error) {
	sql := ``
	sql = ` SELECT * FROM ai_prompt WHERE is_share=1 ORDER BY create_time DESC `

	o := orm.NewOrm()
	_, err = o.Raw(sql).QueryRows(&items)
	return
}

type RespGroupListItem struct {
	GroupId   int64  `description:"目录id"`
	GroupName string `description:"目录名称"`
	AdminId   int    `description:"目录创建者账号ID"`
	IsShare   int8   `description:"是否共享，0私有，1共享"`
	PromptList   []*AiPromptItem
}