schema

type AIAgentRuntime struct {
	gorm.Model

	Uuid              string `json:"uuid" gorm:"unique_index"`
	PersistentSession string `gorm:"index"`
	Name              string `json:"name"`
	Seq               int64  `json:"seq" gorm:"index"`

	TypeName       AIAgentRuntimeType `gorm:"index"`
	QuotedTimeline string             `json:"timeline"`

	QuotedUserInput string `json:"quoted_user_input"`
	ForgeName       string `json:"forge_name"`
}

type AIAgentRuntimeType string

type AIForge struct {
	gorm.Model

	ForgeVerboseName   string
	ForgeName          string `gorm:"unique_index"`
	ForgeContent       string
	ForgeType          string // "yak" or "json"
	ParamsUIConfig     string
	Params             string // cli params
	UserPersistentData string // for user preferences
	Description        string // forge description
	Tools              string // tools
	ToolKeywords       string // tool keywords
	Actions            string
	Tags               string

	InitPrompt       string
	PersistentPrompt string
	PlanPrompt       string
	ResultPrompt     string

	IsTemporary bool // for temporary use, will be cleaned up later
}

type AIMemoryCollection struct {
	gorm.Model

	// 会话ID，每个会话有一个独立的HNSW索引
	SessionID string `json:"session_id" gorm:"unique_index;not null"`

	// HNSW Graph 的二进制序列化数据
	GraphBinary []byte `json:"graph_binary" gorm:"type:blob"`

	// HNSW 参数配置
	M           int     `json:"m" gorm:"default:16"`             // 最大邻居数
	Ml          float64 `json:"ml" gorm:"default:0.25"`          // 层生成因子
	EfSearch    int     `json:"ef_search" gorm:"default:20"`     // 搜索时的候选节点数
	EfConstruct int     `json:"ef_construct" gorm:"default:200"` // 构建时的候选节点数

	// 向量维度（固定为7维 - C.O.R.E. P.A.C.T.）
	Dimension int `json:"dimension" gorm:"default:7"`
}

type AIMemoryEntity struct {
	gorm.Model

	// 记忆条目唯一标识
	MemoryID string `json:"memory_id" gorm:"unique_index;not null"`

	// 会话ID，关联到特定的AI会话
	SessionID string `json:"session_id" gorm:"index;not null"`

	// 记忆内容
	Content string `json:"content" gorm:"type:text;not null"`

	// 领域标签，用于分类和过滤
	Tags StringArray `json:"tags" gorm:"type:text"`

	// 潜在问题列表，用于RAG检索
	PotentialQuestions StringArray `json:"potential_questions" gorm:"type:text"`

	// C.O.R.E. P.A.C.T. Framework 评分 (0.0-1.0)
	C_Score float64 `json:"c_score"` // Connectivity Score 关联度
	O_Score float64 `json:"o_score"` // Origin Score 来源与确定性
	R_Score float64 `json:"r_score"` // Relevance Score 相关性
	E_Score float64 `json:"e_score"` // Emotion Score 情感
	P_Score float64 `json:"p_score"` // Preference Score 个人偏好
	A_Score float64 `json:"a_score"` // Actionability Score 可操作性
	T_Score float64 `json:"t_score"` // Temporality Score 时效性

	// C.O.R.E. P.A.C.T. 向量，用于快速过滤和排序
	CorePactVector FloatArray `json:"core_pact_vector" gorm:"type:text"`
}

type AIYakTool struct {
	gorm.Model

	Name string `json:"name" gorm:"unique_index"`
	// 展示给用户的名称
	VerboseName string `json:"verbose_name"`
	Description string `json:"description" gorm:"type:text;index"`
	Keywords    string `json:"keywords" gorm:"type:text;index"`
	Content     string `json:"content" gorm:"type:text"`
	Params      string `json:"params" gorm:"type:text"`
	Path        string `json:"path" gorm:"type:text;index"`
	Hash        string `json:"hash"`
	IsFavorite  bool   `json:"is_favorite" gorm:"default:false;index"`
}

type AiApiKeys struct {
	gorm.Model
	APIKey        string    `json:"api_key" gorm:"index"`
	AllowedModels string    `json:"allowed_models"`
	InputBytes    int64     `json:"input_bytes"`                // 输入字节数统计
	OutputBytes   int64     `json:"output_bytes"`               // 输出字节数统计
	UsageCount    int64     `json:"usage_count"`                // 使用次数统计
	SuccessCount  int64     `json:"success_count"`              // 成功请求数
	FailureCount  int64     `json:"failure_count"`              // 失败请求数
	LastUsedTime  time.Time `json:"last_used_time"`             // 上次使用时间
	Active        bool      `json:"active" gorm:"default:true"` // 新增：API Key 激活状态
}

type AiCheckpoint struct {
	gorm.Model

	CoordinatorUuid    string           `json:"coordinator_uuid" gorm:"index"`
	Seq                int64            `json:"seq" gorm:"index"`
	Type               AiCheckpointType `json:"type" gorm:"index"`
	RequestQuotedJson  string           `json:"request_quoted_json"`
	ResponseQuotedJson string           `json:"response_quoted_json"`
	Finished           bool             `json:"finished"`

	Hash string `json:"hash" gorm:"unique_index"`
}

type AiCheckpointType string

type AiOutputEvent struct {
	gorm.Model
	CoordinatorId string
	Type          EventType

	NodeId      string
	IsSystem    bool
	IsStream    bool
	IsReason    bool
	IsSync      bool
	StreamDelta []byte
	IsJson      bool
	Content     []byte

	SyncID    string
	EventUUID string

	Timestamp int64

	// task index
	TaskIndex string

	// task uuid
	TaskUUID string
	// disable markdown render
	DisableMarkdown bool
	CallToolID      string

	Processes []*AiProcess `gorm:"many2many:ai_processes_and_events;"`

	ContentType string
	AIService   string
}

type AiProcess struct {
	gorm.Model
	ProcessType string           `json:"process_type"`
	ProcessId   string           `json:"process_id" `
	Events      []*AiOutputEvent `gorm:"many2many:ai_processes_and_events;"`
}

type AiProvider struct {
	gorm.Model

	WrapperName  string `json:"wrapper_name" gorm:"index"`
	ModelName    string `json:"model_name" gorm:"index"`
	TypeName     string `json:"type_name" gorm:"index"`
	DomainOrURL  string `json:"domain_or_url" gorm:"index"`
	APIKey       string `json:"api_key" gorm:"index"`
	NoHTTPS      bool   `json:"no_https"`
	ProviderMode string `json:"provider_mode" gorm:"default:'chat'"` // Provider 模式: "chat" 或 "embedding"

	// 可用性指标
	SuccessCount  int64 `json:"success_count"`  // 成功请求总数
	FailureCount  int64 `json:"failure_count"`  // 失败请求总数
	TotalRequests int64 `json:"total_requests"` // 总请求数

	// 最后一次请求信息
	LastRequestTime   time.Time `json:"last_request_time"`   // 最后一次请求时间
	LastRequestStatus bool      `json:"last_request_status"` // 最后一次请求状态 (true=成功, false=失败)
	LastLatency       int64     `json:"last_latency"`        // 最后一次请求延迟 (毫秒)

	// 健康状态
	IsHealthy             bool      `json:"is_healthy"`                                    // 提供者是否健康
	HealthCheckTime       time.Time `json:"health_check_time"`                             // 最后一次健康检查时间
	IsFirstCheckCompleted bool      `json:"is_first_check_completed" gorm:"default:false"` // 首次健康检查是否完成
}

