youtube-maze/lib/api.mjs

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);
};