From 9978ebf08582fe299b35c0b0b3b123d32f2d8faa Mon Sep 17 00:00:00 2001
From: LuanRT <luan.lrt4@gmail.com>
Date: Sat, 17 Dec 2022 00:54:08 -0300
Subject: [PATCH] refactor(Parser): reduce reliance on localised strings (#258)

---
 src/core/Music.ts                             | 31 ++++----------
 src/core/TabbedFeed.ts                        | 17 ++++++--
 src/parser/classes/C4TabbedHeader.ts          | 41 ++++++++++++-------
 .../classes/ChannelAboutFullMetadata.ts       | 26 ++++++++++--
 src/parser/classes/SectionList.ts             |  2 +-
 src/parser/helpers.ts                         | 12 ++++++
 src/parser/youtube/Channel.ts                 | 19 +++++----
 src/parser/youtube/Search.ts                  |  2 +-
 src/parser/youtube/Settings.ts                |  2 +-
 src/parser/youtube/TimeWatched.ts             |  2 +-
 src/parser/ytmusic/Explore.ts                 |  4 +-
 src/parser/ytmusic/HomeFeed.ts                |  2 +-
 src/parser/ytmusic/Library.ts                 |  2 +-
 src/parser/ytmusic/Playlist.ts                | 16 +++-----
 src/parser/ytmusic/Recap.ts                   |  2 +-
 src/parser/ytmusic/Search.ts                  |  2 +-
 src/parser/ytmusic/TrackInfo.ts               | 20 ++++-----
 17 files changed, 117 insertions(+), 85 deletions(-)

diff --git a/src/core/Music.ts b/src/core/Music.ts
index 41e83bd5..5c3d69d6 100644
--- a/src/core/Music.ts
+++ b/src/core/Music.ts
@@ -11,9 +11,6 @@ import Playlist from '../parser/ytmusic/Playlist';
 import Recap from '../parser/ytmusic/Recap';
 
 import Tab from '../parser/classes/Tab';
-import Tabbed from '../parser/classes/Tabbed';
-import SingleColumnMusicWatchNextResults from '../parser/classes/SingleColumnMusicWatchNextResults';
-import WatchNextTabbedResults from '../parser/classes/WatchNextTabbedResults';
 import SectionList from '../parser/classes/SectionList';
 
 import Message from '../parser/classes/Message';
@@ -235,13 +232,9 @@ class Music {
       parse: true
     });
 
-    const tabs = data.contents.item()
-      .as(SingleColumnMusicWatchNextResults).contents.item()
-      .as(Tabbed).contents.item()
-      .as(WatchNextTabbedResults)
-      .tabs.array().as(Tab);
+    const tabs = data.contents_memo.getType(Tab);
 
-    const tab = tabs.get({ title: 'Up next' });
+    const tab = tabs?.[0];
 
     if (!tab)
       throw new InnertubeError('Could not find target tab.');
@@ -287,20 +280,16 @@ class Music {
       parse: true
     });
 
-    const tabs = data.contents.item()
-      .as(SingleColumnMusicWatchNextResults).contents.item()
-      .as(Tabbed).contents.item()
-      .as(WatchNextTabbedResults)
-      .tabs.array().as(Tab);
+    const tabs = data.contents_memo.getType(Tab);
 
-    const tab = tabs.get({ title: 'Related' });
+    const tab = tabs?.matchCondition((tab) => tab.endpoint.payload.browseEndpointContextSupportedConfigs?.browseEndpointContextMusicConfig?.pageType === 'MUSIC_PAGE_TYPE_TRACK_RELATED');
 
     if (!tab)
       throw new InnertubeError('Could not find target tab.');
 
     const page = await tab.endpoint.call(this.#actions, { client: 'YTMUSIC', parse: true });
 
-    const shelves = page.contents.item().as(SectionList).contents.array().as(MusicCarouselShelf, MusicDescriptionShelf);
+    const shelves = page.contents.item().as(SectionList).contents.as(MusicCarouselShelf, MusicDescriptionShelf);
 
     return shelves;
   }