type AliveHost struct {
	gorm.Model

	Hash string `json:"hash"`

	IP        string `json:"ip"`
	IPInteger int64  `json:"ip_integer"`

	// 设置运行时 ID 为了关联具体漏洞
	RuntimeId string `json:"runtime_id"`
}

type AnalyzedHTTPFlow struct {
	gorm.Model
	ResultId        string          `json:"result_id" gorm:"index"`
	Rule            string          `json:"rule"`
	RuleVerboseName string          `json:"rule_verbose_name"`
	HTTPFlowId      int64           `json:"http_flow_id"`
	ExtractedData   []ExtractedData `json:"extracted_data"`
}

type BackProject struct {
	Project
	FolderName      string `json:"folder_name"`
	ChildFolderName string `json:"child_folder_name"`
}

type Briefing struct {
	gorm.Model

	SourceXmlUrl    string
	RssFeedHash     string
	Hash            string     `gorm:"columns:hash;unique_index"`
	Title           string     `json:"title,omitempty"`
	Description     string     `json:"description,omitempty"`
	Content         string     `json:"content,omitempty"`
	Link            string     `json:"link,omitempty"`
	Updated         string     `json:"updated,omitempty"`
	UpdatedParsed   *time.Time `json:"updatedParsed,omitempty"`
	Published       string     `json:"published,omitempty"`
	PublishedParsed *time.Time `json:"publishedParsed,omitempty"`
	Author          string     `json:"author,omitempty"`
	AuthorEmail     string     `json:"author_email,omitempty"`
	GUID            string     `json:"guid,omitempty"`
	ImageUrl        string     `json:"image_url,omitempty"`
	ImageName       string     `json:"image_name,omitempty"`
	Categories      string     `json:"categories,omitempty"`
	Tags            string     `json:"tags"`
	IsRead          bool       `json:"is_read"`
}

type BroadCastHandler struct {
	// contains filtered or unexported fields
}

type CPE struct {
	Part     string `yaml:"part,omitempty" json:"part"`
	Vendor   string `yaml:"vendor,omitempty" json:"vendor"`
	Product  string `yaml:"product,omitempty" json:"product"`
	Version  string `yaml:"version,omitempty" json:"version"`
	Update   string `yaml:"update,omitempty" json:"update"`
	Edition  string `yaml:"edition,omitempty" json:"edition"`
	Language string `yaml:"language,omitempty" json:"language"`
}

type CodecFlow struct {
	gorm.Model
	FlowName   string
	WorkFlow   []byte
	WorkFlowUI string
}

type Domain struct {
	gorm.Model

	Domain    string `json:"domain" gorm:"index"`
	IPAddr    string `json:"ip_addr"`
	IPInteger int64  `json:"ip_integer"`

	HTTPTitle string

	Hash string `json:"hash" gorm:"unique_index"`

	Tags string `json:"tags"`
}

type ERModelEntity struct {
	gorm.Model

	RepositoryUUID    string      `gorm:"index"`
	EntityName        string      `gorm:"index"`
	Uuid              string      `gorm:"unique_index"`
	Description       string      // 对该实体的简要描述
	EntityType        string      // 实体的类型或类别
	EntityTypeVerbose string      // 实体类型的详细描述
	Attributes        MetadataMap `gorm:"type:text" json:"attributes"`

	RuntimeID string
}

type ERModelRelationship struct {
	gorm.Model

	Uuid           string `gorm:"unique_index"`
	RepositoryUUID string `gorm:"index"`

	SourceEntityID uint
	TargetEntityID uint

	SourceEntityIndex       string
	TargetEntityIndex       string
	RelationshipType        string
	RelationshipTypeVerbose string
	Hash                    string      `gorm:"unique_index"`
	Attributes              MetadataMap `gorm:"type:text" json:"attributes"`

	RuntimeID string
}

type EntityRepository struct {
	gorm.Model

	RAGID          string `gorm:"index"`
	Uuid           string `gorm:"unique_index"`
	EntityBaseName string `gorm:"index"`
	Description    string
}

type EventType string

type ExecHistory struct {
	gorm.Model

	Hash string `gorm:"unique_index"`

	RuntimeId     string `json:"runtime_id" gorm:"unique_index"`
	Script        string `json:"script"`
	ScriptId      string `json:"script_id" gorm:"index"`
	TimestampNano int64  `json:"timestamp"`
	FromYakModule string `json:"from_yak_module" gorm:"index"`
	DurationMs    int64  `json:"duration_ms"`
	Params        string `json:"params"`
	Stdout        string `json:"stdout"`
	Stderr        string `json:"stderr"`
	Ok            bool   `json:"ok"`
	Reason        string `json:"reason"`

	StdoutLen int64 `json:"stdout_len" gorm:"-"`
	StderrLen int64 `json:"stderr_len" gorm:"-"`

	// json
	Messages string `json:"messages"`
}

type ExecResult struct {
	gorm.Model

	YakScriptName string `json:"yak_script_name" gorm:"index"`
	Raw           string `json:"raw"`
}

type ExtractedData struct {
	gorm.Model

	// sourcetype 一般来说是标注数据来源
	SourceType string `gorm:"index"`

	// trace id 表示数据源的 ID
	TraceId string `gorm:"index"`

	// 提取数据的正则数据
	Regexp string

	// 规则 Verbose
	RuleVerbose string

	// UTF8 safe escape
	Data string

	// DataIndex 表示数据的位置
	DataIndex int

	// Length 表示数据的长度
	Length int

	// IsMatchRequest 表示是否是匹配请求
	IsMatchRequest bool

	AnalyzedHTTPFlowId uint `gorm:"index"`
}

type FloatArray []float32

type GeneralRule struct {
	gorm.Model
	*CPE
	RuleName        string `json:"指纹名称" gorm:"unique_index"`
	WebPath         string `json:"web路径"`
	ExtInfo         string
	MatchExpression string              `json:"指纹规则"`
	Groups          []*GeneralRuleGroup `gorm:"many2many:general_rule_and_group;"`
}

type GeneralRuleGroup struct {
	gorm.Model
	GroupName string         `json:"group_name" gorm:"unique_index"`
	Rules     []*GeneralRule `gorm:"many2many:general_rule_and_group;"`
}

