bitmagnet/internal/plugin/core/postgres/config.go

package postgres

import (
	"fmt"
	"slices"
	"strings"

	"github.com/bitmagnet-io/bitmagnet/internal/config/param"
	"go.uber.org/fx"
)

type (
	DSN      string
	Host     string
	Port     uint16
	User     string
	Password string
	Database string
)

var (
	ParamDSN = param.MustNew(
		param.Description[DSN](
			"Postgres Data Source Name (if specified, all other options for this plugin are ignored)",
		),
	)

	ParamHost = param.MustNew(
		param.Description[Host]("Postgres host"),
		param.Default(Host("localhost")),
	)

	ParamPort = param.MustNew(
		param.Description[Port]("Postgres port number"),
		param.PortNumber[Port](),
		param.Default[Port](5432),
	)

	ParamUser = param.MustNew(
		param.Description[User]("Postgres username"),
		param.Default(User("postgres")),
	)

	ParamPassword = param.MustNew(
		param.Description[Password]("Postgres password"),
		param.Default(Password("postgres")),
	)

	ParamDatabase = param.MustNew(
		param.Description[Database]("Postgres database name"),
		param.Default(Database("bitmagnet")),
	)
)

type Config struct {
	fx.In
	DSN      DSN
	Host     Host
	Port     Port
	User     User
	Password Password
	Database Database
}

func (c Config) CreateDSN() DSN {
	if c.DSN != "" {
		return c.DSN
	}

	var values []string

	for k, v := range c.values() {
		values = append(values, fmt.Sprintf("%s=%s", k, v))
	}

	slices.Sort(values)

	return DSN(strings.Join(values, " "))
}

func (c Config) values() map[string]string {
	p := map[string]string{}
	setIfNotEmpty(p, "dbname", c.Database)
	setIfNotEmpty(p, "user", c.User)
	setIfNotEmpty(p, "host", c.Host)
	setIfNotEmpty(p, "port", fmt.Sprintf("%d", c.Port))
	setIfNotEmpty(p, "password", c.Password)

	return p
}

func setIfNotEmpty(m map[string]string, key string, val any) {
	strVal := fmt.Sprintf("%v", val)
	if strVal != "" {
		m[key] = strVal
	}
}