import got from 'got';
import * as cheerio from 'cheerio';
import { TwitterDlArgsSchema, TwitterDLResponseSchema, TwitterDlSchema } from '../types/twitter-v1.js';
import { DEFAULT_HEADERS } from '../constant.js';
import { stringifyCookies, generateTokenId } from './util.js';


export default async function twitterdl(url) {
    TwitterDlArgsSchema.parse([url]);

    const idMatch = url.match(/status\/(\d+)/) || url.match(/(\d+)/);
    if (!idMatch) throw new Error("Invalid Twitter URL: Cannot extract tweet ID.");

    const id = idMatch[1];
    const token = generateTokenId(id);

    try {
        const data = await got(`https://api.redketchup.io/tweetAttachments-v6?id=${encodeURIComponent(token)}`, {
            headers: {
                ...DEFAULT_HEADERS,
                origin: 'https://redketchup.io',
                referer: 'https://redketchup.io/',
            }
        }).json();

        if (!Array.isArray(data)) {
            throw new Error("Unexpected response format: Expected an array.");
        }

        const json = TwitterDLResponseSchema.parse({ includes: { media: data } });

        const videos = json.includes.media
            .filter((m) => m.type === 'video')
            .flatMap((m) => m.variants)
            .filter((v) => v.content_type !== 'application/x-mpegURL');

        return TwitterDlSchema.parse(videos);
    } catch (error) {
        console.error("Error fetching Twitter video:", error.message);
        throw new Error("Failed to fetch Twitter video. Please try again later.");
    }
}