michael/astro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: cookies in redirect

Browse source
This commit is contained in:
Emanuele Stoppa 2023-08-23 11:41:24 +01:00
parent 353dcb23e8
commit 25f448b68e
5 changed files with 19 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
packages/astro/src/core/endpoint/index.ts
View file

@ -48,12 +48,14 @@ export function createAPIContext({
generator: `Astro v${ASTRO_VERSION}`,
props,
redirect(path, status) {
return new Response(null, {
const response = new Response(null, {
status: status || 302,
headers: {
Location: path,
},
});
attachCookiesToResponse(response, context.cookies);
return response;
},
ResponseWithEncoding,
url: new URL(request.url),

4
packages/astro/src/core/render/core.ts
View file

@ -24,12 +24,14 @@ export type RenderPage = {
export async function renderPage({ mod, renderContext, env, cookies }: RenderPage) {
if (routeIsRedirect(renderContext.route)) {
return new Response(null, {
const response = new Response(null, {
status: redirectRouteStatus(renderContext.route, renderContext.request.method),
headers: {
location: redirectRouteGenerate(renderContext.route, renderContext.params),
},
});
attachToResponse(response, cookies);
return response;
}
// Validate the page component before rendering the page

5
packages/astro/test/fixtures/middleware-dev/src/middleware.js vendored
View file

@ -18,14 +18,14 @@ const first = defineMiddleware(async (context, next) => {
return new Response(JSON.stringify(object), {
headers: response.headers,
});
} else if(context.url.pathname === '/clone') {
} else if (context.url.pathname === '/clone') {
const response = await next();
const newResponse = response.clone();
const /** @type {string} */ html = await newResponse.text();
const newhtml = html.replace('<h1>testing</h1>', '<h1>it works</h1>');
return new Response(newhtml, { status: 200, headers: response.headers });
} else {
if(context.url.pathname === '/') {
if (context.url.pathname === '/') {
context.cookies.set('foo', 'bar');
}
@ -38,6 +38,7 @@ const second = defineMiddleware(async (context, next) => {
if (context.request.url.includes('/second')) {
context.locals.name = 'second';
} else if (context.request.url.includes('/redirect')) {
context.cookies.set('redirect', 'cookie redirect set');
return context.redirect('/', 302);
}
return await next();

3
packages/astro/test/fixtures/middleware-dev/src/pages/index.astro vendored
View file

@ -1,5 +1,7 @@
---
const data = Astro.locals;
const redirect = Astro.cookies.get("redirect");
console.log(redirect);
---
<html>
@ -10,5 +12,6 @@ const data = Astro.locals;
<span>Index</span>
<p>{data?.name}</p>
<p>{redirect.value}</p>
</body>
</html>

7
packages/astro/test/middleware.test.js
View file

@ -79,6 +79,13 @@ describe('Middleware in DEV mode', () => {
let html = await res.text();
expect(html).to.contain('<h1>it works</h1>');
});
it('cookies should be preserved when calling a redirect', async () => {
let res = await fixture.fetch('/redirect');
let html = await res.text();
console.log(res.headers.get('set-cookie'));
expect(html).to.contain('<p>cookie redirect set</p>');
});
});
describe('Middleware in PROD mode, SSG', () => {