// ==UserScript==
// @name         Twitter Media Downloader
// @namespace    http://tampermonkey.net/
// @version      1.4
// @description  Download images and videos from Twitter posts and pack them into a ZIP archive with metadata.
// @author       Dramorian
// @license      MIT
// @match        https://twitter.com/*
// @match        https://x.com/*
// @require      https://cdnjs.cloudflare.com/ajax/libs/jszip/3.7.1/jszip.min.js
// @require      https://cdnjs.cloudflare.com/ajax/libs/FileSaver.js/2.0.5/FileSaver.min.js
// @grant        none
// @downloadURL none
// ==/UserScript==

(function() {
    'use strict';

    const BASE_URL = 'https://x.com/i/api/graphql/QuBlQ6SxNAQCt6-kBiCXCQ/TweetDetail';

    function getCookie() {
        const cookies = document.cookie.split(';').reduce((acc, cookie) => {
            const [name, value] = cookie.split('=').map(c => c.trim());
            acc[name] = value;
            return acc;
        }, {});
        return {
            lang: cookies.lang || 'en',
            ct0: cookies.ct0 || ''
        };
    }

    async function fetchTweetData(tweetId) {
        const variables = {
            focalTweetId: tweetId,
            with_rux_injections: false,
            rankingMode: "Relevance",
            includePromotedContent: true,
            withCommunity: true,
            withQuickPromoteEligibilityTweetFields: true,
            withBirdwatchNotes: true,
            withVoice: true
        };

        const features = {
            rweb_tipjar_consumption_enabled: true,
            responsive_web_graphql_exclude_directive_enabled: true,
            verified_phone_label_enabled: false,
            creator_subscriptions_tweet_preview_api_enabled: true,
            responsive_web_graphql_timeline_navigation_enabled: true,
            responsive_web_graphql_skip_user_profile_image_extensions_enabled: false,
            communities_web_enable_tweet_community_results_fetch: true,
            c9s_tweet_anatomy_moderator_badge_enabled: true,
            articles_preview_enabled: true,
            responsive_web_edit_tweet_api_enabled: true,
            graphql_is_translatable_rweb_tweet_is_translatable_enabled: true,
            view_counts_everywhere_api_enabled: true,
            longform_notetweets_consumption_enabled: true,
            responsive_web_twitter_article_tweet_consumption_enabled: true,
            tweet_awards_web_tipping_enabled: false,
            creator_subscriptions_quote_tweet_preview_enabled: false,
            freedom_of_speech_not_reach_fetch_enabled: true,
            standardized_nudges_misinfo: true,
            tweet_with_visibility_results_prefer_gql_limited_actions_policy_enabled: true,
            rweb_video_timestamps_enabled: true,
            longform_notetweets_rich_text_read_enabled: true,
            longform_notetweets_inline_media_enabled: true,
            responsive_web_enhance_cards_enabled: false
        };

        const fieldToggles = {
            withArticleRichContentState: true,
            withArticlePlainText: false,
            withGrokAnalyze: false,
            withDisallowedReplyControls: false
        };

        const url = encodeURI(`${BASE_URL}?variables=${JSON.stringify(variables)}&features=${JSON.stringify(features)}&fieldToggles=${JSON.stringify(fieldToggles)}`);
        const cookies = getCookie();
        const headers = {
            'authorization': 'Bearer AAAAAAAAAAAAAAAAAAAAANRILgAAAAAAnNwIzUejRCOuH5E6I8xnZz4puTs%3D1Zv7ttfk8LF81IUq16cHjhLTvJu4FA33AGWWjCpTnA',
            'x-twitter-active-user': 'yes',
            'x-twitter-client-language': cookies.lang,
            'x-csrf-token': cookies.ct0
        };

        try {
            const response = await fetch(url, {
                method: 'GET',
                headers: headers
            });

            if (!response.ok) {
                const text = await response.text();
                console.error(`Error: ${text}`);
                return [];
            }

            const data = await response.json();
            const tweetEntry = data.data?.threaded_conversation_with_injections_v2?.instructions?.[0]?.entries?.find(n => n.entryId === `tweet-${tweetId}`);
            const tweetResult = tweetEntry?.content?.itemContent?.tweet_results?.result;

            if (!tweetResult) {
                console.error('Tweet result not found');
                return [];
            }

            const media = tweetResult.legacy?.entities?.media;

            if (!media || media.length === 0) {
                return [];
            }

            return media.flatMap(item => {
                if (item.type === 'photo') {
                    return [item.media_url_https + '?name=orig'];
                } else if (item.type === 'video') {
                    const highestQuality = item.video_info.variants
                    .filter(variant => variant.content_type === 'video/mp4')
                    .reduce((max, variant) => variant.bitrate > max.bitrate ? variant : max, {
                        bitrate: 0
                    });

                    return [{
                        url: highestQuality.url,
                        bitrate: highestQuality.bitrate,
                        content_type: highestQuality.content_type
                    }];
                }
                return [];
            });
        } catch (error) {
            console.error('Failed to fetch tweet data:', error);
            return [];
        }
    }

    async function downloadMedia(tweetElement, mediaData) {
        const zip = new JSZip();
        const tweetLinkElement = tweetElement.querySelector('a[href*="/status/"]');
        let tweetLink = tweetLinkElement.href;
        const tweetParts = tweetLink.match(/https:\/\/(?:x\.com|twitter\.com)\/([^\/]+)\/status\/(\d+)/);
        const authorHandle = tweetParts[1];
        const tweetId = tweetParts[2];

        // Normalize the tweet URL to ensure it doesn't include media indexes
        tweetLink = `https://x.com/${authorHandle}/status/${tweetId}`;

        const authorCommentElement = tweetElement.querySelector('div[lang]');
        const authorComment = authorCommentElement ? authorCommentElement.innerText : '';
        const dateElement = tweetElement.querySelector('time');
        const postDateTime = dateElement ? new Date(dateElement.getAttribute('datetime')) : new Date();

        let metadata = `${tweetLink}\n`;
        if (authorComment) {
            metadata += `${authorComment}\n`;
        }
        metadata += `@${authorHandle}\n${postDateTime.toLocaleString()}\n`;

        let mediaIndex = 1;
        for (const media of mediaData) {
            try {
                if (media.content_type === 'video/mp4') {
                    const videoData = await fetch(media.url).then(res => res.blob());
                    const videoName = `${authorHandle}_${tweetId}_${mediaIndex}.mp4`;
                    zip.file(videoName, videoData);
                    metadata += `${media.url}\n`;
                } else {
                    const imageData = await fetch(media).then(res => res.blob());
                    const imageName = `${authorHandle}_${tweetId}_${mediaIndex}.jpg`;
                    zip.file(imageName, imageData);
                    metadata += `${media}\n`;
                }
                mediaIndex++;
            } catch (error) {
                console.error('Failed to fetch media:', error);
            }
        }

        zip.file('metadata.txt', metadata.trim());
        zip.file(`${authorHandle}_${tweetId}.url`, `[InternetShortcut]\nURL=${tweetLink}`);

        const content = await zip.generateAsync({
            type: 'blob'
        });
        saveAs(content, `${authorHandle}_${tweetId}.zip`);
    }

    function addDownloadButton(tweetElement) {
        const tweetLinkElement = tweetElement.querySelector('a[href*="/status/"]');
        if (!tweetLinkElement) return;

        const tweetLink = tweetLinkElement.href;
        const tweetParts = tweetLink.match(/https:\/\/(?:x\.com|twitter\.com)\/([^\/]+)\/status\/(\d+)/);
        const tweetId = tweetParts[2];

        // CSS selectors to determine if media is present
        const mediaSelectors = [
            'a[href*="/photo/1"]',
            'div[role="progressbar"]',
            'button[data-testid="playButton"]',
        ];

        // Check if any media selectors are present
        const hasMedia = mediaSelectors.some(selector => tweetElement.querySelector(selector));

        if (!hasMedia) return;

        // Store the tweetId in a data attribute on the button
        if (tweetElement.querySelector('.download-media-btn')) return;

        const button = document.createElement('button');
        button.innerText = 'Download Media';
        button.className = 'download-media-btn';
        button.style.marginLeft = '10px';
        button.dataset.tweetId = tweetId; // Store the tweet ID
        button.addEventListener('click', () => onDownloadButtonClick(button));

        const actionBar = tweetElement.querySelector('[role="group"]');
        if (actionBar) {
            actionBar.appendChild(button);
        }
    }

    async function onDownloadButtonClick(button) {
        const tweetId = button.dataset.tweetId;

        // Fetch the tweet data when the button is clicked
        const mediaData = await fetchTweetData(tweetId);

        if (mediaData.length === 0) {
            console.warn('No media found for this tweet.');
            return;
        }

        // Find the tweet element containing the button
        const tweetElement = button.closest('article');
        if (tweetElement) {
            await downloadMedia(tweetElement, mediaData);
        }
    }

    const observer = new MutationObserver(mutations => {
        for (const mutation of mutations) {
            for (const addedNode of mutation.addedNodes) {
                if (addedNode.nodeType === Node.ELEMENT_NODE) {
                    const tweetElements = addedNode.matches('article') ? [addedNode] : addedNode.querySelectorAll('article');
                    tweetElements.forEach(tweetElement => addDownloadButton(tweetElement));
                }
            }
        }
    });

    observer.observe(document.body, {
        childList: true,
        subtree: true
    });
})();