From 01e9ae1e32b814c380fff692d823eeafb9da65a0 Mon Sep 17 00:00:00 2001
From: hailey <me@haileyok.com>
Date: Sun, 22 Jun 2025 10:10:05 -0700
Subject: [PATCH] fix: allow handle to be used for repo in listRecords (#18)

---
 .gitignore                         |  1 +
 server/handle_repo_list_records.go | 47 ++++++++++++++++++++++++------
 2 files changed, 39 insertions(+), 9 deletions(-)

diff --git a/.gitignore b/.gitignore
index 8626012..efc81de 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@
 /cocoon
 *.key
 *.secret
+.DS_Store
diff --git a/server/handle_repo_list_records.go b/server/handle_repo_list_records.go
index 3c7da6c..4358ec7 100644
--- a/server/handle_repo_list_records.go
+++ b/server/handle_repo_list_records.go
@@ -2,15 +2,23 @@ package server
 
 import (
 	"strconv"
-	"strings"
 
 	"github.com/Azure/go-autorest/autorest/to"
 	"github.com/bluesky-social/indigo/atproto/data"
+	"github.com/bluesky-social/indigo/atproto/syntax"
 	"github.com/haileyok/cocoon/internal/helpers"
 	"github.com/haileyok/cocoon/models"
 	"github.com/labstack/echo/v4"
 )
 
+type ComAtprotoRepoListRecordsRequest struct {
+	Repo       string `query:"repo" validate:"required"`
+	Collection string `query:"collection" validate:"required,atproto-nsid"`
+	Limit      int64  `query:"limit"`
+	Cursor     string `query:"cursor"`
+	Reverse    bool   `query:"reverse"`
+}
+
 type ComAtprotoRepoListRecordsResponse struct {
 	Cursor  *string                               `json:"cursor,omitempty"`
 	Records []ComAtprotoRepoListRecordsRecordItem `json:"records"`
@@ -38,10 +46,22 @@ func getLimitFromContext(e echo.Context, def int) (int, error) {
 }
 
 func (s *Server) handleListRecords(e echo.Context) error {
-	did := e.QueryParam("repo")
-	collection := e.QueryParam("collection")
-	cursor := e.QueryParam("cursor")
-	reverse := e.QueryParam("reverse")
+	var req ComAtprotoRepoListRecordsRequest
+	if err := e.Bind(&req); err != nil {
+		s.logger.Error("could not bind list records request", "error", err)
+		return helpers.ServerError(e, nil)
+	}
+
+	if err := e.Validate(req); err != nil {
+		return helpers.InputError(e, nil)
+	}
+
+	if req.Limit <= 0 {
+		req.Limit = 50
+	} else if req.Limit > 100 {
+		req.Limit = 100
+	}
+
 	limit, err := getLimitFromContext(e, 50)
 	if err != nil {
 		return helpers.InputError(e, nil)
@@ -51,14 +71,23 @@ func (s *Server) handleListRecords(e echo.Context) error {
 	dir := "<"
 	cursorquery := ""
 
-	if strings.ToLower(reverse) == "true" {
+	if req.Reverse {
 		sort = "ASC"
 		dir = ">"
 	}
 
-	params := []any{did, collection}
-	if cursor != "" {
-		params = append(params, cursor)
+	did := req.Repo
+	if _, err := syntax.ParseDID(did); err != nil {
+		actor, err := s.getActorByHandle(req.Repo)
+		if err != nil {
+			return helpers.InputError(e, to.StringPtr("RepoNotFound"))
+		}
+		did = actor.Did
+	}
+
+	params := []any{did, req.Collection}
+	if req.Cursor != "" {
+		params = append(params, req.Cursor)
 		cursorquery = "AND created_at " + dir + " ?"
 	}
 	params = append(params, limit)
-- 
2.51.2