main.go

package main

import (
	"database/sql"
	"log"
	"net/http"
	"os"

	"github.com/danielmiranda22/social-network-server/internal/api"
	"github.com/danielmiranda22/social-network-server/internal/api/admin"
	"github.com/danielmiranda22/social-network-server/internal/api/chirps"
	"github.com/danielmiranda22/social-network-server/internal/api/health"
	"github.com/danielmiranda22/social-network-server/internal/api/users"
	"github.com/danielmiranda22/social-network-server/internal/database"
	"github.com/joho/godotenv"
	_ "github.com/lib/pq"
)

func main() {
	const filepathRoot = "."
	const port = "8080"
	const apiSecretEnvKey = "API_SECRET"
	const polkaKeyEnvKey = "POLKA_KEY"

	godotenv.Load()

	dbURL := os.Getenv("DB_URL")
	if dbURL == "" {
		log.Fatal("DB_URL environment variable is required")
	}

	apiSecret := os.Getenv(apiSecretEnvKey)
	if apiSecret == "" {
		log.Fatalf("%s environment variable is required", apiSecretEnvKey)
	}

	polkaKey := os.Getenv(polkaKeyEnvKey)
	if polkaKey == "" {
		log.Fatalf("%s environment variable is required", polkaKeyEnvKey)
	}

	dbConn, err := sql.Open("postgres", dbURL)
	if err != nil {
		log.Fatalf("Failed to connect to database: %v", err)
	}
	defer dbConn.Close()

	apiCfg := &api.ApiConfig{
		DB:        database.New(dbConn),
		Platform:  os.Getenv("PLATFORM"),
		APISecret: apiSecret,
		PolkaKey:  polkaKey,
	}

	mux := http.NewServeMux()

	// file server
	mux.Handle("/app/",
		apiCfg.MiddlewareMetricsInc(
			http.StripPrefix("/app", http.FileServer(http.Dir(filepathRoot))),
		),
	)

	// health
	mux.HandleFunc("GET /api/healthz", health.IsHealthHandler)

	// admin
	mux.HandleFunc("GET /admin/metrics", admin.HandlerMetrics(apiCfg))
	mux.HandleFunc("POST /admin/reset",  admin.HandlerReset(apiCfg))

	// users
	mux.HandleFunc("POST /api/users",   users.HandlerCreateUser(apiCfg))
	mux.HandleFunc("PUT /api/users",    users.HandlerUpdateUser(apiCfg))
	mux.HandleFunc("POST /api/login",   users.HandlerLogin(apiCfg))
	mux.HandleFunc("POST /api/refresh", users.HandlerRefresh(apiCfg))
	mux.HandleFunc("POST /api/revoke",  users.HandlerRevoke(apiCfg))

	// webhooks
	mux.HandleFunc("POST /api/polka/webhooks", users.HandlerUpdateChirpyRed(apiCfg))

	// chirps
	mux.HandleFunc("GET /api/chirps",              chirps.HandlerGetAllChirps(apiCfg))
	mux.HandleFunc("GET /api/chirps/{chirpID}",    chirps.HandlerGetChirpByID(apiCfg))
	mux.HandleFunc("POST /api/chirps",             chirps.HandlerCreateChirp(apiCfg))
	mux.HandleFunc("DELETE /api/chirps/{chirpID}", chirps.HandlerDeleteChirpByID(apiCfg))

	server := &http.Server{
		Addr:    ":" + port,
		Handler: mux,
	}

	log.Printf("Serving files from %s on port: %s\n", filepathRoot, port)
	log.Fatal(server.ListenAndServe())
}