118 lines
3.2 KiB
JavaScript
118 lines
3.2 KiB
JavaScript
import debug from 'debug';
|
|
import util from 'util';
|
|
import fetch from 'node-fetch';
|
|
import fs from 'fs';
|
|
|
|
import { WATCH_BASE } from './util.mjs';
|
|
import { retryable, exclusive } from './retry.mjs';
|
|
import { TemporaryError } from './retry.mjs';
|
|
|
|
const log = debug('youtube-maze:api');
|
|
const METADATA_REGEXES = [
|
|
/window\["ytInitialPlayerResponse"\] = (\{.*?\});/,
|
|
/var ytInitialPlayerResponse = (\{.*?\});/,
|
|
];
|
|
|
|
/**
|
|
* Fetch metadata about a YouTube video.
|
|
*
|
|
* @async
|
|
* @param String videoId Identifier of the video to fetch.
|
|
* @return Object The video metadata object.
|
|
*/
|
|
const bareGetMetadata = async (videoId) =>
|
|
{
|
|
const url = util.format(WATCH_BASE, encodeURIComponent(videoId));
|
|
log(`Fetching ${videoId} (${url})`);
|
|
const res = await fetch(url);
|
|
|
|
if (res.status === 429)
|
|
{
|
|
throw new TemporaryError('Too many requests');
|
|
}
|
|
else if (res.status !== 200)
|
|
{
|
|
throw new Error(`Invalid YouTube HTTP status: ${res.status}`);
|
|
}
|
|
|
|
const body = await res.text();
|
|
const searchResults = METADATA_REGEXES.map(regex => body.match(regex));
|
|
const searchResult = searchResults.find(item => item !== null);
|
|
|
|
if (searchResult === null)
|
|
{
|
|
throw new TemporaryError(`Incomplete YouTube response for video ${videoId}`);
|
|
}
|
|
|
|
const metadata = JSON.parse(searchResult[1]);
|
|
|
|
if (metadata.playabilityStatus.status !== 'OK')
|
|
{
|
|
throw new Error(`Video ${videoId} is not available; \
|
|
status: ${metadata.playabilityStatus.status}; \
|
|
reason: "${metadata.playabilityStatus.reason}"`);
|
|
}
|
|
|
|
const actualId = metadata.videoDetails.videoId;
|
|
|
|
if (videoId !== actualId)
|
|
{
|
|
throw new Error(`Video ${videoId} has actual id ${actualId}`);
|
|
}
|
|
|
|
log(`Done fetching ${videoId}`);
|
|
return metadata;
|
|
};
|
|
|
|
export const getMetadata = exclusive(retryable(bareGetMetadata));
|
|
|
|
/**
|
|
* Get video ID and title.
|
|
*
|
|
* @param Object metadata The video metadata object from `getMetadata`.
|
|
* @return Object Containing the video ID and title.
|
|
*/
|
|
export const getVideoInfo = metadata => ({
|
|
videoId: metadata.videoDetails.videoId,
|
|
title: metadata.videoDetails.title,
|
|
});
|
|
|
|
/**
|
|
* Find videos linked from the endscreen of a YouTube video.
|
|
*
|
|
* @param Object metadata The video metadata object from `getMetadata`.
|
|
* @return List of identifiers of linked videos.
|
|
*/
|
|
export const getEndScreenVideos = metadata =>
|
|
{
|
|
if (!('endscreen' in metadata))
|
|
{
|
|
return [];
|
|
}
|
|
|
|
return metadata.endscreen.endscreenRenderer.elements
|
|
.map(elt => elt.endscreenElementRenderer)
|
|
.filter(rdr => 'watchEndpoint' in rdr.endpoint)
|
|
.map(rdr => rdr.endpoint.watchEndpoint.videoId);
|
|
};
|
|
|
|
/**
|
|
* Find videos linked from as cards from a YouTube video.
|
|
*
|
|
* @param Object metadata The video metadata object from `getMetadata`.
|
|
* @return List of identifiers of linked videos.
|
|
*/
|
|
export const getCardVideos = metadata =>
|
|
{
|
|
if (!('cards' in metadata))
|
|
{
|
|
return [];
|
|
}
|
|
|
|
return metadata.cards.cardCollectionRenderer.cards
|
|
.map(card => card.cardRenderer.content)
|
|
.filter(content => 'videoInfoCardContentRenderer' in content)
|
|
.map(content => content.videoInfoCardContentRenderer)
|
|
.map(rdr => rdr.action.watchEndpoint.videoId);
|
|
};
|