type GeneralStorage struct {
	gorm.Model

	Key string `json:"key" gorm:"unique_index"`

	// 经过 JSON + Strconv
	Value string `json:"value"`

	// 过期时间
	ExpiredAt time.Time

	// YAKIT SUBPROC_ENV
	ProcessEnv bool

	// 帮助信息，描述这个变量是干嘛的
	Verbose string

	// 描述变量所在的组是啥
	Group string
}

type GroupInfo struct {
	LayerLevel int
	Key        string
	Neighbors  []string
}

type HTTPFlow struct {
	gorm.Model

	HiddenIndex        string `gorm:"index" json:"hidden_index,omitempty"`
	NoFixContentLength bool   `json:"no_fix_content_length" json:"no_fix_content_length,omitempty"`
	Hash               string `gorm:"unique_index" json:"unique_index,omitempty"`
	IsHTTPS            bool   `json:"is_https,omitempty"`
	Url                string `gorm:"index" json:"url,omitempty"`
	Path               string `json:"path,omitempty"`
	Method             string `json:"method,omitempty"`
	RequestLength      int64  `json:"request_length,omitempty"`
	BodyLength         int64  `json:"body_length,omitempty"`
	ContentType        string `json:"content_type,omitempty"`
	StatusCode         int64  `json:"status_code,omitempty"`
	SourceType         string `json:"source_type,omitempty"`
	Request            string `json:"request,omitempty"`
	Response           string `json:"response,omitempty"`
	Duration           int64  `json:"duration,omitempty"`
	GetParamsTotal     int    `json:"get_params_total,omitempty"`
	PostParamsTotal    int    `json:"post_params_total,omitempty"`
	CookieParamsTotal  int    `json:"cookie_params_total,omitempty"`
	IPAddress          string `json:"ip_address,omitempty"`
	RemoteAddr         string `json:"remote_addr,omitempty"`
	IPInteger          int    `json:"ip_integer,omitempty"`
	Tags               string `json:"tags,omitempty"` // 用来打标！
	Payload            string `json:"payload,omitempty"`

	// Websocket 相关字段
	IsWebsocket bool `json:"is_websocket,omitempty"`
	// 用来计算 websocket hash, 每次连接都不一样，一般来说，内部对象 req 指针足够了
	WebsocketHash string `json:"websocket_hash,omitempty"`

	RuntimeId   string         `json:"runtime_id,omitempty" gorm:"index"`
	FromPlugin  string         `json:"from_plugin,omitempty"`
	ProcessName sql.NullString `json:"process_name,omitempty"`

	// friendly for gorm build instance, not for store
	// 这两个字段不参与数据库存储，但是在序列化的时候，会被覆盖
	// 主要用来标记用户的 Request 和 Response 是否超大
	IsRequestOversize  bool `gorm:"-" json:"is_request_oversize,omitempty"`
	IsResponseOversize bool `gorm:"-" json:"is_response_oversize,omitempty"`

	IsReadTooSlowResponse      bool   `json:"is_read_too_slow_response,omitempty"`
	IsTooLargeResponse         bool   `json:"is_too_large_response,omitempty"`
	TooLargeResponseHeaderFile string `json:"too_large_response_header_file,omitempty"`
	TooLargeResponseBodyFile   string `json:"too_large_response_body_file,omitempty"`
	// 同步到企业端
	UploadOnline bool   `json:"upload_online,omitempty"`
	Host         string `json:"host,omitempty"`
}

type Host struct {
	gorm.Model

	IP        string `json:"ip" gorm:"unique_index"`
	IPInteger int64  `json:"ip_integer"`

	IsInPublicNet bool

	// splite by comma
	Domains string
}

type HotPatchTemplate struct {
	gorm.Model
	Name    string `json:"name"`
	Content string `json:"content"`
	Type    string `json:"type"`
}

type HybridScanTask struct {
	gorm.Model

	TaskId string `gorm:"unique_index"`
	// executing
	// paused
	// done
	Status              string
	Reason              string // user cancel / finished / recover failed so on
	SurvivalTaskIndexes string // 暂停的时候正在执行的任务

	// struct{ https bool; request bytes }[]
	Targets string
	// string[]
	Plugins         string
	TotalTargets    int64
	TotalPlugins    int64
	TotalTasks      int64
	FinishedTasks   int64
	FinishedTargets int64

	ScanConfig []byte

	HybridScanTaskSource string
}

type I18n struct {
	Zh string `json:"zh"`
	En string `json:"en"`
}

type KnowledgeBaseEntry struct {
	gorm.Model

	// 知识库名称
	KnowledgeBaseID int64 `gorm:"not null" json:"knowledge_base_id"`

	RelatedEntityUUIDS string // split by ","

	// 知识标题(和知识库名称应该是联合唯一索引)
	KnowledgeTitle string `gorm:"not null" json:"knowledge_title"`

	// 知识类型（如：CoreConcept、Standard、Guideline等）
	KnowledgeType string `gorm:"index;not null" json:"knowledge_type"`

	// 重要性评分（1-10）
	ImportanceScore int `gorm:"index" json:"importance_score"`

	// 关键词列表，用于快速搜索和分类
	Keywords StringArray `gorm:"type:text" json:"keywords"`

	// 知识详细信息，包含具体内容描述
	KnowledgeDetails string `gorm:"type:text" json:"knowledge_details"`

	// 知识摘要，简要概述
	Summary string `gorm:"type:text" json:"summary"`

	// 来源页码或章节编号
	SourcePage int `gorm:"index" json:"source_page"`

	// 潜在问题列表，这些问题可能与该知识条目相关
	PotentialQuestions StringArray `gorm:"type:text" json:"potential_questions"`

	// 潜在问题向量，用于快速搜索潜在问题
	PotentialQuestionsVector FloatArray `gorm:"type:text" json:"potential_questions_vector"`

	// 唯一标识符，用于在向量索引中唯一标识该知识条目
	HiddenIndex string `gorm:"unique_index"`
}

type KnowledgeBaseInfo struct {
	gorm.Model

	RAGID string `gorm:"index"`
	// 知识库名称(唯一)
	KnowledgeBaseName string `gorm:"unique_index;not null" json:"knowledge_base_name"`

	// 知识库描述
	KnowledgeBaseDescription string `gorm:"type:text" json:"knowledge_base_description"`

	// 知识库类型
	KnowledgeBaseType string `gorm:"index;not null" json:"knowledge_base_type"`

	SerialVersionUID string `gorm:"index" json:"serial_version_uid"`
}

type LoginSession struct {
	gorm.Model

	SessionID string    `json:"session_id" gorm:"index"`
	ExpiresAt time.Time `json:"expires_at"`
}

