PluralKit/services/web-proxy/util.go

package main

import (
	"net/http/httputil"
	"net/url"
	"os"
	"regexp"
	"strings"

	"github.com/prometheus/client_golang/prometheus"
)

var metric = prometheus.NewHistogramVec(
	prometheus.HistogramOpts{
		Name:    "pk_http_requests",
		Buckets: []float64{.1, .25, 1, 2.5, 5, 20},
	},
	[]string{"domain", "method", "status", "route"},
)

func proxyTo(host string) *httputil.ReverseProxy {
	rp := httputil.NewSingleHostReverseProxy(&url.URL{
		Scheme:   "http",
		Host:     host,
		RawQuery: "",
	})
	rp.ModifyResponse = logTimeElapsed
	return rp
}

var systemsRegex = regexp.MustCompile("systems/[^/{}]+")
var membersRegex = regexp.MustCompile("members/[^/{}]+")
var groupsRegex = regexp.MustCompile("groups/[^/{}]+")
var switchesRegex = regexp.MustCompile("switches/[^/{}]+")
var guildsRegex = regexp.MustCompile("guilds/[^/{}]+")
var messagesRegex = regexp.MustCompile("messages/[^/{}]+")

func cleanPath(host, path string) string {
	if host != "api.pluralkit.me" {
		return ""
	}

	path = strings.ToLower(path)

	if !(strings.HasPrefix(path, "/v2") || strings.HasPrefix(path, "/private")) {
		return ""
	}

	path = systemsRegex.ReplaceAllString(path, "systems/{systemRef}")
	path = membersRegex.ReplaceAllString(path, "members/{memberRef}")
	path = groupsRegex.ReplaceAllString(path, "groups/{groupRef}")
	path = switchesRegex.ReplaceAllString(path, "switches/{switchRef}")
	path = guildsRegex.ReplaceAllString(path, "guilds/{guild_id}")
	path = messagesRegex.ReplaceAllString(path, "messages/{message_id}")

	return path
}

func requireEnv(key string) string {
	if val, ok := os.LookupEnv(key); !ok {
		panic("missing `" + key + "` in environment")
	} else {
		return val
	}
}