diff --git a/lib/Innertube.js b/lib/Innertube.js index e1a6443e..c19bf927 100644 --- a/lib/Innertube.js +++ b/lib/Innertube.js @@ -23,30 +23,33 @@ class Innertube extends EventEmitter { async init() { const response = await Axios.get(Constants.urls.YT_BASE_URL, Constants.default_headers(this)).catch((error) => error); - if (response instanceof Error) throw new Error('Could not retrieve Innertube configuration data: ' + response.message); - let innertube_data = JSON.parse('{' + Utils.getStringBetweenStrings(response.data, 'ytcfg.set({', '});') + '}'); - if (innertube_data.INNERTUBE_CONTEXT) { - this.context = innertube_data.INNERTUBE_CONTEXT; - this.key = innertube_data.INNERTUBE_API_KEY; - this.id_token = innertube_data.ID_TOKEN; - this.session_token = innertube_data.XSRF_TOKEN; - this.player_url = innertube_data.PLAYER_JS_URL; - this.logged_in = innertube_data.LOGGED_IN; - this.sts = innertube_data.STS; - this.context.client.hl = 'en'; - this.context.client.gl = 'US'; + if (response instanceof Error) throw new Error(`Could not extract Innertube data: ${response.message}`); + + try { + const innertube_data = JSON.parse(`{${Utils.getStringBetweenStrings(response.data, 'ytcfg.set({', '});')}}`); + if (innertube_data.INNERTUBE_CONTEXT) { + this.context = innertube_data.INNERTUBE_CONTEXT; + this.key = innertube_data.INNERTUBE_API_KEY; + this.id_token = innertube_data.ID_TOKEN; + this.session_token = innertube_data.XSRF_TOKEN; + this.player_url = innertube_data.PLAYER_JS_URL; + this.logged_in = innertube_data.LOGGED_IN; + this.sts = innertube_data.STS; + this.context.client.hl = 'en'; + this.context.client.gl = 'US'; - this.player = new Player(this); - await this.player.init(); + this.player = new Player(this); + await this.player.init(); - if (this.logged_in && this.cookie.length > 1) { - this.auth_apisid = Utils.getStringBetweenStrings(this.cookie, 'PAPISID=', ';'); - this.auth_apisid = Utils.generateSidAuth(this.auth_apisid); + if (this.logged_in && this.cookie.length > 1) { + this.auth_apisid = Utils.getStringBetweenStrings(this.cookie, 'PAPISID=', ';'); + this.auth_apisid = Utils.generateSidAuth(this.auth_apisid); + } + } else { + return this.init(); } - - this.initialized = true; - } else { - this.initialized = false; + } catch (err) { + return this.init(); } return this; } @@ -93,43 +96,42 @@ class Innertube extends EventEmitter { async search(query, options = { period: 'any', order: 'relevance', duration: 'any' }) { if (!query) throw new Error('No query was provided'); - if (!this.initialized) throw new Error('Missing Innertube data.'); + + const response = await Axios.post(`${Constants.urls.YT_BASE_URL}/youtubei/v1/search${this.logged_in && this.cookie.length < 1 ? '' : `?key=${this.key}`}`, JSON.stringify({ context: this.context, params: Constants.filters(options.period + ',' + options.duration + ',' + options.order), query }), Constants.innertube_request_opts({ session: this })).catch((error) => error); + if (response instanceof Error) throw new Error(`Could not search on YouTube: ${response.message}`); - const yt_response = await Axios.post(`${Constants.urls.YT_BASE_URL}/youtubei/v1/search${this.logged_in && this.cookie.length < 1 ? '' : `?key=${this.key}`}`, JSON.stringify({ context: this.context, params: Constants.filters(options.period + ',' + options.duration + ',' + options.order), query }), Constants.innertube_request_opts({ session: this })).catch((error) => error); - if (yt_response instanceof Error) throw new Error('Could not search on YouTube: ' + yt_response.message); - - let content = yt_response.data.contents.twoColumnSearchResultsRenderer.primaryContents.sectionListRenderer.contents[0].itemSectionRenderer.contents; + let content = response.data.contents.twoColumnSearchResultsRenderer.primaryContents.sectionListRenderer.contents[0].itemSectionRenderer.contents; let search_response = {}; search_response.search_metadata = {}; search_response.search_metadata.query = content[0].showingResultsForRenderer ? content[0].showingResultsForRenderer.originalQuery.simpleText : query; search_response.search_metadata.corrected_query = content[0].showingResultsForRenderer ? content[0].showingResultsForRenderer.correctedQueryEndpoint.searchEndpoint.query : query; - search_response.search_metadata.estimated_results = parseInt(yt_response.data.estimatedResults); + search_response.search_metadata.estimated_results = parseInt(response.data.estimatedResults); search_response.videos = content.map((data) => { if (!data.videoRenderer) return; let video = data.videoRenderer; return { title: video.title.runs[0].text, - description: video.detailedMetadataSnippets ? video.detailedMetadataSnippets[0].snippetText.runs.map((item) => item.text).join('') : 'N/A', + description: video.detailedMetadataSnippets && video.detailedMetadataSnippets[0].snippetText.runs.map((item) => item.text).join('') || 'N/A', author: video.ownerText.runs[0].text, id: video.videoId, - url: 'https://youtu.be/' + video.videoId, - channel_url: Constants.urls.YT_BASE_URL + video.ownerText.runs[0].navigationEndpoint.commandMetadata.webCommandMetadata.url, + url: `https://youtu.be/${video.videoId}`, + channel_url: `${Constants.urls.YT_BASE_URL}${video.ownerText.runs[0].navigationEndpoint.commandMetadata.webCommandMetadata.url}`, metadata: { - view_count: video.viewCountText ? video.viewCountText.simpleText : 'N/A', + view_count: video.viewCountText && video.viewCountText.simpleText || 'N/A', short_view_count_text: { - simple_text: video.shortViewCountText ? video.shortViewCountText.simpleText : 'N/A', - accessibility_label: video.shortViewCountText ? (video.shortViewCountText.accessibility ? video.shortViewCountText.accessibility.accessibilityData.label : 'N/A') : 'N/A', + simple_text: video.shortViewCountText && video.shortViewCountText.simpleText || 'N/A', + accessibility_label: video.shortViewCountText && (video.shortViewCountText.accessibility && video.shortViewCountText.accessibility.accessibilityData.label || 'N/A') || 'N/A', }, thumbnails: video.thumbnail.thumbnails, duration: { - seconds: TimeToSeconds(video.lengthText ? video.lengthText.simpleText : '0'), - simple_text: video.lengthText ? video.lengthText.simpleText : 'N/A', - accessibility_label: video.lengthText ? video.lengthText.accessibility.accessibilityData.label : 'N/A' + seconds: TimeToSeconds(video.lengthText && video.lengthText.simpleText || '0'), + simple_text: video.lengthText && video.lengthText.simpleText || 'N/A', + accessibility_label: video.lengthText && video.lengthText.accessibility.accessibilityData.label || 'N/A' }, - published: video.publishedTimeText ? video.publishedTimeText.simpleText : 'N/A', - badges: video.badges ? video.badges.map((item) => item.metadataBadgeRenderer.label) : 'N/A', - owner_badges: video.ownerBadges ? video.ownerBadges.map((item) => item.metadataBadgeRenderer.tooltip) : 'N/A ' + published: video.publishedTimeText && video.publishedTimeText.simpleText || 'N/A', + badges: video.badges && video.badges.map((item) => item.metadataBadgeRenderer.label) || 'N/A', + owner_badges: video.ownerBadges && video.ownerBadges.map((item) => item.metadataBadgeRenderer.tooltip) || 'N/A ' } }; }).filter((video_block) => video_block !== undefined); @@ -149,7 +151,7 @@ class Innertube extends EventEmitter { } else { video_data.getLivechat = () => {}; } - + video_data.like = () => Actions.engage(this, 'like/like', { video_id: id }); video_data.dislike = () => Actions.engage(this, 'like/dislike', { video_id: id }); video_data.removeLike = () => Actions.engage(this, 'like/removelike', { video_id: id }); @@ -161,32 +163,32 @@ class Innertube extends EventEmitter { return video_data; } - + async getComments(video_id, token) { let comment_section_token; - + if (!token) { const data_continuation = await Actions.getContinuation(this, { video_id }); const item_section_renderer = data_continuation.data.contents.twoColumnWatchNextResults.results.results.contents.find((item) => item.itemSectionRenderer); comment_section_token = item_section_renderer.itemSectionRenderer.contents[0].continuationItemRenderer.continuationEndpoint.continuationCommand.token; } - + const response = await Actions.getContinuation(this, { continuation_token: comment_section_token || token }); - if (!response.success) throw new Error('Could not fetch comment section'); - + if (!response.success) throw new Error('Could not fetch comments section'); + const comments_section = { comments: [] }; !token && (comments_section.comment_count = response.data.onResponseReceivedEndpoints[0].reloadContinuationItemsCommand.continuationItems && response.data.onResponseReceivedEndpoints[0].reloadContinuationItemsCommand.continuationItems[0].commentsHeaderRenderer.countText.runs[0].text || 'N/A'); - + let continuation_token; - !token && (continuation_token = response.data.onResponseReceivedEndpoints[1].reloadContinuationItemsCommand.continuationItems.find((item) => item.continuationItemRenderer).continuationItemRenderer.continuationEndpoint.continuationCommand.token) || - (continuation_token = response.data.onResponseReceivedEndpoints[0].appendContinuationItemsAction.continuationItems.find((item) => item.continuationItemRenderer).continuationItemRenderer.continuationEndpoint.continuationCommand.token); - + !token && (continuation_token = response.data.onResponseReceivedEndpoints[1].reloadContinuationItemsCommand.continuationItems.find((item) => item.continuationItemRenderer).continuationItemRenderer.continuationEndpoint.continuationCommand.token) + || (continuation_token = response.data.onResponseReceivedEndpoints[0].appendContinuationItemsAction.continuationItems.find((item) => item.continuationItemRenderer).continuationItemRenderer.continuationEndpoint.continuationCommand.token); + comments_section.getContinuation = () => this.getComments(video_id, continuation_token); - + let contents; - !token && (contents = response.data.onResponseReceivedEndpoints[1].reloadContinuationItemsCommand.continuationItems) || - (contents = response.data.onResponseReceivedEndpoints[0].appendContinuationItemsAction.continuationItems); - + !token && (contents = response.data.onResponseReceivedEndpoints[1].reloadContinuationItemsCommand.continuationItems) + || (contents = response.data.onResponseReceivedEndpoints[0].appendContinuationItemsAction.continuationItems); + contents.forEach((thread) => { if (!thread.commentThreadRenderer) return; const comment = { @@ -207,26 +209,26 @@ class Innertube extends EventEmitter { }; comments_section.comments.push(comment); }); - + return comments_section; } - + async getSubscriptionsFeed() { const response = await Actions.browse(this, 'subscriptions_feed'); if (!response.success) throw new Error('Could not fetch subscriptions feed'); - + const contents = response.data.contents.twoColumnBrowseResultsRenderer.tabs[0].tabRenderer.content.sectionListRenderer.contents; const subscriptions_feed = {}; - + contents.forEach((section) => { if (!section.itemSectionRenderer) return; - + const section_contents = section.itemSectionRenderer.contents[0]; const section_items = section_contents.shelfRenderer.content.gridRenderer.items; - + const key = section_contents.shelfRenderer.title.runs[0].text; subscriptions_feed[key.toLowerCase().replace(/ +/g, '_')] = []; - + section_items.forEach((item) => { const content = { title: item.gridVideoRenderer.title.runs.map((run) => run.text).join(' '), @@ -240,18 +242,18 @@ class Innertube extends EventEmitter { owner_badges: item.gridVideoRenderer.ownerBadges && item.gridVideoRenderer.ownerBadges.map((badge) => badge.metadataBadgeRenderer.tooltip) || 'N/A' } }; - + subscriptions_feed[key.toLowerCase().replace(/ +/g, '_')].push(content); }); }); - + return subscriptions_feed; } async getNotifications() { const response = await Actions.notifications(this, 'get_notification_menu'); if (!response.success) throw new Error('Could not fetch notifications'); - + const contents = response.data.actions[0].openPopupAction.popup.multiPageMenuRenderer.sections[0]; if (!contents.multiPageMenuNotificationSectionRenderer) return { error: 'You don\'t have any notification.' }; return contents.multiPageMenuNotificationSectionRenderer.items.map((notification) => { @@ -278,11 +280,8 @@ class Innertube extends EventEmitter { async requestVideoInfo(id, desktop) { let response; - if (!desktop) { - response = await Axios.get(`${Constants.urls.YT_WATCH_PAGE}?v=${id}&t=8s&pbj=1`, Constants.innertube_request_opts({ session: this, id, desktop: false })).catch((error) => error); - } else { - response = await Axios.post(`${Constants.urls.YT_BASE_URL}/youtubei/v1/player${this.logged_in && this.cookie.length < 1 ? '' : `?key=${this.key}`}`, JSON.stringify(Constants.video_details_reqbody(id, this.sts, this.context)), Constants.innertube_request_opts({ session: this, id, desktop: true })).catch((error) => error); - } + !desktop && (response = await Axios.get(`${Constants.urls.YT_WATCH_PAGE}?v=${id}&t=8s&pbj=1`, Constants.innertube_request_opts({ session: this, id, desktop: false })).catch((error) => error)) + || (response = await Axios.post(`${Constants.urls.YT_BASE_URL}/youtubei/v1/player${this.logged_in && this.cookie.length < 1 ? '' : `?key=${this.key}`}`, JSON.stringify(Constants.video_details_reqbody(id, this.sts, this.context)), Constants.innertube_request_opts({ session: this, id, desktop: true })).catch((error) => error)); if (response instanceof Error) throw new Error('Could not retrieve watch page info: ' + response.message); return response.data; } @@ -351,16 +350,11 @@ class Innertube extends EventEmitter { if (options.type != 'videoandaudio') { let streams; - - if (options.type != 'audio') { - streams = filtered_streams.filter((format) => format.mimeType.includes(options.format || 'mp4') && format.qualityLabel == options.quality); - } else { - streams = filtered_streams.filter((format) => format.mimeType.includes(options.format || 'mp4')); - } - - if (streams == undefined || streams.length == 0) { - streams = filtered_streams.filter((format) => format.quality == 'medium'); - } + + options.type != 'audio' && (streams = filtered_streams.filter((format) => format.mimeType.includes(options.format || 'mp4') && format.qualityLabel == options.quality)) + || (streams = filtered_streams.filter((format) => format.mimeType.includes(options.format || 'mp4'))); + + streams == undefined || streams.length == 0 && (streams = filtered_streams.filter((format) => format.quality == 'medium')); bitrates = streams.map((format) => format.bitrate); url = streams.filter((format) => format.bitrate === Math.max(...bitrates))[0]; @@ -368,7 +362,7 @@ class Innertube extends EventEmitter { const selected_format = options.type == 'videoandaudio' ? filtered_streams[0] : url; if (!selected_format) { - return stream.emit('error', { message: 'Could not find any suitable format.', type: 'FORMATS_UNAVAILABLE' }); + return stream.emit('error', { message: 'Could not find any suitable format.', type: 'FORMAT_UNAVAILABLE' }); } else { stream.emit('info', { video_details, selected_format, formats }); }