photoprism/pkg/http/header/cdn.go

package header

import (
	"net/http"

	"github.com/photoprism/photoprism/pkg/list"
)

// Content Delivery Network (CDN) headers.
const (
	CdnHost         = "Cdn-Host"
	CdnMobileDevice = "Cdn-Mobiledevice"
	CdnServerZone   = "Cdn-Serverzone"
	CdnServerID     = "Cdn-Serverid"
	CdnConnectionID = "Cdn-Connectionid"
)

var (
	// CdnMethods lists HTTP methods allowed via CDN.
	CdnMethods = []string{http.MethodGet, http.MethodHead, http.MethodOptions}
)

// IsCdn checks whether the request seems to come from a CDN.
func IsCdn(req *http.Request) bool {
	if req == nil {
		return false
	} else if req.Header == nil || req.URL == nil {
		return false
	}

	if req.Header.Get(CdnHost) != "" {
		return true
	}

	return false
}

// AbortCdnRequest checks if the request should not be served through a CDN.
func AbortCdnRequest(req *http.Request) bool {
	switch {
	case !IsCdn(req):
		return false
	case req.Header.Get(XAuthToken) != "":
		return true
	case req.URL.Path == "/":
		return true
	default:
		return list.Excludes(CdnMethods, req.Method)
	}
}