dsh-router-core 0.2.0 → 0.3.1

package/src/supplier-config.ts

@@ -29,12 +29,28 @@ interface ConfigFile {
 export class SupplierConfigStore {
   private fp = ''
   private bySupplier = new Map<string, SupplierConfig>()
+  /** 全部供应商 id（加载时同步），用于别名唯一性校验。 */
+  private knownIds = new Set<string>()
 
   constructor(stateFile: string) {
     this.fp = stateFile ? join(dirname(stateFile), 'supplier-config.json') : ''
     this.load()
   }
 
+  /** 同步已知供应商 id 集合（加载供应商后调用）。 */
+  sync(supplierIds: string[]): void {
+    this.knownIds = new Set(supplierIds)
+  }
+
+  /** 各供应商当前生效的别名（未设置别名的用供应商 id）。 */
+  effectiveAliases(): Map<string, string> {
+    const out = new Map<string, string>()
+    for (const id of this.knownIds) {
+      out.set(id, this.bySupplier.get(id)?.alias || id)
+    }
+    return out
+  }
+
   /** 读取某供应商配置(不存在则返回默认并注册)。 */
   get(supplierId: string): SupplierConfig {
     let cfg = this.bySupplier.get(supplierId)
@@ -45,9 +61,21 @@ export class SupplierConfigStore {
     return cfg
   }
 
-  setAlias(supplierId: string, alias: string): void {
-    this.get(supplierId).alias = alias
+  /**
+   * 设置别名。别名必须唯一——它是对外模型全名的前缀（alias/model），
+   * 请求时用别名反查供应商，重复就会指错人。空别名表示用供应商 id（默认值）。
+   * @returns 冲突时返回占用的供应商 id
+   */
+  setAlias(supplierId: string, alias: string): { ok: boolean; conflictWith?: string } {
+    const clean = alias.trim()
+    // 空 = 用默认（供应商 id）；此时要确认没有别人显式占用了本供应商的 id 作别名
+    const effective = clean === '' ? supplierId : clean
+    for (const [id, cur] of this.effectiveAliases()) {
+      if (id !== supplierId && cur === effective) return { ok: false, conflictWith: id }
+    }
+    this.get(supplierId).alias = clean
     this.saveLocked()
+    return { ok: true }
   }
 
   setModelEnabled(supplierId: string, modelId: string, enabled: boolean): void {

package/src/suppliers/loader.ts

@@ -123,7 +123,8 @@ export function wrapModule(instance: SupplierModule, env: SupplierEnv, source: s
         })),
       )
     },
-    getAlias: (): string => env.store.get(instance.id).alias || instance.getAlias(),
+    // 别名留空即用供应商 id（默认值）——别名是用户可改的展示前缀，唯一性由核心校验
+    getAlias: (): string => env.store.get(instance.id).alias || instance.id,
     customModelIds: (): string[] => [...env.store.get(instance.id).custom],
     accounts: (): SupplierAccountNow[] => instance.status().accounts,
     chatOnce: (uid: string, req: ChatRequest): Promise<ChatOnceResult> => instance.chatOnce(uid, req),

package/src/suppliers/nvidia/plugin.ts

@@ -29,7 +29,6 @@ const BASE = 'https://integrate.api.nvidia.com/v1'
 const MODELS_URL = `${BASE}/models`
 const CHAT_URL = `${BASE}/chat/completions`
 /** 默认前缀（用户可在面板改；loader 包装会优先用 store 里的值）。 */
-const DEFAULT_ALIAS = 'nv'
 /** 非聊天用途的模型（护栏/翻译/解析/检测/图像生成等），聊天路由用不上，拉列表时过滤掉。 */
 const NON_CHAT_MODEL = /(guard|safety|moderation|translate|transcription|parse|detector|embed|rerank|diffusion|tts|asr|calibration)/i
 
@@ -68,7 +67,7 @@ export default function factory(env: SupplierEnv): SupplierModule {
 
   /** 当前前缀（与 loader 包装一致：store 覆盖默认值）。 */
   function currentAlias(): string {
-    return env.store.get(id).alias || DEFAULT_ALIAS
+    return env.store.get(id).alias || id
   }
 
   // ---- 模型可用性探测（NIM 的 /v1/models 无状态字段，大量模型已下线/未授权） ----
@@ -218,7 +217,7 @@ export default function factory(env: SupplierEnv): SupplierModule {
       return { id, name, accounts }
     },
     listModels: (force?: boolean): Promise<ModelInfo[]> => allModels(!!force),
-    getAlias: (): string => 'nv',
+    getAlias: (): string => id,
     async addApiKey(input: { name: string; apiKey: string }): Promise<{ ok: boolean; error?: string; account?: { uid: string; nickname: string } }> {
       const key = input.apiKey.trim()
       if (key === '') return { ok: false, error: 'API key 不能为空' }

package/src/suppliers/opencode/plugin.ts

@@ -11,7 +11,6 @@
 import type { ChatRequest, ModelInfo } from '../../router/types.ts'
 import type { AccountState, ChatOnceResult, SupplierEnv, SupplierModule, SupplierStatusNow } from '../contract.ts'
 
-const DEFAULT_ALIAS = 'oc'
 
 export const id = 'opencode'
 export const name = 'OpenCode Free'
@@ -61,7 +60,7 @@ export default function factory(env: SupplierEnv): SupplierModule {
 
     /** 当前前缀（与 loader 包装一致：store 覆盖默认值）。 */
   function currentAlias(): string {
-    return env.store.get(id).alias || DEFAULT_ALIAS
+    return env.store.get(id).alias || id
   }
 
   /** 剥本供应商 alias 前缀（只剥自己的，模型 id 自带的斜杠保留）。 */
@@ -82,7 +81,7 @@ export default function factory(env: SupplierEnv): SupplierModule {
     icon,
     status: (): SupplierStatusNow => ({ id, name, accounts: [] }),
     listModels: (force?: boolean): Promise<ModelInfo[]> => allModels(!!force),
-    getAlias: (): string => 'oc',
+    getAlias: (): string => id,
     // testModel 由 dsh-router 核心统一走 chatOnce 路径（无账号，无需回退）
     lastError: (): string | undefined => lastErr,
     /** 无账号供应商：uid 恒为空，只认 free 模型。 */

package/src/suppliers/openrouter/plugin.ts

@@ -23,7 +23,6 @@ const BASE = 'https://openrouter.ai/api/v1'
 const MODELS_URL = `${BASE}/models`
 const CHAT_URL = `${BASE}/chat/completions`
 /** 默认前缀（用户可在面板改；loader 包装会优先用 store 里的值）。 */
-const DEFAULT_ALIAS = 'or'
 
 /** 免费模型过滤（同 9Router openrouter-free）：pricing 全 0 且 context ≥ 200k。 */
 function isFreeModelMeta(m: { pricing?: { prompt?: string; completion?: string }; context_length?: number }): boolean {
@@ -70,7 +69,7 @@ export default function factory(env: SupplierEnv): SupplierModule {
 
   /** 当前前缀（与 loader 包装一致：store 覆盖默认值）。 */
   function currentAlias(): string {
-    return env.store.get(id).alias || DEFAULT_ALIAS
+    return env.store.get(id).alias || id
   }
 
   async function refreshModels(): Promise<ModelInfo[]> {
@@ -122,7 +121,7 @@ export default function factory(env: SupplierEnv): SupplierModule {
       return { id, name, accounts }
     },
     listModels: (force?: boolean): Promise<ModelInfo[]> => allModels(!!force),
-    getAlias: (): string => 'or',
+    getAlias: (): string => id,
     async addApiKey(input: { name: string; apiKey: string }): Promise<{ ok: boolean; error?: string; account?: { uid: string; nickname: string } }> {
       const key = input.apiKey.trim()
       if (key === '') return { ok: false, error: 'API key 不能为空' }

package/src/suppliers/registry.ts

@@ -121,11 +121,16 @@ export function supplierRoutes(base: string, loaded: LoadedSupplier, store: Supp
       const body = JSON.parse(await readBody(req)) as { alias?: string }
       const alias = body.alias ?? ''
       const clean = alias.trim()
-      if (clean === '' || !/^[A-Za-z0-9_-]+$/.test(clean)) {
+      // 空 = 用供应商 id（默认值），合法；非空则校验字符集
+      if (clean !== '' && !/^[A-Za-z0-9_-]+$/.test(clean)) {
         writeJson(res, 400, { ok: false, error: '前缀只能包含字母、数字、- 和 _' })
         return
       }
-      store.setAlias(s.id, clean)
+      const r = store.setAlias(s.id, clean)
+      if (!r.ok) {
+        writeJson(res, 400, { ok: false, error: `前缀已被供应商 ${JSON.stringify(r.conflictWith ?? '?')} 占用` })
+        return
+      }
       writeJson(res, 200, { ok: true })
     },
   })