type MCPServer struct {
	gorm.Model

	Name    string `gorm:"index;not null" json:"name"` // 服务器名称
	Type    string `gorm:"index;not null" json:"type"` // 服务器类型 (stdio/sse)
	URL     string `gorm:"type:text" json:"url"`       // 服务器URL (for sse type)
	Command string `gorm:"type:text" json:"command"`   // 启动命令 (for stdio type)
	Enable  bool   `gorm:"default:true" json:"enable"` // 是否启用
}

type MapEx[K comparable, V any] map[K]V

type MarkdownDoc struct {
	gorm.Model

	YakScriptId   int64  `json:"yak_script_id" gorm:"index"`
	YakScriptName string `json:"yak_script_name" gorm:"index"`
	Markdown      string `json:"markdown"`
}

type MenuItem struct {
	gorm.Model

	Group         string `json:"group" `
	Verbose       string `json:"verbose"`
	YakScriptName string `json:"yak_script_name"`
	Hash          string `json:"-" gorm:"unique_index"`

	// quoted json
	BatchPluginFilterJson string `json:"batch_plugin_filter_json"`
	Mode                  string `json:"mode"`
	MenuSort              int64  `json:"menu_sort"`
	GroupSort             int64  `json:"group_sort"`
}

type MetadataMap map[string]interface{}

type NaslScript struct {
	gorm.Model
	OriginFileName  string `json:"origin_file_name"`
	Hash            string `json:"hash" gorm:"unique_index"`
	OID             string `json:"oid"`
	CVE             string `json:"cve"`
	ScriptName      string `json:"script_name"`
	Script          string `json:"script"`
	Tags            string `json:"tags,omitempty"`
	Version         string `json:"version"`
	Category        string `json:"category"`
	Family          string `json:"family"`
	Copyright       string `json:"copyright"`
	Dependencies    string `json:"dependencies,omitempty"`
	RequirePorts    string `json:"require_ports,omitempty"`
	RequireUdpPorts string `json:"require_udp_ports,omitempty"`
	ExcludeKeys     string `json:"exclude_keys,omitempty"`
	Xref            string `json:"xref,omitempty"`
	Preferences     string `json:"preferences,omitempty"`
	BugtraqId       string `json:"bugtraqId,omitempty"`
	MandatoryKeys   string `json:"mandatory_keys,omitempty"`
	Timeout         int    `json:"timeout,omitempty"`
	RequireKeys     string `json:"require_keys,omitempty"`
}

type NavigationBar struct {
	gorm.Model
	Group         string `json:"group" `
	YakScriptName string `json:"yak_script_name"`
	Hash          string `json:"-" gorm:"unique_index"`
	Mode          string `json:"mode"`
	VerboseSort   int64  `json:"verbose_sort"`
	GroupSort     int64  `json:"group_sort"`
	Route         string `json:"route"`
	Verbose       string `json:"verbose"`
	GroupLabel    string `json:"group_label"`
	VerboseLabel  string `json:"verbose_label"`
}

type Note struct {
	gorm.Model

	Title   string `json:"title"`
	Content string `json:"content"`
}

type Payload struct {
	gorm.Model

	// Must: payload group
	Group string `json:"group" gorm:"index"`

	// payload folder
	Folder     *string `json:"folder" gorm:"column:folder;default:''"`          // default empty string
	GroupIndex *int64  `json:"group_index" gorm:"column:group_index;default:0"` // default 0

	// strconv Quoted
	// Must: payload data
	Content *string `json:"content"`

	// hit count
	HitCount *int64 `json:"hit_count" gorm:"column:hit_count;default:0"` // default 0

	// the group save in file only contain one payload, and this `payload.IsFile = true` `payload.Content` is filepath
	IsFile *bool `json:"is_file" gorm:"column:is_file;default:false"` // default false

	// Hash string
	Hash string `json:"hash" gorm:"unique_index"`
}

type PluginEnv struct {
	gorm.Model
	Key   string `gorm:"unique_index"`
	Value string
}

type PluginGroup struct {
	gorm.Model

	YakScriptName string `json:"yak_script_name" gorm:"index"`
	Group         string `json:"group"`
	Hash          string `json:"hash" gorm:"unique_index"`
	TemporaryId   string `json:"temporary_id"`
	IsPocBuiltIn  bool   `json:"is_poc_built_in"`
}

type Port struct {
	gorm.Model

	Host        string `json:"host" gorm:"index"`
	IPInteger   int    `json:"ip_integer" gorm:"column:ip_integer" json:"ip_integer"`
	Port        int    `json:"port" gorm:"index"`
	Proto       string `json:"proto"`
	ServiceType string `json:"service_type"`
	State       string `json:"state"`
	Reason      string `json:"reason"`
	Fingerprint string `json:"fingerprint"`
	CPE         string `json:"cpe"`
	HtmlTitle   string `json:"html_title"`
	From        string `json:"from"`
	Hash        string `json:"hash"`
	TaskName    string `json:"task_name"`

	// runtime id 运行时 ID
	RuntimeId string `json:"runtime_id"`
}

type Progress struct {
	gorm.Model
	RuntimeId            string
	CurrentProgress      float64
	YakScriptOnlineGroup string
	// 记录指针
	LastRecordPtr int64
	TaskName      string
	// 额外信息
	ExtraInfo string

	ProgressSource string

	// 任务记录的参数
	ProgressTaskParam []byte

	// 目标 大部分的progress都应该有制定目标，所以尝试提取出来作为单独的数据使用
	Target string
}

type Project struct {
	gorm.Model

	ProjectName string
	Description string

	DatabasePath string

	IsCurrentProject bool
	FolderID         int64
	ChildFolderID    int64
	Type             string
	// Hash string `gorm:"unique_index"`
	// 企业版 项目模块及项目编号
	ExternalModule      string
	ExternalProjectCode string
}

type ProjectGeneralStorage struct {
	gorm.Model

	Key string `json:"key" gorm:"unique_index"`

	// 经过 JSON + Strconv
	Value string `json:"value"`

	// 过期时间
	ExpiredAt time.Time

	// YAKIT SUBPROC_ENV
	ProcessEnv bool

	// 帮助信息，描述这个变量是干嘛的
	Verbose string

	// 描述变量所在的组是啥
	Group string
}

type RAGDocumentType string

type Report struct {
	TitleValue string        `json:"title"`
	OwnerValue string        `json:"owner"`
	FromValue  string        `json:"from"`
	Items      []*ReportItem `json:"items"`
}

type ReportItem struct {
	Type    string `json:"type"`
	Content string `json:"content"`
}

type ReportRecord struct {
	gorm.Model

	Title       string
	PublishedAt time.Time `json:"published_at"`
	Hash        string    `json:"hash" gorm:"unique_index"`
	Owner       string    `json:"owner"`
	From        string    `json:"from"`
	QuotedJson  string    `json:"quoted_json"`
}

