0
Fork 0
mirror of https://github.com/withastro/astro.git synced 2024-12-16 21:46:22 -05:00

fix(assets): Forward headers from the original request to the internal request to the image (#10775)

This commit is contained in:
Erika 2024-04-15 05:06:30 -04:00 committed by GitHub
parent 01cb41763e
commit 0684312145
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 5 deletions

View file

@ -0,0 +1,5 @@
---
"astro": patch
---
Fixes assets endpoint in serverless returning 404 in certain situations where the website might be under a protected route

View file

@ -7,9 +7,12 @@ import { isRemoteAllowed } from '../utils/remotePattern.js';
// @ts-expect-error
import { imageConfig } from 'astro:assets';
async function loadRemoteImage(src: URL) {
async function loadRemoteImage(src: URL, headers: Headers) {
try {
const res = await fetch(src);
const res = await fetch(src, {
// Forward all headers from the original request
headers,
});
if (!res.ok) {
return undefined;
@ -41,15 +44,16 @@ export const GET: APIRoute = async ({ request }) => {
let inputBuffer: ArrayBuffer | undefined = undefined;
const sourceUrl = isRemotePath(transform.src)
const isRemoteImage = isRemotePath(transform.src);
const sourceUrl = isRemoteImage
? new URL(transform.src)
: new URL(transform.src, url.origin);
if (isRemotePath(transform.src) && isRemoteAllowed(transform.src, imageConfig) === false) {
if (isRemoteImage && isRemoteAllowed(transform.src, imageConfig) === false) {
return new Response('Forbidden', { status: 403 });
}
inputBuffer = await loadRemoteImage(sourceUrl);
inputBuffer = await loadRemoteImage(sourceUrl, isRemoteImage ? new Headers() : request.headers);
if (!inputBuffer) {
return new Response('Not Found', { status: 404 });