mirror of
https://github.com/drmext/MonkeyBusiness.git
synced 2024-11-15 02:57:38 +01:00
92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
from urllib.parse import urlunparse, urlencode
|
|
|
|
import uvicorn
|
|
|
|
from fastapi import FastAPI, Request, Response
|
|
|
|
import config
|
|
import modules
|
|
|
|
from core_common import core_process_request, core_prepare_response, E
|
|
|
|
|
|
def urlpathjoin(parts, sep='/'):
|
|
return sep + sep.join([x.lstrip(sep) for x in parts])
|
|
|
|
|
|
server_address = f"{config.ip}:{config.port}"
|
|
server_services_url = urlunparse(('http', server_address, config.services_prefix, None, None, None))
|
|
keepalive_address = "127.0.0.1"
|
|
|
|
app = FastAPI()
|
|
for router in modules.routers:
|
|
app.include_router(router)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("https://github.com/drmext/MonkeyBusiness")
|
|
print(" __ __ _ ")
|
|
print("| \/ | ___ _ __ | | _____ _ _ ")
|
|
print("| |\/| |/ _ \| '_ \| |/ / _ \ | | |")
|
|
print("| | | | (_) | | | | < __/ |_| |")
|
|
print("|_| |_|\___/|_| |_|_|\_\___|\__, |")
|
|
print(" |___/ ")
|
|
print(" ____ _ ")
|
|
print("| __ ) _ _ ___(_)_ __ ___ ___ ___ ")
|
|
print("| _ \| | | / __| | '_ \ / _ \/ __/ __|")
|
|
print("| |_) | |_| \__ \ | | | | __/\__ \__ \\")
|
|
print("|____/ \__,_|___/_|_| |_|\___||___/___/")
|
|
print()
|
|
print(f"<services>{server_services_url}</services>")
|
|
print("<url_slash __type=\"bool\">1</url_slash>")
|
|
print()
|
|
uvicorn.run("pyeamu:app", host=config.ip, port=config.port, reload=True)
|
|
|
|
|
|
@app.post(urlpathjoin([config.services_prefix, "/{gameinfo}/services/get"]))
|
|
async def services_get(request: Request):
|
|
request_info = await core_process_request(request)
|
|
|
|
services = {}
|
|
|
|
for service in modules.routers:
|
|
model_blacklist = services.get('model_blacklist', [])
|
|
model_whitelist = services.get('model_whitelist', [])
|
|
|
|
if request_info['model'] in model_blacklist:
|
|
continue
|
|
|
|
if model_whitelist and request_info['model'] not in model_whitelist:
|
|
continue
|
|
|
|
k = (service.tags[0] if service.tags else service.prefix).strip('/')
|
|
if k not in services:
|
|
services[k] = urlunparse(('http', server_address, service.prefix, None, None, None))
|
|
|
|
keepalive_params = {
|
|
"pa": keepalive_address,
|
|
"ia": keepalive_address,
|
|
"ga": keepalive_address,
|
|
"ma": keepalive_address,
|
|
"t1": 2,
|
|
"t2": 10,
|
|
}
|
|
services["keepalive"] = urlunparse(('http', keepalive_address, "/keepalive", None, urlencode(keepalive_params), None))
|
|
services["ntp"] = urlunparse(('ntp', "pool.ntp.org", "/", None, None, None))
|
|
services["services"] = urlunparse(('http', server_address, "/core", None, None, None))
|
|
|
|
response = E.response(
|
|
E.services(
|
|
expire=10800,
|
|
mode='operation',
|
|
product_domain=1,
|
|
*[E.item(
|
|
name=k,
|
|
url=services[k]
|
|
) for k in services]
|
|
)
|
|
)
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|