routes.go

package routes

import (
	"go-graphql-mongo-server/auth"
	"go-graphql-mongo-server/gqlhandler"
	"net/http"

	"github.com/gorilla/mux"
	"github.com/prometheus/client_golang/prometheus/promhttp"
)

type Route struct {
	Method     string
	Pattern    string
	Handler    http.HandlerFunc
	Middleware []mux.MiddlewareFunc
}

var routes []Route

func initializeRoutes() {
	createLimiterMiddleware()
	setAllRoutes()
}

func registerAPIRoute(method, pattern string, handler http.HandlerFunc, middleware []mux.MiddlewareFunc) {
	pattern = "/api" + pattern
	registerCommonRoute(method, pattern, handler, middleware)
}
func registerCommonRoute(method, pattern string, handler http.HandlerFunc, middleware []mux.MiddlewareFunc) {
	routes = append(routes, Route{method, pattern, handler, middleware}, Route{"OPTIONS", pattern, handler, middleware})
}

func setAllRoutes() {

	registerAPIRoute(
		"POST",
		"/graphql",
		gqlhandler.GraphqlHandler,
		[]mux.MiddlewareFunc{
			limiterMiddleware.Handle,
			auth.Middleware,
		})

	registerAPIRoute(
		"GET",
		"/graphiql",
		gqlhandler.GraphiqlHandler,
		nil,
	)

	registerCommonRoute(
		"GET",
		"/metrics",
		promhttp.Handler().ServeHTTP,
		nil,
	)

	registerCommonRoute(
		"GET",
		"/health",
		healthCheckHandler,
		nil,
	)
}

func NewRouter() (router *mux.Router) {
	initializeRoutes()

	router = mux.NewRouter()

	for _, route := range routes {
		r := router.
			Methods(route.Method).
			Path(route.Pattern)

		var handler http.Handler = route.Handler
		for i := len(route.Middleware) - 1; i >= 0; i-- {
			handler = route.Middleware[i](handler)
		}

		r.Handler(handler)

	}
	return
}