@iasiv5/dsh-quota-watch 0.0.1

package/src/client.mjs

@@ -0,0 +1,369 @@
+import { CLIENT_POLL_INTERVAL_MS, CLIENT_ROUTES } from './shared.mjs'
+
+export const name = 'quota-watch-client'
+export const inject = []
+
+const CARD_SELECTOR = '[data-dsh-quota-watch-card]'
+const COLLAPSE_KEY = 'dsh-quota-watch:collapsed'
+const FETCH_TIMEOUT_MS = 15_000
+
+const COPY = {
+  zh: {
+    title: '套餐监控',
+    refresh: '刷新',
+    collapse: '收起',
+    expand: '展开',
+    loading: '读取中',
+    updated: '更新',
+    notConfigured: '未配置',
+    noPlan: '暂无套餐窗口',
+    failed: '查询失败',
+    stale: '数据可能已过期',
+    requestFailed: '刷新失败，显示上次成功数据',
+    available: '可用额度',
+    used: '本期已用',
+    balanceUnavailable: '余额暂不可用',
+    reset: '重置',
+    windows: { '5h': '5 小时', week: '每周', month: '每月' },
+  },
+  en: {
+    title: 'Quota Watch',
+    refresh: 'Refresh',
+    collapse: 'Collapse',
+    expand: 'Expand',
+    loading: 'Loading',
+    updated: 'Updated',
+    notConfigured: 'Not configured',
+    noPlan: 'No plan windows',
+    failed: 'Query failed',
+    stale: 'Data may be stale',
+    requestFailed: 'Refresh failed; showing last successful data',
+    available: 'Available credits',
+    used: 'Used this cycle',
+    balanceUnavailable: 'Balance unavailable',
+    reset: 'Resets',
+    windows: { '5h': '5 hours', week: 'Weekly', month: 'Monthly' },
+  },
+}
+
+const STYLE_TEXT = `
+[data-dsh-quota-watch-card] { box-sizing: border-box; width: 100%; margin: 0 0 8px; color: var(--dsw-alias-label-primary, inherit); font: inherit; }
+[data-dsh-quota-watch-card] *, [data-dsh-quota-watch-card] *::before, [data-dsh-quota-watch-card] *::after { box-sizing: border-box; }
+.dqw-card { border: 1px solid var(--dsw-alias-border-secondary, rgba(128,128,128,.35)); border-radius: 10px; background: var(--dsw-alias-bg-base, rgba(128,128,128,.08)); padding: 8px; display: flex; flex-direction: column; gap: 7px; }
+.dqw-head { display: flex; align-items: center; gap: 4px; min-width: 0; }
+.dqw-title { flex: 1; min-width: 0; margin: 0; font-size: 12px; font-weight: 600; line-height: 18px; }
+.dqw-button { border: 0; border-radius: 6px; background: transparent; color: inherit; cursor: pointer; font: inherit; font-size: 11px; line-height: 18px; padding: 1px 4px; }
+.dqw-button:hover { background: var(--dsw-alias-bg-hover, rgba(128,128,128,.16)); }
+.dqw-button:disabled { cursor: wait; opacity: .6; }
+.dqw-meta { margin: 0; font-size: 10px; line-height: 14px; opacity: .7; }
+.dqw-provider { border-top: 1px solid var(--dsw-alias-border-secondary, rgba(128,128,128,.25)); padding-top: 6px; display: flex; flex-direction: column; gap: 4px; }
+.dqw-provider-title { margin: 0; font-size: 11px; font-weight: 600; line-height: 16px; }
+.dqw-row { display: flex; flex-direction: column; gap: 2px; }
+.dqw-row-head { display: flex; justify-content: space-between; gap: 5px; font-size: 10px; line-height: 14px; }
+.dqw-bar { display: block; height: 4px; overflow: hidden; border-radius: 4px; background: var(--dsw-alias-bg-tertiary, rgba(128,128,128,.2)); }
+.dqw-bar-fill { display: block; height: 100%; border-radius: inherit; background: var(--dsw-alias-button-primary-fill, #5b8def); }
+.dqw-muted { margin: 0; font-size: 10px; line-height: 14px; opacity: .7; }
+.dqw-error { margin: 0; font-size: 10px; line-height: 14px; color: var(--dsw-alias-label-warning, #b46900); }
+.dqw-credit { margin: 0; font-size: 10px; line-height: 14px; font-variant-numeric: tabular-nums; }
+`
+
+function localeFor(win) {
+  return String(win?.navigator?.language ?? 'zh').toLowerCase().startsWith('zh') ? 'zh' : 'en'
+}
+
+function formatNumber(value, locale) {
+  if (typeof value !== 'number' || !Number.isFinite(value)) return ''
+  return new Intl.NumberFormat(locale === 'zh' ? 'zh-CN' : 'en-US', { maximumFractionDigits: 1 }).format(value)
+}
+
+function formatPercent(value, locale) {
+  if (typeof value !== 'number' || !Number.isFinite(value)) return ''
+  const clamped = Math.max(0, Math.min(100, value))
+  const rounded = clamped >= 10 ? Math.round(clamped) : Number(clamped.toFixed(1))
+  return `${formatNumber(rounded, locale)}%`
+}
+
+function formatTime(value, locale) {
+  if (typeof value !== 'string' || value === '') return ''
+  const date = new Date(value)
+  if (Number.isNaN(date.getTime())) return ''
+  return new Intl.DateTimeFormat(locale === 'zh' ? 'zh-CN' : 'en-US', {
+    dateStyle: 'short',
+    timeStyle: 'short',
+  }).format(date)
+}
+
+function text(doc, tag, className, value) {
+  const node = doc.createElement(tag)
+  if (className) node.className = className
+  node.textContent = String(value ?? '')
+  return node
+}
+
+function progress(doc, label, percent, locale) {
+  const row = doc.createElement('div')
+  row.className = 'dqw-row'
+  const head = doc.createElement('div')
+  head.className = 'dqw-row-head'
+  head.append(text(doc, 'span', '', label), text(doc, 'span', '', formatPercent(percent, locale)))
+  const bar = doc.createElement('span')
+  bar.className = 'dqw-bar'
+  bar.setAttribute('role', 'progressbar')
+  bar.setAttribute('aria-valuemin', '0')
+  bar.setAttribute('aria-valuemax', '100')
+  bar.setAttribute('aria-valuenow', String(Math.max(0, Math.min(100, percent))))
+  const fill = doc.createElement('span')
+  fill.className = 'dqw-bar-fill'
+  fill.style.width = `${Math.max(0, Math.min(100, percent))}%`
+  bar.append(fill)
+  row.append(head, bar)
+  return row
+}
+
+function providerSection(doc, key, heading) {
+  const section = doc.createElement('section')
+  section.className = 'dqw-provider'
+  section.dataset.provider = key
+  section.append(text(doc, 'h4', 'dqw-provider-title', heading))
+  const content = doc.createElement('div')
+  content.dataset.role = 'content'
+  content.className = 'dqw-content'
+  section.append(content)
+  return section
+}
+
+function appendStaleError(doc, content, provider, copy) {
+  if (provider.status === 'stale') content.append(text(doc, 'p', 'dqw-error', `${copy.stale}: ${provider.error ?? ''}`))
+  else if (provider.status === 'error') content.append(text(doc, 'p', 'dqw-error', provider.error ?? copy.failed))
+}
+
+function renderGlm(doc, section, provider, copy, locale) {
+  const content = section.querySelector('[data-role="content"]')
+  content.replaceChildren()
+  appendStaleError(doc, content, provider, copy)
+  if (provider.status === 'missing') {
+    content.append(text(doc, 'p', 'dqw-muted', copy.notConfigured))
+    return
+  }
+  if (provider.status === 'loading') {
+    content.append(text(doc, 'p', 'dqw-muted', copy.loading))
+    return
+  }
+  if (!provider.plan?.windows?.length) {
+    if (provider.status !== 'error') content.append(text(doc, 'p', 'dqw-muted', copy.noPlan))
+    return
+  }
+  const planName = provider.plan.planName
+  if (planName) content.append(text(doc, 'p', 'dqw-muted', planName))
+  for (const window of provider.plan.windows) {
+    const label = window.name || copy.windows[window.key] || window.key
+    if (typeof window.percent === 'number' && Number.isFinite(window.percent)) {
+      content.append(progress(doc, label, window.percent, locale))
+    } else {
+      content.append(text(doc, 'p', 'dqw-muted', label))
+    }
+    const reset = formatTime(window.resetsAt, locale)
+    if (reset) content.append(text(doc, 'p', 'dqw-muted', `${copy.reset}: ${reset}`))
+  }
+}
+
+function renderCopilot(doc, section, provider, copy, locale) {
+  const content = section.querySelector('[data-role="content"]')
+  content.replaceChildren()
+  appendStaleError(doc, content, provider, copy)
+  if (provider.status === 'missing') {
+    content.append(text(doc, 'p', 'dqw-muted', copy.notConfigured))
+    return
+  }
+  if (provider.status === 'loading') {
+    content.append(text(doc, 'p', 'dqw-muted', copy.loading))
+    return
+  }
+  const quota = provider.quota
+  if (quota?.planName) content.append(text(doc, 'p', 'dqw-muted', quota.planName))
+  if (quota?.balanceAvailable === true) {
+    content.append(text(doc, 'p', 'dqw-credit', `${copy.available}: ${formatNumber(quota.remaining, locale)} / ${formatNumber(quota.entitlement, locale)}`))
+    if (typeof quota.percentRemaining === 'number') {
+      content.append(progress(doc, copy.available, quota.percentRemaining, locale))
+    }
+  } else {
+    content.append(text(doc, 'p', 'dqw-muted', copy.balanceUnavailable))
+  }
+  if (typeof quota?.creditsUsed === 'number') {
+    content.append(text(doc, 'p', 'dqw-muted', `${copy.used}: ${formatNumber(quota.creditsUsed, locale)}`))
+  }
+  const reset = formatTime(quota?.resetsAt, locale)
+  if (reset) content.append(text(doc, 'p', 'dqw-muted', `${copy.reset}: ${reset}`))
+}
+
+function footArea(doc) {
+  const sidebar = doc.querySelector('[data-pane="sidebar"], [class*="sidebarCol"]')
+  return sidebar?.querySelector('[class*="footArea"]') ?? undefined
+}
+
+function readCollapsed(win) {
+  try { return win.localStorage.getItem(COLLAPSE_KEY) === '1' } catch { return false }
+}
+
+function writeCollapsed(win, value) {
+  try { win.localStorage.setItem(COLLAPSE_KEY, value ? '1' : '0') } catch { /* Storage may be unavailable. */ }
+}
+
+async function fetchJson(fetchImpl, path, method) {
+  const response = await fetchImpl(path, {
+    ...(method === 'POST' ? { method: 'POST' } : {}),
+    cache: 'no-store',
+    signal: AbortSignal.timeout(FETCH_TIMEOUT_MS),
+  })
+  if (!response.ok) throw new Error(`HTTP ${response.status}`)
+  return response.json()
+}
+
+/** Mount a self-healing sidebar card. Dependencies are injectable for DOM tests. */
+export function mountQuotaCard({
+  doc = globalThis.document,
+  win = globalThis.window,
+  fetchImpl = globalThis.fetch,
+  pollIntervalMs = CLIENT_POLL_INTERVAL_MS,
+} = {}) {
+  if (!doc?.body || typeof fetchImpl !== 'function') return () => {}
+  if (doc.querySelector(CARD_SELECTOR)) return () => {}
+
+  const locale = localeFor(win)
+  const copy = COPY[locale]
+  const container = doc.createElement('div')
+  container.dataset.dshQuotaWatchCard = ''
+  container.dataset.dshPlugin = 'quota-watch'
+  container.dataset.dshPart = 'sidebar-card'
+  const style = text(doc, 'style', '', STYLE_TEXT)
+  const card = doc.createElement('section')
+  card.className = 'dqw-card'
+  card.setAttribute('aria-label', copy.title)
+  const header = doc.createElement('header')
+  header.className = 'dqw-head'
+  header.append(text(doc, 'h3', 'dqw-title', copy.title))
+  const refreshButton = text(doc, 'button', 'dqw-button', '↻')
+  refreshButton.type = 'button'
+  refreshButton.title = copy.refresh
+  refreshButton.setAttribute('aria-label', copy.refresh)
+  refreshButton.dataset.action = 'refresh'
+  const collapseButton = text(doc, 'button', 'dqw-button', '−')
+  collapseButton.type = 'button'
+  collapseButton.dataset.action = 'collapse'
+  const body = doc.createElement('div')
+  body.className = 'dqw-body'
+  const lastUpdated = text(doc, 'p', 'dqw-meta', copy.loading)
+  lastUpdated.dataset.role = 'updated'
+  const glmSection = providerSection(doc, 'glm', 'GLM Coding Plan')
+  const copilotSection = providerSection(doc, 'copilot', 'GitHub Copilot')
+  body.append(lastUpdated, glmSection, copilotSection)
+  header.append(refreshButton, collapseButton)
+  card.append(header, body)
+  container.append(style, card)
+  let collapsed = readCollapsed(win)
+  let snapshot
+  let requestSequence = 0
+  let timer
+  let disposed = false
+  let queuedPlace = false
+
+  const applyCollapsed = () => {
+    body.hidden = collapsed
+    collapseButton.textContent = collapsed ? '+' : '−'
+    collapseButton.title = collapsed ? copy.expand : copy.collapse
+    collapseButton.setAttribute('aria-label', collapseButton.title)
+    collapseButton.setAttribute('aria-expanded', String(!collapsed))
+  }
+
+  const render = (next) => {
+    snapshot = next
+    const glm = next?.providers?.find((provider) => provider.key === 'glm')
+    const copilot = next?.providers?.find((provider) => provider.key === 'copilot')
+    if (glm) renderGlm(doc, glmSection, glm, copy, locale)
+    if (copilot) renderCopilot(doc, copilotSection, copilot, copy, locale)
+    const updated = formatTime(next?.updatedAt ? new Date(next.updatedAt).toISOString() : '', locale)
+    lastUpdated.textContent = updated ? `${copy.updated}: ${updated}` : copy.loading
+  }
+
+  const showTransportError = () => {
+    if (snapshot) lastUpdated.textContent = copy.requestFailed
+    else lastUpdated.textContent = copy.failed
+  }
+
+  const poll = async (force) => {
+    if (disposed) return
+    const sequence = ++requestSequence
+    refreshButton.disabled = force === true
+    try {
+      const next = await fetchJson(fetchImpl, force ? CLIENT_ROUTES.refresh : CLIENT_ROUTES.overview, force ? 'POST' : 'GET')
+      if (sequence === requestSequence && !disposed) render(next)
+    } catch {
+      if (sequence === requestSequence && !disposed) showTransportError()
+    } finally {
+      if (!disposed && sequence === requestSequence) refreshButton.disabled = false
+    }
+  }
+
+  const place = () => {
+    const foot = footArea(doc)
+    if (!foot) return
+    const settings = foot.querySelector('[class*="settingsArea"]')
+    if (settings) {
+      if (container.parentElement !== foot || container.nextElementSibling !== settings) foot.insertBefore(container, settings)
+    } else if (container.parentElement !== foot || foot.lastElementChild !== container) {
+      foot.append(container)
+    }
+  }
+
+  const schedulePlace = () => {
+    if (queuedPlace || disposed) return
+    queuedPlace = true
+    queueMicrotask(() => {
+      queuedPlace = false
+      if (!disposed) place()
+    })
+  }
+
+  const observer = new win.MutationObserver(schedulePlace)
+  observer.observe(doc.body, { childList: true, subtree: true })
+  place()
+  applyCollapsed()
+
+  const startPolling = () => {
+    if (timer !== undefined || doc.visibilityState === 'hidden' || disposed) return
+    void poll(false)
+    timer = win.setInterval(() => { void poll(false) }, pollIntervalMs)
+  }
+  const stopPolling = () => {
+    if (timer !== undefined) win.clearInterval(timer)
+    timer = undefined
+  }
+  const onVisibilityChange = () => {
+    if (doc.visibilityState === 'hidden') stopPolling()
+    else startPolling()
+  }
+  const onRefresh = () => { void poll(true) }
+  const onCollapse = () => {
+    collapsed = !collapsed
+    writeCollapsed(win, collapsed)
+    applyCollapsed()
+  }
+  refreshButton.addEventListener('click', onRefresh)
+  collapseButton.addEventListener('click', onCollapse)
+  doc.addEventListener('visibilitychange', onVisibilityChange)
+  startPolling()
+
+  return () => {
+    disposed = true
+    stopPolling()
+    observer.disconnect()
+    doc.removeEventListener('visibilitychange', onVisibilityChange)
+    refreshButton.removeEventListener('click', onRefresh)
+    collapseButton.removeEventListener('click', onCollapse)
+    container.remove()
+  }
+}
+
+export function apply(ctx) {
+  ctx.effect(() => mountQuotaCard(), 'dsh-quota-watch: sidebar')
+}

