From 736f417e8098d20db6f21a8c6ab61b682c0ac421 Mon Sep 17 00:00:00 2001
From: CasperMcFadden95 <145611964+CasperMcFadden95@users.noreply.github.com>
Date: Sun, 22 Jun 2025 18:16:54 +0000
Subject: [PATCH] Update dzsecurity.py to use _EMBED_REGEX

---
 yt_dlp/extractor/dzsecurity.py | 104 ++++++++++++++++-----------------
 1 file changed, 52 insertions(+), 52 deletions(-)

diff --git a/yt_dlp/extractor/dzsecurity.py b/yt_dlp/extractor/dzsecurity.py
index 8435c41f7..f8c85daa2 100644
--- a/yt_dlp/extractor/dzsecurity.py
+++ b/yt_dlp/extractor/dzsecurity.py
@@ -1,70 +1,70 @@
-import re
 
+from yt_dlp import traverse_obj
 from yt_dlp.extractor.common import InfoExtractor
-from yt_dlp.utils import ExtractorError
+from yt_dlp.utils import smuggle_url, unsmuggle_url
 
 
 class DzsecurityLiveIE(InfoExtractor):
-    _VALID_URL = r'https?://(?:www\.)?(echoroukonline\.com/live(?:-news)?|ennaharonline\.com/live(?:-news)?|elhayat\.dz/%D8%A7%D9%84%D8%A8%D8%AB-%D8%A7%D9%84%D8%AD%D9%8A)'
+    _VALID_URL = r'https?://live\.dzsecurity\.net/live/player/(?P<id>[\w-]+)'
+    _EMBED_REGEX = [rf'<iframe [^>]*\bsrc\s*=\s*[\'"](?P<url>{_VALID_URL})']
 
-    _TESTS = [{
-        'url': 'https://www.ennaharonline.com/live',
-        'info_dict': {
-            'id': 'ennahartv',
-            'title': r're:البث الحي لقناة النهار &#8211; النهار أونلاين',
-            'ext': 'mp4',
-            'live_status': 'is_live',
+    _WEBPAGE_TESTS = [
+        {
+            'url': 'https://www.echoroukonline.com/live',
+            'info_dict': {
+                'id': 'echorouktv',
+                'title': r're:البث الحي لقناة الشروق تي في',
+                'ext': 'mp4',
+                'live_status': 'is_live',
+            },
         },
-        'skip': 'Geo-restricted to Algeria',
-    }, {
-        'url': 'https://www.echoroukonline.com/live',
-        'info_dict': {
-            'id': 'echorouktv',
-            'title': r're:البث الحي لقناة الشروق تي في',
-            'ext': 'mp4',
-            'live_status': 'is_live',
+        {
+            'url': 'https://www.echoroukonline.com/live-news',
+            'info_dict': {
+                'id': 'echorouknews',
+                'title': r're:البث الحي لقناة الشروق نيوز - آخر أخبار الجزائر',
+                'ext': 'mp4',
+                'live_status': 'is_live',
+            },
         },
-    }, {
-        'url': 'https://www.echoroukonline.com/live-news',
-        'info_dict': {
-            'id': 'echorouknews',
-            'title': r're:البث الحي لقناة الشروق نيوز - آخر أخبار الجزائر',
-            'ext': 'mp4',
-            'live_status': 'is_live',
+        {
+            'url': 'https://elhayat.dz/%D8%A7%D9%84%D8%A8%D8%AB-%D8%A7%D9%84%D8%AD%D9%8A/',
+            'info_dict': {
+                'id': 'elhayattv',
+                'title': r're:البث الحي - الحياة',
+                'ext': 'mp4',
+                'live_status': 'is_live',
+            },
         },
-    }, {
-        'url': 'https://elhayat.dz/%D8%A7%D9%84%D8%A8%D8%AB-%D8%A7%D9%84%D8%AD%D9%8A',
-        'info_dict': {
-            'id': 'elhayattv',
-            'title': r're:البث الحي - الحياة',
-            'ext': 'mp4',
-            'live_status': 'is_live',
+        {
+            'url': 'https://www.ennaharonline.com/live',
+            'info_dict': {
+                'id': 'ennahartv',
+                'title': r're:البث الحي لقناة النهار &#8211; النهار أونلاين',
+                'ext': 'mp4',
+                'live_status': 'is_live',
+            },
+            'skip': 'Geo-restricted to Algeria',
         },
-    }]
+    ]
+
+    @classmethod
+    def _extract_embed_urls(cls, url, webpage):
+        for embed_url in super()._extract_embed_urls(url, webpage):
+            yield smuggle_url(embed_url, {'referer': url})
 
     def _real_extract(self, url):
-        webpage = self._download_webpage(url, url, impersonate=True)
+        url, smuggled_data = unsmuggle_url(url, {})
 
-        title = self._html_extract_title(webpage, default='Live Stream')
+        stream_id = self._match_id(url)
 
-        player_url, stream_id = self._html_search_regex(
-            r'https://live\.dzsecurity\.net/live/player/([a-zA-Z0-9_-]+)',
-            webpage,
-            'player URL',
-            group=(0, 1),
-        )
+        title = stream_id
+        referer = smuggled_data['referer']
+        if referer:
+            webpage = self._download_webpage(referer, referer)
+            title = self._html_extract_title(webpage, default=title)
 
-        base_url_match = re.match(r'(https?://[^/]+)', url)
-        if not base_url_match:
-            raise ExtractorError('Failed to extract base URL from input URL')
-
-        base_url = base_url_match.group(1)
-
-        headers = {
-            'Referer': base_url,
-        }
-
-        player_page = self._download_webpage(player_url, player_url, headers=headers)
+        player_page = self._download_webpage(url, stream_id, headers=traverse_obj(smuggled_data, {'Referer': 'referer'}))
 
         m3u8_url = 'https:' + self._search_regex(
             r'src:\s*location\.protocol\s*\+\s*"(//[^"]+\.m3u8\?[^"]+)"',