Merge pull request #1 from bogeychan/main

SaltyAom · web-flow · commit 426208551b88 · 2023-04-25T10:49:35.000+07:00
fix: ignored context headers
diff --git a/bun.lockb b/bun.lockb
diff --git a/example/index.ts b/example/index.ts
@@ -13,7 +13,7 @@ const page = `<!DOCTYPE HTML>
 </html>`
 
 const app = new Elysia()
-    .use(html)
+    .use(html())
     .get('/', () => page)
     .get('/html', ({ html }) => html(page))
     .listen(8080)
diff --git a/package.json b/package.json
@@ -33,13 +33,13 @@
     "release": "npm run build && npm run test && npm publish --access public"
   },
   "peerDependencies": {
-    "elysia": ">= 0.1.0-rc.5"
+    "elysia": ">= 0.4.10"
   },
   "devDependencies": {
     "@types/node": "^18.11.7",
-    "bun-types": "^0.2.2",
+    "bun-types": "^0.5.8",
     "eslint": "^8.26.0",
-    "elysia": "^0.3.0-rc.1",
+    "elysia": "^0.4.10",
     "rimraf": "^3.0.2",
     "typescript": "^4.8.4"
   }
diff --git a/src/index.ts b/src/index.ts
@@ -2,15 +2,24 @@ import { Elysia } from 'elysia'
 
 export const html = () => (app: Elysia) =>
     app
-        .decorate(
-            'html',
-            (value: string) =>
-                new Response(value, {
+        .derive((context) => ({
+            html(value: string) {
+
+                //! error: expect(received).toBe(expected)
+                //! Expected: "text/html"
+                //! Received: "text/html, text/html"
+
+                // context.set.headers['content-type'] = 'text/html';
+                // return new Response(value, { headers: context.set.headers });
+
+                return new Response(value, {
                     headers: {
+                        ...context.set.headers,
                         'content-type': 'text/html'
                     }
                 })
-        )
+            }
+        }))
         .onAfterHandle((context, response) => {
             if (
                 typeof response === 'string' &&
