import { eq } from 'drizzle-orm';
import type { DrizzleClient } from '../client.js';
import { reviewRemovalQueue } from '../schema.js';

export async function setReviewRemovalDecision(
  db: DrizzleClient,
  removalId: string,
  status: 'APPROVED' | 'REJECTED',
  decidedAt: Date,
): Promise<void> {
  await db
    .update(reviewRemovalQueue)
    .set({ status, decidedAt })
    .where(eq(reviewRemovalQueue.id, removalId));
}

export async function setReviewRemovalAiScore(
  db: DrizzleClient,
  removalId: string,
  aiScore: string,
): Promise<void> {
  await db
    .update(reviewRemovalQueue)
    .set({ aiScore })
    .where(eq(reviewRemovalQueue.id, removalId));
}