import { useMutation } from "@tanstack/react-query";
import {
  generateValidationSql,
  runValidationRule,
  suggestValidationRules,
} from "@/lib/tauri";

export function useGenerateValidationSql() {
  return useMutation({
    mutationFn: ({
      connectionId,
      ruleDescription,
    }: {
      connectionId: string;
      ruleDescription: string;
    }) => generateValidationSql(connectionId, ruleDescription),
  });
}

export function useRunValidationRule() {
  return useMutation({
    mutationFn: ({
      connectionId,
      sql,
      sampleLimit,
    }: {
      connectionId: string;
      sql: string;
      sampleLimit?: number;
    }) => runValidationRule(connectionId, sql, sampleLimit),
  });
}

export function useSuggestValidationRules() {
  return useMutation({
    mutationFn: (connectionId: string) => suggestValidationRules(connectionId),
  });
}