import type { IndemnClient } from './client.js';
import type { TestSet, CreateTestSetInput } from './types.js';

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

  async listTestSets(agentId?: string): Promise<TestSet[]> {
    const query = agentId ? { agent_id: agentId } : undefined;
    return this.client.evalsGet<TestSet[]>('/api/v1/test-sets', query);
  }

  async getTestSet(id: string): Promise<TestSet> {
    return this.client.evalsGet<TestSet>(`/api/v1/test-sets/${id}`);
  }

  async createTestSet(input: CreateTestSetInput): Promise<TestSet> {
    return this.client.evalsPost<TestSet>('/api/v1/test-sets', input);
  }

  async updateTestSet(id: string, input: Partial<CreateTestSetInput>): Promise<TestSet> {
    return this.client.evalsPut<TestSet>(`/api/v1/test-sets/${id}`, input);
  }

  async deleteTestSet(id: string): Promise<void> {
    await this.client.evalsDelete(`/api/v1/test-sets/${id}`);
  }

  async listVersions(id: string): Promise<TestSet[]> {
    return this.client.evalsGet<TestSet[]>(`/api/v1/test-sets/${id}/versions`);
  }

  async getVersion(id: string, version: number): Promise<TestSet> {
    return this.client.evalsGet<TestSet>(`/api/v1/test-sets/${id}/versions/${version}`);
  }
}