sub2api测活工具，个人自用分享一下

TianP (好吃到甜) · 2026-05-25 · via LINUX DO - 最新话题
Sub2API 账号模型巡检油猴脚本改版分享

说明：这个脚本是在参考站内一位大佬的油猴脚本基础上改出来的，主要是按我自己的使用场景做了一些调整。原思路来自前辈，感谢站内大佬的分享；如果有不合适的地方可以提醒我修改或删除。
主要修改

这版主要改了两个地方：
支持按分组收集账号并测活
- 脚本会尽量识别当前账号列表页面的分组。
- 拉取账号时只处理当前分组下的账号。
- 如果没识别到分组，会弹窗让你手动填写本次要测活的分组。
增加账号并发数设置
- 面板里可以直接设置账号级并发数。
- 默认并发数是 8。
- 最大限制为 50，个人建议先从 5-15 之间试。
脚本功能

在后台账号列表页右侧注入一个“账号巡检”按钮。
自动尝试捕获当前页面请求里的 Authorization。
如果自动捕获失败，也可以在面板里手动填写 Bearer token。
按当前分组拉取账号列表。
对可调度账号进行模型测活。
已经关闭调度的账号会跳过。
任一测试模型异常时，会自动关闭该账号的 schedulable。
可以设置：
- 单模型超时时间
- 账号并发数
- 测试模型
使用方式

安装 Tampermonkey / 油猴扩展。
新建脚本，把脚本内容粘进去保存。
打开后台账号列表页：
- https://{改成你自己的域名}/admin/accounts
页面右侧会出现“账号巡检”按钮。
展开面板后确认：
- Authorization 是否已自动捕获
- 当前分组是否识别正确
- 并发数是否合适
- 测试模型是否正确
点击“开始巡检”。
注意事项

这个脚本会在模型异常时自动关闭账号调度，建议先小范围、小并发测试。
并发数不要一上来拉太高，避免请求过多导致误判或给服务造成压力。
如果页面没有自动捕获到 Authorization，可以刷新页面后再试，或者手动填写。
如果当前分组没有识别成功，开始时会提示手动填写。
脚本仅用于自己后台账号巡检场景，请确认自己有相关管理权限后再使用。
下面是正文里可以补一句的话：
完整脚本

// ==UserScript==
// @name         Sub2API 账号模型巡检并自动下线
// @namespace    https://sinry.example
// @version      0.1.1
// @description  按当前页面分组批量测试账号模型；任一模型异常时自动关闭账号 schedulable；已关闭调度账号跳过
// @match        https://{改成你自己的域名}/admin/accounts*
// @run-at       document-start
// @grant        none
// ==/UserScript==

