module.go

package pgx

import (
	"fmt"

	"github.com/go-modulus/modulus/logger"
	"github.com/go-modulus/modulus/module"
)

type ModuleConfig struct {
	DSN string `env:"PGX_DSN" comment:"Use this variable to set the DSN for the PGX connection. It overwrites the other PG_* variables. Example: PGX_DSN=postgres://postgres:foobar@localhost:5432/test?sslmode=disable"`

	ConnectionConfig *ConnectionConfig `env:",prefix=PG_"`
}

type ConnectionConfig struct {
	Host     string `env:"HOST, default=localhost"`
	Port     int    `env:"PORT, default=5432"`
	User     string `env:"USER, default=postgres"`
	Password string `env:"PASSWORD, default=foobar"`
	Database string `env:"DB_NAME, default=test"`
	SslMode  string `env:"SSL_MODE, default=disable"`
}

func (c ConnectionConfig) Dsn() string {
	return fmt.Sprintf(
		"postgres://%s:%s@%s:%d/%s?sslmode=%s",
		c.User,
		c.Password,
		c.Host,
		c.Port,
		c.Database,
		c.SslMode,
	)
}

func (c ModuleConfig) Dsn() string {
	if c.DSN != "" {
		return c.DSN
	}

	if c.ConnectionConfig == nil {
		return ""
	}

	return c.ConnectionConfig.Dsn()
}

func NewModule() *module.Module {
	return module.NewModule("modulus/pgx").
		AddProviders(
			NewPgxPool,
		).
		AddDependencies(
			logger.NewModule(),
		).
		InitConfig(ModuleConfig{})
}

func NewManifesto() module.Manifesto {
	return module.NewManifesto(
		NewModule(),
		"github.com/go-modulus/pgx",
		"A wrapper for the pgx package to integrate it into the Modulus framework.",
		"1.0.0",
	)
}