fix(VideoInfo): Fix like count being undefined (#787)

LuanRT · Oct 28, 2024 · 182bf42 · 182bf42
1 parent dba0be7
commit 182bf42
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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;
         }
       }