Recognize live and shorts YouTube URLs (#230)

thibaudgg · Mar 17, 2024 · e1fa9cf · e1fa9cf
1 parent 597dc3c
commit e1fa9cf
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/lib/video_info/providers/youtube.rb b/lib/video_info/providers/youtube.rb
@@ -54,7 +54,7 @@ def thumbnail_maxres
       private
 
       def _url_regex
-        %r{(?:youtube(?:-nocookie)?\.com/(?:[^/]+/.+/|(?:v|e(?:mbed)?)/|
+        %r{(?:youtube(?:-nocookie)?\.com/(?:[^/]+/.+/|(?:v|e(?:mbed)?|live|shorts)/|
            .*[?&]v=)|youtu\.be/)([^"&?/ ]{11})}x
       end
 

diff --git a/spec/lib/video_info/providers/youtube_spec.rb b/spec/lib/video_info/providers/youtube_spec.rb
@@ -17,6 +17,21 @@
         it { is_expected.to be_truthy }
       end
 
+      context "with embed url" do
+        let(:url) { "https://www.youtube.com/embed/JM9NgvjjVng" }
+        it { is_expected.to be_truthy }
+      end
+
+      context "with live url" do
+        let(:url) { "https://www.youtube.com/live/SMhkA07Fbfs" }
+        it { is_expected.to be_truthy }
+      end
+
+      context "with shorts url" do
+        let(:url) { "https://youtube.com/shorts/MOcigdHkzhE" }
+        it { is_expected.to be_truthy }
+      end
+
       context "with other url" do
         let(:url) { "http://google.com/video1" }
         it { is_expected.to be_falsey }