@fish-under-sea/magpie-neartoken 0.2.1 → 0.2.3

package/index.mjs

@@ -12,7 +12,7 @@
 // 站点自己的 /v1/models 按令牌分组过滤，很多账号拿到的是空列表，所以模型目录
 // 直接读 /api/pricing（公开、全量），不靠那个端点。
 
-import { readFileSync } from 'node:fs'
+import { readFileSync, writeFileSync } from 'node:fs'
 import { dirname, join } from 'node:path'
 import { fileURLToPath } from 'node:url'
 
@@ -118,6 +118,9 @@ const hasWord = (id, words) => words.some((w) => id.toLowerCase().includes(w))
  * 按那一行走那一组——这就是"模型后面跟着它的分组"。
  */
 async function catalog(disabled = new Set()) {
+  // 同时把「全部分组」抄一份给页面：页面在沙箱里发不出网络请求，但能加载自己
+  // 目录下的脚本，所以这里写 page/groups.js，模型广场按它重绘。
+  const groupsOut = new Map()
   const env = await call('/api/pricing')
   const rows = Array.isArray(env?.data) ? env.data : []
   const refs = env?.group_refs && typeof env.group_refs === 'object' ? env.group_refs : {}
@@ -132,6 +135,13 @@ async function catalog(disabled = new Set()) {
     const film = hasWord(model, FILMS)
     if (!chat && !drawer && !film) continue
     const all = Array.isArray(row?.enable_groups) && row.enable_groups.length ? row.enable_groups : ['']
+    // 先按「全部分组」计数（关掉的也算），模型广场要显示它有多少个模型
+    for (const raw of all) {
+      const g = String(raw ?? '').trim()
+      if (!g) continue
+      const had = groupsOut.get(g)
+      groupsOut.set(g, { name: g, ratio: 0, models: (had?.models || 0) + 1, ref: '' })
+    }
     const groups = all.filter((g) => !disabled.has(String(g ?? '').trim()))
     if (!groups.length) continue
     for (const raw of groups) {
@@ -156,9 +166,57 @@ async function catalog(disabled = new Set()) {
       }
     }
   }
+  writeGroups([...groupsOut.values()].map((g) => ({
+    ...g,
+    ratio: Number(ratios[g.name]) > 0 ? Number(ratios[g.name]) : 0,
+    ref: refs[g.name] ? String(refs[g.name]) : '',
+  })).sort((a, b) => a.name.localeCompare(b.name, 'zh')))
   return out
 }
 
