support referer header

Author: itsMapleLeaf

src/main.ts

@@ -5,6 +5,7 @@ import { asAbsolutePath } from "./as-absolute-path"
 import type { AssetFile } from "./asset-files"
 import { collectAssetFiles, serveAsset } from "./asset-files"
 import "./browser-globals"
+import { serveRemixResponse } from "./serve-remix-response"
 
 const defaultMode = app.isPackaged ? "production" : process.env.NODE_ENV
 
@@ -96,38 +97,6 @@ async function handleRequest(
   )
 }
 
-async function serveRemixResponse(
-  request: Electron.ProtocolRequest,
-  handleRequest: RequestHandler,
-  context: unknown,
-): Promise<Electron.ProtocolResponse> {
-  const init: RequestInit = {
-    method: request.method,
-    headers: request.headers,
-    referrer: request.referrer,
-  }
-
-  if (request.uploadData) {
-    // concat might not be correct but 🤷
-    init.body = Buffer.concat(request.uploadData.map((data) => data.bytes))
-  }
-
-  const remixRequest = new Request(request.url, init)
-  const response = await handleRequest(remixRequest, context)
-
-  const headers: Record<string, string[]> = {}
-  for (const [key, value] of response.headers) {
-    const values = (headers[key] ??= [])
-    values.push(value)
-  }
-
-  return {
-    data: Buffer.from(await response.arrayBuffer()),
-    headers,
-    statusCode: response.status,
-  }
-}
-
 function purgeRequireCache(prefix: string) {
   for (const key in require.cache) {
     if (key.startsWith(prefix)) {

src/serve-remix-response.test.ts

@@ -0,0 +1,26 @@
+import { describe, expect, it } from "vitest"
+import "./browser-globals"
+import { serveRemixResponse } from "./serve-remix-response"
+
+describe("serveRemixResponse", () => {
+  it("sets referer", async () => {
+    const handler = (request: Request) => {
+      expect(request.headers.get("Referer")).toEqual("http://localhost/def")
+
+      // node-fetch 2 doesn't seem to support this property at all,
+      // so neither will i
+      // expect(request.referrer).toEqual("http://localhost/def")
+
+      return Promise.resolve(new Response())
+    }
+
+    const mockRequest: Electron.ProtocolRequest = {
+      method: "POST",
+      url: "http://localhost/abc",
+      headers: {},
+      referrer: "http://localhost/def",
+    }
+
+    await serveRemixResponse(mockRequest, handler, {})
+  })
+})

src/serve-remix-response.ts

@@ -0,0 +1,35 @@
+import type { RequestHandler } from "@remix-run/server-runtime"
+import "./browser-globals"
+
+export async function serveRemixResponse(
+  request: Electron.ProtocolRequest,
+  handleRequest: RequestHandler,
+  context: unknown,
+): Promise<Electron.ProtocolResponse> {
+  const body = request.uploadData
+    ? Buffer.concat(request.uploadData.map((data) => data.bytes))
+    : undefined
+
+  const remixHeaders = new Headers(request.headers)
+  remixHeaders.append("Referer", request.referrer)
+
+  const remixRequest = new Request(request.url, {
+    method: request.method,
+    headers: remixHeaders,
+    body,
+  })
+
+  const response = await handleRequest(remixRequest, context)
+
+  const headers: Record<string, string[]> = {}
+  for (const [key, value] of response.headers) {
+    const values = (headers[key] ??= [])
+    values.push(value)
+  }
+
+  return {
+    data: Buffer.from(await response.arrayBuffer()),
+    headers,
+    statusCode: response.status,
+  }
+}