From b00163a71c8a3facd0f781a66299b6588e7c69d3 Mon Sep 17 00:00:00 2001
From: Rafael Moraes <50295204+glomatico@users.noreply.github.com>
Date: Fri, 24 Apr 2026 11:18:01 -0300
Subject: [PATCH] Add optional m3u8 wrapper support

---
 gamdl/downloader/song.py |  2 +-
 gamdl/interface/base.py  | 10 +++++++++-
 gamdl/interface/song.py  | 26 ++++++++++++++++++++++++--
 3 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/gamdl/downloader/song.py b/gamdl/downloader/song.py
index fe4a351..ad39424 100644
--- a/gamdl/downloader/song.py
+++ b/gamdl/downloader/song.py
@@ -96,7 +96,7 @@ class AppleMusicSongDownloader:
             staged_path=staged_path,
         )
 
-        if self.base.use_wrapper and not legacy:
+        if self.base.interface.base.use_wrapper and not legacy:
             await self._decrypt_amdecrypt(
                 encrypted_path,
                 staged_path,
diff --git a/gamdl/interface/base.py b/gamdl/interface/base.py
index 282d1d1..6f25bb8 100644
--- a/gamdl/interface/base.py
+++ b/gamdl/interface/base.py
@@ -29,12 +29,16 @@ class AppleMusicBaseInterface:
         itunes_api: ItunesApi,
         cover_format: CoverFormat,
         cover_size: int,
+        use_wrapper: bool,
+        wrapper_m3u8_ip: str,
         cdm: Cdm,
     ) -> None:
         self.apple_music_api = apple_music_api
         self.itunes_api = itunes_api
         self.cover_format = cover_format
         self.cover_size = cover_size
+        self.use_wrapper = use_wrapper
+        self.wrapper_m3u8_ip = wrapper_m3u8_ip
         self.cdm = cdm
 
     @staticmethod
@@ -121,8 +125,10 @@ class AppleMusicBaseInterface:
         apple_music_api: AppleMusicApi,
         cover_format: CoverFormat = CoverFormat.JPG,
         cover_size: int = 1200,
-        itunes_api: ItunesApi | None = None,
+        use_wrapper: bool = False,
+        wrapper_m3u8_ip: str = "127.0.0.1:20020",
         wvd_path: str | None = None,
+        itunes_api: ItunesApi | None = None,
     ):
         itunes_api = itunes_api or await ItunesApi.create(
             storefront=apple_music_api.storefront,
@@ -135,6 +141,8 @@ class AppleMusicBaseInterface:
             itunes_api=itunes_api,
             cover_format=cover_format,
             cover_size=cover_size,
+            use_wrapper=use_wrapper,
+            wrapper_m3u8_ip=wrapper_m3u8_ip,
             cdm=cdm,
         )
         return base
diff --git a/gamdl/interface/song.py b/gamdl/interface/song.py
index 7b2da95..3c839d5 100644
--- a/gamdl/interface/song.py
+++ b/gamdl/interface/song.py
@@ -5,6 +5,7 @@ import json
 import re
 from typing import Callable
 from xml.dom import minidom
+import struct
 from xml.etree import ElementTree
 
 import m3u8
@@ -258,6 +259,25 @@ class AppleMusicSongInterface:
             else:
                 return await self._get_stream_info(song_metadata, codec)
 
+    async def get_wrapper_m3u8(self, adam_id: str) -> str | None:
+        host, port = self.base.wrapper_m3u8_ip.split(":")
+        reader, writer = await asyncio.open_connection(host, port)
+
+        data = struct.pack("B", len(adam_id)) + adam_id.encode()
+        writer.write(data)
+        await writer.drain()
+
+        response = await reader.readuntil(b"\n")
+        m3u8_url = response.decode().strip()
+
+        writer.close()
+        await writer.wait_closed()
+
+        if m3u8_url:
+            return m3u8_url
+
+        return None
+
     async def _get_stream_info(
         self,
         song_metadata: dict,
@@ -272,8 +292,10 @@ class AppleMusicSongInterface:
                 )
             )["data"][0]
 
-        m3u8_master_url = song_metadata["attributes"]["extendedAssetUrls"].get(
-            "enhancedHls"
+        m3u8_master_url = (
+            self.get_wrapper_m3u8(self.base.parse_catalog_media_id(song_metadata))
+            if self.base.use_wrapper
+            else song_metadata["attributes"]["extendedAssetUrls"].get("enhancedHls")
         )
         if not m3u8_master_url:
             return None