Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: throw error if rewrite is attempted after body is used #11258

Merged
merged 2 commits into from
Jun 14, 2024
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions .changeset/healthy-oranges-report.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
---
'astro': patch
---

Throws an error if Astro.rewrite is used after the request body has been used
ascorbic marked this conversation as resolved.
Show resolved Hide resolved
23 changes: 23 additions & 0 deletions packages/astro/src/core/errors/errors-data.ts
Original file line number Diff line number Diff line change
Expand Up @@ -1266,6 +1266,29 @@ export const ServerOnlyModule = {
message: (name: string) => `The "${name}" module is only available server-side.`,
} satisfies ErrorData;


/**
* @docs
* @description
* `Astro.rewrite()` cannot be used if the request body has already been read. If you need to read the body you should first clone the request. For example:
ascorbic marked this conversation as resolved.
Show resolved Hide resolved
*
* ```js
* const data = await Astro.request.clone().formData();
*
* Astro.rewrite("/target")
* ```
*
* @see
* - [Request.clone()](https://developer.mozilla.org/en-US/docs/Web/API/Request/clone)
* - [Astro.rewrite](https://docs.astro.build/en/reference/configuration-reference/#experimentalrewriting)
*/

export const RewriteWithBodyUsed = {
name: 'RewriteWithBodyUsed',
title: 'Cannot use Astro.rewrite after the request body has been read',
message: 'Astro.rewrite() cannot be used if the request body has already been read. If you need to read the body you should first clone the request.',
ascorbic marked this conversation as resolved.
Show resolved Hide resolved
} satisfies ErrorData;

/**
* @docs
* @kind heading
Expand Down
3 changes: 3 additions & 0 deletions packages/astro/src/core/render-context.ts
Original file line number Diff line number Diff line change
Expand Up @@ -542,6 +542,9 @@ export class RenderContext {
* @param oldRequest The old `Request`
*/
#copyRequest(newUrl: URL, oldRequest: Request): Request {
if(oldRequest.bodyUsed) {
throw new AstroError(AstroErrorData.RewriteWithBodyUsed);
}
return new Request(newUrl, {
method: oldRequest.method,
headers: oldRequest.headers,
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
---
---

<form method="post" action="/post/post-body-used">
<input type="text" name="email" value="[email protected]" />
<input type="submit" />
</form>
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
---
let email = ''
if (Astro.request.method === 'POST') {
try {
const data = await Astro.request.formData();
email = data.get('email');
} catch (e) {
console.log(e)
}
}
---

<h1>Post B</h1>

<h2>{email}</h2>
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
---
let data
if (Astro.request.method === 'POST') {
try {
data = await Astro.request.text();
} catch (e) {
console.log(e)
}
}

return Astro.rewrite('/post/post-b')

---

<h1>Post body used</h1>

<h2>{data}</h2>
27 changes: 27 additions & 0 deletions packages/astro/test/rewrite.test.js
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,21 @@ describe('Dev rewrite, hybrid/server', () => {
assert.match($('h1').text(), /Title/);
assert.match($('p').text(), /some-slug/);
});

it('should display an error if a rewrite is attempted after the body has been consumed', async () => {
const formData = new FormData();
formData.append('email', '[email protected]');

const request = new Request('http://example.com/post/post-body-used', {
method: 'POST',
body: formData,
});
const response = await fixture.fetch('/post/post-body-used', request);
const html = await response.text();
const $ = cheerioLoad(html);

assert.equal($('title').text(), 'RewriteWithBodyUsed');
});
});

describe('Build reroute', () => {
Expand Down Expand Up @@ -272,6 +287,18 @@ describe('SSR rewrite, hybrid/server', () => {
assert.match($('h1').text(), /Title/);
assert.match($('p').text(), /some-slug/);
});

it('should return a 500 if a rewrite is attempted after the body has been read', async () => {
const formData = new FormData();
formData.append('email', '[email protected]');

const request = new Request('http://example.com/post/post-body-used', {
method: 'POST',
body: formData,
});
const response = await app.render(request);
assert.equal(response.status, 500);
});
});

describe('Middleware', () => {
Expand Down
Loading