type Risk struct {
	gorm.Model

	Hash string `json:"hash"`

	// essential
	IP        string `json:"ip"`
	IPInteger int64  `json:"ip_integer"`

	// extraTargets
	Url  string `json:"url"`
	Port int    `json:"port"`
	Host string `json:"host"`

	//
	Title           string `json:"title"`
	TitleVerbose    string `json:"title_verbose"`
	Description     string `json:"description"`
	Solution        string `json:"solution"`
	RiskType        string `json:"risk_type"`
	RiskTypeVerbose string `json:"risk_verbose"`
	Parameter       string `json:"parameter"`
	Payload         string `json:"payload"`
	Details         string `json:"details"`
	Severity        string `json:"severity"`

	// 来源于哪个插件？
	FromYakScript string `json:"from_yak_script"`
	YakScriptUUID string `json:"yak_script_uuid"`

	// 等待验证中？
	WaitingVerified bool `json:"waiting_verified"`
	// 用于验证的 ReverseToken
	ReverseToken string `json:"reverse_token"`

	// 设置运行时 ID 为了关联具体漏洞
	RuntimeId      string `json:"runtime_id"`
	QuotedRequest  string `json:"quoted_request"`
	QuotedResponse string `json:"quoted_response"`

	// 潜在威胁：用于输出合规性质的漏洞内容
	IsPotential bool `json:"is_potential"`

	CVE                 string `json:"cve"`
	IsRead              bool   `json:"is_read"`
	Ignore              bool   `json:"ignore"`
	UploadOnline        bool   `json:"upload_online"`
	TaskName            string `json:"task_name"`
	CveAccessVector     string `json:"cve_access_vector"`
	CveAccessComplexity string `json:"cve_access_complexity"`
	Tags                string `json:"tags"`

	// SyntaxFlow
	ResultID    uint   `json:"result_id"`
	Variable    string `json:"variable"`
	ProgramName string `json:"program_name"`
}

type RssFeed struct {
	gorm.Model

	SourceXmlUrl    string
	Hash            string     `gorm:"columns:hash;unique_index"`
	Title           string     `json:"title,omitempty"`
	Description     string     `json:"description,omitempty"`
	Link            string     `json:"link,omitempty"`
	FeedLink        string     `json:"feedLink,omitempty"`
	Updated         string     `json:"updated,omitempty"`
	UpdatedParsed   *time.Time `json:"updatedParsed,omitempty"`
	Published       string     `json:"published,omitempty"`
	PublishedParsed *time.Time `json:"publishedParsed,omitempty"`
	Author          string     `json:"author,omitempty"`
	AuthorEmail     string     `json:"author_email,omitempty"`
	Language        string     `json:"language,omitempty"`
	ImageUrl        string     `json:"image_url,omitempty"`
	ImageName       string     `json:"image_name,omitempty"`
	Copyright       string     `json:"copyright,omitempty"`
	Generator       string     `json:"generator,omitempty"`
	Categories      string     `json:"categories,omitempty"`
	FeedType        string     `json:"feedType"`
	FeedVersion     string     `json:"feedVersion"`
}

type SSADiffCompareType string

type SSADiffResult struct {
	*gorm.Model
	BaseLine string
	Compare  string
	RuleName string // rule name

	BaseLineRiskHash string
	CompareRiskHash  string

	Status string
	//CompareType 比较类型，是custom还是risk
	CompareType    string
	DiffResultKind string //结果类型，是taskID还是program

	Hash string `gorm:"uniqueIndex"`
}

type SSADiffResultKind string

type SSAProject struct {
	gorm.Model
	// 项目基础信息
	ProjectName string             `json:"project_name" gorm:"index;not null;comment:项目名称"`
	Description string             `json:"description,omitempty" gorm:"comment:项目描述"`
	Tags        string             `json:"tags,omitempty" gorm:"comment:项目标签"`
	Language    ssaconfig.Language `json:"language" gorm:"index;comment:项目语言"`
	URL         string             `json:"url,omitempty" gorm:"index;comment:项目源码地址"`
	// 配置选项
	Config []byte `json:"config"`
	Hash   string `json:"hash" gorm:"unique_index"`
}

type SSARisk struct {
	gorm.Model

	Hash string `json:"hash" gorm:"unique_index"`

	// risk info
	Title        string             `json:"title"`
	TitleVerbose string             `json:"title_verbose"`
	Description  string             `json:"description" gorm:"type:longtext"`
	Solution     string             `json:"solution" gorm:"type:longtext"`
	RiskType     string             `json:"risk_type"`
	Details      string             `json:"details"`
	Severity     SyntaxFlowSeverity `json:"severity" gorm:"index"`
	Language     string             `json:"language"`
	// 潜在威胁：用于输出合规性质的漏洞内容
	IsPotential bool `json:"is_potential"`

	// risk info
	CVE                 string      `json:"cve"`
	CWE                 StringArray `gorm:"type:text" json:"cwe"`
	IsRead              bool        `json:"is_read"`
	Ignore              bool        `json:"ignore"`
	UploadOnline        bool        `json:"upload_online"`
	CveAccessVector     string      `json:"cve_access_vector"`
	CveAccessComplexity string      `json:"cve_access_complexity"`
	Tags                string      `json:"tags"`

	// 来源于哪个规则
	FromRule string `json:"from_rule"`
	// 来源于哪个项目
	ProgramName string `json:"program_name" gorm:"index"`
	// file url yakurl
	CodeSourceUrl string `json:"code_source_url" gorm:"index"`
	CodeRange     string `json:"code_range" gorm:"type:longtext"`
	CodeFragment  string `json:"code_fragment" gorm:"type:longtext"`
	// for query risk
	FunctionName string `json:"function_name" gorm:"index"`
	Line         int64  `json:"line" gorm:"index"`
	// 设置运行时 ID 关联 SyntaxflowTask
	RuntimeId string `json:"runtime_id" gorm:"index"`
	// for query result
	ResultID uint64 `json:"result_id"` // result
	Variable string `json:"variable"`  // result/variable
	Index    int64  `json:"index"`     // result/variable/index

	// 最新处置状态
	LatestDisposalStatus string `json:"latest_disposal_status" gorm:"index;default:'not_set'"`
	// Risk 特征哈希，用于区分不同批次扫描的同一个漏洞
	// 包含 SSA 信息(disam、function)、FileUrl、SfRule 信息(ruleId、variableName、level)
	RiskFeatureHash string `json:"risk_feature_hash" gorm:"index"`

	SSAProjectID uint64 `json:"ssa_project_id" gorm:"index"`
}

type SSARiskDisposalStatus string

type SSARiskDisposals struct {
	gorm.Model
	TaskId    string `json:"task_id" gorm:"index"`
	SSARiskID int64  `json:"ssa_risk_id" gorm:"index"`
	// RiskFeatureHash 用于标识风险特征的唯一哈希值,可以用来实现处置的继承
	RiskFeatureHash string `json:"risk_feature_hash" gorm:"index;not null"`
	Status          string `json:"status" gorm:"index"`
	Comment         string `json:"comment" gorm:"type:text"`
}

