64 lines
1.6 KiB
Go
64 lines
1.6 KiB
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 ""
|
|
}
|
|
|
|
if !(strings.HasPrefix(host, "/v2") || strings.HasPrefix(host, "/private")) {
|
|
return ""
|
|
}
|
|
|
|
path = systemsRegex.ReplaceAllString(path, "{systemRef}")
|
|
path = membersRegex.ReplaceAllString(path, "{memberRef}")
|
|
path = groupsRegex.ReplaceAllString(path, "{groupRef}")
|
|
path = switchesRegex.ReplaceAllString(path, "{switchRef}")
|
|
path = guildsRegex.ReplaceAllString(path, "{guild_id}")
|
|
path = messagesRegex.ReplaceAllString(path, "{message_id}")
|
|
|
|
return path
|
|
}
|
|
|
|
func requireEnv(key string) string {
|
|
if val, ok := os.LookupEnv(key); !ok {
|
|
panic("missing `" + key + "` in environment")
|
|
} else {
|
|
return val
|
|
}
|
|
}
|