database

type BackupConfig struct {
	Repositories []*pgbackrest.Repository `json:"repositories"`
	Schedules    []*BackupSchedule        `json:"schedules"`
}

type BackupSchedule struct {
	ID             string             `json:"id"`
	Type           BackupScheduleType `json:"type"`
	CronExpression string             `json:"cron_expression"`
}

type BackupScheduleType string

type BuildServiceHostListParams struct {
	ServiceHostID      string           // Host where the service instance runs
	NodeInstances      []*NodeInstances // All database instances, grouped by node
	TargetNodes        []string         // Optional ordered node filter (from database_connection.target_nodes)
	TargetSessionAttrs string           // Caller-provided: "primary", "prefer-standby", etc.
}

type ConnectionInfo struct {
	AdminHost        string
	AdminPort        int
	PeerHost         string
	PeerPort         int
	PeerSSLCert      string
	PeerSSLKey       string
	PeerSSLRootCert  string
	PatroniPort      int
	ClientAddresses  []string
	ClientPort       int
	InstanceHostname string
}

type ConnectionOptions struct {
	DSN *postgres.DSN
	TLS *tls.Config
}

type Database struct {
	DatabaseID       string
	TenantID         *string
	CreatedAt        time.Time
	UpdatedAt        time.Time
	State            DatabaseState
	Spec             *Spec
	Instances        []*Instance
	ServiceInstances []*ServiceInstance
}

type DatabaseConnection struct {
	TargetNodes        []string `json:"target_nodes,omitempty"`
	TargetSessionAttrs string   `json:"target_session_attrs,omitempty"`
}

type DatabaseState string

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

type Extension struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type ExtraNetworkSpec struct {
	ID         string            `json:"id"`                    // required
	Aliases    []string          `json:"aliases,omitempty"`     // optional
	DriverOpts map[string]string `json:"driver_opts,omitempty"` // optional
}

type ExtraVolumesSpec struct {
	HostPath        string `json:"host_path"`
	DestinationPath string `json:"destination_path"`
}

type HealthCheckResult struct {
	Status    string    `json:"status"`
	Message   string    `json:"message,omitempty"`
	CheckedAt time.Time `json:"checked_at"`
}

type Instance struct {
	InstanceID string          `json:"instance_id"`
	DatabaseID string          `json:"database_id"`
	HostID     string          `json:"host_id"`
	NodeName   string          `json:"node_name"`
	State      InstanceState   `json:"state"`
	Status     *InstanceStatus `json:"status"`
	CreatedAt  time.Time       `json:"created_at"`
	UpdatedAt  time.Time       `json:"updated_at"`
	Error      string          `json:"error,omitempty"`
}

type InstanceResource struct {
	Spec                     *InstanceSpec         `json:"spec"`
	InstanceHostname         string                `json:"instance_hostname"`
	PrimaryInstanceID        string                `json:"primary_instance_id"`
	OrchestratorDependencies []resource.Identifier `json:"dependencies"`
	ConnectionInfo           *ConnectionInfo       `json:"connection_info"`
}

type InstanceResources struct {
	Instance  *InstanceResource
	Resources []*resource.ResourceData
}

type InstanceSpec struct {
	InstanceID       string              `json:"instance_id"`
	TenantID         *string             `json:"tenant_id,omitempty"`
	DatabaseID       string              `json:"database_id"`
	HostID           string              `json:"host_id"`
	DatabaseName     string              `json:"database_name"`
	NodeName         string              `json:"node_name"`
	NodeOrdinal      int                 `json:"node_ordinal"`
	PgEdgeVersion    *host.PgEdgeVersion `json:"pg_edge_version"`
	Port             *int                `json:"port"`
	PatroniPort      *int                `json:"patroni_port"`
	CPUs             float64             `json:"cpus"`
	MemoryBytes      uint64              `json:"memory"`
	DatabaseUsers    []*User             `json:"database_users"`
	BackupConfig     *BackupConfig       `json:"backup_config"`
	RestoreConfig    *RestoreConfig      `json:"restore_config"`
	PostgreSQLConf   map[string]any      `json:"postgresql_conf"`
	ClusterSize      int                 `json:"cluster_size"`
	OrchestratorOpts *OrchestratorOpts   `json:"orchestrator_opts,omitempty"`
	InPlaceRestore   bool                `json:"in_place_restore,omitempty"`
}

