guardrails

type AIPattern struct {
	Name        string         `json:"name"`
	Type        string         `json:"type"` // domain, content, api
	Pattern     *regexp.Regexp `json:"-"`
	PatternStr  string         `json:"pattern"`
	Severity    string         `json:"severity"`
	Description string         `json:"description"`
}

type AuditEventType string

type AuditLogEntry struct {
	// Timestamp 时间戳
	Timestamp time.Time `json:"timestamp"`
	// EventType 事件类型
	EventType AuditEventType `json:"event_type"`
	// ValidatorName 验证器名称
	ValidatorName string `json:"validator_name"`
	// ContentHash 内容哈希（用于追踪，不存储原始内容）
	ContentHash string `json:"content_hash"`
	// Errors 验证错误列表
	Errors []ValidationError `json:"errors,omitempty"`
	// Metadata 附加元数据
	Metadata map[string]any `json:"metadata,omitempty"`
}

type AuditLogFilter struct {
	// StartTime 开始时间
	StartTime *time.Time
	// EndTime 结束时间
	EndTime *time.Time
	// EventTypes 事件类型过滤
	EventTypes []AuditEventType
	// ValidatorNames 验证器名称过滤
	ValidatorNames []string
	// Limit 返回数量限制
	Limit int
	// Offset 偏移量
	Offset int
}

type AuditLogger interface {
	// Log 记录审计日志
	Log(ctx context.Context, entry *AuditLogEntry) error
	// Query 查询审计日志
	Query(ctx context.Context, filter *AuditLogFilter) ([]*AuditLogEntry, error)
	// Count 统计审计日志数量
	Count(ctx context.Context, filter *AuditLogFilter) (int, error)
}

type ChainMode string

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

type ContentFilterConfig struct {
	// BlockedPatterns 禁止的正则模式
	BlockedPatterns []string
	// Replacement 替换文本
	Replacement string
	// CaseSensitive 是否区分大小写
	CaseSensitive bool
}

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

type ContentMatch struct {
	Pattern  string `json:"pattern"`
	Value    string `json:"value"`
	Position int    `json:"position"`
	Length   int    `json:"length"`
}

type Detection struct {
	ID          string         `json:"id"`
	PatternName string         `json:"pattern_name"`
	Type        string         `json:"type"`
	Source      string         `json:"source"`
	Evidence    string         `json:"evidence"`
	Severity    string         `json:"severity"`
	Timestamp   time.Time      `json:"timestamp"`
	UserID      string         `json:"user_id,omitempty"`
	Metadata    map[string]any `json:"metadata,omitempty"`
}

type FailureAction string

type Filter interface {
	// Filter 执行过滤，返回过滤后的内容
	Filter(ctx context.Context, content string) (string, error)
	// Name 返回过滤器名称
	Name() string
}

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

type GuardrailsConfig struct {
	InputValidators  []Validator `json:"-"`
	OutputValidators []Validator `json:"-"`
	OutputFilters    []Filter    `json:"-"`

	// 内置验证器配置
	MaxInputLength      int      `json:"max_input_length"`
	BlockedKeywords     []string `json:"blocked_keywords"`
	PIIDetectionEnabled bool     `json:"pii_detection_enabled"`
	InjectionDetection  bool     `json:"injection_detection"`

	// 失败处理
	OnInputFailure  FailureAction `json:"on_input_failure"`
	OnOutputFailure FailureAction `json:"on_output_failure"`
	MaxRetries      int           `json:"max_retries"`
}

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

type InjectionDetectorConfig struct {
	// CaseSensitive 是否区分大小写
	CaseSensitive bool
	// UseDelimiters 是否使用分隔符隔离
	UseDelimiters bool
	// CustomPatterns 自定义注入模式
	CustomPatterns []string
	// Priority 验证器优先级
	Priority int
	// EnabledLanguages 启用的语言检测，为空则启用所有
	EnabledLanguages []string
}

type InjectionMatch struct {
	Pattern     string `json:"pattern"`
	Description string `json:"description"`
	Severity    string `json:"severity"`
	Position    int    `json:"position"`
	Length      int    `json:"length"`
	MatchedText string `json:"matched_text"`
}

