package node

import (
	"ai-agent/workflow/consts/node"
	"ai-agent/workflow/model/entity"

	"gitea.com/red-future/common/beans"
	"github.com/gogf/gf/v2/frame/g"
)

// CreateNodePromptReq 创建节点提示词请求
type CreateNodePromptReq struct {
	g.Meta     `path:"/create" method:"post" tags:"节点提示词管理" summary:"创建节点提示词" dc:"创建用户自定义节点提示词"`
	NodeType   node.NodeType   `json:"nodeType"   v:"required#节点类型不能为空"`
	Prompt     string          `json:"prompt"     v:"required#提示词不能为空"`
	SourceType node.SourceType `json:"sourceType"`
}

type CreateNodePromptRes struct {
	Id int64 `json:"id,string"`
}

// UpdateNodePromptReq 更新节点提示词请求
type UpdateNodePromptReq struct {
	g.Meta   `path:"/update" method:"put" tags:"节点提示词管理" summary:"更新节点提示词" dc:"更新用户自定义节点提示词"`
	Id       int64         `json:"id"        v:"required#ID不能为空"`
	NodeType node.NodeType `json:"nodeType"`
	Prompt   string        `json:"prompt"`
}

// DeleteNodePromptReq 删除节点提示词请求
type DeleteNodePromptReq struct {
	g.Meta `path:"/delete" method:"delete" tags:"节点提示词管理" summary:"删除节点提示词" dc:"删除用户自定义节点提示词"`
	Id     int64 `json:"id" v:"required#ID不能为空"`
}

// GetNodePromptReq 根据ID查询节点提示词请求
type GetNodePromptReq struct {
	g.Meta  `path:"/get" method:"get" tags:"节点提示词管理" summary:"查询节点提示词详情" dc:"根据ID查询节点提示词详情"`
	Id      int64  `json:"id"`
	Prompt  string `json:"prompt"`
	Creator string `json:"creator"`
}

// ListNodePromptReq 查询节点提示词列表请求
type ListNodePromptReq struct {
	g.Meta   `path:"/list" method:"get" tags:"节点提示词管理" summary:"查询节点提示词列表" dc:"查询当前创建人的节点提示词，包含系统和用户自定义"`
	Page     *beans.Page   `json:"page"`
	Creator  string        `json:"creator"`
	NodeType node.NodeType `json:"nodeType"`
}

// ListMyNodePromptReq 查询当前用户节点提示词列表请求
type ListMyNodePromptReq struct {
	g.Meta   `path:"/listMy" method:"get" tags:"节点提示词管理" summary:"查询当前用户节点提示词列表" dc:"查询当前创建人自己创建的节点提示词列表"`
	Page     *beans.Page   `json:"page"`
	NodeType node.NodeType `json:"nodeType"`
	Creator  string        `json:"creator"`
}

// NodePromptResp 节点提示词响应
type NodePromptResp struct {
	*entity.NodePrompt
}

// ListNodePromptResp 节点提示词列表响应
type ListNodePromptResp struct {
	List  []*entity.NodePrompt `json:"list"`
	Total int                  `json:"total"`
}