From 051709c3c1dc49e551c8b142757d65db515742ac Mon Sep 17 00:00:00 2001
From: Dave Nicolson <david.nicolson@gmail.com>
Date: Sat, 27 Jul 2024 22:13:46 +0200
Subject: [PATCH] User Button class

---
 src/parser/classes/menus/Menu.ts | 5 +++--
 src/parser/youtube/Library.ts    | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/parser/classes/menus/Menu.ts b/src/parser/classes/menus/Menu.ts
index e8daaa8d1..8cc0e95ab 100644
--- 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
index 1f0f3e0d5..ce6536863 100644
--- 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);