@qthinker/opencode-deepseek-search 0.1.1 → 0.2.1

package/src/gui/panel.tsx

@@ -0,0 +1,154 @@
+// The desktop side-panel tab: the same states and copy as the TUI card
+// (status-view.ts), composed for a resizable panel instead of a terminal column.
+import type { JSX } from "solid-js"
+import { For, Show, createMemo, onCleanup, onMount } from "solid-js"
+import { usePlugin } from "@opencode/plugin/gui"
+import type { SearchStatus } from "../rpc.ts"
+import { sidebarRows, type SidebarRows } from "../status-view.ts"
+
+export interface DeepSeekPanelProps {
+  readonly status: SearchStatus | undefined
+  readonly error: string | undefined
+  readonly busy: boolean
+  readonly onToggle: () => void
+  readonly onRefresh: () => void
+}
+
+type Tone = SidebarRows["tone"]
+
+/** Text-presentation symbols, never emoji. */
+const GLYPH: Readonly<Record<Tone, string>> = { on: "●", off: "○", warn: "◐", error: "!", idle: "…" }
+
+/**
+ * No theme tokens exist on the desktop side, so the tone is a minimal palette
+ * per mode; text and rules otherwise inherit the host's color.
+ */
+const INK = {
+  dark: { on: "#63d19a", warn: "#e0ad4f", error: "#ee7b74", off: "#8b9198", idle: "#8b9198", rule: "rgba(255,255,255,0.14)" },
+  light: { on: "#1e7a4c", warn: "#8a5f08", error: "#b03931", off: "#6b7076", idle: "#6b7076", rule: "rgba(0,0,0,0.13)" },
+} as const
+
+const LABEL: Readonly<Record<Tone, string>> = {
+  on: "停用（交还原生搜索）", off: "启用（由 DeepSeek 接管）",
+  warn: "交还原生搜索", error: "重试", idle: "重试",
+}
+
+const ICON = { on: "arrow-undo-down", off: "globe", warn: "arrow-undo-down", error: "refresh", idle: "refresh" } as const
+
+/** The product name is chrome; the state after ` · ` is the message. */
+function headline(title: string, accent: string): JSX.Element {
+  const at = title.indexOf(" · ")
+  if (at < 0) return <span style={{ color: accent }}>{title}</span>
+  return (
+    <>
+      <span style={{ opacity: 0.6, "font-weight": 400 }}>{title.slice(0, at)}</span>
+      <span style={{ opacity: 0.3, "font-weight": 400 }}>{" · "}</span>
+      <span style={{ color: accent }}>{title.slice(at + 3)}</span>
+    </>
+  )
+}
+
+/** `今日 搜索 3 · 抓取 5 · …` → its label and ` · ` chunks, so values wrap as whole stats. */
+function splitLine(line: string): { readonly label: string; readonly chunks: readonly string[] } {
+  const at = line.indexOf(" ")
+  return at < 0 ? { label: line, chunks: [] } : { label: line.slice(0, at), chunks: line.slice(at + 1).split(" · ") }
+}
+
+/** The fix-it row: the env var name is an identifier, so give it the mono font. */
+function help(action: string, mono: string): JSX.Element {
+  const key = "DEEPSEEK_API_KEY"
+  const at = action.indexOf(key)
+  if (at < 0) return <>{action}</>
+  return (
+    <>
+      {action.slice(0, at)}<code style={{ "font-family": mono, "font-size": "0.94em" }}>{key}</code>{action.slice(at + key.length)}
+    </>
+  )
+}
+
+export function DeepSeekPanel(props: DeepSeekPanelProps): JSX.Element {
+  const ctx = usePlugin()
+  const rows = createMemo(() => sidebarRows({ status: props.status, error: props.error, maxColumns: Infinity }))
+  const ink = createMemo(() => INK[ctx.appearance.mode()])
+  const mono = () => ctx.appearance.font("mono")
+
+  // Cross-half `changed` delivery is the one runtime behaviour this repo cannot
+  // verify, so the panel rereads on its own timer while it is mounted — the same
+  // fallback the TUI sidebar uses.
+  onMount(() => {
+    const timer = setInterval(() => props.onRefresh(), 10_000)
+    onCleanup(() => clearInterval(timer))
+  })
+
+  return (
+    <div
+      aria-busy={props.busy || undefined}
+      style={{
+        display: "flex", "flex-direction": "column", height: "100%", "min-width": 0,
+        "box-sizing": "border-box", padding: "16px 16px 24px", "overflow-y": "auto", "overflow-x": "hidden",
+      }}
+    >
+      <header style={{ display: "flex", "align-items": "flex-start", gap: "8px" }}>
+        <span aria-hidden="true" style={{ "flex-shrink": 0, "font-size": "10px", "line-height": "20px", "font-weight": 700, color: ink()[rows().tone] }}>
+          {GLYPH[rows().tone]}
+        </span>
+        <h2 aria-live="polite" style={{ flex: 1, "min-width": 0, margin: 0, "font-size": "13px", "font-weight": 600, "line-height": "20px", "overflow-wrap": "anywhere" }}>
+          {headline(rows().title, ink()[rows().tone])}
+        </h2>
+        <ctx.ui.Tooltip value="刷新用量" placement="bottom">
+          <ctx.ui.IconButton icon="refresh" label="刷新用量" size="small" onClick={props.onRefresh} />
+        </ctx.ui.Tooltip>
+      </header>
+
+      {props.status === undefined ? (
+        rows().lines.length > 0 ? (
+          <p style={{ margin: "14px 0 0", "font-size": "12px", "line-height": 1.6, opacity: 0.7, "overflow-wrap": "anywhere" }}>{rows().lines[0]}</p>
+        ) : (
+          <div style={{ display: "flex", "align-items": "center", gap: "8px", "margin-top": "14px", opacity: 0.7 }}>
+            <ctx.ui.Loader size="small" />
+            <span style={{ "font-size": "12px" }}>正在读取用量…</span>
+          </div>
+        )
+      ) : (
+        <dl style={{ margin: "16px 0 0", "border-top": `1px solid ${ink().rule}`, "border-bottom": `1px solid ${ink().rule}` }}>
+          <For each={rows().lines}>
+            {(line, index) => {
+              const parts = splitLine(line)
+              return (
+                <div style={{ display: "flex", "align-items": "baseline", gap: "12px", padding: "10px 0", "border-top": index() > 0 ? `1px solid ${ink().rule}` : undefined }}>
+                  <dt style={{ "flex-shrink": 0, "font-size": "11px", "letter-spacing": "0.08em", "line-height": "1.9", opacity: 0.62 }}>
+                    {parts.label}
+                  </dt>
+                  <dd style={{
+                    margin: 0, "margin-left": "auto", "min-width": 0, display: "flex", "flex-wrap": "wrap", "justify-content": "flex-end",
+                    "column-gap": "6px", "row-gap": "2px", "font-family": mono(), "font-size": "13px", "line-height": "1.55",
+                    "font-variant-numeric": "tabular-nums", "text-align": "right",
+                  }}>
+                    <For each={parts.chunks}>
+                      {(chunk, i) => (
+                        <span style={{ "min-width": 0, "overflow-wrap": "anywhere", "font-weight": chunk.startsWith("¥") ? 500 : 400, opacity: chunk.startsWith("¥") ? 1 : 0.8 }}>
+                          {chunk}{i() < parts.chunks.length - 1 ? " ·" : ""}
+                        </span>
+                      )}
+                    </For>
+                  </dd>
+                </div>
+              )
+            }}
+          </For>
+        </dl>
+      )}
+
+      <Show when={!rows().clickable}>
+        <p style={{ margin: "16px 0 0", "font-size": "12px", "line-height": 1.6, opacity: 0.68, "overflow-wrap": "anywhere" }}>{help(rows().action, mono())}</p>
+      </Show>
+      <Show when={rows().clickable && rows().tone !== "idle"}>
+        <div style={{ "margin-top": "18px" }}>
+          <ctx.ui.Button variant={rows().tone === "off" || rows().tone === "error" ? "contrast" : "neutral"} icon={ICON[rows().tone]} disabled={props.busy} busy={props.busy} onClick={props.onToggle}>
+            {LABEL[rows().tone]}
+          </ctx.ui.Button>
+        </div>
+      </Show>
+    </div>
+  )
+}

