main.go

package main

import (
	// "fmt"
	"github.com/gin-gonic/gin"
	"url-shortener/src/handlers"
	"url-shortener/src/config"
)

const PORT = config.PORT

func CORSMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        c.Header("Access-Control-Allow-Origin", "*")
        c.Header("Access-Control-Allow-Credentials", "true")
        c.Header("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
        c.Header("Access-Control-Allow-Methods", "POST,HEAD,PATCH, OPTIONS, GET, PUT, DELETE")
        if c.Request.Method == "OPTIONS" {
            c.AbortWithStatus(204)
            return
        }
        c.Next()
    }
}

func main() {
	app := gin.Default()
	app.Use(CORSMiddleware())

	//Init handlers
	httpHandler := handlers.NewHandler()
	
	//Routes
	app.GET("/u/:key", httpHandler.ForwardUrl)
	app.POST("/short-url", httpHandler.CreateShortUrl)
	
	//Start server
	app.Run("localhost:" + PORT)
}