From 430d1449588f1d4b420a5de5dcacd79d48d22563 Mon Sep 17 00:00:00 2001
From: WorldObservationLog <wolc@duck.com>
Date: Tue, 28 May 2024 18:53:41 +0800
Subject: [PATCH] feat: remove mitm feature

---
 src/cmd.py  |  8 --------
 src/mitm.py | 33 ---------------------------------
 2 files changed, 41 deletions(-)
 delete mode 100644 src/mitm.py

diff --git a/src/cmd.py b/src/cmd.py
index 4e9a2b8..645d50f 100644
--- a/src/cmd.py
+++ b/src/cmd.py
@@ -15,7 +15,6 @@ from src.rip import rip_song, rip_album, rip_artist, rip_playlist
 from src.types import GlobalAuthParams
 from src.url import AppleMusicURL, URLType, Song
 from src.utils import get_song_id_from_m3u8
-from src.mitm import start_proxy
 
 
 class NewInteractiveShell:
@@ -50,11 +49,6 @@ class NewInteractiveShell:
                                  default="alac")
         m3u8_parser.add_argument("-f", "--force", default=False, action="store_true")
         subparser.add_parser("exit")
-        mitm_parser = subparser.add_parser("mitm")
-        mitm_parser.add_argument("-c", "--codec",
-                                 choices=["alac", "ec3", "aac", "aac-binaural", "aac-downmix", "ac3"],
-                                 default="alac")
-        mitm_parser.add_argument("-f", "--force", default=False, action="store_true")
 
         logger.remove()
         logger.add(lambda msg: print_formatted_text(ANSI(msg), end=""), colorize=True, level="INFO")
@@ -87,8 +81,6 @@ class NewInteractiveShell:
                 await self.do_download(args.url, args.codec, args.force, args.include)
             case "m3u8":
                 await self.do_m3u8(args.url, args.codec, args.force)
-            case "mitm":
-                await self.do_mitm(args.codec, args.force)
             case "exit":
                 self.loop.stop()
                 sys.exit()
diff --git a/src/mitm.py b/src/mitm.py
deleted file mode 100644
index 3464441..0000000
--- a/src/mitm.py
+++ /dev/null
@@ -1,33 +0,0 @@
-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()