feat: mitm download

src/mitm.py

@@ -0,0 +1,33 @@
+import plistlib
+
+import mitmproxy.http
+from mitmproxy import options
+from mitmproxy.tools import dump
+from loguru import logger
+
+
+class RequestHandler:
+    def __init__(self, callback):
+        self.callback = callback
+
+    def response(self, flow: mitmproxy.http.HTTPFlow):
+        if flow.request.host == "play.itunes.apple.com" and flow.request.path == "/WebObjects/MZPlay.woa/wa/subPlaybackDispatch":
+            data = plistlib.loads(flow.response.content)
+            m3u8 = data["songList"][0]["hls-playlist-url"]
+            flow.response.status_code = 500
+            self.callback(m3u8)
+
+
+async def start_proxy(host, port, callback):
+    opts = options.Options(listen_host=host, listen_port=port, mode=["socks5"])
+
+    master = dump.DumpMaster(
+        opts,
+        with_termlog=False,
+        with_dumper=False,
+    )
+    master.addons.add(RequestHandler(callback))
+
+    logger.info(f"Mitmproxy started at socks5://{host}:{port}")
+
+    await master.run()