config.go

package database

import (
	"fmt"
	"os"

	"entgo.io/ent/dialect"
	"github.com/stablecog/sc-go/log"
	"github.com/stablecog/sc-go/utils"
)

type SqlDBConn interface {
	DSN() string
	Dialect() string
}

type PostgresConn struct {
	Host     string
	Port     int
	Password string
	User     string
	DBName   string
}

func (c *PostgresConn) DSN() string {
	return fmt.Sprintf("postgres://%s:%s@%s:%d/%s", c.User, c.Password, c.Host, c.Port, c.DBName)
}

func (c *PostgresConn) Dialect() string {
	return "pgx"
}

type SqliteConn struct {
	FileName string
	Mode     string
}

func (c *SqliteConn) DSN() string {
	return fmt.Sprintf("file:%s?cache=shared&mode=%s&_fk=1", c.FileName, c.Mode)
}

func (c *SqliteConn) Dialect() string {
	return dialect.SQLite
}

// Gets the DB connection information based on environment variables
func GetSqlDbConn(mock bool) (SqlDBConn, error) {
	if mock {
		return &SqliteConn{FileName: "testing", Mode: "memory"}, nil
	}
	// Use postgres
	postgresDb := utils.GetEnv().PostgresDB
	postgresUser := utils.GetEnv().PostgresUser
	postgresPassword := utils.GetEnv().PostgresPassword
	postgresHost := utils.GetEnv().PostgresHost
	postgresPort := utils.GetEnv().PostgresPort

	if postgresDb == "" || postgresUser == "" || postgresPassword == "" {
		log.Error("Postgres environment variables not set, not sure what to do? so exiting")
		os.Exit(1)
	}
	log.Infof("Using PostgreSQL database %s@%s:%d", postgresUser, postgresHost, postgresPort)
	return &PostgresConn{
		Host:     postgresHost,
		Port:     postgresPort,
		Password: postgresPassword,
		User:     postgresUser,
		DBName:   postgresDb,
	}, nil
}