From e1dc38319cb5974496b9877958d491cec157a3a0 Mon Sep 17 00:00:00 2001
From: Aaron Tan <70739609+aaron-tan@users.noreply.github.com>
Date: Wed, 16 Aug 2023 16:24:46 +1000
Subject: [PATCH] [erocast] Update erocast

Build on code from PR #31631.
- Add playlist info extractor to download playlists
- Refactor code from suggestions by @dirkf
---
 youtube_dl/extractor/erocast.py    | 99 ++++++++++++++++++++++++++++++
 youtube_dl/extractor/extractors.py |  4 ++
 2 files changed, 103 insertions(+)
 create mode 100644 youtube_dl/extractor/erocast.py

diff --git a/youtube_dl/extractor/erocast.py b/youtube_dl/extractor/erocast.py
new file mode 100644
index 000000000..5c1e7ec8c
--- /dev/null
+++ b/youtube_dl/extractor/erocast.py
@@ -0,0 +1,99 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+
+from ..utils import traverse_obj
+
+import re
+
+
+class ErocastIE(InfoExtractor):
+    IE_NAME = 'erocast'
+    _VALID_URL = r'https?://(?:www\.)?erocast\.me/track/(?P<id>[0-9]+)/(?P<display_id>[0-9a-zA-Z_-]+)'
+    _TESTS = [{
+        'url': 'https://erocast.me/track/6508/piano-sample-by-ytdl',
+        'md5': '6764726b2d19161e93c9cf3a9a69800a',
+        'info_dict': {
+            'id': '6508',
+            'ext': 'mp4',
+            'title': 'Piano sample by ytdl',
+            'uploader': 'ytdl',
+        }
+    }, {
+        'url': 'https://erocast.me/track/4254/intimate-morning-with-your-wife',
+        'md5': '45b06c21cf93612dd72a3d764c0bb362',
+        'info_dict': {
+            'id': '4254',
+            'ext': 'mp4',
+            'title': 'Intimate morning  with your wife',
+            'uploader': 'ZLOY_ASMR',
+        }
+    }, {
+        'url': 'https://erocast.me/playlist/2278/rough-hot-and-dirty',
+        'only_matching': True
+    }]
+
+    def _real_extract(self, url):
+        video_id = self._match_id(url)
+        webpage = self._download_webpage(url, video_id)
+
+        # The Song data is in a script tag with the following format:
+        # (see https://github.com/ytdl-org/youtube-dl/issues/31203#issuecomment-1259867716)
+        searchPattern = r'<script\b[^>]*>var\s+song_data_%s\s*=\s*(.+?)</script>' % re.escape(video_id)
+        jsonString = self._html_search_regex(searchPattern, webpage, 'song_data')
+
+        # The data is in JSON format, so we convert the JSON String to a python object
+        # and read the data from this json object
+        jsonObject = self._parse_json(jsonString, None, fatal=False)
+        audio_url = jsonObject['stream_url']
+        title = jsonObject['title']
+        user_name = traverse_obj(jsonObject, ('user', 'name'), expected_type=lambda x: x.strip() or None)
+
+        # The audio url is a m3u8 playlist, so we extract the audio url from this playlist
+        formats = self._extract_m3u8_formats(
+            audio_url, video_id, 'mp4', 'm3u8_native',
+            m3u8_id='hls', fatal=False)
+        self._sort_formats(formats)
+
+        return {
+            'id': video_id,
+            'formats': formats,
+            'title': title,
+            'uploader': user_name,
+        }
+
+
+class ErocastPlaylistIE(InfoExtractor):
+    _VALID_URL = r'https?://(?:www\.)?erocast\.me/playlist/(?P<playlist_id>[0-9]+)/(?P<playlist_title>[0-9a-zA-Z_-]+)'
+    _TESTS = [{
+        'url': 'https://erocast.me/playlist/2278/rough-hot-and-dirty',
+        'info_dict': {
+            'id': '2278',
+            'title': 'rough-hot-and-dirty',
+        },
+        'playlist_mincount': 3,
+    }, {
+        'url': 'https://erocast.me/playlist/1841/dilf',
+        'info_dict': {
+            'id': '1841',
+            'title': 'dilf',
+        },
+        'playlist_mincount': 7,
+    }]
+
+    def _real_extract(self, url):
+        mobj = re.match(self._VALID_URL, url)
+        playlist_id = mobj.group('playlist_id')
+        title = mobj.group('playlist_title')
+
+        webpage = self._download_webpage(url, playlist_id)
+
+        # Get URL for each track in the playlist
+        entries = [self.url_result(
+            'https://erocast.me/track/' + mobj.group('song_id') + '/' + mobj.group('track_url'),
+            ie=ErocastIE.ie_key(), video_id=mobj.group('song_id'))
+            for mobj in re.finditer(
+                r'<script\b[^>]*>var\s+song_data_(?P<song_id>[0-9]+)\s*=\s*.+?\\/track\\/[0-9]+\\/(?P<track_url>.+)\"\,\"stream_url.+?</script>', webpage)]
+
+        return self.playlist_result(entries, playlist_id, title)
diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
index cb39876c2..be85cdcd6 100644
--- a/youtube_dl/extractor/extractors.py
+++ b/youtube_dl/extractor/extractors.py
@@ -358,6 +358,10 @@ from .elpais import ElPaisIE
 from .embedly import EmbedlyIE
 from .engadget import EngadgetIE
 from .eporner import EpornerIE
+from .erocast import (
+    ErocastIE,
+    ErocastPlaylistIE,
+)
 from .eroprofile import EroProfileIE
 from .escapist import EscapistIE
 from .espn import (