yaclty/LightMirrors
1
0
Fork 0
mirror of https://github.com/NoCLin/LightMirrors synced 2025-08-04 17:22:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: yaclty:e6e2eae60048ead8126fe64d3043a127cc61ca77
Branches Tags
yaclty:master
..
compare: yaclty:2e1ae11ad4826b2fb2bf2ec7109f2690612ea405
Branches Tags
yaclty:master

No commits in common. "e6e2eae60048ead8126fe64d3043a127cc61ca77" and "2e1ae11ad4826b2fb2bf2ec7109f2690612ea405" have entirely different histories.
e6e2eae600 ... 2e1ae11ad4

1 changed files with 9 additions and 7 deletions
Show Stats Expand all files Collapse all files

12
src/mirrorsrun/proxy/direct.py
View File

@ -67,7 +67,7 @@ async def direct_proxy(
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
req_headers = request.headers.mutablecopy() req_headers = request.headers.mutablecopy()
for key in req_headers.keys(): for key in req_headers.keys():
if key in ["host"]: if key not in ["user-agent", "accept", "authorization"]:
del req_headers[key] del req_headers[key]
httpx_req: HttpxRequest = client.build_request( httpx_req: HttpxRequest = client.build_request(
@ -79,16 +79,18 @@ async def direct_proxy(
httpx_req = await pre_process_request(request, httpx_req, pre_process) httpx_req = await pre_process_request(request, httpx_req, pre_process)
upstream_response = await client.send( upstream_response = await client.send(
httpx_req, httpx_req, follow_redirects=follow_redirects
follow_redirects=follow_redirects,
) )
res_headers = {key: value for key, value in upstream_response.headers.items()} res_headers = upstream_response.headers
if request.method != "HEAD":
res_headers.pop("content-length", None) res_headers.pop("content-length", None)
res_headers.pop("content-encoding", None) res_headers.pop("content-encoding", None)
logger.info(
f"proxy {request.url} to {target_url} {upstream_response.status_code}"
)
content = upstream_response.content content = upstream_response.content
response = Response( response = Response(
headers=res_headers, headers=res_headers,