type InjectionPattern struct {
	Pattern     *regexp.Regexp
	Description string
	Severity    string
	Language    string // "en", "zh", "universal"
}

type KeywordAction string

type KeywordMatch struct {
	Keyword  string `json:"keyword"`
	Severity string `json:"severity"`
	Position int    `json:"position"`
	Length   int    `json:"length"`
}

type KeywordSeverity struct {
	Keyword  string
	Severity string
}

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

type KeywordValidatorConfig struct {
	// BlockedKeywords 禁止的关键词列表
	BlockedKeywords []string
	// KeywordSeverities 关键词严重级别映射（可选，未配置的使用默认级别）
	KeywordSeverities map[string]string
	// DefaultSeverity 默认严重级别
	DefaultSeverity string
	// Action 处理动作
	Action KeywordAction
	// CaseSensitive 是否区分大小写
	CaseSensitive bool
	// Replacement 过滤模式下的替换文本
	Replacement string
	// Priority 验证器优先级
	Priority int
}

type LengthAction string

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

type LengthValidatorConfig struct {
	// MaxLength 最大长度（字符数）
	MaxLength int
	// Action 超限处理动作
	Action LengthAction
	// Priority 验证器优先级
	Priority int
}

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

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

type OutputValidatorConfig struct {
	// Validators 输出验证器列表
	Validators []Validator
	// Filters 输出过滤器列表
	Filters []Filter
	// SafeReplacement 有害内容的安全替代响应
	SafeReplacement string
	// EnableAuditLog 是否启用审计日志
	EnableAuditLog bool
	// AuditLogger 审计日志记录器
	AuditLogger AuditLogger
	// Priority 验证器优先级
	Priority int
}

type PIIAction string

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

type PIIDetectorConfig struct {
	// Action 处理动作
	Action PIIAction
	// EnabledTypes 启用的 PII 类型，为空则启用所有类型
	EnabledTypes []PIIType
	// CustomPatterns 自定义正则模式
	CustomPatterns map[PIIType]*regexp.Regexp
	// Priority 验证器优先级
	Priority int
}

type PIIMatch struct {
	Type     PIIType `json:"type"`
	Value    string  `json:"value"`
	Masked   string  `json:"masked"`
	Position int     `json:"position"`
	Length   int     `json:"length"`
}

type PIIType string

type ShadowAIConfig struct {
	EnableNetworkMonitor bool          `json:"enable_network_monitor"`
	EnableContentScan    bool          `json:"enable_content_scan"`
	ScanInterval         time.Duration `json:"scan_interval"`
	AlertThreshold       int           `json:"alert_threshold"`
	WhitelistedDomains   []string      `json:"whitelisted_domains"`
	WhitelistedApps      []string      `json:"whitelisted_apps"`
}

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

type ShadowAIStats struct {
	TotalDetections int            `json:"total_detections"`
	BySeverity      map[string]int `json:"by_severity"`
	ByPattern       map[string]int `json:"by_pattern"`
}

type TripwireError struct {
	ValidatorName string
	Result        *ValidationResult
}

type ValidationError struct {
	Code     string `json:"code"`
	Message  string `json:"message"`
	Severity string `json:"severity"` // critical, high, medium, low
	Field    string `json:"field,omitempty"`
}

type ValidationResult struct {
	Valid    bool              `json:"valid"`
	Tripwire bool              `json:"tripwire,omitempty"` // 触发即中断整个 Agent 执行链
	Errors   []ValidationError `json:"errors,omitempty"`
	Warnings []string          `json:"warnings,omitempty"`
	Metadata map[string]any    `json:"metadata,omitempty"`
}

type Validator interface {
	// Validate 执行验证，返回验证结果
	Validate(ctx context.Context, content string) (*ValidationResult, error)
	// Name 返回验证器名称
	Name() string
	// Priority 返回优先级（数字越小优先级越高）
	Priority() int
}

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

type ValidatorChainConfig struct {
	// Mode 执行模式
	Mode ChainMode
}

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