(function () {
  'use strict';

  const CONFIG = {
    apiBase: location.origin,
    pageSize: 100,
    defaultTimeoutMs: 45000,
    defaultConcurrency: 8,
    maxConcurrency: 50,
    prompt: 'hi',
    onlyCheckSchedulable: false,
    stopOnFirstModelFailure: true,
    preferredModels: ['gpt-5.4', 'gpt-4o-mini', 'gpt-4o', 'gpt-4.1', 'gpt-4.1-mini'],
    defaultTestModel: 'gpt-5.4',
    groupParamNames: ['group', 'groups', 'account_group', 'accountGroup'],
    pageAuthTokenKey: 'auth_token',
    authStorageKey: '__sub2api_checker_auth__',
    timeoutStorageKey: '__sub2api_checker_timeout_ms__',
    concurrencyStorageKey: '__sub2api_checker_concurrency__',
    testModelStorageKey: '__sub2api_checker_test_model__',
    currentGroupStorageKey: '__sub2api_checker_current_group__',
    autoDisableStorageKey: '__sub2api_checker_auto_disable__',
  };

  function getCachedAuthToken() {
    const raw =
      localStorage.getItem(CONFIG.pageAuthTokenKey) ||
      sessionStorage.getItem(CONFIG.pageAuthTokenKey) ||
      localStorage.getItem(CONFIG.authStorageKey) ||
      '';
    return raw ? (raw.startsWith('Bearer ') ? raw : `Bearer ${raw}`) : '';
  }

  function clampConcurrency(value) {
    const n = Math.floor(Number(value));
    if (!Number.isFinite(n)) return CONFIG.defaultConcurrency;
    return Math.min(CONFIG.maxConcurrency, Math.max(1, n));
  }

  function readSavedConcurrency() {
    return clampConcurrency(localStorage.getItem(CONFIG.concurrencyStorageKey) || CONFIG.defaultConcurrency);
  }

  function readSavedAutoDisable() {
    return localStorage.getItem(CONFIG.autoDisableStorageKey) !== 'false';
  }

  const state = {
    authHeader: getCachedAuthToken(),
    timeoutMs: Number(localStorage.getItem(CONFIG.timeoutStorageKey) || CONFIG.defaultTimeoutMs),
    concurrency: readSavedConcurrency(),
    testModel: localStorage.getItem(CONFIG.testModelStorageKey) || CONFIG.defaultTestModel,
    autoDisable: readSavedAutoDisable(),
    currentGroup: '',
    running: false,
    stopRequested: false,
    panelReady: false,
    collapsed: true,
    stats: {
      total: 0,
      checked: 0,
      active: 0,
      started: 0,
      ok: 0,
      enabled: 0,
      disabled: 0,
      skipped: 0,
      failed: 0,
    },
  };

  function log(msg, type = 'info') {
    const time = new Date().toLocaleTimeString();
    const line = `[${time}] ${msg}`;
    console[type === 'error' ? 'error' : 'log'](`[sub2api-checker] ${line}`);
    const box = document.querySelector('#sub2api-checker-log');
    if (!box) return;
    const color =
      type === 'error' ? '#ff7875' :
      type === 'warn' ? '#ffd666' :
      type === 'success' ? '#95de64' : '#d9d9d9';
    const row = document.createElement('div');
    row.style.color = color;
    row.textContent = line;
    box.appendChild(row);
    box.scrollTop = box.scrollHeight;
  }

  function saveAuth(auth) {
    if (!auth || typeof auth !== 'string') return;
    const normalized = auth.startsWith('Bearer ') ? auth : `Bearer ${auth}`;
    const changed = state.authHeader !== normalized;
    state.authHeader = normalized;
    localStorage.setItem(CONFIG.authStorageKey, normalized);
    const input = document.querySelector('#sub2api-checker-auth');
    if (input && !input.value) input.value = normalized;
    if (changed) log('已捕获 Authorization', 'success');
  }

  function saveTimeoutMs(timeoutMs) {
    const n = Number(timeoutMs);
    if (!Number.isFinite(n) || n < 1000) return false;
    state.timeoutMs = n;
    localStorage.setItem(CONFIG.timeoutStorageKey, String(n));
    const input = document.querySelector('#sub2api-checker-timeout');
    if (input) input.value = String(Math.floor(n / 1000));
    return true;
  }

  function saveConcurrency(concurrency) {
    const n = clampConcurrency(concurrency);
    state.concurrency = n;
    localStorage.setItem(CONFIG.concurrencyStorageKey, String(n));
    const input = document.querySelector('#sub2api-checker-concurrency');
    if (input) input.value = String(n);
    return true;
  }

  function saveTestModel(model) {
    const normalized = String(model || '').trim();
    if (!normalized) return false;
    state.testModel = normalized;
    localStorage.setItem(CONFIG.testModelStorageKey, normalized);
    const input = document.querySelector('#sub2api-checker-test-model');
    if (input) input.value = normalized;
    return true;
  }

  function saveAutoDisable(enabled) {
    state.autoDisable = !!enabled;
    localStorage.setItem(CONFIG.autoDisableStorageKey, String(state.autoDisable));
    const input = document.querySelector('#sub2api-checker-auto-disable');
    if (input) input.checked = state.autoDisable;
    return true;
  }

  function normalizeGroup(value) {
    if (value === undefined || value === null) return '';
    const text = String(value).trim();
    if (!text) return '';
    if (['全部', '全部分组', 'all', 'null', 'undefined'].includes(text.toLowerCase())) return '';
    return text;
  }

  function getGroupFromUrl(urlLike = location.href) {
    try {
      const url = new URL(String(urlLike), location.origin);
      for (const name of CONFIG.groupParamNames) {
        const value = normalizeGroup(url.searchParams.get(name));
        if (value) return value;
      }
    } catch (_) {}
    return '';
  }

  function getElementMeta(el) {
    const parts = [
      el.getAttribute('name'),
      el.id,
      el.getAttribute('aria-label'),
      el.getAttribute('title'),
      el.getAttribute('placeholder'),
      typeof el.className === 'string' ? el.className : '',
    ];

    if (el.labels) {
      parts.push(...Array.from(el.labels).map((label) => label.textContent || ''));
    }

    const closestLabel = el.closest('label')?.textContent || '';
    if (closestLabel) parts.push(closestLabel);

    const parentText = el.parentElement?.textContent || '';
    if (parentText.length < 120) parts.push(parentText);

    return parts.filter(Boolean).join(' ').toLowerCase();
  }

  function looksLikeGroupControl(el) {
    const meta = getElementMeta(el);
    return meta.includes('group') || meta.includes('分组');
  }

  function getGroupFromDom() {
    const controls = Array.from(document.querySelectorAll('select,input')).filter(looksLikeGroupControl);

    for (const el of controls) {
      const value = normalizeGroup(
        el.tagName === 'SELECT'
          ? el.value || el.selectedOptions?.[0]?.value || el.selectedOptions?.[0]?.textContent || ''
          : el.value || el.getAttribute('value') || ''
      );
      if (value) return value;
    }

    const labelEls = Array.from(document.querySelectorAll('label,.ant-form-item-label,[class*="label"],[class*="Label"]'))
      .filter((el) => /分组|group/i.test(el.textContent || ''));

    for (const label of labelEls) {
      const container = label.closest('.ant-form-item') || label.parentElement;
      const candidates = [
        container?.querySelector('select')?.value,
        container?.querySelector('select')?.selectedOptions?.[0]?.value,
        container?.querySelector('select')?.selectedOptions?.[0]?.textContent,
        container?.querySelector('input')?.value,
        container?.querySelector('.ant-select-selection-item')?.getAttribute('title'),
        container?.querySelector('.ant-select-selection-item')?.textContent,
        container?.querySelector('[class*="singleValue"]')?.textContent,
        container?.querySelector('[class*="selected"]')?.textContent,
      ];

      for (const candidate of candidates) {
        const value = normalizeGroup(candidate);
        if (value) return value;
      }
    }

    return '';
  }

  function readCurrentGroup() {
    return getGroupFromUrl(location.href) || getGroupFromDom();
  }

  function updateGroupDisplay() {
    const el = document.querySelector('#sub2api-checker-current-group');
    if (!el) return;
    el.textContent = state.currentGroup || '未识别';
    el.style.color = state.currentGroup ? '#95de64' : '#ffd666';
  }

  function saveCurrentGroup(group, source = '') {
    const normalized = normalizeGroup(group);
    if (!normalized) return false;

    const changed = state.currentGroup !== normalized;
    state.currentGroup = normalized;
    localStorage.setItem(CONFIG.currentGroupStorageKey, normalized);
    updateGroupDisplay();

    if (source && changed) {
      log(`已识别当前分组：${normalized}（${source}）`, 'success');
    }

    return true;
  }

  async function injectAuthSniffer() {
    while (!document.documentElement) {
      await new Promise((resolve) => setTimeout(resolve, 0));
    }

    const script = document.createElement('script');
    const nonce = document.querySelector('script[nonce]')?.nonce || document.querySelector('script[nonce]')?.getAttribute('nonce');
    if (nonce) script.nonce = nonce;
    script.textContent = `
      (() => {
        const groupParamNames = ${JSON.stringify(CONFIG.groupParamNames)};

        const emit = (auth) => {
          if (!auth) return;
          document.dispatchEvent(new CustomEvent('__sub2api_checker_auth__', { detail: auth }));
        };

        const emitGroup = (urlLike) => {
          try {
            if (!urlLike) return;
            const url = new URL(String(urlLike), location.origin);
            if (!url.pathname.includes('/api/v1/admin/accounts')) return;

            for (const name of groupParamNames) {
              const value = (url.searchParams.get(name) || '').trim();
              if (value) {
                document.dispatchEvent(new CustomEvent('__sub2api_checker_group__', { detail: value }));
                return;
              }
            }
          } catch (_) {}
        };

        const pickAuth = (headersLike) => {
          try {
            if (!headersLike) return '';
            if (headersLike instanceof Headers) {
              return headersLike.get('Authorization') || headersLike.get('authorization') || '';
            }
            if (Array.isArray(headersLike)) {
              for (const [k, v] of headersLike) {
                if (String(k).toLowerCase() === 'authorization') return v || '';
              }
              return '';
            }
            if (typeof headersLike === 'object') {
              for (const key of Object.keys(headersLike)) {
                if (key.toLowerCase() === 'authorization') return headersLike[key] || '';
              }
            }
          } catch (_) {}
          return '';
        };

        const origFetch = window.fetch;
        if (origFetch) {
          window.fetch = function(input, init) {
            const auth =
              pickAuth(init && init.headers) ||
              pickAuth(input && input.headers);
            if (auth) emit(auth);

            const requestUrl =
              typeof input === 'string' || input instanceof URL
                ? String(input)
                : input && input.url;
            emitGroup(requestUrl);

            return origFetch.apply(this, arguments);
          };
        }

        const origOpen = XMLHttpRequest.prototype.open;
        const origSetHeader = XMLHttpRequest.prototype.setRequestHeader;
        XMLHttpRequest.prototype.open = function() {
          this.__sub2apiAuth = '';
          emitGroup(arguments[1]);
          return origOpen.apply(this, arguments);
        };
        XMLHttpRequest.prototype.setRequestHeader = function(name, value) {
          if (String(name).toLowerCase() === 'authorization' && value) {
            this.__sub2apiAuth = value;
            emit(value);
          }
          return origSetHeader.apply(this, arguments);
        };
      })();
    `;
    document.documentElement.appendChild(script);
    script.remove();

    document.addEventListener('__sub2api_checker_auth__', (event) => {
      saveAuth(event.detail);
    });

    document.addEventListener('__sub2api_checker_group__', (event) => {
      saveCurrentGroup(event.detail, '账号列表请求');
    });
  }

  function updateStats() {
    const el = document.querySelector('#sub2api-checker-stats');
    if (!el) return;
    const s = state.stats;
    const queued = Math.max(0, s.total - s.checked - s.active);
    el.textContent = `总数 ${s.total} | 运行中 ${s.active} | 队列 ${queued} | 已处理 ${s.checked} | 正常 ${s.ok} | 已启用 ${s.enabled} | 已关闭 ${s.disabled} | 跳过 ${s.skipped} | 异常 ${s.failed}`;
  }

  function updatePanelCollapsed() {
    const shell = document.querySelector('#sub2api-checker-shell');
    const root = document.querySelector('#sub2api-checker-panel');
    const toggle = document.querySelector('#sub2api-checker-toggle');
    if (!root || !toggle || !shell) return;
    root.style.width = state.collapsed ? '0px' : '420px';
    root.style.opacity = state.collapsed ? '0' : '1';
    root.style.marginRight = state.collapsed ? '0px' : '12px';
    root.style.pointerEvents = state.collapsed ? 'none' : 'auto';
    root.style.transform = state.collapsed ? 'translateX(12px)' : 'translateX(0)';
    toggle.textContent = state.collapsed ? '账号巡检' : '收起';
    toggle.style.borderRadius = state.collapsed ? '10px 0 0 10px' : '10px';
    shell.style.pointerEvents = 'auto';
  }

  function ensurePanel() {
    if (state.panelReady) return;
    state.panelReady = true;

    const shell = document.createElement('div');
    shell.id = 'sub2api-checker-shell';
    shell.style.cssText = `
      position: fixed;
      right: 0;
      top: 120px;
      z-index: 1000000;
      display: flex;
      flex-direction: row;
      align-items: flex-start;
      pointer-events: auto;
    `;
    document.body.appendChild(shell);

    const toggle = document.createElement('button');
    toggle.id = 'sub2api-checker-toggle';
    toggle.style.cssText = `
      padding: 10px 8px;
      border: 0;
      border-radius: 10px 0 0 10px;
      background: #1677ff;
      color: #fff;
      cursor: pointer;
      writing-mode: vertical-rl;
      text-orientation: mixed;
      box-shadow: 0 8px 24px rgba(0,0,0,.25);
      transition: transform .28s ease, box-shadow .28s ease, border-radius .28s ease;
      font: 12px/1.2 -apple-system,BlinkMacSystemFont,Segoe UI,Roboto,Helvetica,Arial,PingFang SC,Microsoft YaHei,sans-serif;
    `;
    toggle.addEventListener('mouseenter', () => {
      toggle.style.transform = 'translateX(-2px)';
      toggle.style.boxShadow = '0 10px 28px rgba(0,0,0,.32)';
    });
    toggle.addEventListener('mouseleave', () => {
      toggle.style.transform = 'translateX(0)';
      toggle.style.boxShadow = '0 8px 24px rgba(0,0,0,.25)';
    });
    toggle.addEventListener('click', () => {
      state.collapsed = !state.collapsed;
      updatePanelCollapsed();
    });
    shell.appendChild(toggle);

    const root = document.createElement('div');
    root.id = 'sub2api-checker-panel';
    root.style.cssText = `
      width: 0;
      opacity: 0;
      overflow: hidden;
      transition: width .28s ease, opacity .22s ease, margin-right .28s ease, transform .28s ease;
      transform: translateX(12px);
    `;
    root.innerHTML = `
      <div id="sub2api-checker-panel-inner" style="
        width:420px;
        background:rgba(16, 18, 27, 0.96);
        color:#fff;
        border:1px solid #30363d;
        border-radius:12px;
        box-shadow:0 8px 24px rgba(0,0,0,.35);
        font:12px/1.5 -apple-system,BlinkMacSystemFont,Segoe UI,Roboto,Helvetica,Arial,PingFang SC,Microsoft YaHei,sans-serif;
        overflow:hidden;
      ">
      <div style="padding:12px 14px;border-bottom:1px solid #30363d;font-weight:700;">Sub2API 账号模型巡检</div>
      <div style="padding:12px 14px;display:flex;flex-direction:column;gap:8px;">
        <label style="display:flex;flex-direction:column;gap:4px;">
          <span>Authorization（优先自动捕获，抓不到再手填）</span>
          <input id="sub2api-checker-auth" type="text" placeholder="Bearer xxxxxx"
            style="width:100%;box-sizing:border-box;padding:8px;border-radius:8px;border:1px solid #434a57;background:#111723;color:#fff;" />
        </label>
        <label style="display:flex;flex-direction:column;gap:4px;">
          <span>单模型超时时间（秒）</span>
          <input id="sub2api-checker-timeout" type="number" min="1" step="1" placeholder="45"
            style="width:100%;box-sizing:border-box;padding:8px;border-radius:8px;border:1px solid #434a57;background:#111723;color:#fff;" />
        </label>
        <label style="display:flex;flex-direction:column;gap:4px;">
          <span>账号并发数（1-${CONFIG.maxConcurrency}，建议 5-15）</span>
          <input id="sub2api-checker-concurrency" type="number" min="1" max="${CONFIG.maxConcurrency}" step="1" placeholder="8"
            style="width:100%;box-sizing:border-box;padding:8px;border-radius:8px;border:1px solid #434a57;background:#111723;color:#fff;" />
        </label>
        <label style="display:flex;flex-direction:column;gap:4px;">
          <span>测试模型</span>
          <input id="sub2api-checker-test-model" type="text" placeholder="gpt-5.4"
            style="width:100%;box-sizing:border-box;padding:8px;border-radius:8px;border:1px solid #434a57;background:#111723;color:#fff;" />
        </label>
        <label style="display:flex;gap:8px;align-items:center;color:#d9d9d9;">
          <input id="sub2api-checker-auto-disable" type="checkbox" style="margin:0;" />
          <span>模型异常时自动关闭账号调度</span>
        </label>
        <div style="display:flex;gap:8px;align-items:center;color:#bfbfbf;">
          <span>当前页面分组：</span>
          <strong id="sub2api-checker-current-group" style="flex:1;color:#ffd666;">未识别</strong>
          <button id="sub2api-checker-refresh-group" style="padding:6px 8px;border:0;border-radius:8px;background:#434a57;color:#fff;cursor:pointer;">重新读取</button>
        </div>
        <div style="display:flex;gap:8px;align-items:center;">
          <button id="sub2api-checker-start" style="flex:1;padding:8px 10px;border:0;border-radius:8px;background:#1677ff;color:#fff;cursor:pointer;">开始巡检</button>
          <button id="sub2api-checker-stop" style="flex:1;padding:8px 10px;border:0;border-radius:8px;background:#fa541c;color:#fff;cursor:pointer;">停止</button>
        </div>
        <div id="sub2api-checker-stats" style="color:#bfbfbf;">总数 0 | 运行中 0 | 队列 0 | 已处理 0 | 正常 0 | 已启用 0 | 已关闭 0 | 跳过 0 | 异常 0</div>
        <div id="sub2api-checker-log" style="height:320px;overflow:auto;background:#0b0f17;border:1px solid #30363d;border-radius:8px;padding:8px;"></div>
      </div>
      </div>
    `;
    shell.appendChild(root);

    const authInput = root.querySelector('#sub2api-checker-auth');
    authInput.value = state.authHeader;
    authInput.addEventListener('change', () => {
      const v = authInput.value.trim();
      if (v) saveAuth(v);
    });

    const timeoutInput = root.querySelector('#sub2api-checker-timeout');
    timeoutInput.value = String(Math.floor(state.timeoutMs / 1000));
    timeoutInput.addEventListener('change', () => {
      const sec = Number(timeoutInput.value || 0);
      if (!saveTimeoutMs(sec * 1000)) {
        timeoutInput.value = String(Math.floor(state.timeoutMs / 1000));
        log('超时时间无效，需大于等于 1 秒', 'error');
        return;
      }
      log(`已设置单模型超时 ${sec} 秒`, 'success');
    });

    const concurrencyInput = root.querySelector('#sub2api-checker-concurrency');
    concurrencyInput.value = String(state.concurrency);
    concurrencyInput.addEventListener('change', () => {
      saveConcurrency(concurrencyInput.value || CONFIG.defaultConcurrency);
      log(`已设置账号并发数 ${state.concurrency}`, 'success');
    });

    const testModelInput = root.querySelector('#sub2api-checker-test-model');
    testModelInput.value = state.testModel;
    testModelInput.addEventListener('change', () => {
      const model = testModelInput.value.trim();
      if (!saveTestModel(model)) {
        testModelInput.value = state.testModel;
        log('测试模型不能为空', 'error');
        return;
      }
      log(`已设置测试模型 ${state.testModel}`, 'success');
    });

    const autoDisableInput = root.querySelector('#sub2api-checker-auto-disable');
    autoDisableInput.checked = state.autoDisable;
    autoDisableInput.addEventListener('change', () => {
      saveAutoDisable(autoDisableInput.checked);
      log(`模型异常时${state.autoDisable ? '会' : '不会'}自动关闭账号调度`, 'success');
    });

    root.querySelector('#sub2api-checker-refresh-group').addEventListener('click', () => {
      const group = readCurrentGroup();
      if (group) {
        saveCurrentGroup(group, '当前页面');
      } else {
        updateGroupDisplay();
        log('未能从当前页面识别分组，开始时会弹窗确认', 'warn');
      }
    });

    root.querySelector('#sub2api-checker-start').addEventListener('click', () => run().catch((err) => {
      log(`运行异常：${err.message}`, 'error');
      state.running = false;
    }));
    root.querySelector('#sub2api-checker-stop').addEventListener('click', () => {
      state.stopRequested = true;
      log('已请求停止，当前请求结束后退出', 'warn');
    });

    updateGroupDisplay();
    updatePanelCollapsed();
  }

  async function waitDomReady() {
    if (document.body) return;
    await new Promise((resolve) => {
      const timer = setInterval(() => {
        if (document.body) {
          clearInterval(timer);
          resolve();
        }
      }, 50);
    });
  }

  async function apiFetch(url, options = {}) {
    const headers = new Headers(options.headers || {});
    if (state.authHeader && !headers.has('Authorization')) {
      headers.set('Authorization', state.authHeader);
    }
    const resp = await fetch(url, {
      ...options,
      headers,
      credentials: 'include',
    });
    return resp;
  }

  async function fetchAccounts() {
    let page = 1;
    const items = [];
    while (true) {
      const url = new URL('/api/v1/admin/accounts', CONFIG.apiBase);
      url.searchParams.set('page', String(page));
      url.searchParams.set('page_size', String(CONFIG.pageSize));
      url.searchParams.set('platform', '');
      url.searchParams.set('type', '');
      url.searchParams.set('status', '');
      url.searchParams.set('privacy_mode', '');
      url.searchParams.set('group', state.currentGroup);
      url.searchParams.set('search', '');
      url.searchParams.set('timezone', Intl.DateTimeFormat().resolvedOptions().timeZone || 'Asia/Shanghai');

      const resp = await apiFetch(url.toString(), {
        headers: { Accept: 'application/json, text/plain, */*' },
      });
      if (!resp.ok) throw new Error(`账号列表请求失败：HTTP ${resp.status}`);
      const json = await resp.json();
      if (json.code !== 0) throw new Error(`账号列表返回异常：${json.message || json.code}`);

      const pageItems = json?.data?.items || [];
      items.push(...pageItems);

      const pages = Number(json?.data?.pages || 1);
      if (page >= pages || pageItems.length === 0) break;
      page += 1;
    }
    return items;
  }

  function getModels(account) {
    const targetModel = String(state.testModel || '').trim();
    if (targetModel) return [targetModel];

    const mapping = account?.credentials?.model_mapping || {};
    const keys = Object.keys(mapping).filter(Boolean);
    if (keys.length <= 1) return keys;

    const preferred = [];
    for (const model of CONFIG.preferredModels) {
      if (keys.includes(model)) preferred.push(model);
    }
    const rest = keys.filter((k) => !preferred.includes(k)).sort();
    return [...preferred, ...rest];
  }

  async function testModel(accountId, modelId) {
    const controller = new AbortController();
    let timer = null;
    const resetTimer = () => {
      clearTimeout(timer);
      timer = setTimeout(() => controller.abort(new Error(`模型 ${modelId} 流式超时`)), state.timeoutMs);
    };

    try {
      resetTimer();
      const resp = await apiFetch(`${CONFIG.apiBase}/api/v1/admin/accounts/${accountId}/test`, {
        method: 'POST',
        headers: {
          Accept: '*/*',
          'Content-Type': 'application/json',
        },
        body: JSON.stringify({ model_id: modelId, prompt: CONFIG.prompt }),
        signal: controller.signal,
      });
      if (!resp.ok) {
        clearTimeout(timer);
        return { ok: false, reason: `HTTP ${resp.status}` };
      }

      const reader = resp.body?.getReader();
      if (!reader) {
        clearTimeout(timer);
        const text = await resp.text();
        return { ok: false, reason: `无响应流：${text.slice(0, 200)}` };
      }

      const decoder = new TextDecoder();
      let buffer = '';

      while (true) {
        resetTimer();
        const { value, done } = await reader.read();
        if (done) break;
        buffer += decoder.decode(value, { stream: true }).replace(/\r/g, '');

        let splitIndex;
        while ((splitIndex = buffer.indexOf('\n\n')) >= 0) {
          const chunk = buffer.slice(0, splitIndex);
          buffer = buffer.slice(splitIndex + 2);
          const dataLines = chunk
            .split('\n')
            .map((line) => line.trim())
            .filter((line) => line.startsWith('data:'))
            .map((line) => line.slice(5).trim());

          for (const line of dataLines) {
            if (!line) continue;
            let event;
            try {
              event = JSON.parse(line);
            } catch (_) {
              continue;
            }
            if (event.type === 'error') {
              clearTimeout(timer);
              return { ok: false, reason: event.error || '未知错误' };
            }
            if (event.type === 'test_complete') {
              clearTimeout(timer);
              return { ok: !!event.success, reason: event.success ? 'success' : 'test_complete=false' };
            }
          }
        }
      }

      clearTimeout(timer);
      return { ok: false, reason: '响应流结束但没有 test_complete' };
    } catch (err) {
      clearTimeout(timer);
      return {
        ok: false,
        reason: err?.name === 'AbortError' ? '请求超时' : (err?.message || String(err)),
      };
    }
  }

  async function setAccountSchedulable(accountId, schedulable) {
    const resp = await apiFetch(`${CONFIG.apiBase}/api/v1/admin/accounts/${accountId}/schedulable`, {
      method: 'POST',
      headers: {
        Accept: 'application/json, text/plain, */*',
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({ schedulable: !!schedulable }),
    });

    if (!resp.ok) {
      return { ok: false, reason: `HTTP ${resp.status}` };
    }
    const json = await resp.json();
    if (json.code !== 0) {
      return { ok: false, reason: json.message || `code=${json.code}` };
    }
    return { ok: true, data: json.data };
  }

  function resetStats() {
    state.stats = {
      total: 0,
      checked: 0,
      active: 0,
      started: 0,
      ok: 0,
      enabled: 0,
      disabled: 0,
      skipped: 0,
      failed: 0,
    };
    updateStats();
    const logBox = document.querySelector('#sub2api-checker-log');
    if (logBox) logBox.innerHTML = '';
  }

  async function ensureAuth() {
    const cached = getCachedAuthToken();
    if (cached) {
      saveAuth(cached);
      return true;
    }
    if (state.authHeader) return true;
    const fromInput = document.querySelector('#sub2api-checker-auth')?.value?.trim();
    if (fromInput) {
      saveAuth(fromInput);
      return true;
    }
    const manual = prompt('没有自动捕获到 Authorization，请粘贴 Bearer token');
    if (!manual) return false;
    saveAuth(manual.trim());
    return true;
  }

  async function ensureGroup() {
    const detected = readCurrentGroup();
    if (detected) {
      return saveCurrentGroup(detected, '当前页面');
    }

    if (state.currentGroup) return true;

    const lastGroup = normalizeGroup(localStorage.getItem(CONFIG.currentGroupStorageKey));
    const manual = prompt(
      `没有识别到当前页面选择的分组，请填写本次测活分组。${lastGroup ? `\n上次识别到：${lastGroup}` : ''}`,
      lastGroup || ''
    );

    if (!manual) return false;
    return saveCurrentGroup(manual.trim(), '手动确认');
  }

  async function processAccount(account) {
    const title = `#${account.id} ${account.name || '(未命名)'}`;

    try {
      if (CONFIG.onlyCheckSchedulable && !account.schedulable) {
        state.stats.skipped += 1;
        log(`${title} 已去掉调度，跳过测试`, 'warn');
        return;
      }

      const models = getModels(account);
      if (!models.length) {
        state.stats.failed += 1;
        if (!state.autoDisable) {
          log(`${title} 没有 model_mapping，但未关闭 schedulable`, 'warn');
          return;
        }

        log(`${title} 没有 model_mapping，准备关闭`, 'error');
        const off = await setAccountSchedulable(account.id, false);
        if (off.ok) {
          state.stats.disabled += 1;
          log(`${title} 已关闭 schedulable`, 'success');
        } else {
          log(`${title} 关闭失败：${off.reason}`, 'error');
        }
        return;
      }

      log(`${title} 开始测试 ${models.length} 个模型`);
      let accountOk = true;
      let failReason = '';
      let testedCount = 0;

      for (const model of models) {
        if (state.stopRequested) break;
        log(`${title} 测试模型 ${model}`);
        const result = await testModel(account.id, model);
        testedCount += 1;
        if (!result.ok) {
          accountOk = false;
          failReason = `模型 ${model} 异常：${result.reason}`;
          log(`${title} ${failReason}`, 'error');
          if (CONFIG.stopOnFirstModelFailure) break;
        } else {
          log(`${title} 模型 ${model} 正常`, 'success');
        }
      }

      if (state.stopRequested && testedCount < models.length && accountOk) {
        state.stats.skipped += 1;
        log(`${title} 因停止请求未完成全部模型测试，未改动 schedulable`, 'warn');
        return;
      }

      if (accountOk) {
        state.stats.ok += 1;
        if (!account.schedulable) {
          const on = await setAccountSchedulable(account.id, true);
          if (on.ok) {
            state.stats.enabled += 1;
            log(`${title} 全部模型正常，已重新启用 schedulable`, 'success');
          } else {
            log(`${title} 模型正常但重新启用失败：${on.reason}`, 'error');
          }
        } else {
          log(`${title} 全部模型正常`, 'success');
        }
      } else {
        state.stats.failed += 1;
        if (!state.autoDisable) {
          log(`${title} 检测到异常但未关闭 schedulable（原因：${failReason}）`, 'warn');
          return;
        }

        const off = await setAccountSchedulable(account.id, false);
        if (off.ok) {
          state.stats.disabled += 1;
          log(`${title} 已关闭 schedulable（原因：${failReason}）`, 'success');
        } else {
          log(`${title} 关闭失败：${off.reason}`, 'error');
        }
      }
    } finally {
      state.stats.checked += 1;
      updateStats();
    }
  }

  async function runWorkerPool(accounts) {
    const concurrency = clampConcurrency(state.concurrency);
    log(`账号级并发 ${concurrency}，单账号内模型按顺序测试`);

    async function worker(workerIndex) {
      while (!state.stopRequested) {
        const index = state.stats.started;
        if (index >= accounts.length) break;

        state.stats.started += 1;
        state.stats.active += 1;
        updateStats();

        try {
          await processAccount(accounts[index]);
        } catch (err) {
          state.stats.failed += 1;
          state.stats.checked += 1;
          log(`工作线程 ${workerIndex} 处理账号异常：${err?.message || String(err)}`, 'error');
          updateStats();
        } finally {
          state.stats.active -= 1;
          updateStats();
        }
      }
    }

    const workerCount = Math.min(concurrency, accounts.length);
    await Promise.all(Array.from({ length: workerCount }, (_, index) => worker(index + 1)));
  }

  async function run() {
    if (state.running) {
      log('已有任务在运行', 'warn');
      return;
    }
    if (!(await ensureAuth())) {
      log('缺少 Authorization，已取消', 'error');
      return;
    }
    if (!(await ensureGroup())) {
      log('缺少测活分组，已取消', 'error');
      return;
    }

    state.running = true;
    state.stopRequested = false;
    resetStats();

    try {
      state.collapsed = false;
      updatePanelCollapsed();
      log(`开始拉取账号列表（分组：${state.currentGroup}）`);
      const accounts = await fetchAccounts();
      state.stats.total = accounts.length;
      updateStats();
      log(`共获取 ${accounts.length} 个账号`, 'success');

      await runWorkerPool(accounts);

      if (state.stopRequested) {
        log('任务已按要求停止', 'warn');
      } else {
        log('巡检完成', 'success');
      }
    } finally {
      state.running = false;
      state.stats.active = 0;
      updateStats();
    }
  }

  injectAuthSniffer();
  waitDomReady().then(() => {
    ensurePanel();

    const group = readCurrentGroup();
    if (group) saveCurrentGroup(group, '当前页面');

    if (state.authHeader) {
      log('脚本已就绪，已从本地缓存 auth_token 读取 Authorization', 'success');
    } else {
      log('脚本已就绪，未发现 auth_token；可刷新页面自动捕获或手动粘贴');
    }
  });
})();
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。原文来自 — 版权归原作者所有。
推荐订阅源

LINUX DO - 最新话题

Sub2API 账号模型巡检油猴脚本改版分享

主要修改

脚本功能

使用方式

注意事项

完整脚本
