From 54df8fc5b25a0c9d159a23f5cf9f1eabce1e6294 Mon Sep 17 00:00:00 2001
From: pukkandan <pukkandan.ytdlp@gmail.com>
Date: Sat, 3 Apr 2021 00:23:46 +0530
Subject: [ViewSource] Add extractor to handle `view-source:`

---
 yt_dlp/extractor/commonprotocols.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'yt_dlp/extractor/commonprotocols.py')

diff --git a/yt_dlp/extractor/commonprotocols.py b/yt_dlp/extractor/commonprotocols.py
index d98331a4e..1fc7779b8 100644
--- a/yt_dlp/extractor/commonprotocols.py
+++ b/yt_dlp/extractor/commonprotocols.py
@@ -1,5 +1,7 @@
 from __future__ import unicode_literals
 
+import re
+
 from .common import InfoExtractor
 from ..compat import (
     compat_urlparse,
@@ -58,3 +60,16 @@ class MmsIE(InfoExtractor):
             'title': title,
             'url': url,
         }
+
+
+class ViewSourceIE(InfoExtractor):
+    IE_DESC = False
+    _VALID_URL = r'view-source:(?P<url>.+)'
+
+    _TEST = {
+        'url': 'view-source:https://www.youtube.com/watch?v=BaW_jenozKc',
+        'only_matching': True
+    }
+
+    def _real_extract(self, url):
+        return self.url_result(re.match(self._VALID_URL, url).group('url'))
-- 
cgit v1.2.3