feat(api): init rust rewrite

services/web-proxy/main.go

@@ -2,25 +2,26 @@ package main
 
 import (
 	"context"
-	"encoding/json"
-	"fmt"
+	"log"
 	"net/http"
 	"net/http/httputil"
+	"net/url"
 	"strconv"
-	"strings"
 	"time"
 
-	"github.com/go-redis/redis/v8"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
-
-	"web-proxy/redis_rate"
 )
 
-var limiter *redis_rate.Limiter
-
-// todo: be able to raise ratelimits for >1 consumers
-var token2 string
+func proxyTo(host string) *httputil.ReverseProxy {
+	rp := httputil.NewSingleHostReverseProxy(&url.URL{
+		Scheme:   "http",
+		Host:     host,
+		RawQuery: "",
+	})
+	rp.ModifyResponse = logTimeElapsed
+	return rp
+}
 
 // todo: this shouldn't be in this repo
 var remotes = map[string]*httputil.ReverseProxy{
@@ -29,22 +30,6 @@ var remotes = map[string]*httputil.ReverseProxy{
 	"sentry.pluralkit.me": proxyTo("[fdaa:0:ae33:a7b:8dd7:0:a:202]:9000"),
 }
 
-func init() {
-	redisHost := requireEnv("REDIS_HOST")
-	redisPassword := requireEnv("REDIS_PASSWORD")
-
-	rdb := redis.NewClient(&redis.Options{
-		Addr:     redisHost,
-		Username: "default",
-		Password: redisPassword,
-	})
-	limiter = redis_rate.NewLimiter(rdb)
-
-	token2 = requireEnv("TOKEN2")
-
-	remotes["dash.pluralkit.me"].ModifyResponse = modifyDashResponse
-}
-
 type ProxyHandler struct{}
 
 func (p ProxyHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
@@ -61,48 +46,6 @@ func (p ProxyHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	if r.Host == "api.pluralkit.me" {
-		// root
-		if r.URL.Path == "" {
-			// api root path redirects to docs
-			http.Redirect(rw, r, "https://pluralkit.me/api/", http.StatusFound)
-			return
-		}
-
-		// CORS headers
-		rw.Header().Add("Access-Control-Allow-Origin", "*")
-		rw.Header().Add("Access-Control-Allow-Methods", "*")
-		rw.Header().Add("Access-Control-Allow-Credentials", "true")
-		rw.Header().Add("Access-Control-Allow-Headers", "Content-Type, Authorization, sentry-trace, User-Agent")
-		rw.Header().Add("Access-Control-Max-Age", "86400")
-
-		if r.Method == http.MethodOptions {
-			rw.WriteHeader(200)
-			return
-		}
-
-		if r.URL.Path == "/" {
-			http.Redirect(rw, r, "https://pluralkit.me/api", http.StatusFound)
-			return
-		}
-
-		if strings.HasPrefix(r.URL.Path, "/v1") {
-			rw.Header().Set("content-type", "application/json")
-			rw.WriteHeader(410)
-			rw.Write([]byte(`{"message":"Unsupported API version","code":0}`))
-		}
-
-		if is_trying_to_use_v1_path_on_v2(r.URL.Path) {
-			rw.WriteHeader(400)
-			rw.Write([]byte(`{"message":"Invalid path for API version","code":0}`))
-			return
-		}
-
-		if is_api_ratelimited(rw, r) {
-			return
-		}
-	}
-
 	startTime := time.Now()
 	r = r.WithContext(context.WithValue(r.Context(), "req-time", startTime))
 
@@ -119,40 +62,14 @@ func logTimeElapsed(resp *http.Response) error {
 		"domain": r.Host,
 		"method": r.Method,
 		"status": strconv.Itoa(resp.StatusCode),
-		"route":  cleanPath(r.Host, r.URL.Path),
+		"route":  r.URL.Path,
 	}).Observe(elapsed.Seconds())
 
-	log, _ := json.Marshal(map[string]interface{}{
-		"remote_ip":   r.Header.Get("Fly-Client-IP"),
-		"method":      r.Method,
-		"host":        r.Host,
-		"route":       r.URL.Path,
-		"route_clean": cleanPath(r.Host, r.URL.Path),
-		"status":      resp.StatusCode,
-		"elapsed":     elapsed.Milliseconds(),
-		"user_agent":  r.Header.Get("User-Agent"),
-	})
-	fmt.Println(string(log))
-
-	// log.Printf("[%s] \"%s %s%s\" %d - %vms %s\n", r.Header.Get("Fly-Client-IP"), r.Method, r.Host, r.URL.Path, resp.StatusCode, elapsed.Milliseconds(), r.Header.Get("User-Agent"))
+	log.Printf("[%s] \"%s %s%s\" %d - %vms %s\n", r.Header.Get("Fly-Client-IP"), r.Method, r.Host, r.URL.Path, resp.StatusCode, elapsed.Milliseconds(), r.Header.Get("User-Agent"))
 
 	return nil
 }
 
-func modifyDashResponse(resp *http.Response) error {
-	r := resp.Request
-
-	// cache built+hashed dashboard js/css files forever
-	is_dash_static_asset := strings.HasPrefix(r.URL.Path, "/assets/") &&
-		(strings.HasSuffix(r.URL.Path, ".js") || strings.HasSuffix(r.URL.Path, ".css") || strings.HasSuffix(r.URL.Path, ".map"))
-
-	if is_dash_static_asset && resp.StatusCode == 200 {
-		resp.Header.Add("Cache-Control", "max-age=31536000, s-maxage=31536000, immutable")
-	}
-
-	return logTimeElapsed(resp)
-}
-
 func main() {
 	prometheus.MustRegister(metric)
 
@@ -161,3 +78,11 @@ func main() {
 
 	http.ListenAndServe(":8080", ProxyHandler{})
 }
+
+var metric = prometheus.NewHistogramVec(
+	prometheus.HistogramOpts{
+		Name:    "pk_http_requests",
+		Buckets: []float64{.1, .25, 1, 2.5, 5, 20},
+	},
+	[]string{"domain", "method", "status", "route"},
+)