+/**
+ * writeGroups 把分组表写成 page/groups.js，给沙箱里的页面用。
+ *
+ * 页面发不出网络请求（CSP connect-src 'none'），但能加载自己目录下的脚本，
+ * 所以这里不推数据、只落一个文件：`<script src="groups.js?t=…">` 一引就有。
+ * 写不进去（插件目录只读）就跳过，页面退回它自己的说明。
+ */
+/**
+ * stampLatest 把 npm 上的最新版本写进 page/latest.js。
+ *
+ * 页面自己查不了（沙箱禁联网），但插件能：这里顺手问一次 registry，页面
+ * 「检查更新」时加载这个脚本就知道新不新。一分钟内不重复问。
+ */
+let latestAskedAt = 0
+async function stampLatest() {
+  if (Date.now() - latestAskedAt < 60_000) return
+  latestAskedAt = Date.now()
+  try {
+    const res = await fetch('https://registry.npmjs.org/@fish-under-sea%2Fmagpie-neartoken/latest', {
+      headers: { accept: 'application/json' },
+    })
+    if (!res.ok) return
+    const body = await res.json()
+    const v = typeof body?.version === 'string' ? body.version.trim() : ''
+    if (!v) return
+    writeFileSync(
+      join(here, 'page', 'latest.js'),
+      'self.NEARTOKEN_LATEST = ' + JSON.stringify(v) + '\nself.NEARTOKEN_LATEST_AT = ' + Date.now() + '\n',
+      'utf8',
+    )
+  } catch {
+    /* 问不到就保持原样：页面会说「没查到」 */
+  }
+}
+function writeGroups(groups) {
+  try {
+    const js = 'self.NEARTOKEN_GROUPS = ' + JSON.stringify(groups) + '\nself.NEARTOKEN_GROUPS_AT = ' + Date.now() + '\n'
+    writeFileSync(join(here, 'page', 'groups.js'), js, 'utf8')
+  } catch {
+    /* 只读安装：页面自己会说没拿到明细 */
+  }
+}
+
 export default async function neartoken(input, options) {
   // 页面写进来的开关：关掉的分组，它们的模型不再出现在 magpie 的列表里。
   // 走的是 magpie 的 plugin options（页面 magpie.options.set），不碰宿主代码。
@@ -241,6 +299,7 @@ export default async function neartoken(input, options) {
       ],
       /** 余额：站点钱包端点，按 magpie 的用量卡显示。 */
       async usage(getAuth) {
+        await stampLatest()
         const auth = await getAuth()
         const key = auth?.access || auth?.key
         if (!key) throw new Error('还没有连接 NearToken')
@@ -291,6 +350,7 @@ export default async function neartoken(input, options) {
        * @returns {Promise<Record<string, object>>}
        */
       async models() {
+        await stampLatest()
         return catalog(disabled)
       },
     },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fish-under-sea/magpie-neartoken",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "NearToken for magpie: a provider whose sign-in is the site's own device authorization, models listed per billing group with their multiplier, balance and vouchers on magpie's usage card, plus a page for spending and top-ups.",
   "type": "module",
   "main": "index.mjs",

package/page/app.js

@@ -23,6 +23,9 @@
   let disabledGroups = new Set()
   /** 当前视角：home / market / logs。 */
   let tab = 'home'
+  /** 插件写下的分组快照（page/groups.js），页面加载脚本取到。 */
+  let groupSnapshot = null
+  let groupsAsked = false
   const pick = { amount: null, method: null, custom: '' }
   /** 正在等宿主的哪个动作。 */
   let pendingAct = ''
@@ -152,8 +155,9 @@
   function tick() {
     tickCount += 1
     const counting = view && view.pending
-    if (counting && tickCount % 2 === 0) load().catch(() => {})
-    else if (!counting && tickCount % 60 === 0) load().catch(() => {})
+    // 等确认时每秒问一次：站点那边一点批准，这边下一拍就能接上
+    if (counting) load().catch(() => {})
+    else if (tickCount % 60 === 0) load().catch(() => {})
     if (counting) paintCountdown()
   }
 
@@ -215,6 +219,9 @@
           if (key !== 'home' && view.configured && !windows.has(span)) {
             call('neartoken:usage', { range: span }).then((w) => { windows.set(span, w); render() }).catch(showError)
           }
+          if (key === 'market' && !groupSnapshot) {
+            loadGroups().then((got) => { if (got) render() })
+          }
         },
       }, label)))
   }
@@ -247,6 +254,11 @@
         onclick: () => act('打开确认页', () => call('neartoken:open')),
       }, '打开确认页'),
       el('button', { disabled: !!pendingAct, onclick: () => copy(p.verificationUriComplete) }, '复制链接'),
