fix: Use status code specified in context

example/index.tsx

@@ -71,3 +71,7 @@ app.handle(new Request('http://localhost:8080/'))
 app.handle(new Request('http://localhost:8080/'))
 	.then((x) => x.headers.toJSON())
 	.then(console.log)
+
+	app.handle(new Request('http://localhost:8080/'))
+	.then((x) => x.status)
+	.then(console.log)

example/statuscode.tsx

@@ -0,0 +1,16 @@
+import { Elysia } from 'elysia'
+import { html } from '../src'
+
+const app = new Elysia()
+	.use(html({ autoDetect: true }))
+	.get('/a', ({ html, set }) => {
+        set.status = 'Forbidden'
+        return html(`<h1>Forbidden!</h1>`)
+    })
+	.compile()
+
+console.log(app.routes[0]?.composed?.toString())
+
+app.handle(new Request('http://localhost:8080/a'))
+	.then((x) => x.status)
+	.then(console.log)

package.json

@@ -52,13 +52,13 @@
     "format": "prettier --write ."
   },
   "peerDependencies": {
-    "elysia": ">= 1.0.2"
+    "elysia": ">= 1.0.6"
   },
   "devDependencies": {
     "@elysiajs/stream": "^0.7.2",
     "@types/bun": "^1.0.4",
     "@types/node": "^20.7.2",
-    "elysia": "1.0.2",
+    "elysia": "1.0.6",
     "eslint": "^8.50.0",
     "rimraf": "^5.0.5",
     "typescript": "^5.2.2"

src/handler.ts

@@ -5,8 +5,10 @@ import { isHtml, isTagHtml } from './utils'
 export function handleHtml(
 	value: string | Readable | Promise<string | Readable>,
 	options: HtmlOptions,
-	hasContentType: boolean
+	hasContentType: boolean,
+	status?: number
 ): Promise<Response | string> | Response | string {
+
 	// Only use promises if value is a promise itself
 	if (value instanceof Promise) {
 		return value.then((v) => handleHtml(v, options, hasContentType))
@@ -24,9 +26,7 @@ export function handleHtml(
 
 		return new Response(
 			value,
-			hasContentType
-				? undefined
-				: { headers: { 'content-type': options.contentType! } }
+			initValue(options, hasContentType, status)
 		)
 	}
 
@@ -60,8 +60,13 @@ export function handleHtml(
 
 	return new Response(
 		stream as any,
-		hasContentType
-			? undefined
-			: { headers: { 'content-type': options.contentType! } }
+		initValue(options, hasContentType, status)
 	)
 }
+
+
+function initValue(options: HtmlOptions, hasContentType: boolean, status?: number) {
+	return hasContentType
+		? { status: status?? 200 }
+		: { headers: { 'content-type': options.contentType! }, status: status?? 200 }
+}

src/html.ts

@@ -1,4 +1,4 @@
-import { Elysia } from 'elysia'
+import { Elysia, StatusMap } from 'elysia'
 import { Readable } from 'node:stream'
 import { renderToStream } from '@kitajs/html/suspense'
 
@@ -17,22 +17,26 @@ export function html(options: HtmlOptions = {}) {
 		name: '@elysiajs/html',
 		seed: options
 	}).derive({ as: 'global' }, ({ set }) => {
+
 		return {
 			html(
 				value: Readable | JSX.Element
 			): Promise<Response | string> | Response | string {
-				return handleHtml(value, options, 'content-type' in set.headers)
+				const status = typeof set.status === 'string' ? StatusMap[set.status] : set.status
+				return handleHtml(value, options, 'content-type' in set.headers, status)
 			},
 			stream<A = any>(
 				value: (this: void, arg: A & { id: number }) => JSX.Element,
 				args: A
 			) {
+				const status = typeof set.status === 'string' ? StatusMap[set.status] : set.status
 				return handleHtml(
 					renderToStream((id) =>
 						(value as Function)({ ...args, id })
 					),
 					options,
-					'content-type' in set.headers
+					'content-type' in set.headers,
+					status
 				)
 			}
 		}
@@ -48,10 +52,13 @@ export function html(options: HtmlOptions = {}) {
 					// @kitajs/html stream
 					(value instanceof Readable && 'rid' in value)
 				) {
+					const status = typeof set.status === 'string' ? StatusMap[set.status] : set.status
+
 					const response = await handleHtml(
 						value,
 						options,
-						'content-type' in set.headers
+						'content-type' in set.headers,
+						status
 					)
 
 					if (response instanceof Response) return response

test/index.test.ts

@@ -69,6 +69,16 @@ describe('HTML', () => {
 			'text/html; charset=utf8'
 		)
 	})
+
+	it('returns user provided status code', async () => {
+		const app = new Elysia().use(html()).get('/', ({ html, set }) => {
+			set.status = 404
+			return html('<h1>Not Found</h1>')
+		})
+
+		const res = await app.handle(req('/'))
+		expect(res.status).toBe(404)
+	})
 })
 
 describe('HTML vs No html - header', () => {