clean up main.go

db/db.go

@@ -21,9 +21,18 @@ func CreateAllTables() error {
 	return CreateSessionsTable()
 }
 
+func GetDbUrl() string {
+	host := os.Getenv("PGHOST")
+	port := os.Getenv("PGPORT")
+	user := os.Getenv("PGUSER")
+	pass := os.Getenv("PGPASSWORD")
+
+	return fmt.Sprintf("postgres://%s:%s@%s:%s", user, pass, host, port)
+}
+
 func CreateDB() error {
 	conn, err := pgx.Connect(context.Background(),
-		"postgres://postgres:postgres@localhost:5432",
+		GetDbUrl(),
 	)
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Cannot connect to PostgreSQL: %v\n", err)

main.go

@@ -2,10 +2,8 @@ package main
 
 import (
 	"context"
-	"errors"
 	"fmt"
 	"os"
-	"path/filepath"
 
 	"muzi/db"
 	"muzi/web"
@@ -13,64 +11,25 @@ import (
 	"github.com/jackc/pgx/v5/pgxpool"
 )
 
-func dirCheck(path string) error {
-	_, err := os.Stat(path)
+func check(msg string, err error) {
 	if err != nil {
-		if errors.Is(err, os.ErrNotExist) {
-			os.MkdirAll(path, os.ModePerm)
-		} else {
-			fmt.Fprintf(os.Stderr, "Error checking dir: %s: %v\n", path, err)
-			return err
+		fmt.Fprintf(os.Stderr, "Error %s: %v\n", msg, err)
+		os.Exit(1)
 	}
 }
-	return nil
-}
 
 func main() {
-	zipDir := filepath.Join(".", "imports", "spotify", "zip")
-	extDir := filepath.Join(".", "imports", "spotify", "extracted")
-
-	dirs := []string{zipDir, extDir}
-	for _, dir := range dirs {
-		err := dirCheck(dir)
-		if err != nil {
-			fmt.Fprintf(os.Stderr, "Error checking dir: %s: %v\n", dir, err)
-			return
-		}
-	}
-	err := db.CreateDB()
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error ensuring muzi DB exists: %v\n", err)
-		return
-	}
+	check("ensuring muzi DB exists", db.CreateDB())
 
+	var err error
 	db.Pool, err = pgxpool.New(
 		context.Background(),
-		"postgres://postgres:postgres@localhost:5432/muzi",
+		fmt.Sprintf(db.GetDbUrl(), "/muzi"),
 	)
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Cannot connect to muzi database: %v\n", err)
-		return
-	}
+	check("connecting to muzi database", err)
 	defer db.Pool.Close()
 
-	err = db.CreateAllTables()
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error ensuring all tables exist: %v\n", err)
-		return
-	}
-
-	err = db.CleanupExpiredSessions()
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error cleaning expired sessions: %v\n", err)
-		return
-	}
-
-	/*
-		err = migrate.ImportSpotify(1)
-		if err != nil {
-			return
-		}
-	*/
+	check("ensuring all tables exist", db.CreateAllTables())
+	check("cleaning expired sessions", db.CleanupExpiredSessions())
 	web.Start()
 }