main.go

package main

import (
	"os"
	"log"
	"net/http"
	"github.com/sharizzle/go-api-template/db"
	"github.com/gorilla/mux"
	"github.com/sharizzle/go-api-template/service/user"
	customRouter "github.com/sharizzle/go-api-template/router"
	"github.com/sharizzle/go-api-template/middleware"
	"github.com/joho/godotenv"
)

func NewRouter() *mux.Router {

	//init router
	router := mux.NewRouter()

	//append user routes
	customRouter.AppRoutes = append(customRouter.AppRoutes, user.Routes)

	for _, route := range customRouter.AppRoutes {

		//create subroute
		routePrefix := router.PathPrefix(route.Prefix).Subrouter()

		//loop through each sub route
		for _, r := range route.SubRoutes {

			var handler http.Handler
			handler = r.HandlerFunc

			//check to see if route should be protected with jwt
			if r.Protected {
				handler = middleware.JWTMiddleware(r.HandlerFunc)
			}

			//attach sub route
			routePrefix.
				Path(r.Pattern).
				Handler(handler).
				Methods(r.Method).
				Name(r.Name)
		}

	}

	return router
}

func main(){

	//init router
	godotenv.Load(".env")
	port := os.Getenv("PORT")
	router := NewRouter()

	//Setup database
	db.DB = db.SetupDB()
	db.DB.AutoMigrate(&user.User{})


	//create http server
	log.Fatal(http.ListenAndServe(":"+port, router))
}