feat: add support for shorts-url (#883)

closes #881
fent · Feb 19, 2021 · 67eb6dd · 67eb6dd
1 parent e14c9da
commit 67eb6dd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/url-utils.js b/lib/url-utils.js
@@ -25,7 +25,7 @@ const validQueryDomains = new Set([
   'music.youtube.com',
   'gaming.youtube.com',
 ]);
-const validPathDomains = /^https?:\/\/(youtu\.be\/|(www\.)?youtube.com\/(embed|v)\/)/;
+const validPathDomains = /^https?:\/\/(youtu\.be\/|(www\.)?youtube.com\/(embed|v|shorts)\/)/;
 exports.getURLVideoID = link => {
   const parsed = new URL(link);
   let id = parsed.searchParams.get('v');

diff --git a/test/url-utils-test.js b/test/url-utils-test.js
@@ -17,6 +17,8 @@ describe('getURLVideoID()', () => {
     assert.strictEqual(id, 'RAW_VIDEOID');
     id = getVideoID('http://youtube.com/embed/RAW_VIDEOID');
     assert.strictEqual(id, 'RAW_VIDEOID');
+    id = getVideoID('http://youtube.com/shorts/RAW_VIDEOID');
+    assert.strictEqual(id, 'RAW_VIDEOID');
     id = getVideoID('https://music.youtube.com/watch?v=RAW_VIDEOID&list=RDAMVMmtLgabce8KQ');
     assert.strictEqual(id, 'RAW_VIDEOID');
     id = getVideoID('https://gaming.youtube.com/watch?v=RAW_VIDEOID');
@@ -55,6 +57,8 @@ describe('getVideoID()', () => {
     assert.strictEqual(id, 'RAW_VIDEOID');
     id = getVideoID('http://youtube.com/embed/RAW_VIDEOID');
     assert.strictEqual(id, 'RAW_VIDEOID');
+    id = getVideoID('http://youtube.com/shorts/RAW_VIDEOID');
+    assert.strictEqual(id, 'RAW_VIDEOID');
     id = getVideoID('_LENGTH_11_');
     assert.strictEqual(id, '_LENGTH_11_');
     assert.throws(() => {