js: Allow passing credentials option to fetch [#397] (#644)

* js: Allow passing credentials option to fetch [#397]
* fix passing custom headers in case where remote archive is < 16 kB
* clean up `any` usage

js/test/adapter.test.ts

@@ -1,5 +1,5 @@
 import assert from "node:assert";
-import { describe, mock, test } from "node:test";
+import { describe, test } from "node:test";
 import { PMTiles, Protocol } from "../src";
 import { mockServer } from "./utils";
 

js/test/utils.ts

@@ -6,21 +6,27 @@ class MockServer {
   etag?: string;
   numRequests: number;
   lastCache?: string;
+  lastRequestHeaders: Headers | null;
+  lastCredentials?: string;
 
   reset() {
     this.numRequests = 0;
     this.etag = undefined;
+    this.lastRequestHeaders = null;
   }
 
   constructor() {
     this.numRequests = 0;
     this.etag = undefined;
+    this.lastRequestHeaders = null;
     const serverBuffer = fs.readFileSync("test/data/test_fixture_1.pmtiles");
     const server = setupServer(
       http.get(
         "http://localhost:1337/example.pmtiles",
         ({ request, params }) => {
           this.lastCache = request.cache;
+          this.lastRequestHeaders = request.headers;
+          this.lastCredentials = request.credentials;
           this.numRequests++;
           const range = request.headers.get("range")?.substr(6).split("-");
           if (!range) {
@@ -35,7 +41,31 @@ class MockServer {
             headers: { etag: this.etag } as HeadersInit,
           });
         }
-      )
+      ),
+      http.get("http://localhost:1337/small.pmtiles", ({ request }) => {
+        this.lastRequestHeaders = request.headers;
+        this.lastCredentials = request.credentials;
+        this.numRequests++;
+        const range = request.headers.get("range")?.substr(6).split("-");
+        if (!range) {
+          throw new Error("invalid range");
+        }
+        const rangeEnd = +range[1];
+        if (rangeEnd >= serverBuffer.length) {
+          return new HttpResponse(null, {
+            status: 416,
+            headers: {
+              "Content-Range": `bytes */${serverBuffer.length}`,
+            },
+          });
+        }
+        const rangeStart = +range[0];
+        const body = serverBuffer.slice(rangeStart, rangeEnd + 1);
+        return new HttpResponse(body, {
+          status: 206,
+          headers: { etag: this.etag } as HeadersInit,
+        });
+      })
     );
     server.listen({ onUnhandledRequest: "error" });
   }

js/test/v3.test.ts

@@ -6,6 +6,7 @@ import { mockServer } from "./utils";
 import {
   BufferPosition,
   Entry,
+  FetchSource,
   PMTiles,
   RangeResponse,
   SharedPromiseCache,
@@ -407,3 +408,36 @@ describe("pmtiles v3", () => {
     });
   });
 });
+
+describe("FetchSource", () => {
+  test("customHeaders are sent with requests", async () => {
+    mockServer.reset();
+    const source = new FetchSource(
+      "http://localhost:1337/example.pmtiles",
+      new Headers({ "X-Custom-Header": "test-value" }),
+      "include"
+    );
+    await source.getBytes(0, 100);
+    assert.strictEqual(
+      mockServer.lastRequestHeaders?.get("x-custom-header"),
+      "test-value"
+    );
+    assert.strictEqual(mockServer.lastCredentials, "include");
+  });
+
+  test("customHeaders are preserved on 416 retry", async () => {
+    mockServer.reset();
+    const source = new FetchSource(
+      "http://localhost:1337/small.pmtiles",
+      new Headers({ "X-Custom-Header": "retry-value" }),
+      "include"
+    );
+    await source.getBytes(0, 16384);
+    assert.strictEqual(mockServer.numRequests, 2);
+    assert.strictEqual(
+      mockServer.lastRequestHeaders?.get("x-custom-header"),
+      "retry-value",
+      "include"
+    );
+  });
+});