import type { IndemnClient } from './client.js';
import type { BotConfiguration, SetConfigInput } from './types.js';

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

  async getConfig(agentId: string): Promise<BotConfiguration> {
    return this.client.copilotGet<BotConfiguration>(
      `/organization/:org_id/ai-studio/bots/${agentId}/configurations`,
    );
  }

  async setConfig(agentId: string, data: SetConfigInput): Promise<BotConfiguration> {
    return this.client.copilotPut<BotConfiguration>(
      `/organization/:org_id/ai-studio/bots/${agentId}/configurations`,
      data,
    );
  }

  async getPrompt(agentId: string): Promise<string | undefined> {
    const config = await this.getConfig(agentId);
    return config.ai_config?.system_prompt;
  }

  async setPrompt(agentId: string, prompt: string): Promise<BotConfiguration> {
    return this.setConfig(agentId, { system_prompt: prompt });
  }
}