56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package env
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/url"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/namsral/flag"
|
|
)
|
|
|
|
var (
|
|
Port int
|
|
Env string
|
|
PublicUrl string
|
|
CommitShortSha string
|
|
CommitLongSha string
|
|
PostgresUrl string
|
|
PostgresUrlWithoutPassword string
|
|
)
|
|
|
|
func Load(filenames ...string) error {
|
|
if err := godotenv.Load(); err != nil {
|
|
return err
|
|
}
|
|
flag.IntVar(&Port, "PORT", 4444, "Server port")
|
|
flag.StringVar(&PublicUrl, "PUBLIC_URL", "", "Base URL")
|
|
flag.StringVar(&PostgresUrl, "POSTGRES_DB", "", "PostgreSQL connection URL")
|
|
flag.StringVar(&Env, "ENV", "development", "Build environment")
|
|
return nil
|
|
}
|
|
|
|
func Parse() {
|
|
flag.Parse()
|
|
CommitLongSha = execCmd("git", "rev-parse", "HEAD")
|
|
CommitShortSha = execCmd("git", "rev-parse", "--short", "HEAD")
|
|
if u, err := url.Parse(PostgresUrl); err == nil {
|
|
if pw, ok := u.User.Password(); ok {
|
|
PostgresUrlWithoutPassword = strings.Replace(PostgresUrl, fmt.Sprintf(":%s@", pw), ":*****@", 1)
|
|
} else {
|
|
PostgresUrlWithoutPassword = PostgresUrl
|
|
}
|
|
}
|
|
}
|
|
|
|
func execCmd(name string, args ...string) string {
|
|
cmd := exec.Command(name, args...)
|
|
stdout, err := cmd.Output()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return strings.TrimSpace(string(stdout))
|
|
}
|