package util

import "github.com/gin-gonic/gin"

func IsSSL(c *gin.Context) bool {
	return c.Request.TLS != nil
}

func IsHttps(c *gin.Context) bool {
	return GetRequestScheme(c) == "https"
}

func IsHttp(c *gin.Context) bool {
	return GetRequestScheme(c) == "http"
}

func GetUserAgent(c *gin.Context) string {
	return c.GetHeader("User-Agent")
}

// websocket TODO
// Upgrade: websocket Connection: ...Upgrade...
func GetRequestScheme(c *gin.Context) string {
	// Can't use `r.Request.URL.Scheme`
	// See: https://groups.google.com/forum/#!topic/golang-nuts/pMUkBlQBDF0
	if IsSSL(c) {
		return "https"
	}
	if scheme := c.Request.Header.Get("X-Forwarded-Proto"); scheme != "" {
		return scheme
	}
	if scheme := c.Request.Header.Get("X-Forwarded-Protocol"); scheme != "" {
		return scheme
	}
	if ssl := c.Request.Header.Get("X-Forwarded-Ssl"); ssl == "on" {
		return "https"
	}
	if scheme := c.Request.Header.Get("X-Url-Scheme"); scheme != "" {
		return scheme
	}
	return "http"
}