feat(parser): Add play_all_button to Shelf (#345)

LuanRT · Mar 12, 2023 · 427db5b · 427db5b
1 parent 2b29244
commit 427db5b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/parser/classes/Shelf.ts b/src/parser/classes/Shelf.ts
@@ -2,6 +2,7 @@ import Text from './misc/Text.js';
 import Parser from '../index.js';
 import NavigationEndpoint from './NavigationEndpoint.js';
 import { YTNode } from '../helpers.js';
+import Button from './Button.js';
 
 class Shelf extends YTNode {
   static type = 'Shelf';
@@ -11,6 +12,7 @@ class Shelf extends YTNode {
   content: YTNode | null;
   icon_type?: string;
   menu?: YTNode | null;
+  play_all_button?: Button | null;
 
   constructor(data: any) {
     super();
@@ -29,6 +31,10 @@ class Shelf extends YTNode {
     if (data.menu) {
       this.menu = Parser.parseItem(data.menu);
     }
+
+    if (data.playAllButton) {
+      this.play_all_button = Parser.parseItem<Button>(data.playAllButton);
+    }
   }
 }