Merge pull request #50 from AudiusProject/mjp-tracks-by-permalink

Add GetTracksByPermalink

Author: rickyrombo

api/dbv1/get_track_ids_by_permalink.sql.go

@@ -0,0 +1,8 @@
+-- name: GetTrackIdsByPermalink :many
+SELECT track_id
+FROM track_routes
+JOIN users ON users.user_id = track_routes.owner_id
+WHERE handle_lc = ANY(@handles::text[])
+    AND slug = ANY(@slugs::text[])
+    AND CONCAT(handle_lc, '/', slug) = ANY(@permalinks::text[]) -- in case of conflicts across users
+;

api/dbv1/queries/get_track_ids_by_permalink.sql

@@ -265,6 +265,14 @@ func decodeIdList(c *fiber.Ctx) []int32 {
 	return ids
 }
 
+func queryMutli(c *fiber.Ctx, key string) []string {
+	var values []string
+	for _, v := range c.Request().URI().QueryArgs().PeekMulti(key) {
+		values = append(values, string(v))
+	}
+	return values
+}
+
 func (app *ApiServer) resolveUserHandleToId(handle string) (int32, error) {
 	if hit, ok := app.resolveHandleCache.Get(handle); ok {
 		return hit, nil

api/server.go

@@ -77,6 +77,7 @@ func TestMain(m *testing.M) {
 	insertFixtures("associated_wallets", connectedWalletsBaseRow, "testdata/connected_wallets_fixtures.csv")
 	insertFixtures("aggregate_user_tips", aggregateUserTipsBaseRow, "testdata/aggregate_user_tips_fixtures.csv")
 	insertFixtures("usdc_purchases", usdcPurchaseBaseRow, "testdata/usdc_purchases_fixtures.csv")
+	insertFixtures("track_routes", map[string]any{}, "testdata/track_routes_fixtures.csv")
 
 	// index to es / os
 

api/server_test.go

@@ -9,3 +9,4 @@ track_id,genre,owner_id,title,is_unlisted,stream_conditions,download_conditions
 302,Electronic,3,Tip Gated Stream,f,"{""tip_user_id"": 3}","{""tip_user_id"": 3}"
 303,Electronic,3,Pay Gated Stream,f,"{""usdc_purchase"": {""price"": 135, ""splits"": [{""user_id"": 3, ""percentage"": 100.0}]}}",
 400,Folk,5,Trending Month Folk,f,,
+500,Experimental,6,track by permalink,f,,

api/testdata/track_fixtures.csv

@@ -0,0 +1,2 @@
+"slug","title_slug","collision_id","owner_id","track_id","is_current","blockhash","blocknumber","txhash"
+"track-by-permalink","track-by-permalink",0,6,500,TRUE,"0x24f1465e4bd8803b79b2cbcfad695363a640623053563ffd20fdeaf4656a7b89",23200013,"0x76dcea2bf98e56f683b95a071f5c87405a864eea87243920e60cbc7b96ad565b"

api/testdata/track_routes_fixtures.csv

@@ -4,4 +4,5 @@ user_id,handle,handle_lc,is_deactivated,wallet,playlist_library
 3,someseller,someseller,f,0x234567890abcdef1,
 4,accesstester,accesstester,f,0x34567890abcdef12,
 5,guyintrending,guyintrending,f,0x34567890abcdef13,
+6,TracksByPermalink,tracksbypermalink,f,0xffffffffff,
 91,badguy,badguy,t,0x4567890abcdef123,

api/testdata/user_fixtures.csv

@@ -1,6 +1,8 @@
 package api
 
 import (
+	"strings"
+
 	"bridgerton.audius.co/api/dbv1"
 	"bridgerton.audius.co/trashid"
 	"github.com/gofiber/fiber/v2"
@@ -10,6 +12,31 @@ func (app *ApiServer) v1Tracks(c *fiber.Ctx) error {
 	myId, _ := trashid.DecodeHashId(c.Query("user_id"))
 	ids := decodeIdList(c)
 
+	// Add permalink ID mappings
+	permalinks := queryMutli(c, "permalink")
+	if len(permalinks) > 0 {
+		handles := make([]string, len(permalinks))
+		slugs := make([]string, len(permalinks))
+		for i, permalink := range permalinks {
+			permalinks[i] = strings.ToLower(strings.TrimPrefix(permalink, "/"))
+			splits := strings.Split(permalinks[i], "/")
+			if len(splits) != 2 {
+				return fiber.NewError(fiber.StatusBadRequest, "Invalid permalink: "+permalink)
+			}
+			handles[i] = splits[0]
+			slugs[i] = splits[1]
+		}
+		newIds, err := app.queries.GetTrackIdsByPermalink(c.Context(), dbv1.GetTrackIdsByPermalinkParams{
+			Handles:    handles,
+			Slugs:      slugs,
+			Permalinks: permalinks,
+		})
+		if err != nil {
+			return err
+		}
+		ids = append(ids, newIds...)
+	}
+
 	tracks, err := app.queries.FullTracks(c.Context(), dbv1.GetTracksParams{
 		MyID: int32(myId),
 		Ids:  ids,

api/v1_tracks.go

@@ -0,0 +1,22 @@
+package api
+
+import (
+	"testing"
+
+	"bridgerton.audius.co/api/dbv1"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetTracksByPermalink(t *testing.T) {
+	var tracksResponse struct {
+		Data []dbv1.FullTrack
+	}
+
+	status, body := testGet(t, "/v1/full/tracks?permalink=/TracksByPermalink/track-by-permalink", &tracksResponse)
+	assert.Equal(t, 200, status)
+
+	jsonAssert(t, body, map[string]string{
+		"data.0.id":    "eYake",
+		"data.0.title": "track by permalink",
+	})
+}