From 1b441f64ef6edcccaafd94f532debc56581d45d7 Mon Sep 17 00:00:00 2001
From: AirBorne04 <>
Date: Thu, 13 Oct 2022 07:32:29 +0200
Subject: [PATCH] in dev server, convert a simple endpoint result into a
 response object

---
 .../src/vite-plugin-astro-server/index.ts     | 27 +++++++------------
 1 file changed, 9 insertions(+), 18 deletions(-)

diff --git a/packages/astro/src/vite-plugin-astro-server/index.ts b/packages/astro/src/vite-plugin-astro-server/index.ts
index ed47f3600af2..049dc4615a63 100644
--- a/packages/astro/src/vite-plugin-astro-server/index.ts
+++ b/packages/astro/src/vite-plugin-astro-server/index.ts
@@ -5,7 +5,7 @@ import type { AstroSettings, ManifestData } from '../@types/astro';
 import { DevelopmentEnvironment, SSROptions } from '../core/render/dev/index';
 
 import { Readable } from 'stream';
-import { getSetCookiesFromResponse } from '../core/cookies/index.js';
+import { attachToResponse, getSetCookiesFromResponse } from '../core/cookies/index.js';
 import { call as callEndpoint } from '../core/endpoint/dev/index.js';
 import {
 	collectErrorMetadata,
@@ -352,7 +352,6 @@ async function handleRoute(
 	// Route successfully matched! Render it.
 	if (route.type === 'endpoint') {
 		const result = await callEndpoint(options);
-    
 		if (result.type === 'response') {
 			if (result.response.headers.get('X-Astro-Response') === 'Not-Found') {
 				const fourOhFourRoute = await matchRoute('/404', env, manifest);
@@ -379,22 +378,14 @@ async function handleRoute(
 			if (computedMimeType) {
 				contentType = computedMimeType;
 			}
-
-      const header: any = {
-        'Content-Type': `${contentType};charset=utf-8`
-      };
-
-			let cookieHeader = [];
-			for (let cok of result.cookies.headers()) {
-				cookieHeader.push(cok);
-			}
-
-      if (cookieHeader.length > 0) {
-        header['Set-Cookie'] = cookieHeader;
-      }
-
-			res.writeHead(200, header);
-			res.end(result.body);
+      const response = new Response(result.body, {
+        status: 200,
+        headers: {
+          'Content-Type': `${contentType};charset=utf-8`,
+        },
+      });
+      attachToResponse(response, result.cookies);
+      await writeWebResponse(res, response);
 		}
 	} else {
 		const result = await renderPage(options);