package database

import (
	"errors"
	"fmt"

	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

var postgresqlDsn = "host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=%s"

func PostgreDialector(configMap map[string]string) (dialector gorm.Dialector, err error) {
	host, okHost := configMap["host"]
	port, okPort := configMap["port"]
	username, okUsername := configMap["username"]
	password, okPwd := configMap["password"]
	dbName, okDbName := configMap["dbname"]
	sslMode, okSSLMode := configMap["sslmode"]
	if !okSSLMode {
		sslMode = "disable"
	}
	timeZone, okTimeZone := configMap["timezone"]
	if !okTimeZone {
		timeZone = "Asia/Shanghai"
	}
	if !okHost || !okPort || !okUsername || !okPwd || !okDbName {
		err = errors.New(fmt.Sprintf("数据库[%s:%s]配置错误", host, port))
		return
	}
	dialector = postgres.Open(fmt.Sprintf(postgresqlDsn, host, username, password, dbName, port, sslMode, timeZone))
	return
}