type InstanceSpecChange struct {
	Previous *InstanceSpec
	Current  *InstanceSpec
}

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

type InstanceState string

type InstanceStatus struct {
	PostgresVersion *string               `json:"postgres_version,omitempty"`
	SpockVersion    *string               `json:"spock_version,omitempty"`
	Addresses       []string              `json:"addresses,omitempty"`
	Port            *int                  `json:"port,omitempty"`
	PatroniState    *patroni.State        `json:"patroni_state,omitempty"`
	Role            *patroni.InstanceRole `json:"role,omitempty"`
	ReadOnly        *string               `json:"read_only,omitempty"`
	PendingRestart  *bool                 `json:"pending_restart,omitempty"`
	PatroniPaused   *bool                 `json:"patroni_paused,omitempty"`
	StatusUpdatedAt *time.Time            `json:"status_updated_at,omitempty"`
	Stopped         *bool                 `json:"stopped,omitempty"`
	Subscriptions   []SubscriptionStatus  `json:"subscriptions,omitempty"`
	Error           *string               `json:"error,omitempty"`
}

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

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

type InstanceUpdateOptions struct {
	InstanceID string        `json:"instance_id"`
	DatabaseID string        `json:"database_id"`
	HostID     string        `json:"host_id"`
	NodeName   string        `json:"node_name"`
	State      InstanceState `json:"state"`
	Error      string        `json:"error,omitempty"`
	Now        time.Time     `json:"now"`
}

type LagTrackerCommitTimestampResource struct {
	// Planner fields
	OriginNode   string `json:"origin_node"`
	ReceiverNode string `json:"receiver_node"`

	// Dependency wiring
	ExtraDependencies []resource.Identifier `json:"dependent_resources,omitempty"`

	// Output (filled at Refresh/Create time)
	CommitTimestamp *time.Time `json:"commit_timestamp,omitempty"`
}

type MCPServiceConfig struct {
	// Optional - LLM proxy for web client (default: false)
	LLMEnabled *bool `json:"llm_enabled,omitempty"`

	// Required when llm_enabled is true
	LLMProvider     string  `json:"llm_provider"`
	LLMModel        string  `json:"llm_model"`
	AnthropicAPIKey *string `json:"anthropic_api_key,omitempty"`
	OpenAIAPIKey    *string `json:"openai_api_key,omitempty"`
	OllamaURL       *string `json:"ollama_url,omitempty"`

	// Optional - security
	AllowWrites *bool            `json:"allow_writes,omitempty"`
	InitToken   *string          `json:"init_token,omitempty"`
	InitUsers   []MCPServiceUser `json:"init_users,omitempty"`

	// Optional - embeddings
	EmbeddingProvider *string `json:"embedding_provider,omitempty"`
	EmbeddingModel    *string `json:"embedding_model,omitempty"`
	EmbeddingAPIKey   *string `json:"embedding_api_key,omitempty"`

	// Optional - LLM tuning (overridable defaults)
	LLMTemperature *float64 `json:"llm_temperature,omitempty"`
	LLMMaxTokens   *int     `json:"llm_max_tokens,omitempty"`

	// Optional - connection pool (overridable defaults)
	PoolMaxConns *int `json:"pool_max_conns,omitempty"`

	// Optional - tool toggles (all enabled by default)
	DisableQueryDatabase       *bool `json:"disable_query_database,omitempty"`
	DisableGetSchemaInfo       *bool `json:"disable_get_schema_info,omitempty"`
	DisableSimilaritySearch    *bool `json:"disable_similarity_search,omitempty"`
	DisableExecuteExplain      *bool `json:"disable_execute_explain,omitempty"`
	DisableGenerateEmbedding   *bool `json:"disable_generate_embedding,omitempty"`
	DisableSearchKnowledgebase *bool `json:"disable_search_knowledgebase,omitempty"`
	DisableCountRows           *bool `json:"disable_count_rows,omitempty"`
}

type MCPServiceUser struct {
	Username string `json:"username"`
	Password string `json:"password"`
}

type Node struct {
	Name             string            `json:"name"`
	HostIDs          []string          `json:"host_ids"`
	PostgresVersion  string            `json:"postgres_version"`
	Port             *int              `json:"port"`
	PatroniPort      *int              `json:"patroni_port"`
	CPUs             float64           `json:"cpus"`
	MemoryBytes      uint64            `json:"memory"`
	PostgreSQLConf   map[string]any    `json:"postgresql_conf"`
	BackupConfig     *BackupConfig     `json:"backup_config"`
	RestoreConfig    *RestoreConfig    `json:"restore_config"`
	OrchestratorOpts *OrchestratorOpts `json:"orchestrator_opts,omitempty"`
	SourceNode       string            `json:"source_node,omitempty"`
}

