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

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,30 @@ 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=…">` 一引就有。
+ * 写不进去（插件目录只读）就跳过，页面退回它自己的说明。
+ */
+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），不碰宿主代码。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fish-under-sea/magpie-neartoken",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "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 = ''
@@ -215,6 +218,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)))
   }
@@ -376,6 +382,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 +430,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) {

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/version.js

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