☰

Repo Gist

Provides GitHub repositories as additional context.

当前为 2025-06-04 提交的版本，查看 最新版本。

  1. // ==UserScript==
  2. // @name         Repo Gist
  3. // @namespace    https://github.com/prudentbird
  4. // @version      0.0.1
  5. // @description  Provides GitHub repositories as additional context.
  6. // @author       Prudent Bird
  7. // @match        https://t3.chat/*
  8. // @match        https://beta.t3.chat/*
  9. // @icon         https://www.google.com/s2/favicons?sz=64&domain=t3.chat
  10. // @grant        GM_getValue
  11. // @grant        GM_setValue
  12. // @grant        unsafeWindow
  13. // @grant        GM_xmlhttpRequest
  14. // @grant        GM_registerMenuCommand
  15. // @run-at       document-idle
  16. // @connect      https://generativelanguage.googleapis.com
  17. // @license      MIT
  18. // ==/UserScript==
  19.  
  20. (function () {
  21.   "use strict";
  22.  
  23.   // --- Configuration and State ---
  24.   let debugMode = false;
  25.   const DB_VERSION = 1;
  26.   const SCRIPT_VERSION = "0.1.0";
  27.   const SCRIPT_NAME = "Repo Gist";
  28.   const DB_NAME = "t3chat_repogist_db";
  29.   const STORE_NAME = "repogist_states";
  30.   const githubSVG = `<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-github-icon lucide-github"><path d="M15 22v-4a4.8 4.8 0 0 0-1-3.5c3 0 6-2 6-5.5.08-1.25-.27-2.48-1-3.5.28-1.15.28-2.35 0-3.5 0 0-1 0-3 1.5-2.64-.5-5.36-.5-8 0C6 2 5 2 5 2c-.3 1.15-.3 2.35 0 3.5A5.403 5.403 0 0 0 4 9c0 3.5 3 5.5 6 5.5-.39.49-.68 1.05-.85 1.65-.17.6-.22 1.23-.15 1.85v4"/><path d="M9 18c-4.51 2-5-2-7-2"/></svg>`;
  31.  
  32.   const GM_STORAGE_KEYS = {
  33.     DEBUG: "debug",
  34.     API_URL: "apiUrl",
  35.     GEMINI_API_KEY: "geminiApiKey",
  36.   };
  37.  
  38.   // --- Utility: Logger ---
  39.   const Logger = {
  40.     log: (...args) => {
  41.       if (debugMode) console.log(`[${SCRIPT_NAME}]`, ...args);
  42.     },
  43.     error: (...args) => console.error(`[${SCRIPT_NAME}]`, ...args),
  44.   };
  45.  
  46.   const getChatId = () => {
  47.     const currentUrl = window.location.href;
  48.     const match = currentUrl.match(/\/chat\/([^/?#]+)/);
  49.     const chatId = match ? match[1] : null;
  50.     if (!chatId) {
  51.       Logger.log("getChatId: No chat ID found in URL", currentUrl);
  52.     }
  53.     return chatId;
  54.   };
  55.  
  56.   let apiUrl = null;
  57.   let geminiApiKey = null;
  58.  
  59.   const ApiKeyModal = {
  60.     _isShown: false,
  61.     _isValidURL: (url) => {
  62.       try {
  63.         new URL(url);
  64.         return true;
  65.       } catch (e) {
  66.         return false;
  67.       }
  68.     },
  69.     show: () => {
  70.       if (document.getElementById(UI_IDS.apiKeyModal) || ApiKeyModal._isShown)
  71.         return;
  72.       ApiKeyModal._isShown = true;
  73.       const wrapper = document.createElement("div");
  74.       wrapper.id = UI_IDS.apiKeyModal;
  75.       wrapper.innerHTML = `
  76.     <div id="${UI_IDS.apiKeyModalContent}">
  77.       <div id="${UI_IDS.apiKeyModalHeader}">
  78.         <div><!-- Icon container -->
  79. <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-cog-icon lucide-cog"><path d="M12 20a8 8 0 1 0 0-16 8 8 0 0 0 0 16Z"/><path d="M12 14a2 2 0 1 0 0-4 2 2 0 0 0 0 4Z"/><path d="M12 2v2"/><path d="M12 22v-2"/><path d="m17 20.66-1-1.73"/><path d="M11 10.27 7 3.34"/><path d="m20.66 17-1.73-1"/><path d="m3.34 7 1.73 1"/><path d="M14 12h8"/><path d="M2 12h2"/><path d="m20.66 7-1.73 1"/><path d="m3.34 17 1.73-1"/><path d="m17 3.34-1 1.73"/><path d="m11 13.73-4 6.93"/></svg>
  80.         </div>
  81.         <div>Enter API Configuration</div><!-- Title -->
  82.         <button id="${UI_IDS.apiKeyModalCloseButton}"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-x-icon lucide-x"><path d="M18 6 6 18"/><path d="m6 6 12 12"/></svg></button>
  83.       </div>
  84.       <div id="${UI_IDS.apiKeyModalInputContainer}">
  85.         <label for="${UI_IDS.apiKeyModalInput}">RepoGist API URL</label>
  86.         <input id="${UI_IDS.apiKeyModalInput}" type="text" placeholder="https://api.repogist.com/ingest" />
  87.         <button id="${UI_IDS.apiKeyModalClearButton}" aria-label="Clear input"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-trash2-icon lucide-trash-2"><path d="M3 6h18"/><path d="M19 6v14c0 1-1 2-2 2H7c-1 0-2-1-2-2V6"/><path d="M8 6V4c0-1 1-2 2-2h4c1 0 2 1 2 2v2"/><line x1="10" x2="10" y1="11" y2="17"/><line x1="14" x2="14" y1="11" y2="17"/></svg></button>
  88.       </div>
  89.       <div id="${UI_IDS.apiKeyModalInputContainer}">
  90.         <label for="${UI_IDS.geminiApiKeyInput}">Gemini API Key</label>
  91.         <input id="${UI_IDS.geminiApiKeyInput}" type="text" placeholder="Enter your Gemini API key" />
  92.         <button id="${UI_IDS.geminiApiKeyClearButton}" aria-label="Clear input"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-trash2-icon lucide-trash-2"><path d="M3 6h18"/><path d="M19 6v14c0 1-1 2-2 2H7c-1 0-2-1-2-2V6"/><path d="M8 6V4c0-1 1-2 2-2h4c1 0 2 1 2 2v2"/><line x1="10" x2="10" y1="11" y2="17"/><line x1="14" x2="14" y1="11" y2="17"/></svg></button>
  93.       </div>
  94.       <button id="${UI_IDS.apiKeyModalSaveButton}">Save</button>
  95.     </div>`;
  96.       document.body.appendChild(wrapper);
  97.       const input = wrapper.querySelector(`#${UI_IDS.apiKeyModalInput}`);
  98.       if (input) {
  99.         input.focus();
  100.       }
  101.       ApiKeyModal._attachEventListeners(wrapper);
  102.     },
  103.     _attachEventListeners: (modalElement) => {
  104.       const urlInput = modalElement.querySelector(
  105.         `#${UI_IDS.apiKeyModalInput}`
  106.       );
  107.       const geminiKeyInput = modalElement.querySelector(
  108.         `#${UI_IDS.geminiApiKeyInput}`
  109.       );
  110.       const saveButton = modalElement.querySelector(
  111.         `#${UI_IDS.apiKeyModalSaveButton}`
  112.       );
  113.       const closeButton = modalElement.querySelector(
  114.         `#${UI_IDS.apiKeyModalCloseButton}`
  115.       );
  116.       const clearButton = modalElement.querySelector(
  117.         `#${UI_IDS.apiKeyModalClearButton}`
  118.       );
  119.       const geminiClearButton = modalElement.querySelector(
  120.         `#${UI_IDS.geminiApiKeyClearButton}`
  121.       );
  122.  
  123.       modalElement.addEventListener("click", (e) => {
  124.         if (e.target === modalElement) {
  125.           ApiKeyModal._isShown = false;
  126.           modalElement.remove();
  127.         }
  128.       });
  129.  
  130.       const updateClearButtonVisibility = (input, button) => {
  131.         if (button) {
  132.           button.style.display = input.value ? "flex" : "none";
  133.         }
  134.       };
  135.  
  136.       updateClearButtonVisibility(urlInput, clearButton);
  137.       updateClearButtonVisibility(geminiKeyInput, geminiClearButton);
  138.  
  139.       if (clearButton) {
  140.         clearButton.addEventListener("click", () => {
  141.           urlInput.value = "";
  142.           urlInput.focus();
  143.           updateClearButtonVisibility(urlInput, clearButton);
  144.         });
  145.       }
  146.  
  147.       if (geminiClearButton) {
  148.         geminiClearButton.addEventListener("click", () => {
  149.           geminiKeyInput.value = "";
  150.           geminiKeyInput.focus();
  151.           updateClearButtonVisibility(geminiKeyInput, geminiClearButton);
  152.         });
  153.       }
  154.  
  155.       urlInput.addEventListener("input", () =>
  156.         updateClearButtonVisibility(urlInput, clearButton)
  157.       );
  158.       geminiKeyInput.addEventListener("input", () =>
  159.         updateClearButtonVisibility(geminiKeyInput, geminiClearButton)
  160.       );
  161.  
  162.       const handleSave = () => {
  163.         const url = urlInput.value.trim();
  164.         const geminiKey = geminiKeyInput.value.trim();
  165.  
  166.         if (url && !ApiKeyModal._isValidURL(url)) {
  167.           alert("Invalid RepoGist API URL");
  168.           return;
  169.         }
  170.  
  171.         if (url) {
  172.           GM_setValue(GM_STORAGE_KEYS.API_URL, url)
  173.             .then(() => {
  174.               apiUrl = url;
  175.               if (geminiKey) {
  176.                 return GM_setValue(GM_STORAGE_KEYS.GEMINI_API_KEY, geminiKey);
  177.               }
  178.             })
  179.             .then(() => {
  180.               if (geminiKey) {
  181.                 geminiApiKey = geminiKey;
  182.               }
  183.               ApiKeyModal._isShown = false;
  184.               modalElement.remove();
  185.             })
  186.             .catch((err) => {
  187.               Logger.error("Failed to save API configuration:", err);
  188.             });
  189.         } else {
  190.           ApiKeyModal._isShown = false;
  191.           modalElement.remove();
  192.         }
  193.       };
  194.  
  195.       saveButton.addEventListener("click", handleSave);
  196.  
  197.       urlInput.addEventListener("keydown", (e) => {
  198.         if (e.key === "Enter") {
  199.           handleSave();
  200.         }
  201.       });
  202.  
  203.       geminiKeyInput.addEventListener("keydown", (e) => {
  204.         if (e.key === "Enter") {
  205.           handleSave();
  206.         }
  207.       });
  208.  
  209.       closeButton.addEventListener("click", () => {
  210.         ApiKeyModal._isShown = false;
  211.         modalElement.remove();
  212.       });
  213.     },
  214.   };
  215.  
  216.   const getRepoNamefromURL = (url) => {
  217.     if (typeof url !== "string") {
  218.       Logger.log("getRepoNamefromURL: Invalid URL type", typeof url);
  219.       return null;
  220.     }
  221.     url = url.replace(/\/+$/, "");
  222.     let match = url.match(/[:\/]([^\/]+)\.git$/);
  223.     if (match) {
  224.       Logger.log("getRepoNamefromURL: Found .git URL match", match[1]);
  225.       return match[1];
  226.     }
  227.     match = url.match(/\/([^\/]+)(?:\/tree\/[^\/]+)?$/);
  228.     if (match) {
  229.       Logger.log("getRepoNamefromURL: Found standard URL match", match[1]);
  230.       return match[1];
  231.     }
  232.     Logger.log("getRepoNamefromURL: No match found for URL", url);
  233.     return null;
  234.   };
  235.  
  236.   const selectors = {
  237.     messageActions: "div.ml-\\[-7px\\].flex.items-center.gap-1",
  238.     searchButton: 'button#search-toggle[aria-label="Enable search"]',
  239.   };
  240.  
  241.   const UI_IDS = {
  242.     apiKeyModal: "api-key-modal",
  243.     apiKeyModalContent: "api-key-modal-content",
  244.     apiKeyModalHeader: "api-key-modal-header",
  245.     apiKeyModalInput: "api-key-modal-input",
  246.     apiKeyModalInputContainer: "api-key-modal-input-container",
  247.     apiKeyModalSaveButton: "api-key-modal-save-button",
  248.     apiKeyModalCloseButton: "api-key-modal-close-button",
  249.     apiKeyModalClearButton: "api-key-modal-clear-button",
  250.     geminiApiKeyInput: "gemini-api-key-input",
  251.     geminiApiKeyClearButton: "gemini-api-key-clear-button",
  252.     importButton: "import-button",
  253.     searchToggle: "search-toggle",
  254.     repoUrlModal: "repo-url-modal",
  255.     repoUrlModalContent: "repo-url-modal-content",
  256.     repoUrlModalHeader: "repo-url-modal-header",
  257.     repoUrlModalDescription: "repo-url-modal-description",
  258.     repoUrlModalInput: "repo-url-modal-input",
  259.     repoUrlModalSaveButton: "repo-url-modal-save-button",
  260.     repoUrlModalCloseButton: "repo-url-modal-close-button",
  261.     repoUrlModalClearButton: "repo-url-modal-clear-button",
  262.     repoUrlModalInputContainer: "repo-url-modal-input-container",
  263.     styleElement: "repo-gist-style",
  264.   };
  265.  
  266.   const CSS_CLASSES = {
  267.     button:
  268.       "inline-flex items-center justify-center whitespace-nowrap font-medium transition-colors focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:opacity-50 [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0 disabled:cursor-not-allowed hover:bg-muted/40 hover:text-foreground disabled:hover:bg-transparent disabled:hover:text-foreground/50 px-3 text-xs -mb-1.5 h-auto gap-2 rounded-full border border-solid border-secondary-foreground/10 py-1.5 pl-2 pr-2.5 text-muted-foreground",
  269.     importButtonLoading: "loading",
  270.     importButtonOn: "on",
  271.   };
  272.  
  273.   const StyleManager = {
  274.     injectGlobalStyles: () => {
  275.       if (document.getElementById(UI_IDS.styleElement)) return;
  276.       const styleEl = document.createElement("style");
  277.       styleEl.id = UI_IDS.styleElement;
  278.       styleEl.textContent = `
  279.       /* Button toggle animation */
  280.       #${UI_IDS.importButton} { position: relative; overflow: hidden; transition: color 0.3s ease; }
  281.       #${UI_IDS.importButton}::before { content: ''; position: absolute; inset: 0; background-color: rgba(219,39,119,0.15); transform: scaleX(0); transform-origin: left; transition: transform 0.3s ease; z-index:-1; }
  282.       #${UI_IDS.importButton}.${CSS_CLASSES.importButtonOn}::before { transform: scaleX(1); }
  283.       #${UI_IDS.importButton} svg { transition: transform 0.3s ease; }
  284.       #${UI_IDS.importButton}.${CSS_CLASSES.importButtonOn} svg { transform: rotate(360deg); }
  285.       
  286.       /* Loading state */
  287.       #${UI_IDS.importButton}.${CSS_CLASSES.importButtonLoading} { opacity: 0.6; position: relative; }
  288.       #${UI_IDS.importButton}.${CSS_CLASSES.importButtonLoading}::after { content: ''; position: absolute; top:50%; left:50%; width:12px; height:12px; margin:-6px 0 0 -6px; border:2px solid currentColor; border-radius:50%; border-top-color:transparent; animation:spin 1s linear infinite; }
  289.       @keyframes spin { to { transform: rotate(360deg); } }
  290.  
  291.       /* Repo URL Modal Styles */
  292.       #${UI_IDS.repoUrlModal} {
  293.         position: fixed;
  294.         inset: 0;
  295.         background: rgba(0,0,0,0.7);
  296.         display: flex;
  297.         align-items: center;
  298.         justify-content: center;
  299.         z-index: 9999;
  300.       }
  301.       #${UI_IDS.repoUrlModalContent} {
  302.         background: #1c1c1e;
  303.         padding: 24px;
  304.         border-radius: 12px;
  305.         width: 500px;
  306.         max-width: 95vw;
  307.         box-sizing: border-box;
  308.         box-shadow: 0 8px 24px rgba(0,0,0,0.3);
  309.       }
  310.       @media (max-width: 600px) {
  311.         #${UI_IDS.repoUrlModalContent} {
  312.           width: 95vw;
  313.           padding: 16px;
  314.         }
  315.       }
  316.       #${UI_IDS.repoUrlModalHeader} {
  317.         display: flex;
  318.         align-items: center;
  319.         margin-bottom: 20px;
  320.         position: relative;
  321.       }
  322.       #${UI_IDS.repoUrlModalHeader} > div:first-child { /* Icon container */
  323.         color: #c62a88;
  324.         margin-right: 12px;
  325.       }
  326.       #${UI_IDS.repoUrlModalHeader} > div:last-child { /* Title container */
  327.         font-size: 22px;
  328.         font-weight: 600;
  329.         color: #fff;
  330.       }
  331.       #${UI_IDS.repoUrlModalCloseButton} {
  332.         position: absolute;
  333.         top: 0;
  334.         right: 0;
  335.         background: none;
  336.         border: none;
  337.         cursor: pointer;
  338.         color: #fff;
  339.         font-size: 24px;
  340.         transition: color 0.3s ease;
  341.       }
  342.       #${UI_IDS.repoUrlModalDescription} {
  343.         color: #999;
  344.         font-size: 14px;
  345.         margin-bottom: 16px;
  346.       }
  347.       #${UI_IDS.repoUrlModalInputContainer} {
  348.         position: relative;
  349.         width: 100%;
  350.         margin-bottom: 16px;
  351.       }
  352.       #${UI_IDS.repoUrlModalInput} {
  353.         width: 100%;
  354.         padding: 12px 36px 12px 12px;
  355.         box-sizing: border-box;
  356.         background: #2a2a2c;
  357.         color: #fff;
  358.         border: 1px solid #333;
  359.         border-radius: 6px;
  360.         outline: none;
  361.         font-size: 14px;
  362.       }
  363.       #${UI_IDS.repoUrlModalClearButton} {
  364.         position: absolute;
  365.         top: 50%;
  366.         right: 10px;
  367.         transform: translateY(-50%);
  368.         width: 20px;
  369.         height: 20px;
  370.         display: flex;
  371.         align-items: center;
  372.         justify-content: center;
  373.         background: none;
  374.         border: none;
  375.         cursor: pointer;
  376.         font-size: 16px;
  377.         font-weight: 500;
  378.         color: #aaa;
  379.         transition: color 0.2s ease;
  380.         padding: 0;
  381.       }
  382.       #${UI_IDS.repoUrlModalClearButton}:hover {
  383.         color: #c62a88;
  384.       }
  385.       #${UI_IDS.repoUrlModalSaveButton} {
  386.         width: 100%;
  387.         padding: 12px;
  388.         background: #a02553;
  389.         border: none;
  390.         border-radius: 6px;
  391.         color: white;
  392.         cursor: pointer;
  393.         font-size: 15px;
  394.         font-weight: 500;
  395.         transition: all 0.2s ease;
  396.       }
  397.       #${UI_IDS.repoUrlModalSaveButton}:hover {
  398.         background: #c62a88;
  399.       }
  400.  
  401.       /* API Key Modal Styles */
  402.        #${UI_IDS.apiKeyModal} {
  403.         position: fixed;
  404.         inset: 0;
  405.         background: rgba(0,0,0,0.7);
  406.         display: flex;
  407.         align-items: center;
  408.         justify-content: center;
  409.         z-index: 9999;
  410.       }
  411.       #${UI_IDS.apiKeyModalContent} {
  412.         background: #1c1c1e;
  413.         padding: 24px;
  414.         border-radius: 12px;
  415.         width: 500px;
  416.         max-width: 95vw;
  417.         box-sizing: border-box;
  418.         box-shadow: 0 8px 24px rgba(0,0,0,0.3);
  419.       }
  420.       @media (max-width: 600px) {
  421.         #${UI_IDS.apiKeyModalContent} {
  422.           width: 95vw;
  423.           padding: 16px;
  424.         }
  425.       }
  426.       #${UI_IDS.apiKeyModalHeader} {
  427.         display: flex;
  428.         align-items: center;
  429.         margin-bottom: 20px;
  430.         position: relative;
  431.       }
  432.       #${UI_IDS.apiKeyModalHeader} > div:first-child { /* Icon container */
  433.         color: #c62a88;
  434.         margin-right: 12px;
  435.       }
  436.       #${UI_IDS.apiKeyModalHeader} > div:last-child { /* Title container */
  437.         font-size: 22px;
  438.         font-weight: 600;
  439.         color: #fff;
  440.       }
  441.       #${UI_IDS.apiKeyModalCloseButton} {
  442.         position: absolute;
  443.         top: 0;
  444.         right: 0;
  445.         background: none;
  446.         border: none;
  447.         cursor: pointer;
  448.         color: #fff;
  449.         font-size: 24px;
  450.         transition: color 0.3s ease;
  451.       }
  452.       #${UI_IDS.apiKeyModalInputContainer} {
  453.         position: relative;
  454.         width: 100%;
  455.         margin-bottom: 16px;
  456.       }
  457.       #${UI_IDS.apiKeyModalInputContainer} label {
  458.         display: block;
  459.         color: #999;
  460.         font-size: 14px;
  461.         margin-bottom: 8px;
  462.       }
  463.       #${UI_IDS.apiKeyModalInput} {
  464.         width: 100%;
  465.         padding: 12px 36px 12px 12px;
  466.         box-sizing: border-box;
  467.         background: #2a2a2c;
  468.         color: #fff;
  469.         border: 1px solid #333;
  470.         border-radius: 6px;
  471.         outline: none;
  472.         font-size: 14px;
  473.       }
  474.       #${UI_IDS.apiKeyModalInput}:focus {
  475.         border-color: #c62a88;
  476.       }
  477.       #${UI_IDS.geminiApiKeyInput} {
  478.         width: 100%;
  479.         padding: 12px 36px 12px 12px;
  480.         box-sizing: border-box;
  481.         background: #2a2a2c;
  482.         color: #fff;
  483.         border: 1px solid #333;
  484.         border-radius: 6px;
  485.         outline: none;
  486.         font-size: 14px;
  487.       }
  488.       #${UI_IDS.geminiApiKeyInput}:focus {
  489.         border-color: #c62a88;
  490.       }
  491.       #${UI_IDS.apiKeyModalClearButton}, #${UI_IDS.geminiApiKeyClearButton} {
  492.         position: absolute;
  493.         top: 68%;
  494.         right: 10px;
  495.         transform: translateY(-50%);
  496.         width: 20px;
  497.         height: 20px;
  498.         display: flex;
  499.         align-items: center;
  500.         justify-content: center;
  501.         background: none;
  502.         border: none;
  503.         cursor: pointer;
  504.         font-size: 16px;
  505.         font-weight: 500;
  506.         color: #aaa;
  507.         transition: color 0.2s ease;
  508.         padding: 0;
  509.       }
  510.       #${UI_IDS.apiKeyModalClearButton}:hover, #${UI_IDS.geminiApiKeyClearButton}:hover {
  511.         color: #c62a88;
  512.       }
  513.       #${UI_IDS.apiKeyModalSaveButton} {
  514.         width: 100%;
  515.         padding: 12px;
  516.         background: #a02553;
  517.         border: none;
  518.         border-radius: 6px;
  519.         color: white;
  520.         cursor: pointer;
  521.         font-size: 15px;
  522.         font-weight: 500;
  523.         transition: all 0.2s ease;
  524.       }
  525.       #${UI_IDS.apiKeyModalSaveButton}:hover {
  526.         background: #c62a88;
  527.       }`;
  528.       document.head.appendChild(styleEl);
  529.     },
  530.   };
  531.  
  532.   const UIManager = {
  533.     importButton: null,
  534.     _createImportButton: () => {
  535.       return new Promise((resolve) => {
  536.         const chatId = getChatId();
  537.         IngestDBManager.getState(chatId)
  538.           .then((state) => {
  539.             const button = document.createElement("button");
  540.             if (state && state.repoUrl) {
  541.               button.innerHTML = `${githubSVG} ${getRepoNamefromURL(
  542.                 state.repoUrl
  543.               )
  544.                 .split("/")
  545.                 .pop()
  546.                 .slice(0, 10)
  547.                 .replace(/^./, (c) => c.toUpperCase())}`;
  548.               button.id = UI_IDS.importButton;
  549.               button.type = "button";
  550.               button.setAttribute("aria-label", "Disable import");
  551.               button.setAttribute("data-state", "open");
  552.               button.className = CSS_CLASSES.button;
  553.               button.dataset.mode = "on";
  554.               button.classList.toggle(CSS_CLASSES.importButtonOn);
  555.             } else {
  556.               button.innerHTML = `${githubSVG} Import Repo`;
  557.               button.id = UI_IDS.importButton;
  558.               button.type = "button";
  559.               button.setAttribute("aria-label", "Enable import");
  560.               button.setAttribute("data-state", "closed");
  561.               button.className = CSS_CLASSES.button;
  562.               button.dataset.mode = "off";
  563.             }
  564.  
  565.             button.addEventListener("click", () => {
  566.               if (!apiUrl || !geminiApiKey) {
  567.                 ApiKeyModal.show();
  568.                 return;
  569.               }
  570.  
  571.               if (!chatId) {
  572.                 alert("No chat ID found, can't import repo");
  573.                 Logger.log("No chat ID found, skipping import button click");
  574.                 return;
  575.               }
  576.  
  577.               if (RepoUrlModal._isShown) {
  578.                 RepoUrlModal._isShown = false;
  579.                 return;
  580.               }
  581.  
  582.               RepoUrlModal.show();
  583.             });
  584.             resolve(button);
  585.           })
  586.           .catch((err) => {
  587.             Logger.error("Failed to create import button:", err);
  588.             resolve(null);
  589.           });
  590.       });
  591.     },
  592.     injectImportButton: () => {
  593.       return new Promise((resolve) => {
  594.         const messageActions = document.querySelectorAll(
  595.           selectors.messageActions
  596.         );
  597.  
  598.         if (!messageActions || messageActions.length === 0) {
  599.           Logger.error("Message Actions not found.");
  600.           resolve(false);
  601.           return;
  602.         }
  603.  
  604.         UIManager._createImportButton()
  605.           .then((button) => {
  606.             UIManager.importButton = button;
  607.             if (!button) {
  608.               Logger.error("Import button not found.");
  609.               resolve(false);
  610.               return;
  611.             }
  612.  
  613.             messageActions.forEach((container) => {
  614.               if (!container.querySelector(`#${UI_IDS.importButton}`)) {
  615.                 container.appendChild(button);
  616.               }
  617.             });
  618.             resolve(true);
  619.           })
  620.           .catch((err) => {
  621.             Logger.error("Failed to inject import button:", err);
  622.             resolve(false);
  623.           });
  624.       });
  625.     },
  626.   };
  627.  
  628.   const RepoUrlModal = {
  629.     _isShown: false,
  630.     _isValidRepoUrl: (url) => {
  631.       const patterns = [
  632.         /^git@[^:]+:.+\.git$/,
  633.         /^https:\/\/[^/]+\/.+\.git$/,
  634.         /^https:\/\/github\.com\/[^/]+\/[^/]+(\/tree\/[^/]+)?$/,
  635.         /^https:\/\/gitlab\.com\/[^/]+\/[^/]+(\/-\/tree\/[^/]+)?$/,
  636.       ];
  637.  
  638.       return patterns.some((pattern) => pattern.test(url));
  639.     },
  640.     show: () => {
  641.       return new Promise((resolve) => {
  642.         const chatId = getChatId();
  643.         IngestDBManager.getState(chatId)
  644.           .then((state) => {
  645.             if (
  646.               document.getElementById(UI_IDS.repoUrlModal) ||
  647.               RepoUrlModal._isShown
  648.             ) {
  649.               resolve();
  650.               return;
  651.             }
  652.             RepoUrlModal._isShown = true;
  653.             const wrapper = document.createElement("div");
  654.             wrapper.id = UI_IDS.repoUrlModal;
  655.             wrapper.innerHTML = `
  656.               <div id="${UI_IDS.repoUrlModalContent}">
  657.                 <div id="${UI_IDS.repoUrlModalHeader}">
  658.                   <div><!-- Icon container -->
  659.                    <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-folder-git2-icon lucide-folder-git-2"><path d="M9 20H4a2 2 0 0 1-2-2V5a2 2 0 0 1 2-2h3.9a2 2 0 0 1 1.69.9l.81 1.2a2 2 0 0 0 1.67.9H20a2 2 0 0 1 2 2v5"/><circle cx="13" cy="12" r="2"/><path d="M18 19c-2.8 0-5-2.2-5-5v8"/><circle cx="20" cy="19" r="2"/></svg>
  660.                   </div>
  661.                   <div>Enter Repo URL</div><!-- Title -->
  662.                   <button id="${UI_IDS.repoUrlModalCloseButton}"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-x-icon lucide-x"><path d="M18 6 6 18"/><path d="m6 6 12 12"/></svg></button>
  663.                 </div>
  664.                 <div id="${UI_IDS.repoUrlModalDescription}">Enter the URL of the GitHub repository you want to import.</div>
  665.                 <div id="${UI_IDS.repoUrlModalInputContainer}">
  666.                   <input id="${UI_IDS.repoUrlModalInput}" type="text" placeholder="https://github.com/username/repo" />
  667.                   <button id="${UI_IDS.repoUrlModalClearButton}" aria-label="Clear input"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-trash2-icon lucide-trash-2"><path d="M3 6h18"/><path d="M19 6v14c0 1-1 2-2 2H7c-1 0-2-1-2-2V6"/><path d="M8 6V4c0-1 1-2 2-2h4c1 0 2 1 2 2v2"/><line x1="10" x2="10" y1="11" y2="17"/><line x1="14" x2="14" y1="11" y2="17"/></svg></button>
  668.                 </div>
  669.                 <button id="${UI_IDS.repoUrlModalSaveButton}">Import</button>
  670.               </div>`;
  671.             document.body.appendChild(wrapper);
  672.             const input = wrapper.querySelector(`#${UI_IDS.repoUrlModalInput}`);
  673.             if (input) {
  674.               input.focus();
  675.               input.value = (state && state.repoUrl) || "";
  676.             }
  677.             RepoUrlModal._attachEventListeners(wrapper);
  678.             resolve();
  679.           })
  680.           .catch((err) => {
  681.             Logger.error("Failed to show repo URL modal:", err);
  682.             resolve();
  683.           });
  684.       });
  685.     },
  686.     _attachEventListeners: (modalElement) => {
  687.       const urlInput = modalElement.querySelector(
  688.         `#${UI_IDS.repoUrlModalInput}`
  689.       );
  690.       const saveButton = modalElement.querySelector(
  691.         `#${UI_IDS.repoUrlModalSaveButton}`
  692.       );
  693.       const closeButton = modalElement.querySelector(
  694.         `#${UI_IDS.repoUrlModalCloseButton}`
  695.       );
  696.       const clearButton = modalElement.querySelector(
  697.         `#${UI_IDS.repoUrlModalClearButton}`
  698.       );
  699.  
  700.       modalElement.addEventListener("click", (e) => {
  701.         if (e.target === modalElement) {
  702.           const urlInput = modalElement.querySelector(
  703.             `#${UI_IDS.repoUrlModalInput}`
  704.           );
  705.           if (urlInput && !urlInput.value) {
  706.             const importButton = document.getElementById(UI_IDS.importButton);
  707.             if (importButton) {
  708.               importButton.classList.remove(CSS_CLASSES.importButtonOn);
  709.               importButton.setAttribute("aria-label", "Enable import");
  710.               importButton.setAttribute("data-state", "closed");
  711.               importButton.innerHTML = `${githubSVG} Import Repo`;
  712.               importButton.dataset.mode = "off";
  713.             }
  714.  
  715.             IngestDBManager.deleteState(getChatId());
  716.           }
  717.           RepoUrlModal._isShown = false;
  718.           modalElement.remove();
  719.         }
  720.       });
  721.  
  722.       const updateClearButtonVisibility = () => {
  723.         if (clearButton) {
  724.           clearButton.style.display = urlInput.value ? "flex" : "none";
  725.         }
  726.       };
  727.  
  728.       updateClearButtonVisibility();
  729.  
  730.       if (clearButton) {
  731.         clearButton.addEventListener("click", () => {
  732.           urlInput.value = "";
  733.           urlInput.focus();
  734.           updateClearButtonVisibility();
  735.         });
  736.       }
  737.  
  738.       urlInput.addEventListener("input", updateClearButtonVisibility);
  739.  
  740.       const handleSave = () => {
  741.         const url = urlInput.value.trim();
  742.         if (url) {
  743.           if (!RepoUrlModal._isValidRepoUrl(url)) {
  744.             alert("Please enter a valid git repository URL.");
  745.             return;
  746.           }
  747.  
  748.           RepoUrlModal._isShown = false;
  749.           modalElement.remove();
  750.  
  751.           const importButton = document.getElementById(UI_IDS.importButton);
  752.           if (importButton) {
  753.             importButton.classList.add(CSS_CLASSES.importButtonLoading);
  754.             importButton.setAttribute("aria-label", "Importing repository...");
  755.             importButton.disabled = true;
  756.           }
  757.  
  758.           new Promise((resolve, reject) => {
  759.             GM_xmlhttpRequest({
  760.               method: "POST",
  761.               url: apiUrl,
  762.               headers: {
  763.                 "Content-Type": "application/json",
  764.               },
  765.               data: JSON.stringify({
  766.                 url,
  767.               }),
  768.               onload: (res) => {
  769.                 if (res.status < 200 || res.status >= 300) {
  770.                   reject(
  771.                     new Error(`API Error (${res.status}): ${res.responseText}`)
  772.                   );
  773.                   return;
  774.                 }
  775.                 try {
  776.                   const data = JSON.parse(res.responseText);
  777.                   if (
  778.                     !data ||
  779.                     typeof data !== "object" ||
  780.                     !data.data ||
  781.                     !data.data.content ||
  782.                     !data.data.normalized ||
  783.                     !data.data.tree ||
  784.                     !data.data.index
  785.                   ) {
  786.                     reject(new Error("Invalid response data from API"));
  787.                     return;
  788.                   }
  789.  
  790.                   IngestDBManager.saveState({
  791.                     chatId: getChatId(),
  792.                     repoUrl: url,
  793.                     repoTree: data.data.tree,
  794.                     repoIndex: data.data.index,
  795.                     repoContent: data.data.content,
  796.                     repoNormalizedContent: data.data.normalized,
  797.                   })
  798.                     .then(() => {
  799.                       if (importButton) {
  800.                         importButton.classList.add(CSS_CLASSES.importButtonOn);
  801.                         importButton.setAttribute(
  802.                           "aria-label",
  803.                           "Disable import"
  804.                         );
  805.                         importButton.setAttribute("data-state", "open");
  806.                         importButton.dataset.mode = "on";
  807.                         importButton.innerHTML = `${githubSVG} ${getRepoNamefromURL(
  808.                           url
  809.                         )
  810.                           .split("/")
  811.                           .pop()
  812.                           .slice(0, 10)
  813.                           .replace(/^./, (c) => c.toUpperCase())}`;
  814.                       }
  815.                       resolve();
  816.                     })
  817.                     .catch((err) => {
  818.                       reject(err);
  819.                     });
  820.                 } catch (err) {
  821.                   reject(err);
  822.                 }
  823.               },
  824.               onerror: (err) => {
  825.                 reject(err);
  826.               },
  827.               ontimeout: () => {
  828.                 reject(new Error("Request timed out"));
  829.               },
  830.               timeout: 30000,
  831.             });
  832.           })
  833.             .catch((err) => {
  834.               if (importButton) {
  835.                 importButton.classList.remove(CSS_CLASSES.importButtonLoading);
  836.                 importButton.classList.remove(CSS_CLASSES.importButtonOn);
  837.                 importButton.setAttribute("aria-label", "Enable import");
  838.                 importButton.setAttribute("data-state", "closed");
  839.                 importButton.innerHTML = `${githubSVG} Import Repo`;
  840.                 importButton.dataset.mode = "off";
  841.                 importButton.disabled = false;
  842.               }
  843.               Logger.error("Error during repo import:", err);
  844.             })
  845.             .finally(() => {
  846.               if (importButton) {
  847.                 importButton.classList.remove(CSS_CLASSES.importButtonLoading);
  848.                 importButton.disabled = false;
  849.                 if (importButton.dataset.mode === "on") {
  850.                   importButton.setAttribute("aria-label", "Disable import");
  851.                 }
  852.               }
  853.             });
  854.         } else {
  855.           alert("Repo URL cannot be empty");
  856.         }
  857.       };
  858.  
  859.       saveButton.addEventListener("click", handleSave);
  860.  
  861.       urlInput.addEventListener("keydown", (e) => {
  862.         if (e.key === "Enter") {
  863.           handleSave();
  864.         }
  865.       });
  866.  
  867.       closeButton.addEventListener("click", () => {
  868.         const urlInput = modalElement.querySelector(
  869.           `#${UI_IDS.repoUrlModalInput}`
  870.         );
  871.         if (urlInput && !urlInput.value) {
  872.           const importButton = document.getElementById(UI_IDS.importButton);
  873.           if (importButton) {
  874.             importButton.classList.remove(CSS_CLASSES.importButtonOn);
  875.             importButton.setAttribute("aria-label", "Enable import");
  876.             importButton.setAttribute("data-state", "closed");
  877.             importButton.innerHTML = `${githubSVG} Import Repo`;
  878.             importButton.dataset.mode = "off";
  879.           }
  880.  
  881.           IngestDBManager.deleteState(getChatId());
  882.         }
  883.         RepoUrlModal._isShown = false;
  884.         modalElement.remove();
  885.       });
  886.     },
  887.   };
  888.  
  889.   const getAllFileNames = (index) => {
  890.     Logger.log(
  891.       "getAllFileNames: Processing index with",
  892.       (index && index.length) || 0,
  893.       "files"
  894.     );
  895.     return index.map((file, idx) => ({
  896.       fileName: file.fileName,
  897.       index: idx,
  898.     }));
  899.   };
  900.  
  901.   const getFileContents = (index, indices) => {
  902.     Logger.log("getFileContents: Requesting contents for indices", indices);
  903.     if (!index || !Array.isArray(indices)) {
  904.       Logger.log("getFileContents: Invalid input", {
  905.         index: !!index,
  906.         indices: !!indices,
  907.       });
  908.       return [];
  909.     }
  910.     const contents = indices
  911.       .map((idx) => {
  912.         const file = index[idx];
  913.         if (file) {
  914.           return {
  915.             fileName: file.fileName,
  916.             content: file.fileContent,
  917.           };
  918.         } else {
  919.           return null;
  920.         }
  921.       })
  922.       .filter(function (item) { return item !== null; });
  923.     Logger.log(
  924.       "getFileContents: Retrieved contents for",
  925.       contents.length,
  926.       "files"
  927.     );
  928.     return contents;
  929.   };
  930.  
  931.   const getRepoTree = () => {
  932.     return new Promise((resolve) => {
  933.       const chatId = getChatId();
  934.       IngestDBManager.getState(chatId)
  935.         .then((state) => {
  936.           if (!state || !state.repoTree) {
  937.             Logger.log("getRepoTree: No repo tree found in state");
  938.             resolve(null);
  939.             return;
  940.           }
  941.           Logger.log("getRepoTree: Retrieved repo tree");
  942.           resolve(state.repoTree);
  943.         })
  944.         .catch((err) => {
  945.           Logger.error("Failed to get repo tree:", err);
  946.           resolve(null);
  947.         });
  948.     });
  949.   };
  950.  
  951.   const generateRelevantContext = (query) => {
  952.     return new Promise((resolve) => {
  953.       Logger.log("generateRelevantContext: Processing query", query);
  954.       const chatId = getChatId();
  955.       IngestDBManager.getState(chatId)
  956.         .then((state) => {
  957.           if (!state || !state.repoIndex) {
  958.             Logger.log("generateRelevantContext: No repo index found in state");
  959.             resolve(null);
  960.             return;
  961.           }
  962.  
  963.           const index = state.repoIndex;
  964.           const fileList = getAllFileNames(index);
  965.           Logger.log(
  966.             "generateRelevantContext: Generated file list with",
  967.             fileList.length,
  968.             "files"
  969.           );
  970.  
  971.           const prompt = `Given the following list of files in a repository and a user query, return the indices of the most relevant files that would help answer the query. Only return the indices of files that are directly relevant.
  972.  
  973. Files in repository:
  974. ${JSON.stringify(fileList, null, 2)}
  975.  
  976. User query: ${query}
  977.  
  978. Return the response in this exact JSON format:
  979. {
  980.   "relevantIndices": [array of indices]
  981. }`;
  982.  
  983.           return new Promise((resolve, reject) => {
  984.             GM_xmlhttpRequest({
  985.               method: "POST",
  986.               url: `https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key=${geminiApiKey}`,
  987.               headers: {
  988.                 "Content-Type": "application/json",
  989.               },
  990.               data: JSON.stringify({
  991.                 contents: [
  992.                   {
  993.                     parts: [{ text: prompt }],
  994.                   },
  995.                 ],
  996.                 generationConfig: {
  997.                   responseMimeType: "application/json",
  998.                   responseSchema: {
  999.                     type: "OBJECT",
  1000.                     properties: {
  1001.                       relevantIndices: {
  1002.                         type: "ARRAY",
  1003.                         items: { type: "NUMBER" },
  1004.                       },
  1005.                     },
  1006.                     propertyOrdering: ["relevantIndices"],
  1007.                   },
  1008.                 },
  1009.               }),
  1010.               onload: (response) => {
  1011.                 if (response.status >= 200 && response.status < 300) {
  1012.                   Logger.log(
  1013.                     "generateRelevantContext: Gemini API call successful"
  1014.                   );
  1015.                   resolve(JSON.parse(response.responseText));
  1016.                 } else {
  1017.                   Logger.error(
  1018.                     "generateRelevantContext: Gemini API error",
  1019.                     response.status
  1020.                   );
  1021.                   reject(new Error(`Gemini API error: ${response.status}`));
  1022.                 }
  1023.               },
  1024.               onerror: (error) => {
  1025.                 Logger.error(
  1026.                   "generateRelevantContext: Gemini API request failed",
  1027.                   error
  1028.                 );
  1029.                 reject(error);
  1030.               },
  1031.             });
  1032.           });
  1033.         })
  1034.         .then((response) => {
  1035.           if (!response) return null;
  1036.           const responseText = response.candidates[0].content.parts[0].text;
  1037.           let relevantIndices;
  1038.           try {
  1039.             relevantIndices = JSON.parse(responseText).relevantIndices;
  1040.             Logger.log(
  1041.               "generateRelevantContext: Parsed relevant indices",
  1042.               relevantIndices
  1043.             );
  1044.           } catch (e) {
  1045.             Logger.error(
  1046.               "generateRelevantContext: Failed to parse Gemini API response",
  1047.               e
  1048.             );
  1049.             return null;
  1050.           }
  1051.  
  1052.           if (!Array.isArray(relevantIndices)) {
  1053.             Logger.error(
  1054.               "generateRelevantContext: Invalid relevantIndices format",
  1055.               relevantIndices
  1056.             );
  1057.             return null;
  1058.           }
  1059.  
  1060.           const relevantFiles = getFileContents(index, relevantIndices);
  1061.           if (!relevantFiles.length) {
  1062.             Logger.log("generateRelevantContext: No relevant files found");
  1063.             return null;
  1064.           }
  1065.  
  1066.           const context = relevantFiles
  1067.             .map(
  1068.               (file) => `File: ${file.fileName}\nContent:\n${file.content}\n`
  1069.             )
  1070.             .join("\n");
  1071.  
  1072.           Logger.log(
  1073.             "generateRelevantContext: Generated context with",
  1074.             relevantFiles.length,
  1075.             "files"
  1076.           );
  1077.           resolve(context);
  1078.         })
  1079.         .catch((error) => {
  1080.           Logger.error(
  1081.             "generateRelevantContext: Error generating context",
  1082.             error
  1083.           );
  1084.           resolve(null);
  1085.         });
  1086.     });
  1087.   };
  1088.  
  1089.   const FetchInterceptor = {
  1090.     originalFetch: null,
  1091.     isIntercepting: false,
  1092.     init: () => {
  1093.       try {
  1094.         if (typeof unsafeWindow === "undefined") {
  1095.           Logger.error("FetchInterceptor: unsafeWindow is not available");
  1096.           return;
  1097.         }
  1098.         const w = unsafeWindow;
  1099.         w.t3ChatIngest = w.t3ChatIngest || { needIngest: false };
  1100.         const originalFetch = w.fetch;
  1101.         FetchInterceptor.originalFetch = originalFetch;
  1102.  
  1103.         w.fetch = (input, initOptions = {}) => {
  1104.           if (FetchInterceptor.isIntercepting) {
  1105.             Logger.log(
  1106.               "FetchInterceptor: Already intercepting, passing through"
  1107.             );
  1108.             return originalFetch.call(w, input, initOptions);
  1109.           }
  1110.  
  1111.           const url = typeof input === "string" ? input : input.url;
  1112.           if (
  1113.             !url.includes("/api/chat") ||
  1114.             url.includes("/api/chat/resume") ||
  1115.             initOptions.method !== "POST"
  1116.           ) {
  1117.             return originalFetch.call(w, input, initOptions);
  1118.           }
  1119.  
  1120.           const chatId = getChatId();
  1121.           IngestDBManager.getState(chatId)
  1122.             .then((state) => {
  1123.               if (!state || !state.repoUrl) {
  1124.                 Logger.log(
  1125.                   "FetchInterceptor: No repo URL in state, passing through"
  1126.                 );
  1127.                 return originalFetch.call(w, input, initOptions);
  1128.               }
  1129.  
  1130.               let data;
  1131.               try {
  1132.                 data = JSON.parse(initOptions.body || "{}");
  1133.               } catch (error) {
  1134.                 Logger.error(
  1135.                   "FetchInterceptor: Failed to parse request body",
  1136.                   error
  1137.                 );
  1138.                 return originalFetch.call(w, input, initOptions);
  1139.               }
  1140.  
  1141.               if (!Array.isArray(data.messages)) {
  1142.                 Logger.log("FetchInterceptor: No messages array in request");
  1143.                 return originalFetch.call(w, input, initOptions);
  1144.               }
  1145.  
  1146.               const messages = data.messages;
  1147.               const lastIdx = messages.length - 1;
  1148.               if (
  1149.                 lastIdx < 0 ||
  1150.                 !messages[lastIdx] ||
  1151.                 messages[lastIdx].role !== "user"
  1152.               ) {
  1153.                 Logger.log("FetchInterceptor: No user message found");
  1154.                 return originalFetch.call(w, input, initOptions);
  1155.               }
  1156.               const originalPrompt = messages[lastIdx].content;
  1157.  
  1158.               if (typeof originalPrompt !== "string") {
  1159.                 Logger.log(
  1160.                   "FetchInterceptor: Invalid prompt type",
  1161.                   typeof originalPrompt
  1162.                 );
  1163.                 return originalFetch.call(w, input, initOptions);
  1164.               }
  1165.  
  1166.               Logger.log(
  1167.                 "FetchInterceptor: Intercepting fetch for ingest enhancement"
  1168.               );
  1169.               FetchInterceptor.isIntercepting = true;
  1170.  
  1171.               return Promise.all([
  1172.                 getRepoTree(),
  1173.                 generateRelevantContext(originalPrompt),
  1174.               ])
  1175.                 .then(([tree, context]) => {
  1176.                   Logger.log("FetchInterceptor: Retrieved repo tree");
  1177.                   Logger.log("FetchInterceptor: Generated context", !!context);
  1178.  
  1179.                   if (context) {
  1180.                     const importInstruction =
  1181.                       "The following information was retrieved from the repository. Please use these results to inform your response:\n";
  1182.                     messages[
  1183.                       lastIdx
  1184.                     ].content = `${importInstruction}\n[Repository Tree]\n${tree}\n\n[Repository Context]\n${context}\n\n[Original Message]\n${originalPrompt}`;
  1185.                     initOptions.body = JSON.stringify(data);
  1186.                     Logger.log(
  1187.                       "FetchInterceptor: Enhanced prompt with repository context"
  1188.                     );
  1189.                   } else {
  1190.                     Logger.log("FetchInterceptor: No context to add to prompt");
  1191.                   }
  1192.  
  1193.                   return originalFetch.call(w, input, initOptions);
  1194.                 })
  1195.                 .catch((error) => {
  1196.                   Logger.error(
  1197.                     "FetchInterceptor: Error during interception",
  1198.                     error
  1199.                   );
  1200.                   return originalFetch.call(w, input, initOptions);
  1201.                 })
  1202.                 .finally(() => {
  1203.                   FetchInterceptor.isIntercepting = false;
  1204.                 });
  1205.             })
  1206.             .catch((error) => {
  1207.               Logger.error("FetchInterceptor: Error getting state", error);
  1208.               return originalFetch.call(w, input, initOptions);
  1209.             });
  1210.         };
  1211.  
  1212.         Logger.log("FetchInterceptor: Initialized successfully");
  1213.       } catch (error) {
  1214.         Logger.error("FetchInterceptor: Failed to initialize", error);
  1215.       }
  1216.     },
  1217.   };
  1218.  
  1219.   const IngestDBManager = {
  1220.     db: null,
  1221.  
  1222.     init: () => {
  1223.       return new Promise((resolve, reject) => {
  1224.         const request = indexedDB.open(DB_NAME, DB_VERSION);
  1225.  
  1226.         request.onerror = () => {
  1227.           Logger.error("Failed to open IndexedDB");
  1228.           reject(request.error);
  1229.         };
  1230.  
  1231.         request.onsuccess = (event) => {
  1232.           IngestDBManager.db = event.target.result;
  1233.           Logger.log("IndexedDB opened successfully");
  1234.           resolve();
  1235.         };
  1236.  
  1237.         request.onupgradeneeded = (event) => {
  1238.           const db = event.target.result;
  1239.           if (!db.objectStoreNames.contains(STORE_NAME)) {
  1240.             const store = db.createObjectStore(STORE_NAME, {
  1241.               keyPath: "chatId",
  1242.             });
  1243.             store.createIndex("repoUrl", "repoUrl", { unique: false });
  1244.             Logger.log("IndexedDB store created");
  1245.           }
  1246.         };
  1247.       });
  1248.     },
  1249.  
  1250.     getAllStates: () => {
  1251.       return new Promise((resolve, reject) => {
  1252.         const transaction = IngestDBManager.db.transaction(
  1253.           [STORE_NAME],
  1254.           "readonly"
  1255.         );
  1256.         const store = transaction.objectStore(STORE_NAME);
  1257.         const request = store.getAll();
  1258.  
  1259.         request.onsuccess = () => resolve(request.result);
  1260.         request.onerror = () => reject(request.error);
  1261.       });
  1262.     },
  1263.  
  1264.     getState: (chatId) => {
  1265.       return new Promise((resolve, reject) => {
  1266.         if (!chatId) {
  1267.           resolve(null);
  1268.           return;
  1269.         }
  1270.  
  1271.         const transaction = IngestDBManager.db.transaction(
  1272.           [STORE_NAME],
  1273.           "readonly"
  1274.         );
  1275.         const store = transaction.objectStore(STORE_NAME);
  1276.         const request = store.get(chatId);
  1277.  
  1278.         request.onsuccess = () => resolve(request.result);
  1279.         request.onerror = () => reject(request.error);
  1280.       });
  1281.     },
  1282.  
  1283.     saveState: (state) => {
  1284.       return new Promise((resolve, reject) => {
  1285.         const transaction = IngestDBManager.db.transaction(
  1286.           [STORE_NAME],
  1287.           "readwrite"
  1288.         );
  1289.         const store = transaction.objectStore(STORE_NAME);
  1290.         const request = store.put(state);
  1291.  
  1292.         request.onsuccess = () => resolve(request.result);
  1293.         request.onerror = () => reject(request.error);
  1294.       });
  1295.     },
  1296.  
  1297.     deleteState: (chatId) => {
  1298.       return new Promise((resolve, reject) => {
  1299.         const transaction = IngestDBManager.db.transaction(
  1300.           [STORE_NAME],
  1301.           "readwrite"
  1302.         );
  1303.         const store = transaction.objectStore(STORE_NAME);
  1304.         const request = store.delete(chatId);
  1305.  
  1306.         request.onsuccess = () => resolve(request.result);
  1307.         request.onerror = () => reject(request.error);
  1308.       });
  1309.     },
  1310.  
  1311.     clearAll: () => {
  1312.       return new Promise((resolve, reject) => {
  1313.         const transaction = IngestDBManager.db.transaction(
  1314.           [STORE_NAME],
  1315.           "readwrite"
  1316.         );
  1317.         const store = transaction.objectStore(STORE_NAME);
  1318.         const request = store.clear();
  1319.  
  1320.         request.onsuccess = () => resolve(request.result);
  1321.         request.onerror = () => reject(request.error);
  1322.       });
  1323.     },
  1324.   };
  1325.  
  1326.   const MenuCommands = {
  1327.     init: () => {
  1328.       return new Promise((resolve) => {
  1329.         GM_registerMenuCommand("Toggle debug logs", () => {
  1330.           GM_getValue(GM_STORAGE_KEYS.DEBUG, false)
  1331.             .then((currentDebug) => {
  1332.               const newDebug = !currentDebug;
  1333.               return GM_setValue(GM_STORAGE_KEYS.DEBUG, newDebug);
  1334.             })
  1335.             .then(() => {
  1336.               debugMode = !debugMode;
  1337.               Logger.log(
  1338.                 `Debug mode toggled to: ${debugMode} via menu. Reloading...`
  1339.               );
  1340.               location.reload();
  1341.             })
  1342.             .catch((err) => {
  1343.               Logger.error("Failed to toggle debug mode:", err);
  1344.             });
  1345.         });
  1346.  
  1347.         GM_registerMenuCommand("Reset Gemini API Key", () => {
  1348.           GM_setValue(GM_STORAGE_KEYS.GEMINI_API_KEY, "")
  1349.             .then(() => {
  1350.               geminiApiKey = null;
  1351.               Logger.log("Gemini API Key reset via menu.");
  1352.               location.reload();
  1353.             })
  1354.             .catch((err) => {
  1355.               Logger.error("Failed to reset Gemini API key:", err);
  1356.             });
  1357.         });
  1358.  
  1359.         GM_registerMenuCommand("Reset RepoGist API URL", () => {
  1360.           GM_setValue(GM_STORAGE_KEYS.API_URL, "")
  1361.             .then(() => {
  1362.               apiUrl = null;
  1363.               Logger.log("RepoGist API URL reset via menu.");
  1364.               location.reload();
  1365.             })
  1366.             .catch((err) => {
  1367.               Logger.error("Failed to reset RepoGist API URL:", err);
  1368.             });
  1369.         });
  1370.  
  1371.         GM_registerMenuCommand("Reset IndexedDB for all chats", () => {
  1372.           IngestDBManager.clearAll()
  1373.             .then(() => {
  1374.               Logger.log("IndexedDB reset via menu.");
  1375.               location.reload();
  1376.             })
  1377.             .catch((err) => {
  1378.               Logger.error("Failed to reset IndexedDB:", err);
  1379.             });
  1380.         });
  1381.  
  1382.         Logger.log("Menu commands registered.");
  1383.         resolve();
  1384.       });
  1385.     },
  1386.   };
  1387.  
  1388.   function main() {
  1389.     try {
  1390.       debugMode = GM_getValue(GM_STORAGE_KEYS.DEBUG, false);
  1391.       Logger.log(
  1392.         `${SCRIPT_NAME} v${SCRIPT_VERSION} starting. Debug mode: ${debugMode}`
  1393.       );
  1394.  
  1395.       FetchInterceptor.init();
  1396.       MenuCommands.init().catch((err) => {
  1397.         Logger.error("Failed to initialize menu commands:", err);
  1398.       });
  1399.       IngestDBManager.init().catch((err) => {
  1400.         Logger.error("Failed to initialize database:", err);
  1401.       });
  1402.       StyleManager.injectGlobalStyles();
  1403.  
  1404.       GM_getValue(GM_STORAGE_KEYS.API_URL)
  1405.         .then((url) => {
  1406.           apiUrl = url;
  1407.           return GM_getValue(GM_STORAGE_KEYS.GEMINI_API_KEY);
  1408.         })
  1409.         .then((key) => {
  1410.           geminiApiKey = key;
  1411.           if (!apiUrl || !geminiApiKey) {
  1412.             Logger.log(
  1413.               "RepoGist API URL or Gemini API Key not found. It will be requested upon first import attempt."
  1414.             );
  1415.           } else {
  1416.             Logger.log("RepoGist API URL and Gemini API Key loaded.");
  1417.           }
  1418.         })
  1419.         .catch((err) => {
  1420.           Logger.error("Failed to load API configuration:", err);
  1421.         });
  1422.  
  1423.       let lastChatId = getChatId();
  1424.       const urlObserver = new MutationObserver(() => {
  1425.         const currentChatId = getChatId();
  1426.         if (currentChatId !== lastChatId) {
  1427.           lastChatId = currentChatId;
  1428.           injectButtonWithRetry().catch((err) => {
  1429.             Logger.error("Failed to inject button:", err);
  1430.           });
  1431.           IngestDBManager.getState(currentChatId)
  1432.             .then((state) => {
  1433.               const importButton = document.getElementById(UI_IDS.importButton);
  1434.               if (importButton) {
  1435.                 if (state && state.repoUrl) {
  1436.                   importButton.innerHTML = `${githubSVG} ${getRepoNamefromURL(
  1437.                     state.repoUrl
  1438.                   )
  1439.                     .split("/")
  1440.                     .pop()
  1441.                     .slice(0, 10)
  1442.                     .replace(/^./, (c) => c.toUpperCase())}`;
  1443.                   importButton.classList.add(CSS_CLASSES.importButtonOn);
  1444.                   importButton.setAttribute("aria-label", "Disable import");
  1445.                   importButton.setAttribute("data-state", "open");
  1446.                   importButton.dataset.mode = "on";
  1447.                 } else {
  1448.                   importButton.innerHTML = `${githubSVG} Import Repo`;
  1449.                   importButton.classList.remove(CSS_CLASSES.importButtonOn);
  1450.                   importButton.setAttribute("aria-label", "Enable import");
  1451.                   importButton.setAttribute("data-state", "closed");
  1452.                   importButton.dataset.mode = "off";
  1453.                 }
  1454.               }
  1455.             })
  1456.             .catch((err) => {
  1457.               Logger.error("Failed to update button state:", err);
  1458.             });
  1459.         }
  1460.       });
  1461.  
  1462.       urlObserver.observe(document.querySelector("title"), {
  1463.         subtree: true,
  1464.         characterData: true,
  1465.         childList: true,
  1466.       });
  1467.  
  1468.       const injectButtonWithRetry = (maxRetries = 5, delay = 1000) => {
  1469.         let retries = 0;
  1470.  
  1471.         const tryInjection = () => {
  1472.           const injectionObserverTargetParent = document.querySelector(
  1473.             selectors.messageActions
  1474.           );
  1475.  
  1476.           if (injectionObserverTargetParent) {
  1477.             UIManager.injectImportButton()
  1478.               .then((success) => {
  1479.                 if (success) {
  1480.                   Logger.log("Successfully injected import button");
  1481.                   return true;
  1482.                 }
  1483.               })
  1484.               .catch((err) => {
  1485.                 Logger.error("Failed to inject button:", err);
  1486.               });
  1487.           }
  1488.  
  1489.           if (retries < maxRetries) {
  1490.             retries++;
  1491.             Logger.log(
  1492.               `Retrying button injection (${retries}/${maxRetries})...`
  1493.             );
  1494.             setTimeout(() => tryInjection(), delay);
  1495.             return false;
  1496.           }
  1497.  
  1498.           return false;
  1499.         };
  1500.  
  1501.         return tryInjection();
  1502.       };
  1503.  
  1504.       injectButtonWithRetry()
  1505.         .then((success) => {
  1506.           if (!success) {
  1507.             Logger.log("Initial button injection failed, setting up observers");
  1508.  
  1509.             const injectionObserver = new MutationObserver(() => {
  1510.               UIManager.injectImportButton()
  1511.                 .then((success) => {
  1512.                   if (success) {
  1513.                     Logger.log("Button injected via mutation observer");
  1514.                   }
  1515.                 })
  1516.                 .catch((err) => {
  1517.                   Logger.error("Failed to inject button:", err);
  1518.                 });
  1519.             });
  1520.  
  1521.             const documentObserver = new MutationObserver(() => {
  1522.               const target = document.querySelector(selectors.messageActions);
  1523.               if (target) {
  1524.                 injectButtonWithRetry()
  1525.                   .then((success) => {
  1526.                     if (success) {
  1527.                       Logger.log("Button injected via document observer");
  1528.                     }
  1529.                   })
  1530.                   .catch((err) => {
  1531.                     Logger.error("Failed to inject button:", err);
  1532.                   });
  1533.               }
  1534.             });
  1535.  
  1536.             const messageActionsContainer = document.querySelector(
  1537.               selectors.messageActions
  1538.             );
  1539.             if (messageActionsContainer) {
  1540.               injectionObserver.observe(messageActionsContainer, {
  1541.                 childList: true,
  1542.                 subtree: true,
  1543.               });
  1544.             }
  1545.  
  1546.             documentObserver.observe(document.body, {
  1547.               childList: true,
  1548.               subtree: true,
  1549.             });
  1550.           }
  1551.         })
  1552.         .catch((err) => {
  1553.           Logger.error("Failed to initialize button injection:", err);
  1554.         });
  1555.  
  1556.       Logger.log(`${SCRIPT_NAME} v${SCRIPT_VERSION} initialized!`);
  1557.     } catch (error) {
  1558.       Logger.error("Failed to initialize:", error);
  1559.     }
  1560.   }
  1561.  
  1562.   main();
  1563. })();