☰

GitZip Lite

Download selected files and folders from GitHub repositories.

目前为 2025-02-12 提交的版本。查看 最新版本。

// ==UserScript==
// @name         GitZip Lite
// @icon         https://www.google.com/s2/favicons?sz=64&domain=github.com
// @namespace    https://github.com/tizee/tempermonkey-gitzip-lite
// @version      1.5
// @description  Download selected files and folders from GitHub repositories.
// @author       tizee
// @match        https://github.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
// @require      https://unpkg.com/[email protected]/dist/powerglitch.min.js
// @grant        GM_xmlhttpRequest
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_addStyle
// @grant        GM_registerMenuCommand
// @connect      api.github.com
// @connect      raw.githubusercontent.com
// @run-at       document-end
// ==/UserScript==

(function () {
    "use strict";

    const itemCollectSelector =
          "div.js-navigation-item, table tbody tr.react-directory-row > td[class$='cell-large-screen']";
    const tokenKey = "githubApiToken";

    const { parseRepoURL, getGitURL, getInfoURL } = {
        parseRepoURL: (repoUrl) => {
            const repoExp = new RegExp(
                "^https://github.com/([^/]+)/([^/]+)(/(tree|blob)/([^/]+)(/(.*))?)?"
            );
            const matches = repoUrl.match(repoExp);

            if (!matches || matches.length === 0) return null;

            const author = matches[1];
            const project = matches[2];
            const branch = matches[5];
            const type = matches[4];
            const path = matches[7] || "";

            const rootUrl = branch
            ? `https://github.com/${author}/${project}/tree/${branch}`
        : `https://github.com/${author}/${project}`;

            if (!type && repoUrl.length - rootUrl.length > 1) {
                return null;
            }

            return {
                author,
                project,
                branch,
                type,
                path,
                inputUrl: repoUrl,
                rootUrl,
            };
        },
        getGitURL: (author, project, type, sha) => {
            if (type === "blob" || type === "tree") {
                const pluralType = type + "s";
                return `https://api.github.com/repos/${author}/${project}/git/${pluralType}/${sha}`;
            }
            return null;
        },
        getInfoURL: (author, project, path, branch) => {
            let url = `https://api.github.com/repos/${author}/${project}/contents/${path}`;
            if (branch) {
                url += `?ref=${branch}`;
            }
            return url;
        },
    };

    // --- GitZip Functions ---

    function base64toBlob(base64Data, contentType) {
        contentType = contentType || "";
        const sliceSize = 1024;
        const byteCharacters = atob(base64Data);
        const bytesLength = byteCharacters.length;
        const slicesCount = Math.ceil(bytesLength / sliceSize);
        const byteArrays = new Array(slicesCount);

        for (let sliceIndex = 0; sliceIndex < slicesCount; ++sliceIndex) {
            const begin = sliceIndex * sliceSize;
            const end = Math.min(begin + sliceSize, bytesLength);

            const bytes = new Array(end - begin);
            for (let offset = begin, i = 0; offset < end; ++i, ++offset) {
                bytes[i] = byteCharacters[offset].charCodeAt(0);
            }
            byteArrays[sliceIndex] = new Uint8Array(bytes);
        }
        return new Blob(byteArrays, { type: contentType });
    }

    function callAjax(url, token) {
        return new Promise(function (resolve, reject) {
            GM_xmlhttpRequest({
                method: "GET",
                url: url,
                headers: {
                    Authorization: token ? "token " + token : undefined,
                    Accept: "application/json",
                },
                onload: function (response) {
                    if (response.status >= 200 && response.status < 300) {
                        try {
                            const jsonResponse = JSON.parse(response.responseText);
                            resolve({ response: jsonResponse });
                        } catch (e) {
                            console.debug("Error parsing JSON:", e);
                            reject(e);
                        }
                    } else {
                        console.debug("Request failed with status:", response.status);
                        reject(response);
                    }
                },
                onerror: function (error) {
                    console.debug("Request failed:", error);
                    reject(error);
                },
            });
        });
    }

    // New dedicated function for binary downloads
    function downloadFile(url, token) {
        return new Promise(function (resolve, reject) {
            GM_xmlhttpRequest({
                method: "GET",
                url: url,
                responseType: "arraybuffer",
                headers: {
                    Authorization: token ? "token " + token : undefined,
                    Accept: "application/octet-stream",
                },
                onload: function (response) {
                    if (response.status >= 200 && response.status < 300) {
                        resolve(new Uint8Array(response.response));
                    } else {
                        reject(new Error(`Download failed: ${response.status}`));
                    }
                },
                onerror: reject,
            });
        });
    }

    // --- End GitZip Functions ---

    function addCheckboxes() {
        const fileRows = document.querySelectorAll(itemCollectSelector);
        fileRows.forEach((row) => {
            if (row.querySelector(".gitziplite-check-wrap")) return;

            // Ensure the row is relatively positioned
            row.style.position = "relative";

            const checkboxContainer = document.createElement("div");
            checkboxContainer.classList.add("gitziplite-check-wrap");
            checkboxContainer.style.position = "absolute";
            checkboxContainer.style.left = "4px";
            checkboxContainer.style.top = "50%";
            checkboxContainer.style.transform = "translateY(-50%)";
            checkboxContainer.style.display = "flex";
            checkboxContainer.style.alignItems = "center";
            checkboxContainer.style.height = "100%";
            checkboxContainer.style.display = "none";

            const checkbox = document.createElement("input");
            checkbox.type = "checkbox";
            checkbox.classList.add("gitziplite-checkbox");

            checkboxContainer.appendChild(checkbox);

            // Find the first element to insert before.  Handles both file and directory rows.
            const insertBeforeElement = row.firstChild;
            if (insertBeforeElement) {
                row.insertBefore(checkboxContainer, insertBeforeElement);
            } else {
                row.appendChild(checkboxContainer); // Fallback if no children exist
            }

            // Add event listeners for hover
            row.addEventListener("mouseenter", () => {
                checkboxContainer.style.display = "flex";
            });

            row.addEventListener("mouseleave", () => {
                if (!checkbox.checked) {
                    checkboxContainer.style.display = "none";
                }
            });

            row.addEventListener("dblclick", () => {
                console.debug("double click", row, checkbox);
                if (checkbox.checked) {
                    checkboxContainer.style.display = "none";
                } else {
                    checkboxContainer.style.display = "flex";
                }
                checkbox.checked = !checkbox.checked;
                checkbox.dispatchEvent(new Event("change"));
            });

            // Add event listener for checkbox change
            checkbox.addEventListener("change", () => {
                let link;
                if (row.tagName === "TD") {
                    link = row.querySelector("a[href]");
                } else {
                    link = row.querySelector("a[href]");
                }

                if (link) {
                    const title = link.textContent.trim();
                    const message = checkbox.checked
                    ? `Selected: ${title}`
            : `Unselected: ${title}`;
                    logMessage(message);
                }
            });
        });
    }

    let logWindow;
    let logToggleButton;
    let downloadButton;

    // Define default button styles
    const defaultButtonStyle = `
        border: 1px solid #ccc;
        padding: 0.3rem 0.6rem;
        cursor: pointer;
        margin-bottom: 0.5rem;
    `;

    function createDownloadButton() {
        // Main container
        const mainContainer = document.createElement("div");
        mainContainer.style.position = "fixed";
        mainContainer.style.bottom = "1rem";
        mainContainer.style.right = "1rem";
        mainContainer.style.zIndex = "1000";

        // Log Window
        logWindow = document.createElement("textarea");
        logWindow.setAttribute("aria-label", "Log Window");
        logWindow.style.width = "100%";
        logWindow.style.height = "10rem";
        logWindow.style.marginBottom = "0.5rem";
        logWindow.style.resize = "none";
        logWindow.style.overflow = "auto";
        logWindow.readOnly = true;
        logWindow.hidden = true;
        logWindow.style.border = "1px solid #ccc";
        logWindow.style.padding = "0.2rem";

        // Log Toggle Button
        logToggleButton = document.createElement("button");
        logToggleButton.textContent = "Show Log";
        logToggleButton.style.cssText = defaultButtonStyle;
        logToggleButton.addEventListener("click", () => {
            logWindow.hidden = !logWindow.hidden;
            logToggleButton.textContent = logWindow.hidden ? "Show Log" : "Hide Log";
        });

        // Download Button
        downloadButton = document.createElement("button");
        downloadButton.textContent = "Download Selected";
        downloadButton.style.cssText = defaultButtonStyle;
        downloadButton.addEventListener("click", downloadSelected);

        // Assemble the UI
        const form = document.createElement("div");
        form.style.display = "flex";
        form.style.flexDirection = "column";
        form.style.gap = "0.5rem";

        form.appendChild(logToggleButton);
        form.appendChild(logWindow);
        form.appendChild(downloadButton);

        mainContainer.appendChild(form);
        document.body.appendChild(mainContainer);
    }

    function logMessage(message) {
        logWindow.value += message + "\n";
        logWindow.scrollTop = logWindow.scrollHeight; // Auto-scroll to bottom
    }

    /**
   * Collects selected files and folders from the DOM.
   * @returns {{files: [], folders: []}} - An object containing arrays of selected files and folders.
   */
    function collectSelectedItems() {
        const selectedFiles = [];
        const selectedFolders = [];
        const checkboxes = document.querySelectorAll(
            ".gitziplite-checkbox:checked"
        );

        checkboxes.forEach((checkbox) => {
            const row = checkbox.parentNode.parentNode; // Direct parent access
            if (!row) {
                console.warn("Could not find a parent row for a selected checkbox.");
                return; // Skip to the next checkbox
            }
            console.debug(row);
            let link;

            if (row.tagName === "TD") {
                link = row.querySelector("a[href]");
            } else {
                link = row.querySelector("a[href]");
            }

            if (link) {
                const href = link.href;
                const title = link.textContent.trim();
                const resolved = parseRepoURL(href);
                if (resolved && resolved.type === "blob") {
                    selectedFiles.push({ href: href, title: title });
                } else if (resolved && resolved.type === "tree") {
                    selectedFolders.push({ href: href, title: title });
                }
            }
        });

        return { files: selectedFiles, folders: selectedFolders };
    }

    /**
   * Zips the given contents and triggers a download.
   * @param {Array<{path: string, content: string}>} allContents - Array of file contents to zip.
   * @param {object} resolvedUrl - Parsed URL information of the repository.
   */
    function zipAndDownload(allContents, resolvedUrl) {
        if (allContents.length === 1) {
            // Handle single file download
            const singleItem = allContents[0];
            console.debug(singleItem);
            if (singleItem.isBinary) {
                // Create Blob directly from Uint8Array
                const blob = new Blob([singleItem.content], {
                    type: "application/octet-stream",
                });
                saveAs(blob, singleItem.path);
            } else {
                // Handle base64 encoded text files
                const blob = base64toBlob(singleItem.content, "");
                saveAs(blob, singleItem.path);
            }
        } else {
            // Handle zip archive creation
            try {
                const currDate = new Date();
                const dateWithOffset = new Date(
                    currDate.getTime() - currDate.getTimezoneOffset() * 60000
                );
                window.JSZip.defaults.date = dateWithOffset;

                const zip = new window.JSZip();
                allContents.forEach((item) => {
                    if (item.isBinary) {
                        // Add binary file as Uint8Array
                        zip.file(item.path, item.content, {
                            createFolders: true,
                            binary: true,
                            date: dateWithOffset,
                        });
                    } else {
                        // Add base64 encoded file
                        zip.file(item.path, item.content, {
                            createFolders: true,
                            base64: true,
                            date: dateWithOffset,
                        });
                    }
                });

                zip.generateAsync({ type: "blob" }).then((content) => {
                    saveAs(
                        content,
                        [resolvedUrl.project]
                        .concat(resolvedUrl.path.split("/"))
                        .join("-") + ".zip"
                    );
                });
            } catch (error) {
                console.debug("Error zipping files:", error);
                logMessage("Error zipping files.");
            }
        }
    }

    async function downloadSelected() {
        const { files: selectedFiles, folders: selectedFolders } =
              collectSelectedItems();

        if (selectedFiles.length === 0 && selectedFolders.length === 0) {
            logMessage("No files or folders selected.");
            return;
        }

        const resolvedUrl = parseRepoURL(window.location.href);
        if (!resolvedUrl) {
            logMessage("Could not resolve repository URL.");
            return;
        }

        const githubToken = GM_getValue(tokenKey);

        if (!githubToken) {
            logMessage(
                "GitHub API token is not set. Please set it in the Tampermonkey dashboard."
            );
            return;
        }

        const allContents = [];

        async function processFolder(folder, pathPrefix = "") {
            logMessage(`Processing folder: ${folder.title}`);
            const folderResolvedUrl = parseRepoURL(folder.href);
            const apiUrl = getInfoURL(
                folderResolvedUrl.author,
                folderResolvedUrl.project,
                folderResolvedUrl.path,
                folderResolvedUrl.branch
            );

            try {
                const xmlResponse = await callAjax(apiUrl, githubToken);
                const folderContents = xmlResponse.response;

                for (const item of folderContents) {
                    const itemPath = pathPrefix + "/" + item.name;
                    if (item.type === "file") {
                        logMessage(`Processing file: ${itemPath}`);
                        const fileInfoUrl = getInfoURL(
                            folderResolvedUrl.author,
                            folderResolvedUrl.project,
                            folderResolvedUrl.path + "/" + item.name,
                            folderResolvedUrl.branch
                        );
                        const fileXmlResponse = await callAjax(fileInfoUrl, githubToken);
                        const fileContent = fileXmlResponse.response;
                        allContents.push({
                            path: itemPath,
                            content: fileContent.content,
                        });
                    } else if (item.type === "dir") {
                        await processFolder(
                            { href: folder.href + "/" + item.name, title: item.name },
                            itemPath
                        );
                    }
                }
            } catch (error) {
                console.debug("Error fetching folder:", folder.title, error);
                logMessage(`Error fetching folder: ${folder.title}`);
            }
        }

        for (const folder of selectedFolders) {
            await processFolder(folder, folder.title);
        }

        for (const file of selectedFiles) {
            logMessage(`Processing file: ${file.title}`);
            const fileResolvedUrl = parseRepoURL(file.href);
            const infoUrl = getInfoURL(
                fileResolvedUrl.author,
                fileResolvedUrl.project,
                fileResolvedUrl.path,
                fileResolvedUrl.branch
            );
            logMessage(`file info url: ${infoUrl}`);
            console.debug(`file info url: ${infoUrl}`);
            try {
                const xmlResponse = await callAjax(infoUrl, githubToken);
                const fileContent = xmlResponse.response;

                if (fileContent.encoding === "base64" && fileContent.content) {
                    allContents.push({
                        path: file.title,
                        content: fileContent.content,
                        isBinary: false,
                    });
                } else if (fileContent.download_url) {
                    // Handle binary file with dedicated download function
                    const binaryData = await downloadFile(
                        fileContent.download_url,
                        githubToken
                    );
                    allContents.push({
                        path: file.title,
                        content: binaryData,
                        isBinary: true,
                    });
                }
            } catch (error) {
                console.debug("Error fetching file:", file.title, error);
                logMessage(`Error fetching file: ${file.title}`);
                return;
            }
        }

        zipAndDownload(allContents, resolvedUrl);
        logMessage("Download complete.");
    }

    // Register menu command for setting token
    GM_registerMenuCommand("Set GitHub API Token", () => {
        const token = prompt("Enter your GitHub API token:");
        if (token) {
            GM_setValue(tokenKey, token);
            alert("Token saved successfully!");
        }
    });

    function onDomLoaded() {
        addCheckboxes();
        createDownloadButton();
    }

    function onUrlChange() {
        addCheckboxes();
    }

    // Initialize
    onDomLoaded();
    // Glitch Animation
    PowerGlitch.glitch(logToggleButton, {
        playMode: "click",
        timing: {
            duration: 400,
            easing: "ease-in-out",
        },
        shake: {
            velocity: 20,
            amplitudeX: 0,
            amplitudeY: 0.1,
        },
    });
    PowerGlitch.glitch(logWindow, {
        playMode: "hover",
        timing: {
            duration: 450,
            easing: "ease-in-out",
        },
        shake: {
            velocity: 20,
            amplitudeX: 0.1,
            amplitudeY: 0,
        },
    });
    PowerGlitch.glitch(downloadButton, {
        playMode: "click",
        timing: {
            duration: 400,
            easing: "ease-in-out",
        },
    });

    // Observe GitHub repository page URL changes (e.g., navigating into a new directory)
    const observer = new MutationObserver(onUrlChange);
    observer.observe(document.body, { childList: true, subtree: true });
})();

QingJ © 2025

镜像随时可能失效，请加Q群300939539或关注我们的公众号极客氢云获取最新地址