package middleware

import (
	"openflare/common"
	"strings"

	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
)

func CORS() gin.HandlerFunc {
	config := cors.DefaultConfig()
	config.AllowCredentials = true
	config.AllowHeaders = []string{"Origin", "Content-Length", "Content-Type", "Authorization", "X-Agent-Token", "Accept"}
	config.AllowOriginFunc = func(origin string) bool {
		serverAddr := strings.TrimRight(common.ServerAddress, "/")
		if serverAddr == "" {
			return true
		}
		if origin == serverAddr {
			return true
		}
		return false
	}
	return cors.New(config)
}