@@ -318,13 +307,9 @@ class Music {
       parse: true
     });
 
-    const tabs = data.contents.item()
-      .as(SingleColumnMusicWatchNextResults).contents.item()
-      .as(Tabbed).contents.item()
-      .as(WatchNextTabbedResults)
-      .tabs.array().as(Tab);
+    const tabs = data.contents_memo.getType(Tab);
 
-    const tab = tabs.get({ title: 'Lyrics' });
+    const tab = tabs?.matchCondition((tab) => tab.endpoint.payload.browseEndpointContextSupportedConfigs?.browseEndpointContextMusicConfig?.pageType === 'MUSIC_PAGE_TYPE_TRACK_LYRICS');
 
     if (!tab)
       throw new InnertubeError('Could not find target tab.');
@@ -334,7 +319,7 @@ class Music {
     if (page.contents.item().key('type').string() === 'Message')
       throw new InnertubeError(page.contents.item().as(Message).text, video_id);
 
-    const section_list = page.contents.item().as(SectionList).contents.array();
+    const section_list = page.contents.item().as(SectionList).contents;
     return section_list.firstOfType(MusicDescriptionShelf);
   }
 
diff --git a/src/core/TabbedFeed.ts b/src/core/TabbedFeed.ts
index 8c53d6ff..ceda6b24 100644
--- a/src/core/TabbedFeed.ts
+++ b/src/core/TabbedFeed.ts
@@ -17,7 +17,7 @@ class TabbedFeed extends Feed {
     return this.#tabs.map((tab) => tab.title.toString());
   }
 
