VueTube/NUXT/plugins/innertube.js

// Code specific to working with the innertube API
// https://www.youtube.com/youtubei/v1

import { Http } from '@capacitor-community/http';
import { getBetweenStrings } from './utils';
import constants from './constants';

class Innertube {

    //--- Initiation ---//

    constructor(ErrorCallback) {
        this.ErrorCallback = ErrorCallback || undefined;
        this.retry_count = 0
    }

    checkErrorCallback() {
        return typeof this.ErrorCallback === "function"
    }

    async initAsync() {
        const html = await Http.get({ url: constants.URLS.YT_URL, params: { hl: "en" } }).catch((error) => error);
        try {
            if (html instanceof Error && this.checkErrorCallback) this.ErrorCallback(html.message, true);
            try {
                const data = JSON.parse(getBetweenStrings(html.data, 'ytcfg.set(', ');'));
                if (data.INNERTUBE_CONTEXT) {
                    this.key = data.INNERTUBE_API_KEY;
                    this.context = data.INNERTUBE_CONTEXT;
                    this.logged_in = data.LOGGED_IN;

                    this.context.client = constants.INNERTUBE_CLIENT(this.context.client)
                    this.header = constants.INNERTUBE_HEADER(this.context.client)
                }

            } catch (err) {
                console.log(err)
                if (this.checkErrorCallback) this.ErrorCallback(err, true)
                if (this.retry_count >= 10) { this.initAsync() } else { if (this.checkErrorCallback) this.ErrorCallback("Failed to retrieve Innertube session", true); }
            }
        } catch (error) {
            this.ErrorCallback(error, true)
        };
    };

    static async createAsync(ErrorCallback) {
        const created = new Innertube(ErrorCallback);
        await created.initAsync();
        return created;
    }

    //--- API Calls ---//

    async browseAsync(action_type) {
        let data = { context: this.context }

        switch (action_type) {
            case 'recommendations':
                data.browseId = 'FEwhat_to_watch'
                break;
            case 'playlist':
                data.browseId = args.browse_id
                break;
            default:
        }

        console.log(data)

        const response = await Http.post({
            url: `${constants.URLS.YT_BASE_API}/browse?key=${this.key}`,
            data: data,
            headers: { "Content-Type": "application/json" }
        }).catch((error) => error);

        if (response instanceof Error) return { success: false, status_code: response.status, message: response.message };

        return {
            success: true,
            status_code: response.status,
            data: response.data
        };
    }

    static getThumbnail(id, resolution) {
            if (resolution == "max"){
                const url = `https://img.youtube.com/vi/${id}/maxresdefault.jpg`
                let img = new Image();
                img.src = url
                img.onload = function(){
                    if (img.height !== 120) return url
                };
            }
            return `https://img.youtube.com/vi/${id}/mqdefault.jpg`
        }
    

    async getVidAsync(id) {
        let data = { context: this.context, videoId: id }
        const response = await Http.get({
            url: `https://m.youtube.com/watch?v=${id}&pbj=1`,
            params: {},
            headers: Object.assign(this.header, {
                referer: `https://m.youtube.com/watch?v=${id}`, 
                'x-youtube-client-name': constants.YT_API_VALUES.CLIENT_WEB, 
                'x-youtube-client-version': constants.YT_API_VALUES.VERSION_WEB})
        }).catch((error) => error);

        const responseMobile = await Http.post({
            url: `${constants.URLS.YT_BASE_API}/player?key=${this.key}`,
            data: data,
            headers: constants.INNERTUBE_HEADER(this.context)
        }).catch((error) => error);

        if (response instanceof Error) return { success: false, status_code: response.response.status, message: response.message };

        return {
            success: true,
            status_code: response.status,
            data: {webOutput: response.data, appOutput: responseMobile.data}
        };
    }

    // Simple Wrappers
    async getRecommendationsAsync() {
        const rec = await this.browseAsync("recommendations");
        console.log(rec.data)
        return rec;
    }

    async VidInfoAsync(id) {
        let response = await this.getVidAsync(id)
        
        if (response.success && (response.data.webOutput[2].playerResponse?.playabilityStatus?.status == ("ERROR" || undefined))) 
            throw new Error(`Could not get information for video: ${response[2].playerResponse?.playabilityStatus?.status} - ${response[2].playerResponse?.playabilityStatus?.reason}`)
        const responseWeb = response.data.webOutput
        const responseApp = response.data.appOutput
        const details = responseWeb[2].playerResponse?.videoDetails
        const microformat = responseWeb[2].playerResponse?.microformat?.playerMicroformatRenderer
        const renderedPanels = responseWeb[3].response?.engagementPanels
        const columnUI = responseWeb[3].response?.contents.singleColumnWatchNextResults?.results?.results
        const resolutions = responseApp.streamingData

        console.log((columnUI.contents).length)

        return {
            id: details.videoId,
            title: details.title || microformat.title?.runs[0].text,
            isLive: details.isLiveContent || microformat.liveBroadcastDetails?.isLiveNow || false,
            channelName: details.author || microformat.ownerChannelName,
            channelUrl: microformat.ownerProfileUrl,
            availableResolutions: resolutions?.formats,
            availableResolutionsAdaptive: resolutions?.adaptiveFormats,
            metadata: {
                description: microformat.description?.runs[0].text,
                descriptionShort: details.shortDescription,
                thumbnails: details.thumbnails?.thumbnails || microformat.thumbnails?.thumbnails,
                isFamilySafe: microformat.isFamilySafe,
                availableCountries: microformat.availableCountries,
                liveBroadcastDetails: microformat.liveBroadcastDetails,
                uploadDate: microformat.uploadDate,
                publishDate: microformat.publishDate,
                isPrivate: details.isPrivate,
                viewCount: details.viewCount || microformat.viewCount,
                lengthSeconds: details.lengthSeconds || microformat.lengthSeconds,
                likes: parseInt(columnUI?.contents[1]
                    .slimVideoMetadataSectionRenderer?.contents[1].slimVideoActionBarRenderer?.buttons[0]
                    .slimMetadataToggleButtonRenderer?.button?.toggleButtonRenderer?.defaultText?.accessibility?.accessibilityData?.label?.replace(/\D/g, '')) // Yes. I know.
            },
            renderedData: {
                description: renderedPanels[0].engagementPanelSectionListRenderer?.content.structuredDescriptionContentRenderer?.items[1].expandableVideoDescriptionBodyRenderer?.descriptionBodyText.runs,
                recommendations: columnUI?.contents[(columnUI.contents).length -1].itemSectionRenderer?.contents
            }
        }

    }


}

export default Innertube