type NodeInstances struct {
	NodeName      string          `json:"node_name"`
	SourceNode    string          `json:"source_node"`
	Instances     []*InstanceSpec `json:"instances"`
	RestoreConfig *RestoreConfig  `json:"restore_config"`
}

type NodeResource struct {
	Name              string   `json:"name"`
	InstanceIDs       []string `json:"instance_ids"`
	PrimaryInstanceID string   `json:"primary_instance_id"`
}

type Orchestrator interface {
	GenerateInstanceResources(spec *InstanceSpec) (*InstanceResources, error)
	GenerateInstanceRestoreResources(spec *InstanceSpec, taskID uuid.UUID) (*InstanceResources, error)
	GenerateServiceInstanceResources(spec *ServiceInstanceSpec) (*ServiceInstanceResources, error)
	GetInstanceConnectionInfo(ctx context.Context, databaseID, instanceID string) (*ConnectionInfo, error)
	GetServiceInstanceStatus(ctx context.Context, serviceInstanceID string) (*ServiceInstanceStatus, error)
	CreatePgBackRestBackup(ctx context.Context, w io.Writer, instanceID string, options *pgbackrest.BackupOptions) error
	ValidateInstanceSpecs(ctx context.Context, changes []*InstanceSpecChange) ([]*ValidationResult, error)
	StopInstance(ctx context.Context, instanceID string) error
	StartInstance(ctx context.Context, instanceID string) error
}

type OrchestratorOpts struct {
	Swarm *SwarmOpts `json:"docker,omitempty"`
}

type PortMapping struct {
	Name          string `json:"name"`
	ContainerPort int    `json:"container_port"`
	HostPort      *int   `json:"host_port,omitempty"`
}

type PostgRESTServiceConfig struct {
	DBSchemas                string  `json:"db_schemas"`   // default: "public"
	DBAnonRole               string  `json:"db_anon_role"` // default: "pgedge_application_read_only"
	DBPool                   int     `json:"db_pool"`      // default: 10, range: 1-30
	MaxRows                  int     `json:"max_rows"`     // default: 1000, range: 1-10000
	JWTSecret                *string `json:"jwt_secret,omitempty"`
	JWTAud                   *string `json:"jwt_aud,omitempty"`
	JWTRoleClaimKey          *string `json:"jwt_role_claim_key,omitempty"`
	ServerCORSAllowedOrigins *string `json:"server_cors_allowed_origins,omitempty"`
}

type ReplicationSlotAdvanceFromCTSResource struct {
	ProviderNode   string `json:"provider_node"`   // slot lives here
	SubscriberNode string `json:"subscriber_node"` // target/receiver node
}

type ReplicationSlotCreateResource struct {
	DatabaseName   string `json:"database_name"`
	ProviderNode   string `json:"provider_node"`
	SubscriberNode string `json:"subscriber_node"`
}

type ReplicationSlotResource struct {
	ProviderNode   string `json:"provider_node"`
	SubscriberNode string `json:"subscriber_node"`
}

type RestoreConfig struct {
	SourceDatabaseID   string                 `json:"source_database_id"`
	SourceNodeName     string                 `json:"source_node_name"`
	SourceDatabaseName string                 `json:"source_database_name"`
	Repository         *pgbackrest.Repository `json:"repository"`
	RestoreOptions     map[string]string      `json:"restore_options"`
}

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

type ServiceConnectionInfo struct {
	Hosts              []ServiceHostEntry
	TargetSessionAttrs string
}

type ServiceHostEntry struct {
	Host string
	Port int
}

type ServiceInstance struct {
	ServiceInstanceID string                 `json:"service_instance_id"`
	ServiceID         string                 `json:"service_id"`
	DatabaseID        string                 `json:"database_id"`
	HostID            string                 `json:"host_id"`
	State             ServiceInstanceState   `json:"state"`
	Status            *ServiceInstanceStatus `json:"status,omitempty"`
	// Credentials is only populated during provisioning workflows. It is not
	// persisted to etcd and will be nil when read from the store.
	Credentials *ServiceUser `json:"credentials,omitempty"`
	CreatedAt   time.Time    `json:"created_at"`
	UpdatedAt   time.Time    `json:"updated_at"`
	Error       string       `json:"error,omitempty"`
}

