import type {
  SpawnEngine,
  SpawnHandle,
  SpawnRequest,
} from "../engine/interface.js";

export class AgentManager {
  private readonly running = new Map<string, SpawnHandle>();

  constructor(private readonly engine: SpawnEngine) {}

  spawn(spec: SpawnRequest): SpawnHandle {
    const handle = this.engine.spawn(spec);
    this.running.set(handle.id, handle);
    void handle
      .wait()
      .catch(() => undefined)
      .finally(() => {
        this.running.delete(handle.id);
      });
    return handle;
  }

  async abort(id: string): Promise<void> {
    await this.running.get(id)?.abort();
  }

  async abortAll(): Promise<void> {
    await Promise.all(
      [...this.running.values()].map((handle) => handle.abort()),
    );
  }

  getRunningIds(): string[] {
    return [...this.running.keys()];
  }
}