main.go

package main

import (
	"errors"
	"fmt"
	"log/slog"
	"net/http"
	"os"

	"github.com/Suryarpan/chat-api/internal/apiconf"
	"github.com/Suryarpan/chat-api/internal/auth"
	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
	"github.com/go-chi/cors"
	"github.com/jackc/pgx/v5/pgxpool"
)

func setUpMiddlewares(r *chi.Mux, cp *pgxpool.Pool) error {
	if r == nil {
		return errors.New("please provide a router")
	} else if cp == nil {
		return errors.New("please provide a connection pool")
	}

	r.Use(apiconf.Logger)
	r.Use(apiconf.ApiConfigure(cp))
	r.Use(middleware.Recoverer)
	r.Use(middleware.CleanPath)
	r.Use(middleware.AllowContentType("application/json", "text/xml"))
	r.Use(middleware.ContentCharset("", "UTF-8", "Latin-1"))
	r.Use(cors.Handler(cors.Options{
		AllowedOrigins:   []string{"https://*", "http://*"},
		AllowedMethods:   []string{"GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"},
		AllowedHeaders:   []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token"},
		ExposedHeaders:   []string{"Link"},
		AllowCredentials: false,
		MaxAge:           300, // Maximum value not ignored by any of major browsers
	}))
	r.Use(middleware.StripSlashes)
	return nil
}

func routeSetup(r *chi.Mux) error {
	if r == nil {
		return errors.New("please provide a router")
	}
	// auth api setup
	r.Mount("/auth", AuthRouter())
	// user api setup
	r.Mount("/user", UserRouter())
	// chat data setup
	r.With(auth.Authentication).Mount("/message", MessageRouter())
	// admin setup
	return nil
}

func main() {
	logger := slog.New(*apiconf.LoggerConfig())
	slog.SetDefault(logger)
	// DB Setup
	connPool, err := apiconf.SetupPool()
	if err != nil {
		panic(fmt.Sprintf("Error: could not create connection pool: %v", err))
	}
	defer connPool.Close()
	// Auth Setup
	err = auth.SetupAuth()
	if err != nil {
		panic(fmt.Sprintf("Error: could not setup auth: %v", err))
	}

	// router setup
	mainRouter := chi.NewRouter()

	err = setUpMiddlewares(mainRouter, connPool)
	if err != nil {
		panic(fmt.Sprintf("Error: could not setup middlewares: %v", err))
	}
	// base api setup
	apiV1router := chi.NewRouter()
	apiV1router.Get("/health", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Hello World"))
	})

	err = routeSetup(apiV1router)
	if err != nil {
		panic(fmt.Sprintf("Error: could not mount the sub routes: %v", err))
	}
	// router run
	mainRouter.Mount("/api/v1", apiV1router)
	port, ok := os.LookupEnv("CHAT_API_PORT")
	if !ok {
		panic("Error: could not find CHAT_API_PORT environment variable")
	}
	logger.Info("starting router", "port", port)

	server := &http.Server{
		Handler: mainRouter,
		Addr:    ":" + port,
	}
	err = server.ListenAndServe()
	if err != nil {
		slog.Error(fmt.Sprintf("%v", err))
	}
}