package aitemplatemoduleser

import (
	"91porn-server/common/log"
	"91porn-server/models/cache/aitemplatemoduledata"
	"91porn-server/models/commod"
	"91porn-server/models/v/aitemplatemodulemod"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

type AppQueryListReq struct {
	commod.Page
}

type AppListRes struct {
	Total   int64                                       `json:"total"`
	HasNext bool                                        `json:"hasNext"`
	List    []*aitemplatemodulemod.AiTemplateModuleInfo `json:"list"`
}

// GetList 获取列表
func (p *AppQueryListReq) GetList() AppListRes {
	var res AppListRes
	var err error
	filter := bson.M{}

	sort := bson.D{{"_id", -1}}
	// 获取列表
	var data []aitemplatemodulemod.AiTemplateModule
	data, res.Total, res.HasNext, err = aitemplatemoduledata.GetListFromCache(filter, int64(p.Skip()), int64(p.Limit()), sort)
	if err != nil {
		log.Error("获取AI模版模块列表列表数据错误", log.Any("Params", *p), log.E(err))
		return res
	}
	res.List = aitemplatemoduledata.FormatAppDataList(data)

	return res
}

type AppQueryAllListReq struct {
}

type AppAllListRes struct {
	List []*aitemplatemodulemod.AiTemplateModuleInfo `json:"list"`
}

// GetList 获取列表
func (p *AppQueryAllListReq) GetList() AppAllListRes {
	var res AppAllListRes
	var err error
	filter := bson.M{"status": 1}
	sort := bson.D{{"sortNum", -1}}
	// 获取列表
	data, err := aitemplatemodulemod.GetAll(filter, sort)
	if err != nil {
		log.Error("获取AI模版模块列表列表数据错误", log.Any("Params", *p), log.E(err))
		return res
	}

	res.List = aitemplatemoduledata.FormatAppDataList(data)
	return res
}

type AppQueryInfoReq struct {
	ID string `json:"id" form:"id"` // id
}
type AppQueryInfoRes = *aitemplatemodulemod.AiTemplateModuleInfo

// GetInfo 获取详情
func (p *AppQueryInfoReq) GetInfo() (res AppQueryInfoRes, err error) {
	oid, err := primitive.ObjectIDFromHex(p.ID)
	if err != nil {
		return res, err
	}

	var item aitemplatemodulemod.AiTemplateModule
	item, err = aitemplatemoduledata.GetInfoFromCache(oid)
	if err != nil {
		log.Error("获取AI模版模块列表详情数据错误", log.Any("ID", p.ID), log.E(err))
		return
	}

	res = aitemplatemoduledata.FormatAppData(item)
	return
}