package/src/index.ts

@@ -14,6 +14,7 @@
 
 import { Plugin } from "@opencode/plugin"
 import type { Result as ToolResult, ToolContext } from "@opencode/plugin/promise/tool"
+import type { RpcRegistration } from "@opencode/plugin/promise/rpc"
 import {
   deepseekFetch,
   deepseekSearch,
@@ -23,6 +24,7 @@ import {
   validateUrl,
   type SearchResult,
 } from "./deepseek.ts"
+import { DeepSeekRpc, type ModeSwitchResult, type SearchStatus } from "./rpc.ts"
 import {
   USAGE_KEY,
   applyTally,
@@ -32,6 +34,8 @@ import {
   formatUsage,
   formatYuan,
   parseLedger,
+  sumDays,
+  usageSummaryLines,
   type Tally,
   type UsageLedger,
 } from "./usage.ts"
@@ -118,10 +122,19 @@ export default Plugin.define({
       chain = run.catch(() => {})
       return run
     }
+    // Filled in after the tool transforms are registered; `tally` can run before
+    // that, so this stays optional. `changed` tells the sidebars to re-read `status`.
+    let rpcRegistration: RpcRegistration<typeof DeepSeekRpc> | undefined
+    // Nested `changed` events are cheap, but they are also not worth failing a
+    // search or a mode switch over, so every caller ignores the outcome.
+    const notifyChanged = (): void => {
+      void rpcRegistration?.events.emit("changed", {}).catch(() => {})
+    }
     const tally = (kind: Tally): void => {
       void withLedger(async (ledger) => {
         applyTally(ledger, kind, pricePerThousand)
         await ctx.storage.set(USAGE_KEY, ledger)
+        notifyChanged()
         const today = ledger.days[dayKey()] ?? emptyDay()
         console.error(
           `[deepseek-search] usage: today search=${today.search} fetch=${today.fetch} ${formatYuan(today.costMicro)}`,
@@ -212,6 +225,32 @@ export default Plugin.define({
           : " · 未找到内置 websearch 工具，实际未能接管"
         : " · 内置 websearch 保持原生")
 
+    // Shared by the `deepseek_mode` tool and the RPC `setMode` method. The
+    // returned strings are the tool's documented output, word for word.
+    const switchMode = async (action: Exclude<ModeAction, "status">): Promise<string> => {
+      if (!pristine) {
+        return (
+          `未找到内置 ${SEARCH_TOOL} 工具，无法在运行时切换。` +
+          `请改用配置文件把 options.mode 设为 "${action === "config" ? configMode : action}"。`
+        )
+      }
+      if (action === "config") {
+        await ctx.storage.remove(MODE_KEY)
+        overrideMode = undefined
+        await applySearchMode(configMode)
+        notifyChanged()
+        return `${describeMode()}\n\n已清除持久化开关，回到配置文件设置（${configMode}）。`
+      }
+      await ctx.storage.set(MODE_KEY, action)
+      overrideMode = action
+      await applySearchMode(action)
+      notifyChanged()
+      return (
+        `${describeMode()}\n\n已持久化：重启 OpenCode 后仍为 ${action}。` +
+        '要恢复配置文件设置，调用 deepseek_mode 并传 mode: "config"。'
+      )
+    }
+
     // ---------- tools ----------
     await ctx.tool.transform((editor) => {
       const builtin = editor.get(SEARCH_TOOL)
@@ -314,31 +353,50 @@ export default Plugin.define({
         execute: async (input) => {
           const action = readModeAction(input)
           if (action === "status") return { content: describeMode() }
-          if (!pristine) {
-            return {
-              content:
-                `未找到内置 ${SEARCH_TOOL} 工具，无法在运行时切换。` +
-                `请改用配置文件把 options.mode 设为 "${action === "config" ? configMode : action}"。`,
-            }
-          }
-          if (action === "config") {
-            await ctx.storage.remove(MODE_KEY)
-            overrideMode = undefined
-            await applySearchMode(configMode)
-            return { content: `${describeMode()}\n\n已清除持久化开关，回到配置文件设置（${configMode}）。` }
-          }
-          await ctx.storage.set(MODE_KEY, action)
-          overrideMode = action
-          await applySearchMode(action)
-          return {
-            content:
-              `${describeMode()}\n\n已持久化：重启 OpenCode 后仍为 ${action}。` +
-              '要恢复配置文件设置，调用 deepseek_mode 并传 mode: "config"。',
-          }
+          return { content: await switchMode(action) }
         },
       })
     })
 
+    // ---------- RPC contract (TUI / desktop sidebars) ----------
+    // Both sidebars read `status` and call `setMode`; nothing else crosses the
+    // process boundary, so the plugin's state stays the single source of truth.
+    rpcRegistration = await ctx.rpc.register(DeepSeekRpc, {
+      // Reads through `withLedger` so it observes the same serialized chain.
+      status: async (): Promise<SearchStatus> =>
+        withLedger((ledger): SearchStatus => {
+          const todayKey = dayKey()
+          return {
+            mode,
+            source: overrideMode ? "override" : "config",
+            patched: searchToolPatched,
+            hasKey: Boolean(apiKey),
+            pricePerThousand,
+            ...usageSummaryLines(ledger, todayKey),
+            // Same ledger read as the strings above, exposed as numbers so the
+            // sidebars can lay them out for their own width.
+            counts: {
+              today: sumDays(ledger, (k) => k === todayKey),
+              month: sumDays(ledger, (k) => k.startsWith(todayKey.slice(0, 7))),
+              total: sumDays(ledger, () => true),
+            },
+          }
+        }),
+      // The input is plain JSON Schema, hence `unknown`: report, never throw.
+      setMode: async (input: unknown): Promise<ModeSwitchResult> => {
+        let action: ModeAction
+        try {
+          action = readModeAction(input)
+        } catch (e) {
+          return { ok: false, mode, message: e instanceof Error ? e.message : String(e) }
+        }
+        if (action === "status") return { ok: false, mode, message: "status 只读取当前模式，不做切换。" }
+        const message = await switchMode(action)
+        // `!pristine` means switchMode could not apply the change; its message says why.
+        return { ok: Boolean(pristine), mode, message }
+      },
+    })
+
     // Load marker for diagnostics; never logs the key itself.
     console.error(
       `[deepseek-search] loaded: mode=${mode} source=${overrideMode ? "override" : "config"} ` +

package/src/rpc.ts

@@ -0,0 +1,160 @@
+// RPC contract shared by the server plugin (handler side) and the TUI / desktop
+// sidebars (client side).
+//
+// Type-only imports only: this module must stay loadable from all three
+// runtimes (Node type-stripping for the server and the TUI, a CJS bundle for
+// the desktop app), so it carries no runtime dependency.
+//
+// The schemas are plain JSON Schema, so `Rpc.Input` / `Rpc.Output` degrade to
+// `unknown` on the client. `readStatus` / `readModeResult` narrow that back
+// defensively, in the style of `parseLedger` (src/usage.ts).
+
+import type { Rpc } from "@opencode/plugin/rpc"
+
+export const DEEPSEEK_RPC_ID = "deepseek-search"
+
+// Shared by the today/month/total slots of `status.counts`.
+const USAGE_COUNTS_SCHEMA = {
+  type: "object",
+  properties: {
+    search: { type: "integer" },
+    fetch: { type: "integer" },
+    empty: { type: "integer" },
+    failed: { type: "integer" },
+    costMicro: { type: "integer" },
+  },
+  required: ["search", "fetch", "empty", "failed", "costMicro"],
+  additionalProperties: false,
+} as const
+
+export const DeepSeekRpc = {
+  id: DEEPSEEK_RPC_ID,
+  methods: {
+    status: {
+      input: { type: "object", properties: {}, additionalProperties: false },
+      output: {
+        type: "object",
+        properties: {
+          mode: { type: "string", enum: ["deepseek", "native"] },
+          source: { type: "string", enum: ["override", "config"] },
+          patched: { type: "boolean" },
+          hasKey: { type: "boolean" },
+          pricePerThousand: { type: "number" },
+          today: { type: "string" },
+          month: { type: "string" },
+          total: { type: "string" },
+          counts: {
+            type: "object",
+            properties: {
+              today: USAGE_COUNTS_SCHEMA,
+              month: USAGE_COUNTS_SCHEMA,
+              total: USAGE_COUNTS_SCHEMA,
+            },
+            required: ["today", "month", "total"],
+            additionalProperties: false,
+          },
+        },
+        required: ["mode", "source", "patched", "hasKey", "pricePerThousand", "today", "month", "total"],
+        additionalProperties: false,
+      },
+    },
+    setMode: {
+      input: {
+        type: "object",
+        properties: {
+          mode: { type: "string", enum: ["deepseek", "native", "config"] },
+        },
+        required: ["mode"],
+        additionalProperties: false,
+      },
+      output: {
+        type: "object",
+        properties: {
+          ok: { type: "boolean" },
+          mode: { type: "string" },
+          message: { type: "string" },
+        },
+        required: ["ok", "mode", "message"],
+        additionalProperties: false,
+      },
+    },
+  },
+  events: {
+    // Emitted after the ledger or the mode changes.
+    changed: { schema: { type: "object" as const, properties: {} } },
+  },
+} satisfies Rpc.PortableDefinition
+
+export interface UsageCounts {
+  search: number
+  fetch: number
+  empty: number
+  failed: number
+  costMicro: number
+}
+
+export interface SearchStatus {
+  mode: "deepseek" | "native"
+  source: "override" | "config"
+  patched: boolean
+  hasKey: boolean
+  pricePerThousand: number
+  today: string
+  month: string
+  total: string
+  /** Structured counters, so each surface can format for its own width. Optional for older servers. */
+  counts?: { today: UsageCounts; month: UsageCounts; total: UsageCounts }
+}
+
+export interface ModeSwitchResult {
+  ok: boolean
+  mode: string
+  message: string
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return typeof value === "object" && value !== null && !Array.isArray(value)
+}
+
+function isMode(value: unknown): value is SearchStatus["mode"] {
+  return value === "deepseek" || value === "native"
+}
+
+/** Strict read of one counter bucket: all five fields, integers ≥ 0, finite cost. */
+function readUsageCounts(value: unknown): UsageCounts | undefined {
+  if (!isRecord(value)) return undefined
+  const { search, fetch, empty, failed, costMicro } = value
+  for (const count of [search, fetch, empty, failed]) {
+    if (typeof count !== "number" || !Number.isInteger(count) || count < 0) return undefined
+  }
+  if (typeof costMicro !== "number" || !Number.isFinite(costMicro) || costMicro < 0) return undefined
+  return { search, fetch, empty, failed, costMicro } as UsageCounts
+}
+
+/** Defensive read of an RPC `status` result. Never throws. */
+export function readStatus(raw: unknown): SearchStatus | undefined {
+  if (!isRecord(raw)) return undefined
+  const { mode, source, patched, hasKey, pricePerThousand, today, month, total, counts } = raw
+  if (!isMode(mode)) return undefined
+  if (source !== "override" && source !== "config") return undefined
+  if (typeof patched !== "boolean" || typeof hasKey !== "boolean") return undefined
+  if (typeof pricePerThousand !== "number" || !Number.isFinite(pricePerThousand)) return undefined
+  if (typeof today !== "string" || typeof month !== "string" || typeof total !== "string") return undefined
+  const status: SearchStatus = { mode, source, patched, hasKey, pricePerThousand, today, month, total }
+  // Old servers omit `counts`; a malformed triple is dropped without failing the rest.
+  if (isRecord(counts)) {
+    const todayCounts = readUsageCounts(counts.today)
+    const monthCounts = readUsageCounts(counts.month)
+    const totalCounts = readUsageCounts(counts.total)
+    if (todayCounts && monthCounts && totalCounts) status.counts = { today: todayCounts, month: monthCounts, total: totalCounts }
+  }
+  return status
+}
+
+/** Defensive read of an RPC `setMode` result. Never throws. */
+export function readModeResult(raw: unknown): ModeSwitchResult | undefined {
+  if (!isRecord(raw)) return undefined
+  const { ok, mode, message } = raw
+  if (typeof ok !== "boolean" || typeof mode !== "string" || typeof message !== "string") return undefined
+  return { ok, mode, message }
+}