package dto

import (
	"rag/consts/document"

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

// CreateDocumentReq 创建文件请求
type CreateDocumentReq struct {
	g.Meta `path:"/create" method:"post" tags:"文件管理" summary:"创建文件" dc:"创建文件"`

	DatasetId int64  `json:"datasetId" v:"required#数据集ID不能为空"`
	Title     string `json:"title" v:"required#标题不能为空"`
	Format    string `json:"format" v:"required#格式不能为空"`
	FileSize  int64  `json:"fileSize" v:"required#大小不能为空"`
	FilePath  string `json:"filePath" v:"required#路径不能为空"`
}

// CreateDocumentRes 创建文件响应
type CreateDocumentRes struct {
	Id int64 `json:"id,string"`
}

// UpdateDocumentReq 更新文件请求
type UpdateDocumentReq struct {
	g.Meta `path:"/update" method:"put" tags:"文件管理" summary:"更新文件" dc:"更新文件"`

	Id           int64                 `json:"id" v:"required#ID不能为空"`
	Status       document.Status       `json:"status"`
	VectorStatus document.VectorStatus `json:"vectorStatus"`
	ChunkCount   int64                 `json:"chunkCount"`
}

// DeleteDocumentReq 删除文件请求
type DeleteDocumentReq struct {
	g.Meta `path:"/delete" method:"delete" tags:"文件管理" summary:"删除文件" dc:"删除文件"`

	Id int64 `json:"id" v:"required#ID不能为空"`
}

// GetDocumentReq 获取文件请求
type GetDocumentReq struct {
	g.Meta `path:"/get" method:"get" tags:"文件管理" summary:"获取文件详情" dc:"获取文件详情"`

	Id        int64  `json:"id" v:"required#ID不能为空"`
	DatasetId int64  `json:"datasetId"`
	Title     string `json:"title"`
}

type GetDocumentRes struct {
	*DocumentVO
	ImgAddressPrefix string `json:"imgAddressPrefix"`
}

// ListDocumentReq 文件列表请求
type ListDocumentReq struct {
	g.Meta `path:"/list" method:"get" tags:"文件管理" summary:"获取文件列表" dc:"分页查询文件列表，支持多条件筛选"`

	Page      *beans.Page     `json:"page"`
	DatasetId int64           `json:"datasetId"`
	Keyword   string          `json:"keyword" dc:"关键词搜索"`
	Title     string          `json:"title" dc:"文件标题"`
	Status    document.Status `json:"status"`
}

// ListDocumentRes 文件列表响应
type ListDocumentRes struct {
	List  []*DocumentVO `json:"list"`
	Total int           `json:"total"`
}

type DocumentVO struct {
	Id           int64                 `json:"id,string" dc:"id"`
	DatasetId    int64                 `json:"datasetId,string"`
	Title        string                `json:"title" dc:"文件标题"`
	Format       string                `orm:"format" json:"format" dc:"文件格式"`
	FilePath     string                `orm:"file_path" json:"filePath" dc:"文件存储路径"`
	Status       document.Status       `json:"status" dc:"状态1启用/0停用"`
	VectorStatus document.VectorStatus `json:"vectorStatus" dc:"向量化状态 状态: 1 待定, 2 处理, 3 完成, 4 失败"`
	ChunkCount   int64                 `json:"chunkCount" dc:"分块数"`
	FileSize     int64                 `json:"fileSize" dc:"文件大小"`
	CreatedAt    *gtime.Time           `json:"createdAt" dc:"创建时间"`
	UpdatedAt    *gtime.Time           `json:"updatedAt" dc:"更新时间"`
}

// DocumentVectorReq 处理文件请求(向量化)
type DocumentVectorReq struct {
	g.Meta `path:"/vectorization" method:"post" tags:"文件管理" summary:"文件向量化处理" dc:"文件向量化处理"`

	Id        int64 `json:"id" v:"required#ID不能为空"`
	DatasetId int64 `json:"datasetId" v:"required#数据集ID不能为空"`
}

type DocumentVectorRPC struct {
	Id          int64           `json:"id" dc:"id"`
	DatasetId   int64           `json:"datasetId" dc:"所属数据集ID"`
	DocumentId  int64           `json:"documentId" dc:"文件ID"`
	ContentHash string          `json:"contentHash" dc:"内容hash"`
	Vector      pgvector.Vector `json:"vector" dc:"向量"`
}

type VectorSemanticSplitReq struct {
	g.Meta `path:"/vectorSemanticSplit" method:"post" tags:"文件管理" summary:"向量化生成" dc:"向量化生成"`

	Id int64 `json:"id" v:"required#ID不能为空"`
}

type SearchRecursiveSplitReq struct {
	g.Meta `path:"/searchRecursiveSplit" method:"post" tags:"文件管理" summary:"全文检索生成" dc:"全文检索生成"`

	Id int64 `json:"id" v:"required#ID不能为空"`
}

type KeywordExtractReq struct {
	g.Meta `path:"/keywordExtract" method:"post" tags:"文件管理" summary:"关键词提取" dc:"关键词提取"`

	Id int64 `json:"id" v:"required#ID不能为空"`
}