mirror of
https://github.com/PaiGramTeam/EASY_PROXY.git
synced 2024-11-30 02:43:58 +00:00
feat: support proxy browser
This commit is contained in:
parent
83638b0323
commit
a9d5a76781
@ -1 +1,4 @@
|
||||
proxy = "socks5://127.0.0.1:7676"
|
||||
cookies = ""
|
||||
old_uid = 0
|
||||
uid = 0
|
||||
|
52
proxy.py
Normal file
52
proxy.py
Normal file
@ -0,0 +1,52 @@
|
||||
from typing import Dict
|
||||
|
||||
from fastapi import FastAPI
|
||||
from starlette.requests import Request
|
||||
from starlette.responses import Response
|
||||
from env import proxy, cookies, old_uid, uid
|
||||
from main import req_client
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
|
||||
def rewrite_headers(method: str, old_headers: Dict[str, str]) -> Dict[str, str]:
|
||||
remove_keys = ["host", "cookie"]
|
||||
headers = {}
|
||||
for k, v in old_headers.items():
|
||||
if k.lower() not in remove_keys:
|
||||
headers[k] = v
|
||||
if method not in ["OPTIONS"]:
|
||||
headers["cookie"] = cookies
|
||||
headers["x-rpc-language"] = "zh-cn"
|
||||
headers["x-rpc-lang"] = "zh-cn"
|
||||
return headers
|
||||
|
||||
|
||||
async def get_proxy(req: Request) -> Response:
|
||||
path = req.path_params.get("path")
|
||||
if not path:
|
||||
return Response(status_code=400, content="path is required")
|
||||
host = req.headers.get("host")
|
||||
query = str(req.query_params).replace(str(old_uid), str(uid))
|
||||
method = req.method
|
||||
headers = rewrite_headers(method, dict(req.headers))
|
||||
try:
|
||||
body = await req.body()
|
||||
except Exception as e:
|
||||
return Response(status_code=400, content=f"get request body info error: {e}")
|
||||
q = "?" + query if query else ""
|
||||
target_url = f"https://{host}/" + path + q
|
||||
return await req_client(method, target_url, headers, body, proxy)
|
||||
|
||||
|
||||
app.add_route(
|
||||
"/{path:path}",
|
||||
get_proxy,
|
||||
methods=["OPTIONS", "HEAD", "GET", "POST", "PUT", "PATCH", "DELETE"],
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
|
||||
uvicorn.run(app, host="0.0.0.0", port=5677)
|
Loading…
Reference in New Issue
Block a user