import { apiFetch } from "@/lib/auth/api-client";
import { CreateTreatmentInput, Treatment } from "./types";

export function fetchColonyTreatments(colonyId: string): Promise<Treatment[]> {
  return apiFetch<Treatment[]>(`/colonies/${colonyId}/treatments`);
}

export function createTreatment(input: CreateTreatmentInput): Promise<Treatment> {
  return apiFetch<Treatment>("/treatments", {
    method: "POST",
    body: JSON.stringify(input),
  });
}