import { and, eq, inArray } from 'drizzle-orm'
import type { ProviderType } from '#shared/types'

const TYPE_NAME: Record<ProviderType, string> = {
  'openai-compatible': 'OpenAI 兼容',
  'anthropic': 'Anthropic',
  'google': 'Gemini'
}

/** 未填写名称时，用 Base URL 的域名作为连接名称 */
export function defaultProviderName(type: ProviderType, baseUrl: string) {
  try {
    if (baseUrl.trim()) return new URL(baseUrl.trim()).host
  }
  catch {}
  return TYPE_NAME[type]
}

export function normalizeModelIds(ids: string[]) {
  return [...new Set(ids.map(s => s.trim()).filter(Boolean))]
}

/**
 * 把连接下的模型同步为给定列表：新增缺少的，删除不在列表中的。
 * 被删除模型的 AI 成员会因外键 ON DELETE SET NULL 变为“未指定模型”。
 */
export function syncProviderModels(providerId: string, modelIds: string[]) {
  const db = useDb()
  const wanted = normalizeModelIds(modelIds)
  const existing = db.select().from(tables.models).where(eq(tables.models.providerId, providerId)).all()
  const toDelete = existing.filter(m => !wanted.includes(m.modelId)).map(m => m.id)
  const toAdd = wanted.filter(id => !existing.some(m => m.modelId === id))
  if (toDelete.length) {
    db.delete(tables.models).where(and(eq(tables.models.providerId, providerId), inArray(tables.models.id, toDelete))).run()
  }
  if (toAdd.length) {
    db.insert(tables.models).values(toAdd.map(modelId => ({ id: newId(), providerId, label: modelId, modelId }))).run()
  }
  return { added: toAdd.length, removed: toDelete.length }
}