Explicitly proxy some ambigious routes. (#30)

Author: stereosteve

api/server.go

@@ -117,6 +117,26 @@ func NewApiServer(config config.Config) *ApiServer {
 	app.Use(app.resolveMyIdMiddleware)
 	app.Use(app.authMiddleware)
 
+	// some not-yet-implemented routes will match handlers below
+	// and won't fall thru to python reverse proxy handler
+	// so add some exclusions here to make `bridge.audius.co` less broken
+	// todo: implement these endpoints in bridgerton.
+	{
+		app.Use("/v1/full/users/top", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/users/genre/top", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/users/subscribers", BalancerForward(config.PythonUpstreams))
+
+		app.Use("/v1/full/playlists/top", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/playlists/trending", BalancerForward(config.PythonUpstreams))
+
+		app.Use("/v1/full/tracks/best_new_releases", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/tracks/feeling_lucky", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/tracks/most_loved", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/tracks/remixables", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/tracks/usdc-purchase", BalancerForward(config.PythonUpstreams))
+		app.Use("/v1/full/tracks/trending/underground", BalancerForward(config.PythonUpstreams))
+	}
+
 	v1 := app.Group("/v1")
 	v1Full := app.Group("/v1/full")
 
@@ -173,23 +193,7 @@ func NewApiServer(config config.Config) *ApiServer {
 	app.Static("/", "./static")
 
 	// proxy unhandled requests thru to existing discovery API
-	{
-		upstreams := []string{
-			"https://discoveryprovider.audius.co",
-			"https://discoveryprovider2.audius.co",
-			"https://discoveryprovider3.audius.co",
-		}
-		if os.Getenv("ENV") == "stage" {
-			upstreams = []string{
-				"https://discoveryprovider.staging.audius.co",
-				"https://discoveryprovider2.staging.audius.co",
-				"https://discoveryprovider3.staging.audius.co",
-				"https://discoveryprovider5.staging.audius.co",
-			}
-		}
-
-		app.Use(BalancerForward(upstreams))
-	}
+	app.Use(BalancerForward(config.PythonUpstreams))
 
 	// gracefully handle 404
 	// (this won't get hit so long as above proxy is in place)
@@ -210,6 +214,7 @@ type ApiServer struct {
 
 func (app *ApiServer) home(c *fiber.Ctx) error {
 	return c.JSON(fiber.Map{
+		"env":     config.Cfg.Env,
 		"started": app.started,
 		"uptime":  time.Since(app.started).Truncate(time.Second).String(),
 	})

config/config.go

@@ -7,15 +7,35 @@ import (
 )
 
 type Config struct {
+	Env                string
 	DbUrl              string
 	DelegatePrivateKey string
 	AxiomToken         string
 	AxiomDataset       string
+	PythonUpstreams    []string
 }
 
 var Cfg = Config{
+	Env:                os.Getenv("ENV"),
 	DbUrl:              os.Getenv("discoveryDbUrl"),
 	DelegatePrivateKey: os.Getenv("delegatePrivateKey"),
 	AxiomToken:         os.Getenv("axiomToken"),
 	AxiomDataset:       os.Getenv("axiomDataset"),
 }
+
+func init() {
+	if os.Getenv("ENV") == "stage" {
+		Cfg.PythonUpstreams = []string{
+			"https://discoveryprovider.staging.audius.co",
+			"https://discoveryprovider2.staging.audius.co",
+			"https://discoveryprovider3.staging.audius.co",
+			"https://discoveryprovider5.staging.audius.co",
+		}
+	} else {
+		Cfg.PythonUpstreams = []string{
+			"https://discoveryprovider.audius.co",
+			"https://discoveryprovider2.audius.co",
+			"https://discoveryprovider3.audius.co",
+		}
+	}
+}