import { defineAction } from "@agent-native/core";
import { getRequestUserEmail } from "@agent-native/core/server/request-context";
import { assertAccess } from "@agent-native/core/sharing";
import { eq } from "drizzle-orm";
import { z } from "zod";

import { getDb, schema } from "../server/db/index.js";
import {
  assertDerivedAudienceAccess,
  nowIso,
  parseJson,
  serializeProposal,
  stableJson,
  writeKnowledgeRecord,
} from "../server/lib/brain.js";
import type { WriteKnowledgeInput } from "../server/lib/brain.js";
import { booleanishSchema } from "./_schemas.js";

export default defineAction({
  description:
    "Approve a pending Brain proposal and publish its knowledge payload.",
  schema: z.object({
    proposalId: z.string().min(1),
    reviewerNotes: z.string().optional(),
    publishCanonical: booleanishSchema
      .optional()
      .describe(
        "Override whether approval mirrors the resulting knowledge into context/company-brain workspace resources.",
      ),
  }),
  run: async ({ proposalId, reviewerNotes, publishCanonical }) => {
    const access = await assertAccess("brain-proposal", proposalId, "editor");
    const proposal = access.resource;
    await assertDerivedAudienceAccess(proposal);
    if (proposal.status !== "pending") {
      throw new Error(`Proposal ${proposalId} is already ${proposal.status}`);
    }
    const payload = parseJson<WriteKnowledgeInput>(proposal.payloadJson, {
      title: proposal.title,
      body: proposal.body,
      evidence: [],
      proposalMode: "never",
    });
    const nextPayload: WriteKnowledgeInput = {
      ...payload,
      publishCanonical: publishCanonical ?? payload.publishCanonical,
      proposalMode: "never",
    };
    const result = await writeKnowledgeRecord(nextPayload, {
      bypassProposal: true,
    });
    await getDb()
      .update(schema.brainProposals)
      .set({
        status: "approved",
        reviewerNotes: reviewerNotes ?? null,
        payloadJson: stableJson(nextPayload),
        reviewedBy: getRequestUserEmail() ?? null,
        reviewedAt: nowIso(),
        updatedAt: nowIso(),
      })
      .where(eq(schema.brainProposals.id, proposalId));
    const [updated] = await getDb()
      .select()
      .from(schema.brainProposals)
      .where(eq(schema.brainProposals.id, proposalId))
      .limit(1);
    return { proposal: serializeProposal(updated), result };
  },
});