+      el('button', {
+        class: 'ghost',
+        disabled: !!pendingAct,
+        onclick: () => { load().catch(() => {}) },
+      }, '我已确认，立即刷新'),
       el('button', {
         class: 'ghost',
         disabled: !!pendingAct,
@@ -376,6 +388,39 @@
     return card('充值', ...parts)
   }
 
+  /**
+   * loadGroups 取分组快照。
+   *
+   * 页面在沙箱里发不出网络请求（CSP connect-src 'none'），但能加载自己目录下的
+   * 脚本；插件（index.mjs）每次拉目录都会写一份 page/groups.js，这里引一下就有。
+   */
+  function loadGroups(force) {
+    if (groupsAsked && !force) return Promise.resolve(groupSnapshot)
+    groupsAsked = true
+    return new Promise((resolve) => {
+      const s = document.createElement('script')
+      s.src = 'groups.js?t=' + Date.now()
+      s.onload = () => {
+        groupSnapshot = Array.isArray(self.NEARTOKEN_GROUPS) ? self.NEARTOKEN_GROUPS : null
+        resolve(groupSnapshot)
+      }
+      s.onerror = () => resolve(null)
+      document.head.append(s)
+    })
+  }
+
+  /** retryGroups 让插件重新拉一次目录（刷新模型表也顺手刷新了这份快照）。 */
+  async function retryGroups() {
+    groupSnapshot = null
+    render()
+    try {
+      await bridge.call('neartoken:refresh', {})
+    } catch {
+      /* 刷新失败也让下面的读取照跑一次 */
+    }
+    await loadGroups(true)
+    render()
+  }
   /** 分组落在哪一类：低价 / 满血 / 图片 / 视频 / 语音 / 向量 / 其他。 */
   function bucketOf(name) {
     const n = String(name || '')
@@ -391,19 +436,18 @@
 
   /** 模型广场：按站点返回的分组重绘，分组即为开关。 */
   function marketCard() {
-    const list = Array.isArray(view.groupList) ? view.groupList : null
+    const list = Array.isArray(view.groupList) && view.groupList.length ? view.groupList : groupSnapshot
     const head = el('div', { class: 'row' },
       el('h2', { class: 'grow', text: '分组', style: 'margin:0' }),
       el('a', { class: 'link', href: 'https://neartoken.cc/pricing', target: '_blank', rel: 'noreferrer' }, '站点广场 ↗'))
     if (!list || !list.length) {
       return card(null, head,
         el('p', { class: 'note' },
-          '宿主还没把分组明细交给页面（`neartoken:state` 目前只回分组数）。'),
-        el('p', { class: 'note' },
-          '开关已接好：写进 magpie 的 plugin options，provider 插件按它过滤模型表；',
-          '宿主一回 groupList，这里就按站点分组重绘。先去 ',
-          el('a', { class: 'link', href: 'https://neartoken.cc/pricing', target: '_blank', rel: 'noreferrer' }, '站点广场'),
-          ' 看分组与倍率。'))
+          '还没拿到分组表。点「重新读取」——插件会去站点拉一次 /api/pricing，',
+          '写成 page/groups.js，页面加载它就有数据（不需要宿主改动）。'),
+        el('div', { class: 'row', style: 'margin-top:10px' },
+          el('button', { class: 'ghost', onclick: () => retryGroups() }, '重新读取'),
+          el('a', { class: 'link', href: 'https://neartoken.cc/pricing', target: '_blank', rel: 'noreferrer', style: 'align-self:center' }, '站点广场 ↗')) )
     }
     const buckets = new Map()
     for (const g of list) {
@@ -604,6 +648,77 @@
     act('断开连接', () => call('neartoken:disconnect'))
   })
 
