refactor: split out response header manager

Author: JacobLinCool

src/cloudflare-worker/handler.ts

@@ -10,6 +10,7 @@ import {
 } from "../core";
 import { Cache } from "./cache";
 import demo from "./demo";
+import Header from "./headers";
 import { booleanize, normalize } from "./utils";
 
 const router = Router();
@@ -130,14 +131,10 @@ async function generate(config: Record<string, string>): Promise<Response> {
     const cache_header =
         `max-age=${cache_time}` + (cache_time <= 0 ? ", no-store, no-cache" : ", public");
 
-    return new Response(await generator.generate(sanitized), {
-        headers: {
-            "Content-Type": "image/svg+xml",
-            "Access-Control-Allow-Origin": "*",
-            "Access-Control-Allow-Credentials": "true",
-            "Cache-Control": cache_header,
-        },
-    });
+    const headers = new Header().add("cors", "svg");
+    headers.set("cache-control", cache_header);
+
+    return new Response(await generator.generate(sanitized), { headers });
 }
 
 // handle path variable
@@ -153,11 +150,7 @@ router.get(
 router.get("*", async ({ query }: { query: Record<string, string> }) => {
     if (!query.username) {
         return new Response(demo, {
-            headers: {
-                "Content-Type": "text/html",
-                "Access-Control-Allow-Origin": "*",
-                "Access-Control-Allow-Credentials": "true",
-            },
+            headers: new Header().add("cors", "html"),
         });
     }
 

src/cloudflare-worker/headers.ts

@@ -0,0 +1,36 @@
+const source = {
+    cors: {
+        "Access-Control-Allow-Origin": "*",
+        "Access-Control-Allow-Methods": "*",
+        "Access-Control-Allow-Headers": "*",
+        "Access-Control-Allow-Credentials": "true",
+    },
+    json: {
+        "Content-Type": "application/json",
+    },
+    html: {
+        "Content-Type": "text/html",
+    },
+    text: {
+        "Content-Type": "text/plain",
+    },
+    svg: {
+        "Content-Type": "image/svg+xml",
+    },
+};
+
+export default class Header extends Headers {
+    constructor(headers?: Headers) {
+        super(headers);
+    }
+
+    add(...types: (keyof typeof source)[]): Headers {
+        for (const type of types) {
+            for (const [key, value] of Object.entries(source[type])) {
+                this.set(key, value);
+            }
+        }
+
+        return this;
+    }
+}

src/cloudflare-worker/index.ts

@@ -1,4 +1,5 @@
 import { handle } from "./handler";
+import Header from "./headers";
 
 export default {
     async fetch(
@@ -12,7 +13,7 @@ export default {
             console.error(err);
             return new Response((err as Error).message, {
                 status: 500,
-                headers: { "Content-Type": "text/plain" },
+                headers: new Header().add("cors", "text"),
             });
         }
     },