main.go

package main

import (
	"flag"
	"log"
	"net/http"
	"os"

	"github.com/go-chi/chi/v5"
	"github.com/joho/godotenv"
	"github.com/veryordinary11/Go-web-server/database"
)

type apiConfig struct {
	fileserverHits int
	DB             *database.DB
	jwtSecret      string
	polkaKey       string
}

func main() {
	// Define the --debug flag
	var debug bool
	flag.BoolVar(&debug, "debug", false, "Enable debug mode")
	flag.Parse()

	// Load data from .env file
	godotenv.Load()
	jwtSecret := os.Getenv("JWT_SECRET")
	polkaKey := os.Getenv("POLKA_KEY")

	var path string
	if debug {
		log.Println("Debug mode enabled")
		path = "database_test.json"
	} else {
		path = "database.json"
	}

	const filepathRoot = "."
	const port = "8080"

	// Create a new database connection
	db, err := database.NewDB(path)
	if err != nil {
		log.Fatalf("Error creating database: %v", err)
	}

	apiCfg := &apiConfig{
		fileserverHits: 0,
		DB:             db,
		jwtSecret:      jwtSecret,
		polkaKey:       polkaKey,
	}

	apiRouter := createAPIRouter(apiCfg)
	adminRouter := createAdminRouter(apiCfg)

	r := chi.NewRouter()

	r.Mount("/api", apiRouter)
	r.Mount("/admin", adminRouter)

	fsHandler := apiCfg.middlewareMetricsInc(http.StripPrefix("/app", http.FileServer(http.Dir(filepathRoot))))

	r.Handle("/app", fsHandler)
	r.Handle("/app/*", fsHandler)

	corsRouter := middlewareCors(r)

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

	err = server.ListenAndServe()
	if err != nil {
		panic(err)
	}
}