import { Context } from 'cordis'
import LlmService from '@deepseek-ai/dsh-llm'
import SessionStore from '@deepseek-ai/dsh-session'
import SystemPrompt from '@deepseek-ai/dsh-system-prompt'
import ToolRegistry from '@deepseek-ai/dsh-tools'
import AgentRegistry, { type Agent } from '@deepseek-ai/dsh-agent'
import AgentLoop from '@deepseek-ai/dsh-agent-loop'
import * as LlmDeepSeek from '@deepseek-ai/dsh-llm-deepseek'
import * as ToolCordis from '@deepseek-ai/dsh-tool-cordis'

/**
 * Shared harness for the cordis-agent e2e suite: the agent spine with the real
 * DeepSeek adapter and the real `@deepseek-ai/dsh-tool-cordis` plugin, so a
 * live model can mount plugins into the very context the test observes. Lives
 * outside the *.e2e.ts pattern so importing it never re-registers another
 * file's tests.
 */

const PERSONA = 'You are cordis-agent, a self-referential harness demo. '
  + 'Your cordis_* tools operate on the live cordis runtime you run inside: '
  + 'cordis_inspect to look around, cordis_mount to add a plugin, cordis_unmount '
  + 'to clean one up. Follow the tool descriptions exactly and report results briefly.'

export async function cordisHarness(): Promise<Context> {
  const ctx = new Context()
  await ctx.plugin(LlmService)
  await ctx.plugin(SessionStore)
  await ctx.plugin(SystemPrompt, { persona: PERSONA })
  await ctx.plugin(ToolRegistry)
  await ctx.plugin(AgentRegistry)
  await ctx.plugin(AgentLoop, { agents: [] })
  await ctx.plugin(LlmDeepSeek, { models: ['deepseek-v4-flash'] })
  await ctx.plugin(ToolCordis)
  return ctx
}

export function waitForIdle(ctx: Context, agent: Agent): Promise<void> {
  return new Promise((resolve) => {
    const dispose = ctx.on('agent/status', (subject, status) => {
      if (subject === agent && status === 'idle') {
        dispose()
        resolve()
      }
    })
  })
}