common

type EtcdCreds struct {
	InstanceID string `json:"instance_id"`
	DatabaseID string `json:"database_id"`
	HostID     string `json:"host_id"`
	NodeName   string `json:"node_name"`
	ParentID   string `json:"parent_id"`
	OwnerUID   int    `json:"owner_uid"`
	OwnerGID   int    `json:"owner_gid"`
	Username   string `json:"username"`
	Password   string `json:"password"`
	CaCert     []byte `json:"ca_cert"`
	ClientCert []byte `json:"client_cert"`
	ClientKey  []byte `json:"client_key"`
}

type GenerateOptions struct {
	// SystemAddresses are IPs, hostnames, or CIDR ranges that pgedge or
	// patroni_replicator connections will originate from.
	SystemAddresses []string
	// ExtraHbaEntries are orchestrator-specific entries to include in the
	// pg_hba.conf.
	ExtraHbaEntries []hba.Entry
	// EnableFastBasebackup enables basebackup's "fast" checkpoint option when
	// bootstrapping this instance from another existing instance.
	EnableFastBasebackup bool
}

type PatroniCluster struct {
	DatabaseID string `json:"database_id"`
	NodeName   string `json:"node_name"`
}

type PatroniConfig struct {
	InstanceID string                  `json:"instance_id"`
	HostID     string                  `json:"host_id"`
	NodeName   string                  `json:"node_name"`
	Generator  *PatroniConfigGenerator `json:"generator"`
	ParentID   string                  `json:"parent_id"`
	OwnerUID   int                     `json:"owner_uid"`
	OwnerGID   int                     `json:"owner_gid"`
}

type PatroniConfigGenerator struct {
	// ArchiveCommand sets the Postgres archive command parameter.
	ArchiveCommand string `json:"archive_command,omitempty"`
	// ClusterSize is the number of nodes in the Spock cluster. This is used for
	// the tunable Postgres parameters.
	ClusterSize int `json:"cluster_size"`
	// CPUs is the number of CPUs allocated for this instance. This is used for
	// the tunable Postgres parameters.
	CPUs float64 `json:"cpus,omitempty"`
	// DatabaseID is the Database's ID.
	DatabaseID string `json:"database_id"`
	// DataDir is the Postgres data directory.
	DataDir string `json:"data_dir"`
	// EtcdCertsDir is the Etcd certificates directory.
	EtcdCertsDir string `json:"etcd_certs_dir"`
	// FQDN is the fully-qualified domain name for this instance. This name must
	// be reachable by sibling instances within the Spock node.
	FQDN string `json:"fqdn"`
	// InstanceID is this instance's ID.
	InstanceID string `json:"instance_id"`
	// LogType sets the Patroni log type.
	LogType patroni.LogType `json:"log_type"`
	// MemoryBytes is the amount of memory that is allocated for this instance.
	// This is used for the tunable Postgres parameters.
	MemoryBytes uint64 `json:"memory_bytes,omitempty"`
	// NodeName is the Spock node name.
	NodeName string `json:"node_name"`
	// NodeOrdinal is the ordinal part of the Spock node name, e.g. for 'n1'
	// this would be '1'. This is used to configure the Snowflake and LOLOR
	// extensions.
	NodeOrdinal int `json:"node_ordinal"`
	// OrchestratorParameters are additional parameters to be provided by the
	// orchestrator implementation.
	OrchestratorParameters map[string]any `json:"orchestrator_parameters,omitempty"`
	// PatroniAllowlist is a user-specified list of addresses, hostnames, or
	// CIDR ranges to include in the allowlist for Patroni's REST API.
	PatroniAllowlist []string `json:"patroni_allowlist"`
	// PatroniPort is the port that Patroni will listen on.
	PatroniPort int `json:"patroni_port"`
	// PostgresCertsDir is the Postgres certificates directory.
	PostgresCertsDir string `json:"postgres_certs_dir"`
	// PostgresPort is the port that Postgres will listen on.
	PostgresPort int `json:"postgres_port"`
	// RestoreCommand is an alternate command to use to bootstrap this instance.
	RestoreCommand string `json:"restore_command"`
	// SpecParameters are user-specified Postgres parameters that are included
	// in the database spec.
	SpecParameters map[string]any `json:"spec_parameters,omitempty"`
	// TenantID is an optional tenant ID that is associated with this instance.
	TenantID *string `json:"tenant_id,omitempty"`
}

type PatroniConfigGeneratorOptions struct {
	// Instance is the instance spec for this instance.
	Instance *database.InstanceSpec
	// HostCPUs is the total number of CPUs available on the host. This is used for
	// the tunable Postgres parameters.
	HostCPUs float64
	// HostMemoryBytes is the total amount of memory available on the host. This
	// is used for the tunable Postgres parameters.
	HostMemoryBytes uint64
	// FQDN is the fully-qualified domain name for this instance. This name must
	// be reachable by sibling instances within the Spock node.
	FQDN string
	// LogType sets the Patroni log type.
	LogType patroni.LogType
	// OrchestratorParameters are additional parameters to be provided by the
	// orchestrator implementation.
	OrchestratorParameters map[string]any
	// PatroniPort is the port that Patroni will listen on.
	PatroniPort int
	// PostgresPort is the port that Postgres will listen on.
	PostgresPort int
	// Paths is used to compute the paths of directories and executables.
	Paths database.InstancePaths
}

type PatroniMember struct {
	DatabaseID string `json:"database_id"`
	NodeName   string `json:"node_name"`
	InstanceID string `json:"instance_id"`
}

type PgBackRestConfig struct {
	InstanceID   string                   `json:"instance_id"`
	HostID       string                   `json:"host_id"`
	DatabaseID   string                   `json:"database_id"`
	NodeName     string                   `json:"node_name"`
	Repositories []*pgbackrest.Repository `json:"repositories"`
	ParentID     string                   `json:"parent_id"`
	Type         pgbackrest.ConfigType    `json:"type"`
	OwnerUID     int                      `json:"owner_uid"`
	OwnerGID     int                      `json:"owner_gid"`
	Paths        database.InstancePaths   `json:"paths"`
	Port         int                      `json:"port"`
}

type PgBackRestStanza struct {
	DatabaseID string `json:"database_id"`
	NodeName   string `json:"node_name"`
}

type PgServiceConf struct {
	ParentID   string   `json:"parent_id"`
	HostID     string   `json:"host_id"`
	InstanceID string   `json:"instance_id"`
	NodeNames  []string `json:"node_names"`
	OwnerUID   int      `json:"owner_uid"`
	OwnerGID   int      `json:"owner_gid"`
}

type PostgresCerts struct {
	InstanceID        string   `json:"instance_id"`
	HostID            string   `json:"host_id"`
	InstanceAddresses []string `json:"instance_addresses"`
	ParentID          string   `json:"parent_id"`
	OwnerUID          int      `json:"owner_uid"`
	OwnerGID          int      `json:"owner_gid"`
	CaCert            []byte   `json:"ca_cert"`
	ServerCert        []byte   `json:"server_cert"`
	ServerKey         []byte   `json:"server_key"`
	SuperuserCert     []byte   `json:"superuser_cert"`
	SuperuserKey      []byte   `json:"superuser_key"`
	ReplicationCert   []byte   `json:"replication_cert"`
	ReplicationKey    []byte   `json:"replication_key"`
}