-  async getTab(title: string) {
+  async getTabByName(title: string) {
     const tab = this.#tabs.find((tab) => tab.title.toLowerCase() === title.toLowerCase());
 
     if (!tab)
@@ -28,8 +28,19 @@ class TabbedFeed extends Feed {
 
     const response = await tab.endpoint.call(this.#actions);
 
-    if (!response)
-      throw new InnertubeError('Failed to call endpoint');
+    return new TabbedFeed(this.#actions, response.data, false);
+  }
+
+  async getTabByURL(url: string) {
+    const tab = this.#tabs.find((tab) => tab.endpoint.metadata.url?.split('/').pop() === url);
+
+    if (!tab)
+      throw new InnertubeError(`Tab "${url}" not found`);
+
+    if (tab.selected)
+      return this;
+
+    const response = await tab.endpoint.call(this.#actions);
 
     return new TabbedFeed(this.#actions, response.data, false);
   }
diff --git a/src/parser/classes/C4TabbedHeader.ts b/src/parser/classes/C4TabbedHeader.ts
index fa246ea1..899cb7dc 100644
--- a/src/parser/classes/C4TabbedHeader.ts
+++ b/src/parser/classes/C4TabbedHeader.ts
@@ -1,20 +1,28 @@
 import Parser from '../index';
 import Author from './misc/Author';
-import Thumbnail from './misc/Thumbnail';
 import Text from './misc/Text';
+import Thumbnail from './misc/Thumbnail';
+
+import type Button from './Button';
+import type ChannelHeaderLinks from './ChannelHeaderLinks';
+import type SubscribeButton from './SubscribeButton';
+
 import { YTNode } from '../helpers';
 
 class C4TabbedHeader extends YTNode {
   static type = 'C4TabbedHeader';
 
-  author;
-  banner;
-  tv_banner;
-  mobile_banner;
-  subscribers;
-  sponsor_button;
-  subscribe_button;
-  header_links;
+  author: Author;
+  banner: Thumbnail[];
+  tv_banner: Thumbnail[];
+  mobile_banner: Thumbnail[];
+  subscribers: Text;
+  videos_count: Text;
+  sponsor_button: Button | null;
+  subscribe_button: SubscribeButton | null;
+  header_links: ChannelHeaderLinks | null;
+  channel_handle: Text;
+  channel_id: string;
 
   constructor(data: any) {
     super();
@@ -23,13 +31,16 @@ class C4TabbedHeader extends YTNode {
       navigationEndpoint: data.navigationEndpoint
     }, data.badges, data.avatar);
 
-    this.banner = data.banner ? Thumbnail.fromResponse(data.banner) : [];
-    this.tv_banner = data.tvBanner ? Thumbnail.fromResponse(data.tvBanner) : [];
-    this.mobile_banner = data.mobileBanner ? Thumbnail.fromResponse(data.mobileBanner) : [];
+    this.banner = Thumbnail.fromResponse(data.banner);
+    this.tv_banner = Thumbnail.fromResponse(data.tvBanner);
+    this.mobile_banner = Thumbnail.fromResponse(data.mobileBanner);
     this.subscribers = new Text(data.subscriberCountText);
-    this.sponsor_button = data.sponsorButton ? Parser.parseItem(data.sponsorButton) : undefined;
-    this.subscribe_button = data.subscribeButton ? Parser.parseItem(data.subscribeButton) : undefined;
-    this.header_links = data.headerLinks ? Parser.parse(data.headerLinks) : undefined;
+    this.videos_count = new Text(data.videosCountText);
+    this.sponsor_button = Parser.parseItem<Button>(data.sponsorButton);
+    this.subscribe_button = Parser.parseItem<SubscribeButton>(data.subscribeButton);
+    this.header_links = Parser.parseItem<ChannelHeaderLinks>(data.headerLinks);
+    this.channel_handle = new Text(data.channelHandleText);
+    this.channel_id = data.channelId;
   }
 }
 
diff --git a/src/parser/classes/ChannelAboutFullMetadata.ts b/src/parser/classes/ChannelAboutFullMetadata.ts
index 80e74e82..d74b8648 100644
--- a/src/parser/classes/ChannelAboutFullMetadata.ts
+++ b/src/parser/classes/ChannelAboutFullMetadata.ts
@@ -1,7 +1,11 @@
+import Parser from '../index';
+
+import Text from './misc/Text';
 import Thumbnail from './misc/Thumbnail';
 import NavigationEndpoint from './NavigationEndpoint';
-import Text from './misc/Text';
-import Parser from '../index';
+
+import type Button from './Button';
+
 import { YTNode } from '../helpers';
 
 class ChannelAboutFullMetadata extends YTNode {
@@ -11,13 +15,20 @@ class ChannelAboutFullMetadata extends YTNode {
   name: Text;
   avatar: Thumbnail[];
   canonical_channel_url: string;
+
+  primary_links: {
+    endpoint: NavigationEndpoint;
+    icon: Thumbnail[];
+    title: Text;
+  }[];
+
   views: Text;
   joined: Text;
   description: Text;
   email_reveal: NavigationEndpoint;
   can_reveal_email: boolean;
   country: Text;
-  buttons;
+  buttons: Button[];
 
   constructor(data: any) {
     super();
@@ -25,13 +36,20 @@ class ChannelAboutFullMetadata extends YTNode {
     this.name = new Text(data.title);
     this.avatar = Thumbnail.fromResponse(data.avatar);
     this.canonical_channel_url = data.canonicalChannelUrl;
+
+    this.primary_links = data.primaryLinks.map((link: any) => ({
+      endpoint: new NavigationEndpoint(link.navigationEndpoint),
+      icon: Thumbnail.fromResponse(link.icon),
+      title: new Text(link.title)
+    }));
+
     this.views = new Text(data.viewCountText);
     this.joined = new Text(data.joinedDateText);
     this.description = new Text(data.description);
     this.email_reveal = new NavigationEndpoint(data.onBusinessEmailRevealClickCommand);
     this.can_reveal_email = !data.signInForBusinessEmail;
     this.country = new Text(data.country);
-    this.buttons = Parser.parse(data.actionButtons);
+    this.buttons = Parser.parseArray<Button>(data.actionButtons);
   }
 }
 
diff --git a/src/parser/classes/SectionList.ts b/src/parser/classes/SectionList.ts
index b8b2af4b..e9dc3b4a 100644
--- a/src/parser/classes/SectionList.ts
+++ b/src/parser/classes/SectionList.ts
@@ -16,7 +16,7 @@ class SectionList extends YTNode {
     }
 
     // TODO: this should be Parser#parseArray
-    this.contents = Parser.parse(data.contents);
+    this.contents = Parser.parseArray(data.contents);
 
     if (data.continuations) {
       if (data.continuations[0].nextContinuationData) {
diff --git a/src/parser/helpers.ts b/src/parser/helpers.ts
index 59b2758c..f6891cd6 100644
--- a/src/parser/helpers.ts
+++ b/src/parser/helpers.ts
@@ -359,6 +359,10 @@ export type ObservedArray<T extends YTNode = YTNode> = Array<T> & {
      * Returns all objects that match the rule.
      */
     getAll: (rule: object, del_items?: boolean) => T[];
+    /**
+     * Returns the first object to match the condition.
+     */
+    matchCondition: (condition: (node: T) => boolean) => T | undefined;
     /**
      * Removes the item at the given index.
      */
@@ -412,6 +416,14 @@ export function observe<T extends YTNode>(obj: Array<T>) {
         );
       }
 
+      if (prop == 'matchCondition') {
+        return (condition: (node: T) => boolean) => (
+          target.find((obj) => {
+            return condition(obj);
+          })
+        );
+      }
+
       if (prop == 'filterType') {
         return (...types: YTNodeConstructor<YTNode>[]) => {
           return observe(target.filter((node: YTNode) => {
diff --git a/src/parser/youtube/Channel.ts b/src/parser/youtube/Channel.ts
index 30afd344..858b412e 100644
--- a/src/parser/youtube/Channel.ts
+++ b/src/parser/youtube/Channel.ts
@@ -9,12 +9,13 @@ import MicroformatData from '../classes/MicroformatData';
 import SubscribeButton from '../classes/SubscribeButton';
 import Tab from '../classes/Tab';
 
-import { InnertubeError } from '../../utils/Utils';
 import FeedFilterChipBar from '../classes/FeedFilterChipBar';
 import ChipCloudChip from '../classes/ChipCloudChip';
 import FilterableFeed from '../../core/FilterableFeed';
 import Feed from '../../core/Feed';
 
+import { InnertubeError } from '../../utils/Utils';
+
 export default class Channel extends TabbedFeed {
   header;
   metadata;
@@ -70,37 +71,37 @@ export default class Channel extends TabbedFeed {
   }
 
   async getHome() {
-    const tab = await this.getTab('Home');
+    const tab = await this.getTabByURL('featured');
     return new Channel(this.actions, tab.page, true);
   }
 
   async getVideos() {
-    const tab = await this.getTab('Videos');
+    const tab = await this.getTabByURL('videos');
     return new Channel(this.actions, tab.page, true);
   }
 
   async getShorts() {
-    const tab = await this.getTab('Shorts');
+    const tab = await this.getTabByURL('shorts');
     return new Channel(this.actions, tab.page, true);
   }
 
   async getLiveStreams() {
-    const tab = await this.getTab('Live');
+    const tab = await this.getTabByURL('streams');
     return new Channel(this.actions, tab.page, true);
   }
 
   async getPlaylists() {
-    const tab = await this.getTab('Playlists');
+    const tab = await this.getTabByURL('playlists');
     return new Channel(this.actions, tab.page, true);
   }
 
   async getCommunity() {
-    const tab = await this.getTab('Community');
+    const tab = await this.getTabByURL('community');
     return new Channel(this.actions, tab.page, true);
   }
 
   async getChannels() {
-    const tab = await this.getTab('Channels');
+    const tab = await this.getTabByURL('channels');
     return new Channel(this.actions, tab.page, true);
   }
 
@@ -109,7 +110,7 @@ export default class Channel extends TabbedFeed {
    * Note that this does not return a new {@link Channel} object.
    */
   async getAbout() {
-    const tab = await this.getTab('About');
+    const tab = await this.getTabByURL('about');
     return tab.memo.getType(ChannelAboutFullMetadata)?.[0];
   }
 
diff --git a/src/parser/youtube/Search.ts b/src/parser/youtube/Search.ts
index 94fd75b2..ccdcabe7 100644
--- a/src/parser/youtube/Search.ts
+++ b/src/parser/youtube/Search.ts
@@ -20,7 +20,7 @@ export default class Search extends Feed {
     super(actions, data, already_parsed);
 
     const contents =
-      this.page.contents_memo.getType(SectionList)?.[0]?.contents?.array() ||
+      this.page.contents_memo.getType(SectionList)?.[0]?.contents ||
       this.page.on_response_received_commands?.[0].contents;
 
     this.results = contents.firstOfType(ItemSection)?.contents;
diff --git a/src/parser/youtube/Settings.ts b/src/parser/youtube/Settings.ts
index f4651484..61954afb 100644
--- a/src/parser/youtube/Settings.ts
+++ b/src/parser/youtube/Settings.ts
@@ -31,7 +31,7 @@ class Settings {
     if (!tab)
       throw new InnertubeError('Target tab not found');
 
-    const contents = tab.content?.as(SectionList).contents.array().as(ItemSection);
+    const contents = tab.content?.as(SectionList).contents.as(ItemSection);
 
     this.introduction = contents?.shift()?.contents?.get({ type: 'PageIntroduction' })?.as(PageIntroduction);
 
diff --git a/src/parser/youtube/TimeWatched.ts b/src/parser/youtube/TimeWatched.ts
index b49d809c..75543d57 100644
--- a/src/parser/youtube/TimeWatched.ts
+++ b/src/parser/youtube/TimeWatched.ts
@@ -19,7 +19,7 @@ class TimeWatched {
     if (!tab)
       throw new InnertubeError('Could not find target tab.');
 
-    this.contents = tab.content?.as(SectionList).contents.array().as(ItemSection);
+    this.contents = tab.content?.as(SectionList).contents.as(ItemSection);
   }
 
   get page(): ParsedResponse {
diff --git a/src/parser/ytmusic/Explore.ts b/src/parser/ytmusic/Explore.ts
index c5d15c7f..29a8a390 100644
--- a/src/parser/ytmusic/Explore.ts
+++ b/src/parser/ytmusic/Explore.ts
@@ -28,8 +28,8 @@ class Explore {
     if (!section_list)
       throw new InnertubeError('Target tab did not have any content.');
 
-    this.top_buttons = section_list.contents.array().firstOfType(Grid)?.items.as(MusicNavigationButton) || ([] as MusicNavigationButton[]);
-    this.sections = section_list.contents.array().getAll({ type: 'MusicCarouselShelf' }) as MusicCarouselShelf[];
+    this.top_buttons = section_list.contents.firstOfType(Grid)?.items.as(MusicNavigationButton) || ([] as MusicNavigationButton[]);
+    this.sections = section_list.contents.getAll({ type: 'MusicCarouselShelf' }) as MusicCarouselShelf[];
   }
 
   get page(): ParsedResponse {
diff --git a/src/parser/ytmusic/HomeFeed.ts b/src/parser/ytmusic/HomeFeed.ts
index bdeb43f4..1fb39693 100644
--- a/src/parser/ytmusic/HomeFeed.ts
+++ b/src/parser/ytmusic/HomeFeed.ts
@@ -33,7 +33,7 @@ class HomeFeed {
     }
 
     this.#continuation = tab.content?.as(SectionList).continuation;
-    this.sections = tab.content?.as(SectionList).contents.array().as(MusicCarouselShelf);
+    this.sections = tab.content?.as(SectionList).contents.as(MusicCarouselShelf);
   }
 
   /**
diff --git a/src/parser/ytmusic/Library.ts b/src/parser/ytmusic/Library.ts
index 7035a9d7..3350bd63 100644
--- a/src/parser/ytmusic/Library.ts
+++ b/src/parser/ytmusic/Library.ts
@@ -30,7 +30,7 @@ class Library {
     const section_list = this.#page.contents_memo.getType(SectionList)?.[0];
 
     this.header = section_list?.header?.item().as(MusicSideAlignedItem);
-    this.contents = section_list?.contents?.array().as(Grid, MusicShelf);
+    this.contents = section_list?.contents?.as(Grid, MusicShelf);
 
     this.#continuation = this.contents?.find((list: Grid | MusicShelf) => list.continuation)?.continuation;
   }
diff --git a/src/parser/ytmusic/Playlist.ts b/src/parser/ytmusic/Playlist.ts
index cc1e178f..1f7f70a8 100644
--- a/src/parser/ytmusic/Playlist.ts
+++ b/src/parser/ytmusic/Playlist.ts
@@ -61,7 +61,7 @@ class Playlist {
   /**
    * Retrieves related playlists
    */
-  async getRelated() {
+  async getRelated(): Promise<MusicCarouselShelf> {
     let section_continuation = this.#page.contents_memo.get('SectionList')?.[0].as(SectionList).continuation;
 
     while (section_continuation) {
@@ -74,19 +74,15 @@ class Playlist {
       const section_list = data.continuation_contents?.as(SectionListContinuation);
       const sections = section_list?.contents?.as(MusicCarouselShelf, MusicShelf);
 
-      const related = sections?.filter(
-        (section) =>
-          section.is(MusicCarouselShelf) ? section.header?.title.toString() === 'Related playlists' :
-            section.title.toString() === 'Related playlists'
-      )[0];
+      const related = sections?.matchCondition((section) => section.is(MusicCarouselShelf))?.as(MusicCarouselShelf);
 
       if (related)
-        return related.contents || [];
+        return related;
 
       section_continuation = section_list?.continuation;
     }
 
-    return [];
+    throw new InnertubeError('Target section not found.');
   }
 
   async getSuggestions(refresh = true) {
@@ -115,9 +111,7 @@ class Playlist {
       const section_list = page.continuation_contents?.as(SectionListContinuation);
       const sections = section_list?.contents?.as(MusicCarouselShelf, MusicShelf);
 
-      const suggestions = sections?.filter(
-        (section) => section.is(MusicShelf) && section.title.toString() === 'Suggestions'
-      )[0] as MusicShelf | undefined;
+      const suggestions = sections?.matchCondition((section) => section.is(MusicShelf))?.as(MusicShelf);
 
       return {
         items: suggestions?.contents || [],
diff --git a/src/parser/ytmusic/Recap.ts b/src/parser/ytmusic/Recap.ts
index e8e9c0fb..cc912a9f 100644
--- a/src/parser/ytmusic/Recap.ts
+++ b/src/parser/ytmusic/Recap.ts
@@ -37,7 +37,7 @@ class Recap {
     if (!tab)
       throw new InnertubeError('Target tab not found');
 
-    this.sections = tab.content?.as(SectionList).contents.array().as(ItemSection, MusicCarouselShelf, Message);
+    this.sections = tab.content?.as(SectionList).contents.as(ItemSection, MusicCarouselShelf, Message);
   }
 
   /**
diff --git a/src/parser/ytmusic/Search.ts b/src/parser/ytmusic/Search.ts
index b50e72ef..746aa2ce 100644
--- a/src/parser/ytmusic/Search.ts
+++ b/src/parser/ytmusic/Search.ts
@@ -49,7 +49,7 @@ class Search {
 
     this.header = tab_content.hasKey('header') ? tab_content.header?.item().as(ChipCloud) : null;
 
-    const shelves = tab_content.contents.array().as(MusicShelf, ItemSection);
+    const shelves = tab_content.contents.as(MusicShelf, ItemSection);
     const item_section = shelves.firstOfType(ItemSection);
 
     this.did_you_mean = item_section?.contents?.firstOfType(DidYouMean) || null;
diff --git a/src/parser/ytmusic/TrackInfo.ts b/src/parser/ytmusic/TrackInfo.ts
index 47cff7d1..725ae7cc 100644
--- a/src/parser/ytmusic/TrackInfo.ts
+++ b/src/parser/ytmusic/TrackInfo.ts
@@ -4,9 +4,7 @@ import Constants from '../../utils/Constants';
 import { InnertubeError } from '../../utils/Utils';
 
 import Tab from '../classes/Tab';
-import Tabbed from '../classes/Tabbed';
 import WatchNextTabbedResults from '../classes/WatchNextTabbedResults';
-import SingleColumnMusicWatchNextResults from '../classes/SingleColumnMusicWatchNextResults';
 import MicroformatData from '../classes/MicroformatData';
 import PlayerOverlay from '../classes/PlayerOverlay';
 import PlaylistPanel from '../classes/PlaylistPanel';
@@ -70,8 +68,7 @@ class TrackInfo {
     this.#playback_tracking = info.playback_tracking;
 
     if (next) {
-      const single_col = next.contents.item().as(SingleColumnMusicWatchNextResults);
-      const tabbed_results = single_col.contents.item().as(Tabbed).contents.item().as(WatchNextTabbedResults);
+      const tabbed_results = next.contents_memo.getType(WatchNextTabbedResults)?.[0];
 
       this.tabs = tabbed_results.tabs.array().as(Tab);
       this.current_video_endpoint = next.current_video_endpoint;
@@ -84,14 +81,17 @@ class TrackInfo {
   /**
    * Retrieves contents of the given tab.
    */
-  async getTab(title: string) {
+  async getTab(title_or_page_type: string) {
     if (!this.tabs)
       throw new InnertubeError('Could not find any tab');
 
-    const target_tab = this.tabs.get({ title });
+    const target_tab =
+      this.tabs.get({ title: title_or_page_type }) ||
+      this.tabs.matchCondition((tab) => tab.endpoint.payload.browseEndpointContextSupportedConfigs?.browseEndpointContextMusicConfig?.pageType === title_or_page_type) ||
+      this.tabs?.[0];
 
     if (!target_tab)
-      throw new InnertubeError(`Tab "${title}" not found`, { available_tabs: this.available_tabs });
+      throw new InnertubeError(`Tab "${title_or_page_type}" not found`, { available_tabs: this.available_tabs });
 
     if (target_tab.content)
       return target_tab.content;
@@ -101,7 +101,7 @@ class TrackInfo {
     if (page.contents.item().key('type').string() === 'Message')
       return page.contents.item().as(Message);
 
-    return page.contents.item().as(SectionList).contents.array();
+    return page.contents.item().as(SectionList).contents;
   }
 
   /**
@@ -140,7 +140,7 @@ class TrackInfo {
    * Retrieves related content.
    */
   async getRelated(): Promise<ObservedArray<MusicCarouselShelf | MusicDescriptionShelf>> {
-    const tab = await this.getTab('Related') as ObservedArray<MusicDescriptionShelf | MusicDescriptionShelf>;
+    const tab = await this.getTab('MUSIC_PAGE_TYPE_TRACK_RELATED') as ObservedArray<MusicDescriptionShelf | MusicDescriptionShelf>;
     return tab;
   }
 
@@ -148,7 +148,7 @@ class TrackInfo {
    * Retrieves lyrics.
    */
   async getLyrics(): Promise<MusicDescriptionShelf | undefined> {
-    const tab = await this.getTab('Lyrics') as ObservedArray<MusicCarouselShelf | MusicDescriptionShelf>;
+    const tab = await this.getTab('MUSIC_PAGE_TYPE_TRACK_LYRICS') as ObservedArray<MusicCarouselShelf | MusicDescriptionShelf>;
     return tab.firstOfType(MusicDescriptionShelf);
   }