demoseed

type AgentManifest struct {
	ID        string `yaml:"id"`
	Name      string `yaml:"name"`
	Level     string `yaml:"level"`
	APIKeyEnv string `yaml:"api_key_env"`
}

type B5Manifest struct {
	AgentID      string   `yaml:"agent_id"`
	DatasourceID string   `yaml:"datasource_id"`
	PolicyID     string   `yaml:"policy_id"`
	Statements   []string `yaml:"statements"`
}

type DatasourceManifest struct {
	ID            string `yaml:"id"`
	Name          string `yaml:"name"`
	DBType        string `yaml:"db_type"`
	Host          string `yaml:"host"`
	Port          int    `yaml:"port"`
	Database      string `yaml:"database"`
	Username      string `yaml:"username"`
	PasswordEnv   string `yaml:"password_env"`
	ConnLimit     int    `yaml:"conn_limit"`
	StmtTimeoutMS int    `yaml:"stmt_timeout_ms"`
	RowLimit      int    `yaml:"row_limit"`
}

type DatasourcePinger interface {
	Ping(ctx context.Context, datasource model.Datasource, password string) error
}

type FeatureEnroller interface {
	EnrollPostgresPolicySelect(context.Context, model.Datasource, []byte, string, executor.Limits) (executor.PostgresPolicyEnrollment, error)
	EnrollPostgresPolicyDML(context.Context, model.Datasource, []byte, string, executor.Limits) (executor.PostgresPolicyEnrollment, error)
}

type LookupEnv func(string) (string, bool)

type Manifest struct {
	Version     int                  `yaml:"version"`
	Banner      string               `yaml:"banner"`
	QPSPerAgent int                  `yaml:"qps_per_agent"`
	Datasources []DatasourceManifest `yaml:"datasources"`
	Agents      []AgentManifest      `yaml:"agents"`
	MaskRules   []MaskRuleManifest   `yaml:"mask_rules"`
	Policies    []PolicyManifest     `yaml:"policies"`
	B5          B5Manifest           `yaml:"b5"`
	Rules       []RuleManifest       `yaml:"rules"`
}

type MaskRuleManifest struct {
	ID            string `yaml:"id"`
	DatasourceID  string `yaml:"datasource_id"`
	SchemaName    string `yaml:"schema_name"`
	TableName     string `yaml:"table_name"`
	ColumnName    string `yaml:"column_name"`
	SensitiveType string `yaml:"sensitive_type"`
	Algo          string `yaml:"algo"`
}

type PolicyManifest struct {
	ID            string `yaml:"id"`
	AgentID       string `yaml:"agent_id"`
	DatasourceID  string `yaml:"datasource_id"`
	ObjectType    string `yaml:"object_type"`
	ObjectName    string `yaml:"object_name"`
	Columns       string `yaml:"columns,omitempty"`
	EnrollmentSQL string `yaml:"enrollment_sql,omitempty"`
	Action        string `yaml:"action"`
}

type RuleManifest struct {
	ID      string `yaml:"id"`
	Enabled bool   `yaml:"enabled"`
}

type Secrets struct {
	Passwords map[string]string
	APIKeys   map[string]string
}

type Summary struct {
	Datasources int
	Agents      int
	MaskRules   int
	Policies    int
	Rules       int
	Audits      int
	Approvals   int
	B5Grants    int
	B2Mode      string
}