cobalt/src/modules/processing/services/tiktok.js

import { genericUserAgent } from "../../config.js";

let userAgent = genericUserAgent.split(' Chrome/1')[0],
    config = {
    tiktok: {
        short: "https://vt.tiktok.com/",
        api: "https://api2.musical.ly/aweme/v1/feed/?aweme_id={postId}&version_code=262&app_name=musical_ly&channel=App&device_id=null&os_version=14.4.2&device_platform=iphone&device_type=iPhone9&region=US&carrier_region=US",
    },
    douyin: {
        short: "https://v.douyin.com/",
        api: "https://www.iesdouyin.com/aweme/v1/web/aweme/detail/?aweme_id={postId}",
    }
}

function selector(j, h, id) {
    if (!j) return false;
    let t;
    switch (h) {
        case "tiktok":
            t = j["aweme_list"].filter((v) => { if (v["aweme_id"] === id) return true })[0];
            break;
        case "douyin":
            t = j['aweme_detail'];
            break;
    }
    if (t.length < 3) return false;
    return t;
}

export default async function(obj) {
    let postId = obj.postId ? obj.postId : false;

    if (!postId) {
        let html = await fetch(`${config[obj.host]["short"]}${obj.id}`, {
            redirect: "manual",
            headers: { "user-agent": userAgent }
        }).then((r) => { return r.text() }).catch(() => { return false });
        if (!html) return { error: 'ErrorCouldntFetch' };

        if (html.slice(0, 17) === '<a href="https://' && html.includes('/video/')) {
            postId = html.split('/video/')[1].split('?')[0].replace("/", '')
        } else if (html.slice(0, 32) === '<a href="https://m.tiktok.com/v/' && html.includes('/v/')) {
            postId = html.split('/v/')[1].split('.html')[0].replace("/", '')
        }
    }
    if (!postId) return { error: 'ErrorCantGetID' };

    let detail;
    detail = await fetch(config[obj.host]["api"].replace("{postId}", postId), {
        headers: {"user-agent": "TikTok 26.2.0 rv:262018 (iPhone; iOS 14.4.2; en_US) Cronet"}
    }).then((r) => { return r.json() }).catch(() => { return false });

    detail = selector(detail, obj.host, postId);
    if (!detail) return { error: 'ErrorCouldntFetch' };

    let video, videoFilename, audioFilename, isMp3, audio, images, filenameBase = `${obj.host}_${postId}`;
    if (obj.host === "tiktok") {
        images = detail["image_post_info"] ? detail["image_post_info"]["images"] : false
    } else {
        images = detail["images"] ? detail["images"] : false
    }

    if (!obj.isAudioOnly && !images) {
        video = obj.host === "tiktok" ? detail["video"]["download_addr"]["url_list"][0] : detail["video"]["play_addr"]["url_list"][2].replace("/play/", "/playwm/");
        videoFilename = `${filenameBase}_video.mp4`;
        if (obj.noWatermark) {
            video = obj.host === "tiktok" ? detail["video"]["play_addr"]["url_list"][0] : detail["video"]["play_addr"]["url_list"][0];
            videoFilename = `${filenameBase}_video_nw.mp4` // nw - no watermark
        }
    } else {
        let fallback = obj.host === "douyin" ? detail["video"]["play_addr"]["url_list"][0].replace("playwm", "play") : detail["video"]["play_addr"]["url_list"][0];
        audio = fallback;
        audioFilename = `${filenameBase}_audio_fv`;  // fv - from video
        if (obj.fullAudio || fallback.includes("music")) {
            audio = detail["music"]["play_url"]["url_list"][0]
            audioFilename = `${filenameBase}_audio`
        }
        if (audio.slice(-4) === ".mp3") isMp3 = true;
    }

    if (video) return {
        urls: video,
        filename: videoFilename
    }
    if (images && obj.isAudioOnly) return {
        urls: audio,
        audioFilename: audioFilename,
        isAudioOnly: true,
        isMp3: isMp3
    }
    if (images) {
        let imageLinks = [];
        for (let i in images) {
            let sel = obj.host === "tiktok" ? images[i]["display_image"]["url_list"] : images[i]["url_list"];
            sel = sel.filter((p) => { if (p.includes(".jpeg?")) return true; })
            imageLinks.push({url: sel[0]})
        }
        return {
            picker: imageLinks,
            urls: audio,
            audioFilename: audioFilename,
            isAudioOnly: true,
            isMp3: isMp3
        }
    }
    if (audio) return {
        urls: audio,
        audioFilename: audioFilename,
        isAudioOnly: true,
        isMp3: isMp3
    }
}