User Button class

LuanRT · Jul 27, 2024 · 051709c · 051709c
1 parent 0f8f92a
commit 051709c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/parser/classes/menus/Menu.ts b/src/parser/classes/menus/Menu.ts
@@ -2,18 +2,19 @@ import { Parser } from '../../index.js';
 import type { ObservedArray } from '../../helpers.js';
 import { YTNode } from '../../helpers.js';
 import type { RawNode } from '../../index.js';
+import Button from '../Button.js';
 
 export default class Menu extends YTNode {
   static type = 'Menu';
 
   items: ObservedArray<YTNode>;
-  top_level_buttons: ObservedArray<YTNode>;
+  top_level_buttons: ObservedArray<Button>;
   label?: string;
 
   constructor(data: RawNode) {
     super();
     this.items = Parser.parseArray(data.items);
-    this.top_level_buttons = Parser.parseArray(data.topLevelButtons);
+    this.top_level_buttons = Parser.parseArray(data.topLevelButtons, Button);
 
     if (Reflect.has(data, 'accessibility') && Reflect.has(data.accessibility, 'accessibilityData')) {
       this.label = data.accessibility.accessibilityData.label;

diff --git a/src/parser/youtube/Library.ts b/src/parser/youtube/Library.ts
@@ -33,7 +33,7 @@ class Library extends Feed<IBrowseResponse> {
   }
 
   async #getAll(shelf: Shelf): Promise<Playlist | History | Feed> {
-    if (!shelf.menu?.as(Menu).hasKey('top_level_buttons'))
+    if (!shelf.menu?.as(Menu).top_level_buttons)
       throw new InnertubeError(`The ${shelf.title.text} shelf doesn't have more items`);
 
     const button = shelf.menu.as(Menu).top_level_buttons.firstOfType(Button);