From 182bf42d3cdf5148c667e958cd4d4eaa091a6d68 Mon Sep 17 00:00:00 2001
From: absidue <48293849+absidue@users.noreply.github.com>
Date: Mon, 28 Oct 2024 20:08:11 +0100
Subject: [PATCH] fix(VideoInfo): Fix like count being undefined (#787)

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

diff --git a/src/parser/classes/menus/Menu.ts b/src/parser/classes/menus/Menu.ts
index 8cc0e95a..a57c5315 100644
--- a/src/parser/classes/menus/Menu.ts
+++ b/src/parser/classes/menus/Menu.ts
@@ -3,18 +3,20 @@ import type { ObservedArray } from '../../helpers.js';
 import { YTNode } from '../../helpers.js';
 import type { RawNode } from '../../index.js';
 import Button from '../Button.js';
+import ButtonView from '../ButtonView.js';
+import SegmentedLikeDislikeButtonView from '../SegmentedLikeDislikeButtonView.js';
 
 export default class Menu extends YTNode {
   static type = 'Menu';
 
   items: ObservedArray<YTNode>;
-  top_level_buttons: ObservedArray<Button>;
+  top_level_buttons: ObservedArray<Button | ButtonView | SegmentedLikeDislikeButtonView>;
   label?: string;
 
   constructor(data: RawNode) {
     super();
     this.items = Parser.parseArray(data.items);
-    this.top_level_buttons = Parser.parseArray(data.topLevelButtons, Button);
+    this.top_level_buttons = Parser.parseArray(data.topLevelButtons, [ Button, ButtonView, SegmentedLikeDislikeButtonView ]);
 
     if (Reflect.has(data, 'accessibility') && Reflect.has(data.accessibility, 'accessibilityData')) {
       this.label = data.accessibility.accessibilityData.label;
diff --git a/src/parser/youtube/History.ts b/src/parser/youtube/History.ts
index e1b40f78..8498d97d 100644
--- a/src/parser/youtube/History.ts
+++ b/src/parser/youtube/History.ts
@@ -1,6 +1,7 @@
 import Feed from '../../core/mixins/Feed.js';
 import ItemSection from '../classes/ItemSection.js';
 import BrowseFeedActions from '../classes/BrowseFeedActions.js';
+import Button from '../classes/Button.js';
 
 import type { Actions, ApiResponse } from '../../core/index.js';
 import type { IBrowseResponse } from '../types/index.js';
@@ -37,7 +38,7 @@ export default class History extends Feed<IBrowseResponse> {
       for (const content of section.contents) {
         const video = content as Video;
         if (video.id === video_id && video.menu) {
-          feedbackToken = video.menu.top_level_buttons[0].endpoint.payload.feedbackToken;
+          feedbackToken = video.menu.top_level_buttons[0].as(Button).endpoint.payload.feedbackToken;
           break;
         }
       }