stdpgxfx

type Config struct {
	// MainDatabaseURL configures the database connection string for the main connection.
	MainDatabaseURL string `env:"MAIN_DATABASE_URL,required"`
	// IamAuth enables RDS IAM authentication. When enabled, the password on every
	// connection attempt is replaced by a freshly built IAM auth token. The signing
	// region is derived per-connection from the RDS hostname (e.g.,
	// "mycluster.cluster-xxx.eu-central-1.rds.amazonaws.com" → "eu-central-1"),
	// which means a derived "ro" pool that points at a different regional endpoint
	// (e.g., a reader endpoint in another region) signs tokens against that region
	// automatically. The hostname must match the standard regional RDS pattern;
	// non-RDS hostnames (e.g., custom domains, RDS Proxy aliases, the global
	// writer endpoint) are not supported.
	IamAuth bool `env:"IAM_AUTH"`
	// Hosts overrides the host for named pools. Useful for Aurora Global Database
	// deployments where the secondary region's reader pool should connect to a
	// local regional reader endpoint while the writer pool stays on the primary
	// region's cluster endpoint.
	//
	// Format: comma-separated <pool>:<host> pairs.
	//
	//	STDPGX_HOSTS=ro:mycluster.cluster-ro-def.ap-southeast-1.rds.amazonaws.com
	//	STDPGX_HOSTS=ro:ro-host.example.com,rw:rw-host.example.com
	//
	// An override wins over name-based conventions (i.e., when a pool's host is
	// overridden, the cluster-ro auto-rewrite is skipped). The IAM signing region
	// is then derived from the overridden host, so cross-region setups Just Work.
	Hosts map[string]string `env:"HOSTS"`
}

type Deriver func(logs *zap.Logger, base *pgxpool.Config) *pgxpool.Config

type Driver[DBT any] interface {
	NewPool(pcfg *pgxpool.Config) (DBT, error)
	Close(pool DBT) error
}

type EndRole struct {
	Username string
	Password string
}

type Params struct {
	fx.In

	Config    Config
	AwsConfig aws.Config `optional:"true"`
	Logs      *zap.Logger
}

type PgtestdbTestMigratorParams struct {
	fx.In
	Migrator pgtestdb.Migrator
	Role     *pgtestdb.Role `optional:"true"`
	EndRole  *EndRole       `optional:"true"`
}

type Result struct {
	fx.Out

	PoolConfig *pgxpool.Config
}

type TestMigrator interface {
	Migrate(
		tb testing.TB,
		cfg Config,
		pcfg *pgxpool.Config,
	) (*pgxpool.Config, error)
}

type TestMigratorFunc func(
	tb testing.TB,
	cfg Config,
	pcfg *pgxpool.Config,
) (*pgxpool.Config, error)