type ScreenRecorder struct {
	gorm.Model

	// 保存到本地的路径
	Filename  string
	NoteInfo  string
	Project   string
	Hash      string `json:"hash" gorm:"unique_index"`
	VideoName string
	Cover     string `gorm:"type:longtext"`
	Duration  string
}

type ScriptOrRule interface {
	GetScriptName() string
	GetContent() string
	GetType() string
}

type SlicesEx[K comparable] []K

type Snippets struct {
	gorm.Model
	SnippetId   string `gorm:"uniqueIndex;not null" json:"snippet_id"`
	SnippetName string `gorm:"type:text;not null" json:"snippet_name"`
	SnippetBody string `gorm:"type:text;not null" json:"snippet_data"`
	SnippetDesc string `gorm:"type:text" json:"snippet_desc"`

	SnippetLevel SnippetsLevel `gorm:"type:text;not null" json:"snippet_level"`
	SnippetState SnippetsType  `gorm:"type:text;not null" json:"snippet_state"`
}

type SnippetsLevel string

type SnippetsType string

type StringArray []string

type SyntaxFlowDescInfo struct {
	// RuleId 关联的规则ID
	RuleId string `json:"rule_id"`

	// Title 告警标题（英文）
	Title string `json:"title"`

	// TitleZh 告警标题（中文）
	TitleZh string `json:"title_zh"`

	// Description 告警详细描述
	// 说明检测到的问题和影响
	Description string `json:"description"`

	// Solution 解决方案
	// 提供修复建议和最佳实践
	Solution string `json:"solution"`

	// Tag 告警标签
	Tag string `json:"tag"`

	// Severity 严重程度
	// 可选值: info（信息）、low（低危）、middle（中危）、high（高危）、critical（严重）
	Severity SyntaxFlowSeverity `json:"severity"`

	// Purpose 告警目的
	// 可选值: audit（审计）、vuln（漏洞）、config（配置）、security（安全）
	Purpose SyntaxFlowRulePurposeType `json:"purpose"`

	// OnlyMsg 是否仅使用自定义消息
	// true: 只使用 Msg 字段的内容
	// false: 使用完整的结构化信息（Title + Description）
	OnlyMsg bool `json:"only_msg"`

	// Msg 自定义消息内容
	// 当 OnlyMsg=true 时，直接使用此消息作为告警内容
	Msg string `json:"msg"`

	// CVE 关联的 CVE 编号
	CVE string `json:"cve"`

	// CWE 关联的 CWE 列表
	CWE StringArray `json:"cwe"`

	// RiskType 风险类型分类
	RiskType string

	// ExtraInfo 额外信息
	// 键值对形式存储其他自定义信息
	ExtraInfo map[string]string `json:"extra_info"`
}

type SyntaxFlowGroup struct {
	gorm.Model
	GroupName string            `json:"group_name" gorm:"unique_index"`
	IsBuildIn bool              `json:"is_build_in" gorm:"index"`
	Rules     []*SyntaxFlowRule `gorm:"many2many:syntax_flow_rule_and_group;"`
}

type SyntaxFlowRule struct {
	gorm.Model

	// RuleId 规则唯一标识符（UUID）
	// 用于全局唯一标识一个规则，支持跨数据库/跨平台同步
	RuleId string `gorm:"unique_index"`

	// Version 规则版本号
	// 用于版本管理和更新检测，格式如 "1.0.0"、"2.1.3" 等
	// 下载规则时会比较本地和在线版本，决定是否需要更新
	Version string

	// Hash 规则内容哈希值
	// 基于 RuleId、RuleName、Content、Tag、OpCodes 计算
	// 用于快速检测规则内容是否变化，确保数据一致性
	Hash string `json:"hash" gorm:"unique_index"`

	// RuleName 规则名称（唯一）
	// 用于标识和查询规则，如 "java-sql-injection"、"runtime-command-exec"
	// 必须唯一，不可重复
	RuleName string `gorm:"unique_index"`

	// Title 规则标题（英文）
	// 简短描述规则用途，如 "SQL Injection Detection"
	Title string

	// TitleZh 规则标题（中文）
	// 中文标题，如 "SQL注入检测"
	TitleZh string

	// Language 规则适用的编程语言
	// 如 "java"、"php"、"javascript" 等
	// 支持的语言列表见 ssaconfig.GetAllSupportedLanguages()
	Language ssaconfig.Language

	// Purpose 规则用途/目的
	// 可选值: audit（审计）、vuln（漏洞）、config（配置）、security（安全）
	Purpose SyntaxFlowRulePurposeType

	// Tag 规则标签
	// 用于分类和筛选，可包含多个标签，如 "injection,security,owasp"
	Tag string

	// CWE 通用弱点枚举列表
	// Common Weakness Enumeration，如 ["CWE-89", "CWE-564"]
	// 用于标准化漏洞分类
	CWE StringArray `gorm:"type:text" json:"cwe"`

	// CVE 通用漏洞编号
	// Common Vulnerabilities and Exposures，如 "CVE-2021-12345"
	CVE string

	// RiskType 风险类型
	// 用于区分不同类型的安全风险
	RiskType string

	// Type 规则类型
	// 可选值: "sf"（SyntaxFlow）、"yak"（Yak 脚本）
	Type SyntaxFlowRuleType

	// Severity 严重程度
	// 可选值: info（信息）、low（低危）、middle（中危）、high（高危）、critical（严重）
	Severity SyntaxFlowSeverity

	// IsBuildInRule 是否为内置规则
	// true: 系统自带规则，通常不可修改
	// false: 用户自定义规则
	IsBuildInRule bool

	// Verified 规则是否已验证
	// true: 规则已经过测试和验证，质量有保证
	// false: 未验证的规则，可能需要进一步测试
	Verified bool

	// NeedUpdate 规则相较于远端是否有改动
	// true: 远端下载到本地后有新的改动（需要进行冲突检测）
	// false: 远端下载到本地后没有改动（可以直接覆盖最新版本）
	NeedUpdate bool

	// Content 规则主体内容（最重要）
	// SyntaxFlow 规则代码或 Yak 脚本代码
	// 这是规则的核心，定义了检测逻辑
	// 示例:
	//   desc(title: "SQL注入")
	//   executeQuery(* as $sql) as $result;
	//   alert $result;
	Content string

	// Description 规则详细描述
	// 说明规则的检测目标、原理等
	// 如 "检测使用字符串拼接构造 SQL 查询的安全风险"
	Description string

	// AlertDesc 告警描述映射
	// 键为变量名（如 "$result"），值为对应的告警信息
	// 用于为不同的检测结果提供不同的告警描述
	AlertDesc MapEx[string, *SyntaxFlowDescInfo] `gorm:"type:text"`

	// Solution 解决方案
	// 提供修复建议，如 "使用预编译语句（PreparedStatement）"
	Solution string

	// AllowIncluded 是否允许被其他规则包含
	// true: 可以作为库规则被其他规则引用
	// false: 独立规则，不能被包含
	AllowIncluded bool

	// IncludedName 包含时使用的名称
	// 当 AllowIncluded=true 时，其他规则通过此名称引用本规则
	IncludedName string

	// OpCodes 操作码（本地使用）
	// 编译后的字节码或中间表示，用于优化执行性能
	OpCodes string

	// Groups 规则所属的分组列表
	// 多对多关系，一个规则可以属于多个分组
	// 通过中间表 syntax_flow_rule_and_group 关联
	Groups []*SyntaxFlowGroup `gorm:"many2many:syntax_flow_rule_and_group;"`
}

