cors.go

package gin

import (
	krakendcors "github.com/devopsfaith/krakend-cors"
	"github.com/devopsfaith/krakend/config"
	"github.com/gin-gonic/gin"
	"gopkg.in/gin-contrib/cors.v1"
)

// New returns a gin.HandlerFunc with the CORS configuration provided in the ExtraConfig
func New(e config.ExtraConfig) gin.HandlerFunc {
	tmp := krakendcors.ConfigGetter(e)
	if tmp == nil {
		return nil
	}
	cfg, ok := tmp.(krakendcors.Config)
	if !ok {
		return nil
	}

	var allowAllOrigins bool
	if len(cfg.AllowOrigins) == 0 {
		allowAllOrigins = true
	} else {
		for _, origin := range cfg.AllowOrigins {
			if origin == "*" {
				allowAllOrigins = true
				cfg.AllowOrigins = nil
				break
			}
		}
	}

	return cors.New(cors.Config{
		AllowAllOrigins:  allowAllOrigins,
		AllowOrigins:     cfg.AllowOrigins,
		AllowMethods:     cfg.AllowMethods,
		AllowHeaders:     cfg.AllowHeaders,
		ExposeHeaders:    cfg.ExposeHeaders,
		AllowCredentials: cfg.AllowCredentials,
		MaxAge:           cfg.MaxAge,
	})
}