-- Additive AI operational records. No existing rows are modified or deleted.
CREATE TYPE "AiRunStatus" AS ENUM ('STARTED', 'COMPLETED', 'FAILED', 'CANCELLED');

CREATE TABLE "ai_runs" (
    "id" TEXT NOT NULL,
    "requestId" TEXT NOT NULL,
    "userId" TEXT NOT NULL,
    "model" TEXT NOT NULL,
    "requestClass" TEXT NOT NULL,
    "analysisMode" TEXT NOT NULL,
    "status" "AiRunStatus" NOT NULL DEFAULT 'STARTED',
    "fallbackUsed" BOOLEAN NOT NULL DEFAULT false,
    "toolNames" TEXT[] DEFAULT ARRAY[]::TEXT[],
    "toolCount" INTEGER NOT NULL DEFAULT 0,
    "inputTokens" INTEGER NOT NULL DEFAULT 0,
    "outputTokens" INTEGER NOT NULL DEFAULT 0,
    "totalTokens" INTEGER NOT NULL DEFAULT 0,
    "latencyMs" INTEGER,
    "errorCategory" TEXT,
    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
    "completedAt" TIMESTAMP(3),
    CONSTRAINT "ai_runs_pkey" PRIMARY KEY ("id")
);

CREATE UNIQUE INDEX "ai_runs_requestId_key" ON "ai_runs"("requestId");
CREATE INDEX "ai_runs_userId_idx" ON "ai_runs"("userId");
CREATE INDEX "ai_runs_createdAt_idx" ON "ai_runs"("createdAt");
CREATE INDEX "ai_runs_status_idx" ON "ai_runs"("status");
CREATE INDEX "ai_runs_model_idx" ON "ai_runs"("model");

CREATE TABLE "ai_premium_budgets" (
    "id" TEXT NOT NULL,
    "userId" TEXT NOT NULL,
    "day" TIMESTAMP(3) NOT NULL,
    "plan" "PlanCode" NOT NULL,
    "premiumRequests" INTEGER NOT NULL DEFAULT 0,
    "updatedAt" TIMESTAMP(3) NOT NULL,
    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
    CONSTRAINT "ai_premium_budgets_pkey" PRIMARY KEY ("id")
);

CREATE UNIQUE INDEX "ai_premium_budgets_userId_day_key" ON "ai_premium_budgets"("userId", "day");
CREATE INDEX "ai_premium_budgets_userId_idx" ON "ai_premium_budgets"("userId");
CREATE INDEX "ai_premium_budgets_day_idx" ON "ai_premium_budgets"("day");

CREATE TABLE "ai_rate_limit_buckets" (
    "id" TEXT NOT NULL,
    "userId" TEXT NOT NULL,
    "windowStart" TIMESTAMP(3) NOT NULL,
    "count" INTEGER NOT NULL DEFAULT 0,
    "updatedAt" TIMESTAMP(3) NOT NULL,
    CONSTRAINT "ai_rate_limit_buckets_pkey" PRIMARY KEY ("id")
);

CREATE UNIQUE INDEX "ai_rate_limit_buckets_userId_windowStart_key" ON "ai_rate_limit_buckets"("userId", "windowStart");
CREATE INDEX "ai_rate_limit_buckets_userId_idx" ON "ai_rate_limit_buckets"("userId");
CREATE INDEX "ai_rate_limit_buckets_windowStart_idx" ON "ai_rate_limit_buckets"("windowStart");

ALTER TABLE "ai_runs" ADD CONSTRAINT "ai_runs_userId_fkey" FOREIGN KEY ("userId") REFERENCES "users"("id") ON DELETE CASCADE ON UPDATE CASCADE;
ALTER TABLE "ai_premium_budgets" ADD CONSTRAINT "ai_premium_budgets_userId_fkey" FOREIGN KEY ("userId") REFERENCES "users"("id") ON DELETE CASCADE ON UPDATE CASCADE;
ALTER TABLE "ai_rate_limit_buckets" ADD CONSTRAINT "ai_rate_limit_buckets_userId_fkey" FOREIGN KEY ("userId") REFERENCES "users"("id") ON DELETE CASCADE ON UPDATE CASCADE;