+  // ---------- 检查更新 ----------
+
+  /** cmp 比两段版本号：a 比 b 新返回正数。只认数字段，够用了。 */
+  function cmpVersion(a, b) {
+    const pa = String(a).split(/[.-]/).map((x) => parseInt(x, 10) || 0)
+    const pb = String(b).split(/[.-]/).map((x) => parseInt(x, 10) || 0)
+    for (let i = 0; i < Math.max(pa.length, pb.length); i += 1) {
+      const d = (pa[i] || 0) - (pb[i] || 0)
+      if (d) return d
+    }
+    return 0
+  }
+
+  /**
+   * loadLatest 取 npm 上的最新版本。
+   *
+   * 页面自己查不了，插件会把结果写成 page/latest.js；这里用 script 引入，
+   * 加 ?t= 破缓存，所以点一次「检查更新」就是现问一次。
+   */
+  function loadLatest() {
+    return new Promise((resolve) => {
+      const s = document.createElement('script')
+      s.src = 'latest.js?t=' + Date.now()
+      s.onload = () => resolve(typeof self.NEARTOKEN_LATEST === 'string' ? self.NEARTOKEN_LATEST : '')
+      s.onerror = () => resolve('')
+      document.head.append(s)
+    })
+  }
+
+  let checking = false
+  async function checkUpdate() {
+    if (checking) return
+    checking = true
+    const btn = $('updates')
+    const note = $('update-note')
+    const was = btn.textContent
+    btn.disabled = true
+    btn.textContent = '检查中…'
+    note.hidden = true
+    const latest = await loadLatest()
+    const mine = self.NEARTOKEN_VERSION || ''
+    checking = false
+    btn.disabled = false
+    btn.textContent = was
+    if (!latest) {
+      note.className = 'update-note warn'
+      note.textContent = '没查到最新版本：插件还没联网拉过一次，稍后再试。'
+      note.hidden = false
+      return
+    }
+    if (mine && cmpVersion(latest, mine) > 0) {
+      note.className = 'update-note new'
+      note.textContent = '有新版本 v' + latest + '（当前 v' + mine + '）'
+      const link = el('a', {
+        class: 'link',
+        href: 'https://github.com/Fish-under-sea/magpie-neartoken/releases',
+        target: '_blank',
+        rel: 'noreferrer',
+        style: 'margin-left:8px',
+      }, '去下载 ↗')
+      note.append(link)
+      note.hidden = false
+      return
+    }
+    note.className = 'update-note'
+    note.textContent = '已是最新 v' + (mine || latest)
+    note.hidden = false
+  }
+
+  $('updates').addEventListener('click', checkUpdate)
+
   // ---------- 画图 ----------
 
   /** stackedChart 画一张按模型堆叠的面积图；模型多于 5 个时余下的合并。 */

package/page/groups.js

@@ -0,0 +1,2 @@
+// 由插件在拉取目录时覆盖写入；这里只是占位，免得页面加载时报 404。
+self.NEARTOKEN_GROUPS = self.NEARTOKEN_GROUPS || null

package/page/icon.svg

@@ -1,5 +1,28 @@
-<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" width="24" height="24" fill="none" stroke="currentColor" stroke-width="1.9" stroke-linecap="round" stroke-linejoin="round">
-  <circle cx="10.8" cy="12.4" r="7"/>
-  <path d="M8.3 15.4V9.4l5 6V9.4"/>
-  <path d="M19.3 3.2l.62 1.48 1.48.62-1.48.62-.62 1.48-.62-1.48L17.2 5.3l1.48-.62z" fill="currentColor" stroke="none"/>
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 256 256" width="256" height="256">
+  <defs>
+    <radialGradient id="body" cx="38%" cy="30%" r="80%">
+      <stop offset="0" stop-color="#FFEFB8"/>
+      <stop offset="55%" stop-color="#FFD166"/>
+      <stop offset="100%" stop-color="#FFA63A"/>
+    </radialGradient>
+    <linearGradient id="leaf" x1="0" y1="0" x2="1" y2="1">
+      <stop offset="0" stop-color="#A9EBC6"/>
+      <stop offset="100%" stop-color="#39A377"/>
+    </linearGradient>
+  </defs>
+  <!-- 叶与梗 -->
+  <path d="M128 66c-4-19 6-31 26-35 3 19-7 31-26 35z" fill="url(#leaf)" stroke="#2B7F5C" stroke-width="7" stroke-linejoin="round"/>
+  <path d="M125 74c2-9 2-16 1-23" stroke="#2B7F5C" stroke-width="8" stroke-linecap="round" fill="none"/>
+  <!-- 果身 -->
+  <circle cx="128" cy="152" r="82" fill="url(#body)" stroke="#FFF8E9" stroke-width="10"/>
+  <!-- 脸 -->
+  <ellipse cx="104" cy="150" rx="7.5" ry="9" fill="#4A3B2A"/>
+  <ellipse cx="152" cy="150" rx="7.5" ry="9" fill="#4A3B2A"/>
+  <circle cx="101.5" cy="147" r="2.4" fill="#FFFDF4"/>
+  <circle cx="149.5" cy="147" r="2.4" fill="#FFFDF4"/>
+  <path d="M118 170c6 7 14 7 20 0" stroke="#4A3B2A" stroke-width="7" stroke-linecap="round" fill="none"/>
+  <ellipse cx="80" cy="168" rx="13" ry="8" fill="#FF8E7A" opacity="0.55"/>
+  <ellipse cx="176" cy="168" rx="13" ry="8" fill="#FF8E7A" opacity="0.55"/>
+  <!-- 高光 -->
+  <ellipse cx="97" cy="118" rx="24" ry="15" fill="#FFFDF3" opacity="0.5" transform="rotate(-28 97 118)"/>
 </svg>

