From 5fe4198ff2929cab49aa13277715a72507b3ecdb Mon Sep 17 00:00:00 2001
From: hannahhoward <hannah@hannahhoward.net>
Date: Sun, 27 Oct 2024 10:42:38 -0700
Subject: [PATCH 1/2] feat(pdp): add handler to check for piece cid status

---
 pdp/handlers.go        |  3 +++
 pdp/handlers_upload.go | 57 +++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 59 insertions(+), 1 deletion(-)

diff --git a/pdp/handlers.go b/pdp/handlers.go
index 3b81c9ee5..94b6cfc69 100644
--- a/pdp/handlers.go
+++ b/pdp/handlers.go
@@ -104,6 +104,9 @@ func Routes(r *chi.Mux, p *PDPService) {
 	// POST /pdp/piece
 	r.Post(path.Join(PDPRoutePath, "/piece"), p.handlePiecePost)
 
+	// GET /pdp/piece/
+	r.Get(path.Join(PDPRoutePath, "/piece/"), p.handleFindPiece)
+
 	// PUT /pdp/piece/upload/{uploadUUID}
 	r.Put(path.Join(PDPRoutePath, "/piece/upload/{uploadUUID}"), p.handlePieceUpload)
 }
diff --git a/pdp/handlers_upload.go b/pdp/handlers_upload.go
index 597cfe7d7..9be9c93b8 100644
--- a/pdp/handlers_upload.go
+++ b/pdp/handlers_upload.go
@@ -12,6 +12,7 @@ import (
 	"net/http"
 	"os"
 	"path"
+	"strconv"
 
 	"github.com/go-chi/chi/v5"
 	"github.com/google/uuid"
@@ -170,7 +171,7 @@ func (p *PDPService) handlePiecePost(w http.ResponseWriter, r *http.Request) {
 					return false, fmt.Errorf("failed to insert into parked_piece_refs: %w", err)
 				}
 
-				// Create a new 'pdp_piece_uploads' entry pointing to the 'pdp_piecerefs' entry
+				// Create a new 'pdp_piece_uploads' entry pointing to the 'parked_piece_refs' entry
 				uploadUUID = uuid.New()
 				_, err = tx.Exec(`
                 INSERT INTO pdp_piece_uploads (id, service, piece_cid, notify_url, piece_ref, check_hash_codec, check_hash, check_size)
@@ -442,3 +443,57 @@ func (p *PDPService) handlePieceUpload(w http.ResponseWriter, r *http.Request) {
 	// Respond with 204 No Content
 	w.WriteHeader(http.StatusNoContent)
 }
+
+// handle find piece allows one to look up a pdp piece by its original post data as
+// query parameters
+func (p *PDPService) handleFindPiece(w http.ResponseWriter, r *http.Request) {
+	// Verify that the request is authorized using ECDSA JWT
+	serviceID, err := p.verifyJWTToken(r)
+	if err != nil {
+		http.Error(w, "Unauthorized: "+err.Error(), http.StatusUnauthorized)
+		return
+	}
+
+	// Parse query parameters
+
+	sizeString := r.URL.Query().Get("size")
+	size, err := strconv.ParseInt(sizeString, 10, 64)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("errors parsing size: %s", err.Error()), 400)
+		return
+	}
+	req := PieceHash{
+		Name: r.URL.Query().Get("name"),
+		Hash: r.URL.Query().Get("hash"),
+		Size: size,
+	}
+
+	ctx := r.Context()
+
+	pieceCid, havePieceCid, err := req.commp(ctx, p.db)
+	if err != nil {
+		http.Error(w, "Failed to process request: "+err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	// upload either not complete or does not exist
+	if !havePieceCid {
+		http.NotFound(w, r)
+		return
+	}
+	response := struct {
+		Service  string `json:"service"`
+		PieceCID string `json:"piece_cid"`
+	}{
+		Service:  serviceID,
+		PieceCID: pieceCid.String(),
+	}
+
+	// encode response
+	w.Header().Set("Content-Type", "application/json")
+	err = json.NewEncoder(w).Encode(response)
+	if err != nil {
+		http.Error(w, "Failed to write response: "+err.Error(), http.StatusInternalServerError)
+		return
+	}
+}

From f230bf4f71474d0705e143908a5a3197f9f9a8e4 Mon Sep 17 00:00:00 2001
From: hannahhoward <hannah@hannahhoward.net>
Date: Sun, 27 Oct 2024 16:28:48 -0700
Subject: [PATCH 2/2] feat(pdp): add handler to check for upload exists

---
 pdp/handlers_upload.go | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/pdp/handlers_upload.go b/pdp/handlers_upload.go
index 9be9c93b8..fe4cb0f59 100644
--- a/pdp/handlers_upload.go
+++ b/pdp/handlers_upload.go
@@ -448,7 +448,7 @@ func (p *PDPService) handlePieceUpload(w http.ResponseWriter, r *http.Request) {
 // query parameters
 func (p *PDPService) handleFindPiece(w http.ResponseWriter, r *http.Request) {
 	// Verify that the request is authorized using ECDSA JWT
-	serviceID, err := p.verifyJWTToken(r)
+	_, err := p.verifyJWTToken(r)
 	if err != nil {
 		http.Error(w, "Unauthorized: "+err.Error(), http.StatusUnauthorized)
 		return
@@ -481,11 +481,24 @@ func (p *PDPService) handleFindPiece(w http.ResponseWriter, r *http.Request) {
 		http.NotFound(w, r)
 		return
 	}
+
+	// Verify that a 'parked_pieces' entry exists for the given 'piece_cid'
+	var count int
+	err = p.db.QueryRow(ctx, `
+    SELECT count(*) FROM parked_pieces WHERE piece_cid = $1 AND long_term = TRUE AND complete = TRUE
+  `, pieceCid.String()).Scan(&count)
+	if err != nil {
+		http.Error(w, "Database error", http.StatusInternalServerError)
+		return
+	}
+	if count == 0 {
+		http.NotFound(w, r)
+		return
+	}
+
 	response := struct {
-		Service  string `json:"service"`
 		PieceCID string `json:"piece_cid"`
 	}{
-		Service:  serviceID,
 		PieceCID: pieceCid.String(),
 	}