type ServiceInstanceResources struct {
	ServiceInstance *ServiceInstance
	Resources       []*resource.ResourceData
}

type ServiceInstanceSpec struct {
	ServiceInstanceID  string
	ServiceSpec        *ServiceSpec
	PgEdgeVersion      *host.PgEdgeVersion // Database version, used for compatibility validation
	DatabaseID         string
	DatabaseName       string
	HostID             string
	CohortMemberID     string
	Credentials        *ServiceUser
	DatabaseNetworkID  string
	NodeName           string             // Database node name (for ServiceUserRole PrimaryExecutor routing)
	DatabaseHosts      []ServiceHostEntry // Ordered list of Postgres host:port entries
	TargetSessionAttrs string             // libpq target_session_attrs value
	Port               *int               // Service instance published port (optional, 0 = random)
	DatabaseNodes      []*NodeInstances   // All database nodes; used to create per-node ServiceUserRole resources
}

type ServiceInstanceState string

type ServiceInstanceStateUpdate struct {
	DatabaseID string                 `json:"database_id,omitempty"`
	State      ServiceInstanceState   `json:"state"`
	Status     *ServiceInstanceStatus `json:"status,omitempty"`
	Error      string                 `json:"error,omitempty"`
}

type ServiceInstanceStatus struct {
	ContainerID  *string            `json:"container_id,omitempty"`
	ImageVersion *string            `json:"image_version,omitempty"`
	Addresses    []string           `json:"addresses,omitempty"`
	Ports        []PortMapping      `json:"ports,omitempty"`
	HealthCheck  *HealthCheckResult `json:"health_check,omitempty"`
	LastHealthAt *time.Time         `json:"last_health_at,omitempty"`
	ServiceReady *bool              `json:"service_ready,omitempty"`
}

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

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

type ServiceInstanceUpdateOptions struct {
	ServiceInstanceID string               `json:"service_instance_id"`
	ServiceID         string               `json:"service_id"`
	DatabaseID        string               `json:"database_id"`
	HostID            string               `json:"host_id"`
	State             ServiceInstanceState `json:"state"`
	Error             string               `json:"error,omitempty"`
}

type ServiceSpec struct {
	ServiceID          string              `json:"service_id"`
	ServiceType        string              `json:"service_type"`
	Version            string              `json:"version"`
	HostIDs            []string            `json:"host_ids"`
	Config             map[string]any      `json:"config"`
	Port               *int                `json:"port,omitempty"`
	CPUs               *float64            `json:"cpus,omitempty"`
	MemoryBytes        *uint64             `json:"memory,omitempty"`
	OrchestratorOpts   *OrchestratorOpts   `json:"orchestrator_opts,omitempty"`
	DatabaseConnection *DatabaseConnection `json:"database_connection,omitempty"`
}

type ServiceUser struct {
	Username string `json:"username"` // Format: "svc_{service_id}_{mode}"
	Password string `json:"password"` // 32-character cryptographically random string
	Role     string `json:"role"`     // Database role, e.g., "pgedge_application_read_only" or "pgedge_application"
}

type Spec struct {
	DatabaseID       string            `json:"database_id"`
	TenantID         *string           `json:"tenant_id,omitempty"`
	DatabaseName     string            `json:"database_name"`
	PostgresVersion  string            `json:"postgres_version"`
	SpockVersion     string            `json:"spock_version"`
	Port             *int              `json:"port"`
	PatroniPort      *int              `json:"patroni_port"`
	CPUs             float64           `json:"cpus"`
	MemoryBytes      uint64            `json:"memory"`
	Nodes            []*Node           `json:"nodes"`
	DatabaseUsers    []*User           `json:"database_users"`
	Services         []*ServiceSpec    `json:"services,omitempty"`
	BackupConfig     *BackupConfig     `json:"backup_config"`
	RestoreConfig    *RestoreConfig    `json:"restore_config"`
	PostgreSQLConf   map[string]any    `json:"postgresql_conf"`
	OrchestratorOpts *OrchestratorOpts `json:"orchestrator_opts,omitempty"`
}

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