package/page/index.html

@@ -14,11 +14,14 @@
       <h1>NearToken</h1>
       <p class="sub" id="site">neartoken.cc</p>
     </div>
-    <span class="chip ver" id="version" title="插件版本">v—</span>
-    <a class="ghost link" id="updates" href="https://github.com/Fish-under-sea/magpie-neartoken/releases" target="_blank" rel="noreferrer">检查更新</a>
-    <span class="chip" id="status"><span class="dot"></span><span id="status-text">正在读取</span></span>
-    <button class="ghost" id="disconnect" hidden>断开连接</button>
+    <div class="actions">
+      <span class="chip ver" id="version" title="插件版本">v—</span>
+      <button class="ghost" id="updates" type="button">检查更新</button>
+      <span class="chip" id="status"><span class="dot"></span><span id="status-text">正在读取</span></span>
+      <button class="ghost" id="disconnect" type="button" hidden>断开连接</button>
+    </div>
   </header>
+  <p class="update-note" id="update-note" hidden></p>
   <main id="view" aria-live="polite">
     <div class="card"><p class="brief">正在读取 NearToken 的状态…</p></div>
   </main>

package/page/latest.js

@@ -0,0 +1,2 @@
+// 由插件在拉取目录时覆盖写入；这里只是占位。
+self.NEARTOKEN_LATEST = self.NEARTOKEN_LATEST || ''

package/page/style.css

@@ -457,4 +457,19 @@ select {
   padding: 3px 8px;
 }
 
-.ghost.link { text-decoration: none; display: inline-flex; align-items: center; }
+.ghost.link { text-decoration: none; display: inline-flex; align-items: center; }
+/* 头部按钮组：版本、检查更新、状态、断开四件一样高，右对齐 */
+.head { align-items: center; gap: 12px; }
+
+.actions { display: flex; align-items: center; gap: 8px; margin-left: auto; }
+.actions .chip { height: 28px; display: inline-flex; align-items: center; gap: 6px; padding: 0 10px; }
+.actions .ghost { height: 28px; display: inline-flex; align-items: center; }
+.actions .ghost[disabled] { opacity: 0.6; }
+
+.update-note {
+  margin: 8px 0 0;
+  font-size: 11.5px;
+  color: var(--nt-muted);
+}
+.update-note.new { color: var(--nt-accent); }
+.update-note.warn { color: var(--nt-warn, #e0a44a); }

package/page/version.js

@@ -1,2 +1,2 @@
 // 由 scripts/stamp.mjs 生成，别手改：版本号在 package.json。
-self.NEARTOKEN_VERSION = "0.2.1"
+self.NEARTOKEN_VERSION = "0.2.3"