nexo-aio-downloader/src/lib/tiktok.js

Download video/photo from multiple sites

const axios = require('axios');
const cheerio = require('cheerio');
const { CookieJar } = require('tough-cookie');
const { wrapper } = require('axios-cookiejar-support');
const { successResponse, errorResponse } = require('./utils');

async function tiktokDownloader(url) {
    const jar = new CookieJar();

    const apiClient = axios.create({
        jar,
        withCredentials: true,
        headers: {
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36 Edg/135.0.0.0',
        },
    });
    wrapper(apiClient);

    try {
        const htmlResponse = await apiClient.get(url);
        const html = htmlResponse.data;

        const $ = cheerio.load(html);
        const scriptContent = $('#__UNIVERSAL_DATA_FOR_REHYDRATION__').html();
        if (!scriptContent) throw new Error('Script tag #__UNIVERSAL_DATA_FOR_REHYDRATION__ not found.');

        const jsonData = JSON.parse(scriptContent);
        const itemStruct = jsonData?.__DEFAULT_SCOPE__?.['webapp.video-detail']?.itemInfo?.itemStruct;
        if (!itemStruct) throw new Error('itemStruct not found within the JSON data.');

        const videoUrlToDownload = itemStruct.video?.downloadAddr || itemStruct.video?.playAddr;

        const importantData = {
            videoId: itemStruct.id,
            description: itemStruct.desc,
            createTime: itemStruct.createTime,
            videoUrl: videoUrlToDownload,
            videoInfo: {
                size: null,
                duration: itemStruct.video?.duration,
                width: itemStruct.video?.width,
                height: itemStruct.video?.height,
                definition: itemStruct.video?.definition,
                coverUrl: itemStruct.video?.cover,
                subtitles: itemStruct.video?.subtitleInfos?.map(sub => ({
                    language: sub.LanguageCodeName,
                    url: sub.Url,
                    format: sub.Format,
                    source: sub.Source,
                })) || [],
            },
            author: {
                id: itemStruct.author?.id,
                uniqueId: itemStruct.author?.uniqueId,
                nickname: itemStruct.author?.nickname,
                avatarThumb: itemStruct.author?.avatarThumb,
            },
            music: {
                id: itemStruct.music?.id,
                title: itemStruct.music?.title,
                authorName: itemStruct.music?.authorName,
                playUrl: itemStruct.music?.playUrl,
                isOriginal: itemStruct.music?.original,
            },
            stats: {
                likes: itemStruct.statsV2?.diggCount ?? itemStruct.stats?.diggCount,
                shares: itemStruct.statsV2?.shareCount ?? itemStruct.stats?.shareCount,
                comments: itemStruct.statsV2?.commentCount ?? itemStruct.stats?.commentCount,
                plays: itemStruct.statsV2?.playCount ?? itemStruct.stats?.playCount,
                collects: itemStruct.statsV2?.collectCount ?? itemStruct.stats?.collectCount,
                reposts: itemStruct.statsV2?.repostCount,
            },
            locationCreated: itemStruct.locationCreated,
            videoBuffer: null,
        };

        if (videoUrlToDownload) {
            try {
                const videoResponse = await apiClient.get(videoUrlToDownload, {
                    responseType: 'arraybuffer',
                    headers: { Referer: url, Range: 'bytes=0-' },
                });

                if (videoResponse.status === 200 || videoResponse.status === 206) {
                    importantData.videoBuffer = Buffer.from(videoResponse.data);
                    importantData.videoInfo.size = videoResponse.data.length;
                } else {
                    console.warn(`Failed to download video. Status: ${videoResponse.status}`);
                }
            } catch (videoError) {
                console.error(`Error downloading video: ${videoError.message}`);
            }
        } else {
            console.warn('Video download URL not found in the data.');
        }

        return successResponse(importantData);

    } catch (error) {
        console.error('Error during main process:', error.message);
        return errorResponse(error.message || 'Something went wrong');
    }
}

module.exports = tiktokDownloader;