fix: fetchNoCors and http_request data handling (#664)

This commit is contained in:
Lukas Senionis
2024-08-03 20:24:25 +03:00
committed by GitHub
parent adf5a76392
commit 3e50ce65a7
2 changed files with 17 additions and 11 deletions
+7 -5
View File
@@ -141,10 +141,10 @@ class Utilities:
# Loosely based on https://gist.github.com/mosquito/4dbfacd51e751827cda7ec9761273e95#file-proxy-py
async def http_request(self, req: Request) -> StreamResponse:
if req.headers.get('X-Decky-Auth', '') != helpers.get_csrf_token() and req.query.get('auth', '') != helpers.get_csrf_token():
if req.query['auth'] != helpers.get_csrf_token():
return Response(text='Forbidden', status=403)
url = req.headers["X-Decky-Fetch-URL"] if "X-Decky-Fetch-URL" in req.headers else unquote(req.query.get('fetch_url', ''))
url = unquote(req.query['fetch_url'])
self.logger.info(f"Preparing {req.method} request to {url}")
headers = dict(req.headers)
@@ -180,7 +180,11 @@ class Utilities:
body = await req.read() # TODO can this also be streamed?
async with ClientSession() as web:
# We disable auto-decompress so that the body is completely forwarded to the
# JS engine for it to do the decompression. Otherwise we need need to clear
# the Content-Encoding header in the response headers, however that would
# defeat the point of this proxy.
async with ClientSession(auto_decompress=False) as web:
async with web.request(req.method, url, headers=headers, data=body, ssl=helpers.get_ssl_context()) as web_res:
res = StreamResponse(headers=web_res.headers, status=web_res.status)
if web_res.headers.get('Transfer-Encoding', '').lower() == 'chunked':
@@ -190,8 +194,6 @@ class Utilities:
self.logger.debug(f"Starting stream for {url}")
async for data in web_res.content.iter_any():
await res.write(data)
if data:
await res.drain()
self.logger.debug(f"Finished stream for {url}")
return res
+10 -6
View File
@@ -528,18 +528,22 @@ class PluginLoader extends Logger {
// Same syntax as fetch but only supports the url-based syntax and an object for headers since it's the most common usage pattern
fetchNoCors(input: string, init?: DeckyRequestInit | undefined): Promise<Response> {
const headers: { [name: string]: string } = {
...(init?.headers as { [name: string]: string }),
'X-Decky-Auth': deckyAuthToken,
'X-Decky-Fetch-URL': input,
const { headers: initHeaders = {}, ...restOfInit } = init || {};
const getPrefixedHeaders = () => {
let prefixedInitHeaders: { [name: string]: any } = {};
for (const [key, value] of Object.entries(initHeaders)) {
prefixedInitHeaders[`X-Decky-Header-${key}`] = value;
}
return prefixedInitHeaders;
};
const headers: { [name: string]: string } = getPrefixedHeaders();
if (init?.excludedHeaders) {
headers['X-Decky-Fetch-Excluded-Headers'] = init.excludedHeaders.join(', ');
}
return fetch('http://127.0.0.1:1337/fetch', {
...init,
return fetch(this.getExternalResourceURL(input), {
...restOfInit,
credentials: 'include',
headers,
});