fix: Add custom header options to gateway methods

PinataCloud · Sep 24, 2024 · d5423a0 · d5423a0
1 parent 3f8b4b8
commit d5423a0
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 8 deletions.
diff --git a/src/core/gateway/createSignedURL.ts b/src/core/gateway/createSignedURL.ts
@@ -71,13 +71,22 @@ export const createSignedURL = async (
 		endpoint = config.endpointUrl;
 	}
 
+	let headers: Record<string, string>;
+
+	if (config.customHeaders && Object.keys(config.customHeaders).length > 0) {
+		headers = { ...config.customHeaders };
+	} else {
+		headers = {
+			"Content-Type": "application/json",
+			Authorization: `Bearer ${config.pinataJwt}`,
+			Source: "sdk/createSignURL",
+		};
+	}
+
 	try {
 		const request = await fetch(`${endpoint}/files/sign`, {
 			method: "POST",
-			headers: {
-				"Content-Type": "application/json",
-				Authorization: `Bearer ${config?.pinataJwt}`,
-			},
+			headers: headers,
 			body: payload,
 		});
 

diff --git a/src/core/gateway/getCid.ts b/src/core/gateway/getCid.ts
@@ -90,12 +90,21 @@ export const getCid = async (
 		endpoint = config.endpointUrl;
 	}
 
+	let headers: Record<string, string>;
+
+	if (config.customHeaders && Object.keys(config.customHeaders).length > 0) {
+		headers = { ...config.customHeaders };
+	} else {
+		headers = {
+			"Content-Type": "application/json",
+			Authorization: `Bearer ${config.pinataJwt}`,
+			Source: "sdk/getCid",
+		};
+	}
+
 	const signedUrlRequest = await fetch(`${endpoint}/files/sign`, {
 		method: "POST",
-		headers: {
-			"Content-Type": "application/json",
-			Authorization: `Bearer ${config?.pinataJwt}`,
-		},
+		headers: headers,
 		body: payload,
 	});