☰

ChatGPT Model Usage Tracker (ultimate+DR) v3.5.1

本地统计 ChatGPT 各模型总量 + 当日增量（按本地时区归零）

// ==UserScript==
// @name         ChatGPT Model Usage Tracker (ultimate+DR) v3.5.1
// @namespace    https://example.local
// @version      3.5.1
// @description  本地统计 ChatGPT 各模型总量 + 当日增量（按本地时区归零）
// @match        https://chat.openai.com/*
// @match        https://chatgpt.com/*
// @run-at       document-start
// @grant        unsafeWindow
// ==/UserScript==

(() => {
  /******************** 0. 配置 ************************/
  const KEY_TOTAL   = '__cgpt_usage__';
  const KEY_DAILY   = '__cgpt_usage_daily__';
  const KEY_DAYFLAG = '__cgpt_usage_day__';

  const seenDR      = new Set();
  const seenMessage = new Set();

  /******************** 1. 数据层 **********************/
  const stats  = JSON.parse(localStorage.getItem(KEY_TOTAL)   || '{}');
  let   daily  = JSON.parse(localStorage.getItem(KEY_DAILY)   || '{}');
  let   dayFlag = localStorage.getItem(KEY_DAYFLAG) || '';

  // **本地时区日期**
  const todayStr = () => new Date().toLocaleDateString('en-CA'); // YYYY-MM-DD

  const ensureToday = () => {
    const now = todayStr();
    if (now !== dayFlag) {
      dayFlag = now;
      daily   = {};
      localStorage.setItem(KEY_DAILY,   JSON.stringify(daily));
      localStorage.setItem(KEY_DAYFLAG, dayFlag);
    }
  };

  const bump = (model) => {
    ensureToday();
    stats[model] = (stats[model] || 0) + 1;
    daily[model] = (daily[model] || 0) + 1;
    localStorage.setItem(KEY_TOTAL, JSON.stringify(stats));
    localStorage.setItem(KEY_DAILY, JSON.stringify(daily));
    render();
  };

  /******************** 2. UI *************************/
  let box;
  const mountUI = () => {
    if (!document.body || (box && document.body.contains(box))) return;

    box = document.createElement('div');
    Object.assign(box.style, {
      position:'fixed', right:'16px', bottom:'16px', zIndex:99999,
      background:'rgba(0,0,0,.78)', color:'#fff', font:'13px/1.4 monospace',
      padding:'8px 14px', borderRadius:'8px', whiteSpace:'pre-wrap',
      userSelect:'none', cursor:'pointer', maxWidth:'300px'
    });
    box.onclick = () => {
      if (confirm('清空统计数据？')) {
        for (const k in stats) delete stats[k];
        for (const k in daily) delete daily[k];
        localStorage.setItem(KEY_TOTAL,'{}');
        localStorage.setItem(KEY_DAILY,'{}');
        render();
      }
    };
    document.body.appendChild(box);
    render();
  };

  const render = () => {
    if (!box) return;
    ensureToday();
    const arrow = '▲';
    const green = (txt) => `<span style="color:#4caf50">${txt}</span>`;
    const lines = Object.keys(stats).length
      ? Object.entries(stats)
          .sort((a,b)=>b[1]-a[1])
          .map(([m,c])=>{
            const inc = daily[m] ? ` ${green(arrow + '(+' + daily[m] + ')')}` : '';
            return `${m}: ${c}${inc}`;
          })
      : ['暂无数据'];
    box.innerHTML = '📊 模型统计<br>' + lines.join('<br>');
  };

  document.addEventListener('DOMContentLoaded', mountUI);
  setInterval(mountUI, 3000);

  /******************** 3. fetch/XHR 拦截（与 v3.5 相同） ********************/
  const origFetch = unsafeWindow.fetch;
  unsafeWindow.fetch = async function(input, init = {}) {
    const req = input instanceof Request ? input : new Request(input, init);
    if (req.method !== 'POST' || req.url.startsWith('chrome-extension')) {
      return origFetch.apply(this, arguments);
    }
    let bodyStr = '';
    try { bodyStr = await req.clone().text(); } catch(_) {}
    hook(bodyStr);
    return origFetch.apply(this, arguments);
  };

  (function() {
    const origOpen = XMLHttpRequest.prototype.open;
    const origSend = XMLHttpRequest.prototype.send;
    const urlCache = new WeakMap();

    XMLHttpRequest.prototype.open = function(m,u,...rest) {
      urlCache.set(this, {method:m,url:u});
      return origOpen.call(this, m, u, ...rest);
    };
    XMLHttpRequest.prototype.send = function(body) {
      const {method} = urlCache.get(this) || {};
      if (method?.toUpperCase() === 'POST') {
        readBody(body).then(hook);
      }
      return origSend.call(this, body);
    };
  })();

  /******************** 5. 主逻辑（保持不变） **********************/
  async function hook(bodyStr){
    if (!bodyStr) return;

    let payload;
    try { payload = JSON.parse(bodyStr); } catch { return; }

    if (Array.isArray(payload.system_hints) &&
        payload.system_hints.includes('research')) {
    // >>> 新增：仅统计真正由用户发起的深度研究请求 <<<
     const firstMsg = payload.messages?.[0];
     const role = firstMsg?.author?.role ?? firstMsg?.role;
     if (role !== 'user') return;          // 后台请求，直接忽略

     const cid = payload.conversation_id;
     if (cid && !seenDR.has(cid)) {
       bump('deep_research');
       seenDR.add(cid);
    }
    return;
    }

    if (!payload.model) return;

    const firstMsg = payload.messages?.[0];
    if (!firstMsg) return;

    const role  = firstMsg.author?.role ?? firstMsg.role;
    const parts = firstMsg.content?.parts;
    if (role !== 'user')       return;
    if (!Array.isArray(parts) || parts.length === 0) return;

    const msgId = firstMsg.id;
    if (msgId && seenMessage.has(msgId)) return;
    if (msgId) seenMessage.add(msgId);

    bump(payload.model);
  }

  async function readBody(b){
    if (!b) return '';
    if (typeof b === 'string') return b;
    if (b instanceof FormData || b instanceof URLSearchParams) return b.toString();
    if (b?.clone) {
      try { return await b.clone().text(); } catch {}
    }
    return '';
  }

  console.log('[USAGE-TRACKER] v3.5.1 injected');
})();