main.go

package main

import (
    "fmt"
    "log"
    "os"

    "github.com/gofiber/fiber/v2"
    "github.com/joho/godotenv"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"

    "fiber-app/routes"
)

func initDB() (*gorm.DB, error) {
    dsn :=  os.Getenv("DATABASE_URL") // Use viper to get DATABASE_URL
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    if err != nil {
        return nil, err
    }

    err = db.AutoMigrate(&routes.Member{})
    if err != nil {
        return nil, err
    }

    return db, nil
}

func main() {
    err := godotenv.Load() // Load environment variables from .env file
    if err != nil {
        log.Fatal("Error loading .env file")
    }

    app := fiber.New()

    db, err := initDB()
    if err != nil {
        log.Fatalf("Failed to connect to the database: %v", err)
    }

    routes.SetupRoutes(app, db)

    portStr := os.Getenv("PORT") // Use os.Getenv to get PORT
    port := 3000 // Default port
    if portStr != "" {
        fmt.Sscanf(portStr, "%d", &port) // Convert string to integer
    }

    log.Fatal(app.Listen(fmt.Sprintf(":%d", port))) // Use %d for integer formatting
}