import type { Provider, ProviderName } from "../types/provider"
import type { ConcurrencyConfig } from "../types/concurrency"
import { SupermemoryProvider } from "./supermemory"
import { Mem0Provider } from "./mem0"
import { ZepProvider } from "./zep"
import { FilesystemProvider } from "./filesystem"
import { RAGProvider } from "./rag"
import { CogmemAiProvider } from "./cogmemai"

const providers: Record<ProviderName, new () => Provider> = {
  supermemory: SupermemoryProvider,
  mem0: Mem0Provider,
  zep: ZepProvider,
  filesystem: FilesystemProvider,
  rag: RAGProvider,
  cogmemai: CogmemAiProvider,
}

export function createProvider(name: ProviderName): Provider {
  const ProviderClass = providers[name]
  if (!ProviderClass) {
    throw new Error(`Unknown provider: ${name}. Available: ${Object.keys(providers).join(", ")}`)
  }
  return new ProviderClass()
}

export function getAvailableProviders(): ProviderName[] {
  return Object.keys(providers) as ProviderName[]
}

export function getProviderInfo(name: ProviderName): {
  name: string
  displayName: string
  concurrency: ConcurrencyConfig | null
} {
  const provider = createProvider(name)
  return {
    name,
    displayName: name.charAt(0).toUpperCase() + name.slice(1),
    concurrency: provider.concurrency || null,
  }
}

export { SupermemoryProvider, Mem0Provider, ZepProvider, FilesystemProvider, RAGProvider, CogmemAiProvider }