import BunSQLite from "@moduletrace/bun-sqlite";
import { Harnesses, isInstallWideHarness, type HarnessKind } from "@/src/dict/harness-dict";
import AppError from "@/src/utils/app-error";
import assertHarnessModel from "./assert-harness-model";
import assertHarnessProvider from "./assert-harness-provider";

type Params = {
    /**
     * A workspace the caller has checked belongs to the session user.
     */
    workspace_id: number;
    harness: HarnessKind;
    /**
     * Drop the workspace's setting: each bot uses its own or the default
     * again.
     */
    inherit: boolean;
    model_id: number | null;
    /**
     * Limits the harness to one provider; null: any.
     */
    provider_id?: number | null;
    fallback: boolean;
    enabled: boolean;
    options?: Record<string, unknown>;
};

/**
 * Saves (or, with `inherit`, removes) a workspace's setting for one
 * harness, which wins over its bots' own. Same rules as the default: the
 * model must be available and fit the job, and text can't be turned off.
 */
export default async function saveWorkspaceHarnessSetting({
    workspace_id,
    harness,
    inherit,
    model_id,
    provider_id = null,
    fallback,
    enabled,
    options,
}: Params) {
    if (inherit) {
        await BunSQLite.sql({
            sql: `DELETE FROM workspace_harness_settings WHERE workspace_id = ? AND harness = ?`,
            values: [workspace_id, harness],
        });
        return;
    }

    if (isInstallWideHarness({ harness })) {
        const title = Harnesses.find((h) => h.value == harness)?.title || harness;

        throw new AppError(`${title} is set for the whole install, in Settings, under Models.`);
    }

    if (harness == "text" && !enabled) {
        throw new AppError(`The text model can't be turned off.`);
    }

    if (provider_id) {
        await assertHarnessProvider({ provider_id });
    }

    if (model_id) {
        await assertHarnessModel({ harness, model_id, provider_id });
    }

    const now = Date.now();

    const res = await BunSQLite.sql({
        sql: `INSERT INTO workspace_harness_settings (created_at, updated_at, workspace_id, harness, model_id, provider_id, fallback, enabled, options)
            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
            ON CONFLICT(workspace_id, harness) DO UPDATE SET updated_at = excluded.updated_at,
            model_id = excluded.model_id, provider_id = excluded.provider_id, fallback = excluded.fallback,
            enabled = excluded.enabled, options = excluded.options`,
        values: [
            now,
            now,
            workspace_id,
            harness,
            model_id,
            provider_id,
            fallback ? 1 : 0,
            enabled ? 1 : 0,
            JSON.stringify(options || {}),
        ],
    });

    if (!res.success) {
        throw new Error(`Couldn't save the workspace's model setting: ${res.error}`);
    }
}