/**
 * IPC route definitions for auto-approve threshold reads/writes.
 *
 * Exposes gateway-owned threshold data to the assistant daemon over
 * the IPC socket.
 */

import { eq, sql } from "drizzle-orm";
import { z } from "zod";

import { getGatewayDb } from "../db/connection.js";
import {
  autoApproveThresholds,
  conversationThresholdOverrides,
} from "../db/schema.js";
import type { IpcRoute } from "./server.js";

const GLOBAL_DEFAULTS = {
  interactive: "medium",
  autonomous: "low",
  headless: "none",
};

const GetConversationThresholdSchema = z.object({
  conversationId: z.string().min(1),
});

const SetConversationThresholdSchema = z.object({
  conversationId: z.string().min(1),
  threshold: z.enum(["none", "low", "medium", "high"]),
});

export const thresholdRoutes: IpcRoute[] = [
  {
    method: "get_global_thresholds",
    handler: () => {
      const db = getGatewayDb();
      const row = db
        .select()
        .from(autoApproveThresholds)
        .where(eq(autoApproveThresholds.id, 1))
        .get();

      if (!row) return GLOBAL_DEFAULTS;

      return {
        interactive: row.interactive,
        autonomous: row.autonomous,
        headless: row.headless,
      };
    },
  },
  {
    method: "get_conversation_threshold",
    schema: GetConversationThresholdSchema,
    handler: (params?: Record<string, unknown>) => {
      const conversationId = params?.conversationId as string;
      const db = getGatewayDb();
      const row = db
        .select()
        .from(conversationThresholdOverrides)
        .where(
          eq(conversationThresholdOverrides.conversationId, conversationId),
        )
        .get();

      if (!row) return null;
      return { threshold: row.threshold };
    },
  },
  {
    method: "set_conversation_threshold",
    schema: SetConversationThresholdSchema,
    handler: (params?: Record<string, unknown>) => {
      const parsed = SetConversationThresholdSchema.parse(params ?? {});
      const db = getGatewayDb();
      db.insert(conversationThresholdOverrides)
        .values({
          conversationId: parsed.conversationId,
          threshold: parsed.threshold,
        })
        .onConflictDoUpdate({
          target: conversationThresholdOverrides.conversationId,
          set: {
            threshold: parsed.threshold,
            updatedAt: sql`datetime('now')`,
          },
        })
        .run();
      return {
        conversationId: parsed.conversationId,
        threshold: parsed.threshold,
      };
    },
  },
];