feat(info): add age_restricted and no_embed_allowed keys to video…

… info' closese #321
fent · Aug 2, 2018 · 7fc2df9 · 7fc2df9
1 parent dcb7281
commit 7fc2df9
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 1 deletion.
diff --git a/lib/info.js b/lib/info.js
@@ -125,6 +125,7 @@ function gotConfig(id, options, additional, config, fromEmbed, callback) {
     } else if (info.status === 'fail') {
       if (info.errorcode === '150' && config.args) {
         info = config.args;
+        info.no_embed_allowed = true;
       } else {
         return callback(
           new Error(`Code ${info.errorcode}: ${util.stripHTML(info.reason)}`));
@@ -156,6 +157,7 @@ function gotConfig(id, options, additional, config, fromEmbed, callback) {
 
     // Add additional properties to info.
     Object.assign(info, additional);
+    info.age_restricted = fromEmbed;
 
     if (info.formats.length ||
         config.args.dashmpd || info.dashmpd || info.hlsvp) {

diff --git a/test/files/videos/pJk0p-98Xzc-vevo/expected_info.json b/test/files/videos/pJk0p-98Xzc-vevo/expected_info.json
diff --git a/test/info-test.js b/test/info-test.js
@@ -171,6 +171,7 @@ describe('ytdl.getInfo()', () => {
         assert.ifError(err);
         scope.done();
         assert.equal(info.formats.length, expectedInfo.formats.length);
+        assert.ok(info.age_restricted);
         done();
       });
     });

diff --git a/typings/index.d.ts b/typings/index.d.ts
@@ -182,6 +182,8 @@ declare module 'ytdl-core' {
       description: string;
       related_videos: relatedVideo[]
       video_url: string;
+      no_embed_allowed?: boolean;
+      age_restricted: boolean;
     }
 
     type relatedVideo = {