type SyntaxFlowRulePurposeType string

type SyntaxFlowRuleType string

type SyntaxFlowScanTask struct {
	gorm.Model
	TaskId   string `gorm:"unique_index"`
	Programs string `gorm:"index"`
	//ProjectName string `gorm:"index"`
	ProjectId uint64 `gorm:"index"`
	// rules
	RulesCount int64

	Status string // executing / done / paused / error
	Reason string // user cancel / finished / recover failed so on

	Kind SyntaxflowResultKind `json:"kind",gorm:"index"` // debug / scan / query

	// 扫描批次
	ScanBatch uint64 `gorm:"index"`
	// query execute
	FailedQuery  int64 // query failed
	SkipQuery    int64 // language not match, skip this rule
	SuccessQuery int64
	// risk
	RiskCount     int64
	InfoCount     int64
	LowCount      int64
	WarningCount  int64
	CriticalCount int64
	HighCount     int64

	// query process
	TotalQuery int64

	// config
	Config []byte `gorm:"type:text"` // new data
}

type SyntaxFlowSeverity string

type SyntaxflowResultKind string

type TrafficPacket struct {
	gorm.Model

	SessionUuid string `gorm:"index"`

	LinkLayerType        string
	NetworkLayerType     string
	TransportLayerType   string
	ApplicationLayerType string
	Payload              string

	// QuotedRaw contains the raw bytes of the packet, quoted such that it can be
	// caution: QuotedRaw is (maybe) not an utf8-valid string
	// quoted-used for save to database
	QuotedRaw string

	EthernetEndpointHardwareAddrSrc string
	EthernetEndpointHardwareAddrDst string
	IsIpv4                          bool
	IsIpv6                          bool
	NetworkEndpointIPSrc            string
	NetworkEndpointIPDst            string
	TransportEndpointPortSrc        int
	TransportEndpointPortDst        int
}

type TrafficSession struct {
	gorm.Model

	Uuid string `gorm:"index"`

	// Traffic SessionType Means a TCP Session / ICMP Request-Response / UDP Request-Response
	// DNS Request-Response
	// HTTP Request-Response
	// we can't treat Proto as any transport layer proto or application layer proto
	// because we can't know the proto of a packet before we parse it
	//
	// just use session type as a hint / verbose to group some frames(packets).
	//
	// 1. tcp (reassembled)
	// 2. udp (try figure out request-response)
	// 3. dns
	// 4. http (flow)
	// 5. icmp (request-response)
	// 6. sni (tls client hello)
	SessionType string `gorm:"index"`

	DeviceName string `gorm:"index"`
	DeviceType string

	// LinkLayer physical layer
	IsLinkLayerEthernet bool
	LinkLayerSrc        string
	LinkLayerDst        string

	// NetworkLayer network layer
	IsIpv4          bool
	IsIpv6          bool
	NetworkSrcIP    string
	NetworkSrcIPInt int64
	NetworkDstIP    string
	NetworkDstIPInt int64

	// TransportLayer transport layer
	IsTcpIpStack          bool
	TransportLayerSrcPort int
	TransportLayerDstPort int

	// TCP State Flags
	// PDU Reassembled
	IsTCPReassembled bool
	// TCP SYN Detected? If so, it's a new TCP Session
	// 'half' means we haven't seen a FIN or RST
	IsHalfOpen bool
	// TCP FIN Detected
	IsClosed bool
	// TCP RST Detected
	IsForceClosed bool

	// TLS ClientHello
	HaveClientHello bool
	SNI             string
}

type TrafficTCPReassembledFrame struct {
	gorm.Model

	SessionUuid string `gorm:"index"`
	QuotedData  string
	Seq         int64
	Timestamp   int64
	Source      string
	Destination string
}

type VectorStoreCollection struct {
	gorm.Model

	// 集合名称，在系统中唯一
	Name string `gorm:"unique_index;" json:"name"`

	// 集合描述信息
	Description string `gorm:"type:text" json:"description"`

	// 使用的嵌入模型名称
	ModelName string `gorm:"index" json:"model_name"`

	// 向量维度，所有文档的嵌入向量必须具有相同的维度
	Dimension int `gorm:"not null" json:"dimension"`

	// HNSW 算法参数配置
	M                int     `gorm:"default:16" json:"m"`                        // 最大邻居数，影响图的连接密度
	Ml               float64 `gorm:"default:0.25" json:"ml"`                     // 层生成因子，控制层级分布
	EfSearch         int     `gorm:"default:20" json:"ef_search"`                // 搜索时的候选节点数
	EfConstruct      int     `gorm:"default:200" json:"ef_construct"`            // 构建时的候选节点数
	DistanceFuncType string  `gorm:"default:'cosine'" json:"distance_func_type"` // 距离函数类型（cosine、euclidean等）

	// PQ 算法参数配置
	EnablePQMode bool `gorm:"default:false" json:"enable_pq_mode"`

	Archived bool `gorm:"default:false" json:"archived"`

	UUID  string
	RAGID string `gorm:"index"`

	GraphBinary    []byte `gorm:"type:blob" json:"graph_binary"`
	CodeBookBinary []byte `gorm:"type:blob" json:"code_book_binary"`

	SerialVersionUID string `gorm:"index" json:"serial_version_uid"`
}

type VectorStoreDocument struct {
	gorm.Model

	// entity / relationship / knowledge
	DocumentType    RAGDocumentType
	EntityID        string `gorm:"index"`
	RelatedEntities string // text split by ","

	DocumentID string `gorm:"uniqueIndex:idx_document_id_collection_id;index;idx_document_id;not null" json:"document_id"`
	UID        []byte `gorm:"blob"`

	// 所属集合的ID，建立外键关系
	CollectionID uint `gorm:"uniqueIndex:idx_document_id_collection_id;not null" json:"collection_id"`

	// 所属集合的 UUID，唯一值
	CollectionUUID string `gorm:"uniqueIndex"`

	// 文档元数据，以 JSON 格式存储，包含原始文本、来源等信息
	Metadata MetadataMap `gorm:"type:text" json:"metadata"`

	// 文档的嵌入向量，以 JSON 格式存储
	PQCode    []byte     `gorm:"type:text" json:"pq_code"`
	Embedding FloatArray `gorm:"type:text" json:"embedding"`

	// 文档的原始文本
	Content string `gorm:"type:text" json:"content"`

	// HNSW 算法中节点存在的最高层级
	MaxLayer int `gorm:"default:0" json:"max_layer"`

	RuntimeID string
}

