hiling_go/api/server.go

package api

import (
	"fmt"

	db "git.nochill.in/nochill/hiling_go/db/sqlc"
	"git.nochill.in/nochill/hiling_go/util"
	"git.nochill.in/nochill/hiling_go/util/token"
	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
)

type Server struct {
	Config     util.Config
	Store      db.Store
	TokenMaker token.Maker
	Router     *gin.Engine
}

func NewServer(config util.Config, store db.Store) (*Server, error) {
	tokenMaker, err := token.NewPasetoMaker(config.TokenSymmetricKey)
	if err != nil {
		return nil, fmt.Errorf("cannot create token maker: %w", err)
	}

	server := &Server{
		Config:     config,
		Store:      store,
		TokenMaker: tokenMaker,
	}

	server.getRoutes()
	return server, nil
}

func (server *Server) getRoutes() {
	router := gin.Default()

	router.Use(cors.New(cors.Config{
		AllowOrigins:     []string{"http://localhost:5173"},
		AllowCredentials: true,
		AllowHeaders:     []string{"Content-Type", "Content-Length", "Accept-Encoding", "X-CSRF-Token", "Authorization", "accept", "origin", "Cache-Control", "X-Requested-With"},
		AllowMethods:     []string{"POST", "PUT", "GET", "DELETE", "PATCH"},
	}))

	// router.Use(CORSMiddleware())

	router.POST("/user/signup", server.createUser)
	router.POST("/user/login", server.login)
	router.POST("/user/logout", server.logout)

	// LOCATION
	router.POST("/locations", server.createLocation)
	router.GET("/locations/recent", server.getListRecentLocationsWithRatings)
	router.GET("/locations/top-ratings", server.getTopListLocations)
	router.GET("/locations", server.getListLocations)
	router.GET("/location/:location_id", server.getLocation)
	router.GET("/location/tags/:location_id", server.getTagsByLocation)
	router.GET("/location/reviews", server.getListLocationReviews)

	//IMAGES
	router.GET("/images/location", server.getAllImagesByLocation)

	// REQUIRE AUTH TOKEN
	authRoutes := router.Use(authMiddleware(server.TokenMaker))
	authRoutes.POST("/review/location", server.createReview)
	authRoutes.GET("/user/review/location/:location_id", server.getUserReviewByLocation)

	server.Router = router
}

func (server *Server) Start(address string) error {
	return server.Router.Run(address)
}