import type { IndemnClient } from './client.js';
import type { Agent, CreateAgentInput, UpdateAgentInput } from './types.js';

export class AgentsSDK {
  constructor(private client: IndemnClient) {}

  async listAgents(): Promise<Agent[]> {
    return this.client.copilotGet<Agent[]>(
      '/organization/:org_id/ai-studio/v2/bots',
    );
  }

  async getAgent(id: string): Promise<Agent> {
    const result = await this.client.copilotGet<Agent | Agent[]>(
      `/organization/:org_id/ai-studio/v2/bots/${id}`,
    );
    // v2 endpoint returns array even for single bot
    return Array.isArray(result) ? result[0] : result;
  }

  async createAgent(input: CreateAgentInput): Promise<Agent> {
    return this.client.copilotPost<Agent>(
      '/organization/:org_id/ai-studio/bots',
      { name: input.name, channels: input.channels || ['WEB'] },
    );
  }

  async updateAgent(id: string, input: UpdateAgentInput): Promise<Agent> {
    return this.client.copilotPut<Agent>(
      `/organization/:org_id/ai-studio/bots/${id}`,
      input,
    );
  }

  async deleteAgent(id: string): Promise<void> {
    await this.client.copilotDelete(
      `/organization/:org_id/ai-studio/bots/${id}`,
    );
  }

  async cloneAgent(id: string): Promise<Agent> {
    return this.client.copilotPost<Agent>(
      '/organization/:org_id/ai-studio/v2/bots/clone',
      { source_id_bot: id },
    );
  }
}