import type { IndemnClient } from './client.js';
import type { Rubric, CreateRubricInput } from './types.js';

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

  async listRubrics(agentId?: string): Promise<Rubric[]> {
    const query = agentId ? { agent_id: agentId } : undefined;
    return this.client.evalsGet<Rubric[]>('/api/v1/rubrics', query);
  }

  async getRubric(id: string): Promise<Rubric> {
    return this.client.evalsGet<Rubric>(`/api/v1/rubrics/${id}`);
  }

  async createRubric(input: CreateRubricInput): Promise<Rubric> {
    return this.client.evalsPost<Rubric>('/api/v1/rubrics', input);
  }

  async updateRubric(id: string, input: Partial<CreateRubricInput>): Promise<Rubric> {
    return this.client.evalsPut<Rubric>(`/api/v1/rubrics/${id}`, input);
  }

  async deleteRubric(id: string): Promise<void> {
    await this.client.evalsDelete(`/api/v1/rubrics/${id}`);
  }

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

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