package/src/core/adapters.mjs

@@ -0,0 +1,163 @@
+const GLM_HOSTS = Object.freeze({
+  'zai-coding-cn': 'open.bigmodel.cn',
+  zai: 'api.z.ai',
+  'zai-coding': 'api.z.ai',
+})
+
+const GLM_WINDOW_KEYS = Object.freeze({ 3: '5h', 5: 'month', 6: 'week' })
+const COPILOT_USER_URL = 'https://api.github.com/copilot_internal/user'
+const COPILOT_HEADERS = Object.freeze({
+  accept: 'application/json',
+  'user-agent': 'GitHubCopilotChat/0.35.0',
+  'editor-version': 'vscode/1.107.0',
+  'editor-plugin-version': 'copilot-chat/0.35.0',
+  'copilot-integration-id': 'vscode-chat',
+})
+
+function asRecord(value) {
+  return typeof value === 'object' && value !== null && !Array.isArray(value) ? value : undefined
+}
+
+function asString(value) {
+  return typeof value === 'string' && value.trim() !== '' ? value.trim() : undefined
+}
+
+function asNumber(value) {
+  if (typeof value === 'number' && Number.isFinite(value)) return value
+  if (typeof value === 'string' && value.trim() !== '') {
+    const parsed = Number(value)
+    return Number.isFinite(parsed) ? parsed : undefined
+  }
+  return undefined
+}
+
+function clampPercent(value) {
+  return Math.max(0, Math.min(100, value))
+}
+
+function toIso(value) {
+  if (typeof value === 'number' && Number.isFinite(value) && value > 0) {
+    const milliseconds = value < 1e12 ? value * 1000 : value
+    const date = new Date(milliseconds)
+    return Number.isNaN(date.getTime()) ? undefined : date.toISOString()
+  }
+  const text = asString(value)
+  if (text === undefined) return undefined
+  if (/^\d+$/.test(text)) return toIso(Number(text))
+  const date = new Date(text)
+  return Number.isNaN(date.getTime()) ? undefined : date.toISOString()
+}
+
+function glmPercent(row, kind) {
+  if (kind === 'CREDIT_LIMIT') {
+    const current = asNumber(row.currentValue)
+    const total = asNumber(row.usage)
+    if (current !== undefined && total !== undefined && total > 0) {
+      return clampPercent((current / total) * 100)
+    }
+  }
+  const reported = asNumber(row.percentage)
+  return reported === undefined ? undefined : clampPercent(reported)
+}
+
+/** Build a fixed-origin GLM Coding Plan quota request. */
+export function buildGlmProbe({ provider, apiKey } = {}) {
+  const host = GLM_HOSTS[provider]
+  const key = asString(apiKey)
+  if (host === undefined || key === undefined) return undefined
+  return {
+    url: `https://${host}/api/monitor/usage/quota/limit`,
+    headers: {
+      authorization: key,
+      'accept-language': 'en-US,en',
+    },
+  }
+}
+
+/** Normalize GLM's provider-defined quota windows without relying on row order. */
+export function parseGlmPlan(status, body) {
+  if (status !== 200) return undefined
+  const root = asRecord(body)
+  if (root?.success !== true) return undefined
+  const data = asRecord(root.data)
+  if (!Array.isArray(data?.limits)) return undefined
+
+  const windows = []
+  for (const candidate of data.limits) {
+    const row = asRecord(candidate)
+    if (row === undefined) continue
+    const kind = asString(row.type)
+    if (kind === 'TIME_LIMIT') continue
+    if (kind !== 'TOKENS_LIMIT' && kind !== 'CREDIT_LIMIT') continue
+    const unit = asNumber(row.unit)
+    const key = GLM_WINDOW_KEYS[unit]
+    if (key === undefined) continue
+    const percent = glmPercent(row, kind)
+    const resetsAt = toIso(row.nextResetTime)
+    windows.push({
+      key,
+      ...(percent !== undefined ? { percent } : {}),
+      ...(resetsAt !== undefined ? { resetsAt } : {}),
+    })
+  }
+
+  if (windows.length === 0) return undefined
+  const planName = asString(data.level)
+  return { ...(planName !== undefined ? { planName } : {}), windows }
+}
+
+/** Build the Copilot user-quota request using the raw GitHub OAuth grant. */
+export function buildCopilotProbe({ githubOAuthToken } = {}) {
+  const token = asString(githubOAuthToken)
+  if (token === undefined) return undefined
+  return {
+    url: COPILOT_USER_URL,
+    headers: {
+      authorization: `Bearer ${token}`,
+      ...COPILOT_HEADERS,
+    },
+  }
+}
+
+/**
+ * Normalize the Copilot user's premium-interaction snapshot.
+ * Remaining credits are never derived from entitlement minus credits_used.
+ */
+export function parseCopilotUser(status, body) {
+  if (status !== 200) return undefined
+  const root = asRecord(body)
+  const snapshots = asRecord(root?.quota_snapshots)
+  const snapshot = asRecord(snapshots?.premium_interactions)
+  if (snapshot === undefined) return undefined
+
+  const tokenBasedBilling = typeof snapshot.token_based_billing === 'boolean'
+    ? snapshot.token_based_billing
+    : typeof root?.token_based_billing === 'boolean'
+      ? root.token_based_billing
+      : undefined
+  const entitlement = asNumber(snapshot.entitlement)
+  const reportedRemaining = asNumber(snapshot.remaining)
+  const balanceAvailable = tokenBasedBilling === true
+    && snapshot.unlimited === false
+    && entitlement !== undefined
+    && entitlement > 0
+    && reportedRemaining !== undefined
+    && reportedRemaining >= 0
+  const percentReported = asNumber(snapshot.percent_remaining)
+  const creditsUsed = tokenBasedBilling === true ? asNumber(snapshot.credits_used) : undefined
+  const resetsAt = toIso(root?.quota_reset_date_utc) ?? toIso(root?.quota_reset_date)
+  const planName = asString(root?.copilot_plan)
+
+  return {
+    ...(planName !== undefined ? { planName } : {}),
+    ...(tokenBasedBilling !== undefined ? { tokenBasedBilling } : {}),
+    balanceAvailable,
+    ...(balanceAvailable ? {
+      remaining: reportedRemaining,
+      entitlement,
+      ...(percentReported !== undefined ? { percentRemaining: clampPercent(percentReported) } : {}),
+    } : {}),
+    ...(creditsUsed !== undefined && creditsUsed >= 0 ? { creditsUsed } : {}),
+    ...(resetsAt !== undefined ? { resetsAt } : {}),
+  }
+}