mmcghen
87d4b4ba78
Introduces a DNS server (dnsserver.py) that redirects Nintendo domains, a Flask server (server.py) for serving Gen4/Gen5 Pokémon event files and logging unknown requests, and supporting batch scripts for starting/stopping services. Adds stunnel configuration and certificates for SSL proxying, as well as initial requirements and log handling. Removes obsolete server.log and restructures event file serving and logging for better maintainability.
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
from flask import Flask, request, Response, send_from_directory
|
|
import os, random, datetime, xml.etree.ElementTree as ET
|
|
|
|
app = Flask(__name__)
|
|
|
|
BASE_DIR = os.path.dirname(__file__)
|
|
LOG_DIR = os.path.join(BASE_DIR, "ac_logs")
|
|
EVENTS_GEN4 = os.path.join(BASE_DIR, "wonder_gen4")
|
|
EVENTS_GEN5 = os.path.join(BASE_DIR, "wonder_gen5")
|
|
|
|
os.makedirs(LOG_DIR, exist_ok=True)
|
|
|
|
# ===== Logging helper =====
|
|
def save_raw_request(prefix: str, data: bytes, headers: dict):
|
|
ts = datetime.datetime.utcnow().strftime("%Y%m%dT%H%M%S%fZ")
|
|
filename = f"{prefix}_{ts}.bin"
|
|
with open(os.path.join(LOG_DIR, filename), "wb") as f:
|
|
f.write(data)
|
|
with open(os.path.join(LOG_DIR, filename + ".meta"), "w", encoding="utf-8") as f:
|
|
for k, v in headers.items():
|
|
f.write(f"{k}: {v}\n")
|
|
|
|
# ===== Gen 4: random event =====
|
|
@app.route("/gen4/event", methods=["GET"])
|
|
def gen4_event():
|
|
files = [f for f in os.listdir(EVENTS_GEN4) if f.lower().endswith((".pgt", ".pcd"))]
|
|
if not files:
|
|
return Response("<error>No Gen4 events found</error>", mimetype="application/xml")
|
|
|
|
chosen = random.choice(files)
|
|
print(f"Serving Gen4 event: {chosen}")
|
|
return send_from_directory(EVENTS_GEN4, chosen, mimetype="application/octet-stream")
|
|
|
|
# ===== Gen 5: random event =====
|
|
@app.route("/gen5/event", methods=["GET"])
|
|
def gen5_event():
|
|
files = [f for f in os.listdir(EVENTS_GEN5) if f.lower().endswith(".pgf")]
|
|
if not files:
|
|
return Response("<error>No Gen5 events found</error>", mimetype="application/xml")
|
|
|
|
chosen = random.choice(files)
|
|
print(f"Serving Gen5 event: {chosen}")
|
|
return send_from_directory(EVENTS_GEN5, chosen, mimetype="application/octet-stream")
|
|
|
|
# ===== Catch-all =====
|
|
@app.route('/', defaults={'path': ''}, methods=['GET','POST'])
|
|
@app.route('/<path:path>', methods=['GET','POST'])
|
|
def catch_all(path):
|
|
raw = request.get_data()
|
|
headers = dict(request.headers)
|
|
save_raw_request(f"unknown_{path}", raw, headers)
|
|
print(f"Unhandled request: /{path}")
|
|
return Response("<error>Unhandled path</error>", mimetype="application/xml")
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=8080)
|