type WebFuzzerConfig struct {
	gorm.Model
	PageId string `gorm:"unique_index"`
	Type   string `gorm:"string"`
	Config string `gorm:"string"`
}

type WebFuzzerLabel struct {
	gorm.Model
	Label string `json:"label"`
	// 模版数据唯一标识，用来兼容做对比
	DefaultDescription string `json:"default_description"`
	Description        string `json:"description"`
	Hash               string `gorm:"unique_index"`
}

type WebFuzzerResponse struct {
	gorm.Model

	WebFuzzerTaskId int    `json:"web_fuzzer_task_id" gorm:"index"`
	OK              bool   `json:"ok"`
	Request         string `json:"request"`
	Content         string `json:"content"`
	Payload         string `json:"payload"`
	Url             string `json:"url"`
	StatusCode      int    `json:"status_code"`
	DurationMs      int    `json:"duration_ms"`
	Timestamp       int64  `json:"timestamp"`
	HiddenIndex     string `json:"hidden_index" gorm:"index"`
}

type WebFuzzerTask struct {
	gorm.Model

	// 原始请求 json+quote
	RawFuzzTaskRequest string `json:"raw_fuzz_task_request"`

	// 对应前端的组织形式
	FuzzerIndex    string `json:"fuzzer_index"`
	FuzzerTabIndex string `json:"fuzzer_tab_index"`

	// HTTP 数据流总量
	HTTPFlowTotal        int    `json:"http_flow_total"`
	HTTPFlowSuccessCount int    `json:"http_flow_success_count"`
	HTTPFlowFailedCount  int    `json:"http_flow_failed_count"`
	Ok                   bool   `json:"ok"`
	Reason               string `json:"reason"` // if not ok
	Host                 string `json:"host"`
	Port                 int    `json:"port"`
	// retry 相关
	RetryRootID uint `json:"retry_root_id"`
}

type WebShell struct {
	gorm.Model
	Url string `json:"url" gorm:"index" `
	// pass=payload
	Pass string `json:"pass"`
	// 加密密钥
	SecretKey string `json:"secret_key" gorm:"not null"`
	// 加密模式
	EncryptedMode string `json:"enc_mode" gorm:"column:enc_mode"`
	// 字符集编码
	Charset string `json:"charset" gorm:"default:'UTF-8'"`
	// 冰蝎还是哥斯拉,或者是其他
	ShellType string `json:"shell_type"`
	// 脚本语言
	ShellScript      string `json:"shell_script"`
	Headers          string `json:"headers" gorm:"type:json"`
	Posts            string `json:"posts" gorm:"type:json"`
	Status           bool   `json:"status"`
	Tag              string `json:"tag"`
	Proxy            string `json:"proxy"`
	Remark           string `json:"remark"`
	Hash             string `json:"hash"`
	PacketCodecName  string `json:"packet_codec_name"`
	PayloadCodecName string `json:"payload_codec_name"`
	Os               string `json:"os"`         //操作系统
	Timeout          int64  `json:"timeout"`    //超时时间
	Retry            int64  `json:"retry"`      //重连次数
	BlockSize        int64  `json:"block_size"` //分块大小
	MaxSize          int64  `json:"max_size"`   //上传的最大数量
	IsSession        bool   `json:"is_session"` //是否是session类型
}

type WebsocketFlow struct {
	gorm.Model

	// WebsocketFlow 过来的应该有 WebsocketHash
	WebsocketRequestHash string `json:"websocket_request_hash" gorm:"index"`

	FrameIndex  int    `json:"frame_index" gorm:"index"`
	FromServer  bool   `json:"from_server"`
	QuotedData  string `json:"quoted_data"`
	MessageType string `json:"message_type"`
	Tags        string `json:"tags"`

	Hash string `json:"hash"`
}

type YakScript struct {
	gorm.Model

	ScriptName string `json:"script_name" gorm:"unique_index"`
	Type       string `json:"type" gorm:"index"`
	Content    string `json:"content"`
	Level      string `json:"level"`
	Params     string `json:"params"`
	Help       string `json:"help"`
	Author     string `json:"author"`
	Tags       string `json:"tags,omitempty"`
	Ignored    bool   `json:"ignore"`

	// 加载本地的数据
	FromLocal bool   `json:"from_local"`
	LocalPath string `json:"local_path"`

	// History string
	IsHistory bool `json:"is_history"`

	// Force Interactive
	// Means that this script will be executed in interactive mode
	// cannot load as a plugin or a module by mix caller
	ForceInteractive bool `json:"force_interactive"`

	FromStore bool `json:"from_store"`

	IsGeneralModule      bool   `json:"is_general_module"`
	GeneralModuleVerbose string `json:"general_module_verbose"`
	GeneralModuleKey     string `json:"general_module_key"`
	FromGit              string `json:"from_git"`

	// 这个是自动填写的，一般不需要自己来填写
	// 条件是 Params 中有一个名字为 target 的必填参数
	IsBatchScript bool `json:"is_batch_script"`
	IsExternal    bool `json:"is_external"`

	EnablePluginSelector bool   `json:"enable_plugin_selector"`
	PluginSelectorTypes  string `json:"plugin_selector_types"`

	// Online ID: 线上插件的 ID
	OnlineId           int64  `json:"online_id"`
	OnlineScriptName   string `json:"online_script_name"`
	OnlineContributors string `json:"online_contributors"`
	OnlineIsPrivate    bool   `json:"online_is_private"`

	// 这个插件所属用户 ID
	UserId int64 `json:"user_id"`
	// 这个插件的 UUID
	Uuid           string `json:"uuid"`
	HeadImg        string `json:"head_img"`
	OnlineBaseUrl  string `json:"online_base_url"`
	BaseOnlineId   int64  `json:"BaseOnlineId"`
	OnlineOfficial bool   `json:"online_official"`
	OnlineGroup    string `json:"online_group"`

	IsCorePlugin bool `json:"is_core_plugin"` // 判断是否是核心插件
	// 废弃字段
	RiskType string `json:"risk_type"`
	// 漏洞详情 建议，描述，cwe
	RiskDetail string `json:"risk_detail"`
	// 漏洞类型-补充说明 废弃
	RiskAnnotation string `json:"risk_annotation"`
	// 协作者
	CollaboratorInfo string `json:"collaborator_info"`

	PluginEnvKey string `json:"plugin_env_key"`

	SkipUpdate bool `json:"skip_update"`
	// contains filtered or unexported fields
}