type Store struct {
	Spec                  *SpecStore
	Database              *DatabaseStore
	Instance              *InstanceStore
	InstanceStatus        *InstanceStatusStore
	InstanceSpec          *InstanceSpecStore
	ServiceInstance       *ServiceInstanceStore
	ServiceInstanceStatus *ServiceInstanceStatusStore
	// contains filtered or unexported fields
}

type StoredDatabase struct {
	storage.StoredValue
	DatabaseID string        `json:"database_id"`
	TenantID   *string       `json:"tenant_id,omitempty"`
	CreatedAt  time.Time     `json:"created_at"`
	UpdatedAt  time.Time     `json:"updated_at"`
	State      DatabaseState `json:"state"`
}

type StoredInstance struct {
	storage.StoredValue
	InstanceID string        `json:"instance_id"`
	DatabaseID string        `json:"database_id"`
	HostID     string        `json:"host_id"`
	NodeName   string        `json:"node_name"`
	State      InstanceState `json:"state"`
	CreatedAt  time.Time     `json:"created_at"`
	UpdateAt   time.Time     `json:"updated_at"`
	Error      string        `json:"error,omitempty"`
}

type StoredInstanceSpec struct {
	storage.StoredValue
	Spec *InstanceSpec `json:"spec"`
}

type StoredInstanceStatus struct {
	storage.StoredValue
	DatabaseID string          `json:"database_id"`
	InstanceID string          `json:"instance_id"`
	Status     *InstanceStatus `json:"status"`
}

type StoredServiceInstance struct {
	storage.StoredValue
	ServiceInstanceID string               `json:"service_instance_id"`
	ServiceID         string               `json:"service_id"`
	DatabaseID        string               `json:"database_id"`
	HostID            string               `json:"host_id"`
	State             ServiceInstanceState `json:"state"`
	CreatedAt         time.Time            `json:"created_at"`
	UpdatedAt         time.Time            `json:"updated_at"`
	Error             string               `json:"error,omitempty"`
}

type StoredServiceInstanceStatus struct {
	storage.StoredValue
	DatabaseID        string                 `json:"database_id"`
	ServiceInstanceID string                 `json:"service_instance_id"`
	Status            *ServiceInstanceStatus `json:"status"`
}

type StoredSpec struct {
	storage.StoredValue
	*Spec
}

type SubscriptionResource struct {
	SubscriberNode    string                `json:"subscriber_node"`
	ProviderNode      string                `json:"provider_node"`
	Disabled          bool                  `json:"disabled"`
	SyncStructure     bool                  `json:"sync_structure"`
	SyncData          bool                  `json:"sync_data"`
	ExtraDependencies []resource.Identifier `json:"dependent_subscriptions"`
	NeedsUpdate       bool                  `json:"needs_update"`
}

type SubscriptionStatus struct {
	ProviderNode string `json:"provider_node"`
	Name         string `json:"name"`
	Status       string `json:"status"`
}

type SwarmOpts struct {
	ExtraVolumes  []ExtraVolumesSpec `json:"extra_volumes,omitempty"`
	ExtraNetworks []ExtraNetworkSpec `json:"extra_networks,omitempty"`
	ExtraLabels   map[string]string  `json:"extra_labels,omitempty"` // optional, used for custom labels on the swarm service
}

type SwitchoverResource struct {
	HostID     string               `json:"host_id"`
	InstanceID string               `json:"instance_id"`
	TargetRole patroni.InstanceRole `json:"target_role"`
}

type SyncEventResource struct {
	ProviderNode      string                `json:"provider_node"`
	SubscriberNode    string                `json:"subscriber_node"`
	SyncEventLsn      string                `json:"sync_event_lsn"`
	ExtraDependencies []resource.Identifier `json:"extra_dependencies"`
}

type User struct {
	Username   string   `json:"username"`
	Password   string   `json:"password"`
	DBOwner    bool     `json:"db_owner,omitempty"`
	Attributes []string `json:"attributes,omitempty"`
	Roles      []string `json:"roles,omitempty"`
}

type ValidationResult struct {
	InstanceID string   `json:"instance_id"`
	HostID     string   `json:"host_id"`
	NodeName   string   `json:"node_name"`
	Valid      bool     `json:"valid"`
	Errors     []string `json:"errors"`
}

type WaitForSyncEventResource struct {
	SubscriberNode string `json:"subscriber_node"`
	ProviderNode   string `json:"provider_node"`
}