Reset HTTP request body after reading the stream during webhook validation (#64)

Author: shonun1

client_test.go

@@ -1744,6 +1744,11 @@ func TestValidateWebhook(t *testing.T) {
 	isValid, err := replicate.ValidateWebhookRequest(req, testSecret)
 	require.NoError(t, err)
 	assert.True(t, isValid)
+
+	// Ensure that the request body is available after validation
+	bodyBytes, err := io.ReadAll(req.Body)
+	require.NoError(t, err)
+	assert.Equal(t, body, string(bodyBytes))
 }
 
 func TestGetDeployment(t *testing.T) {

webhook.go

@@ -1,6 +1,7 @@
 package replicate
 
 import (
+	"bytes"
 	"context"
 	"crypto/hmac"
 	"crypto/sha256"
@@ -80,6 +81,9 @@ func ValidateWebhookRequest(req *http.Request, secret WebhookSigningSecret) (boo
 	if err != nil {
 		return false, fmt.Errorf("failed to read request body: %w", err)
 	}
+	defer req.Body.Close()
+
+	req.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
 	body := string(bodyBytes)
 
 	signedContent := fmt.Sprintf("%s.%s.%s", id, timestamp, body)