// ==UserScript==
// @name         Gemini Model Swapper (Flash 3 Preview)
// @namespace    http://tampermonkey.net/
// @version      1.0
// @description  尝试拦截网络请求，将 Gemini 模型 ID 替换为 Gemini 3 Flash Preview (仅供逆向测试，不保证服务器端可用)
// @author       BunnHack_Replica
// @match        https://business.gemini.google/*
// @grant        none
// @run-at       document-start
// ==/UserScript==

(function() {
    'use strict';

    // ================= 配置区域 =================
    // 目标模型 ID (你想替换成的模型)
    const TARGET_MODEL = "gemini-3-flash-preview";

    // 原始模型 ID (你当前账号默认使用的模型 ID)
    const SOURCE_MODELS = [
        "gemini-3-flash-preview",
        "gemini-3-flash-preview",
        "gemini-3-flash-preview-latest"
    ];
    // ===========================================

    const GREEN_LOG = "color: #4caf50; font-weight: bold;";

    // 1. 拦截 Fetch 请求 (Gemini 主要使用 Fetch)
    const originalFetch = window.fetch;
    window.fetch = async function(input, init) {
        if (init && init.body && typeof init.body === 'string') {
            let bodyChanged = false;
            let newBody = init.body;
            SOURCE_MODELS.forEach(source => {
                if (newBody.includes(source)) {
                    // 执行替换
                    newBody = newBody.replaceAll(source, TARGET_MODEL);
                    bodyChanged = true;
                    // 模仿控制台输出风格
                    console.log(`%c[Match] Replacing "${source}" with "${TARGET_MODEL}"`, GREEN_LOG);
                    console.log(`%c[Request] widgetStreamAssist (Intercepted)`, "font-weight: bold:");
                }
            });
            if (bodyChanged) {
                init.body = newBody;
            }
        }
        return originalFetch.apply(this, arguments);
    };

    // 2. 拦截 XMLHttpRequest (以防万一)
    const originalOpen = XMLHttpRequest.prototype.open;
    const originalSend = XMLHttpRequest.prototype.send;
    XMLHttpRequest.prototype.send = function(body) {
        if (body && typeof body === 'string') {
            let bodyChanged = false;
            let newBody = body;
            SOURCE_MODELS.forEach(source => {
                if (newBody.includes(source)) {
                    newBody = newBody.replaceAll(source, TARGET_MODEL);
                    bodyChanged = true;
                    console.log(`%c[Match-XHR] Replacing "${source}" with "${TARGET_MODEL}"`, GREEN_LOG);
                }
            });
            if (bodyChanged) {
                return originalSend.call(this, newBody);
            }
        }
        return originalSend.call(this, body);
    };

    console.log("%cGemini Model Swapper Loaded.", "color: orange; font-size: 12px;");
})();