import { defineAction } from "@agent-native/core";
import { getRequestUserEmail } from "@agent-native/core/server";
import { getUserSetting, putUserSetting } from "@agent-native/core/settings";
import { z } from "zod";

import { normalizeSignature } from "../shared/signature.js";
import type { UserSettings } from "../shared/types.js";

const settingsSchema = z.object({
  name: z.string().optional().describe("Display name for local fallback mail"),
  signature: z
    .string()
    .optional()
    .describe("Markdown/plain-text signature to add to new drafts"),
  writingStyle: z
    .string()
    .optional()
    .describe("Short notes describing how generated drafts should sound"),
});

export default defineAction({
  description:
    "Update the user's mail drafting settings, including signature and writing style.",
  schema: settingsSchema,
  run: async (args) => {
    const ownerEmail = getRequestUserEmail();
    if (!ownerEmail) throw new Error("no authenticated user");
    const current =
      ((await getUserSetting(ownerEmail, "mail-settings")) as
        | Partial<UserSettings>
        | undefined) ?? {};

    const updates: Partial<UserSettings> = {};
    if (args.name !== undefined) updates.name = args.name.trim();
    if (args.signature !== undefined) {
      updates.signature = normalizeSignature(args.signature);
    }
    if (args.writingStyle !== undefined) {
      updates.writingStyle = args.writingStyle.trim();
    }

    const next = {
      ...current,
      email: current.email || ownerEmail,
      ...updates,
    };
    await putUserSetting(ownerEmail, "mail-settings", next);
    return {
      name: next.name ?? "",
      email: next.email || ownerEmail,
      signature: next.signature ?? "",
      writingStyle: next.writingStyle ?? "",
    };
  },
});