func getEnv(key string, defaultValue string) string {
    value := os.Getenv(key)
    if value == "" {
        return defaultValue
    }
    return value
}

func main() {
    env := getEnv("ENV", "dev")

    // Do not load .env file in Railway production
    if _, exists := os.LookupEnv("RAILWAY_ENVIRONMENT"); !exists {
        if err := godotenv.Load(); err != nil {
            log.Fatal("error loading .env file:", err)
        }
    }

    // Initialize logger
    var logger *slog.Logger

    if env == "prod" {
        logger = slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
            Level: slog.LevelInfo,
        }))
    } else {
        logger = slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
            Level: slog.LevelDebug,
        }))
    }
    utils.SetLogger(logger)

    e := echo.New()
    e.Validator = utils.NewCustomValidator()

    // Initialize DB connection
    var connStr string

    // Check for Railway's DATABASE_URL
    if dbURL := getEnv("DATABASE_URL", ""); dbURL != "" {
        connStr = dbURL
    } else {
        // Fallback to individual connection parameters
        connStr = fmt.Sprintf(
            "postgres://%s:%s@%s:%s/%s?sslmode=%s",
            getEnv("DB_USER", ""),
            getEnv("DB_PASSWORD", ""),
            getEnv("DB_HOST", ""),
            getEnv("DB_PORT", ""),
            getEnv("DB_NAME", ""),
            getEnv("DB_SSLMODE", ""),
        )
    }

    db, err := sql.Open("postgres", connStr)
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // Run migrations
    if err := migrate.RunMigrations(db, "db/migrations"); err != nil {
        log.Fatal(err)
    }

    // Initialize repository with DB connection
    userRepo := repository.NewUserRepository(db)
    passwordResetTokenRepo := repository.NewPasswordResetTokenRepository(db)

    userService := service.NewUserService(userRepo, passwordResetTokenRepo)
    userHandler := handler.NewUserHandler(userService)

    // Register routes
    handler.RegisterRoutes(e, &handler.RouteConfig{
        UserHandler:    userHandler,
        AuthMiddleware: handler.AuthMiddleware,
        BaseMiddleware: []echo.MiddlewareFunc{
            handler.RequestLoggerMiddleware,
            handler.ErrorCodeHandlingMiddleware,
        },
    })

    // Start the server using environment variable for port
    port := getEnv("PORT", "8080")

    fmt.Println("Env port", port)
    e.Logger.Fatal(e.Start(":" + port))

}