import { defineAction } from "@agent-native/core";
import { getRequestUserEmail } from "@agent-native/core/server";
import { eq, and } from "drizzle-orm";
import { z } from "zod";

import { db, schema } from "../server/db/index.js";

export default defineAction({
  description: "Update an existing exercise",
  schema: z.object({
    id: z.coerce.number().describe("Exercise ID"),
    name: z.string().optional().describe("Exercise name"),
    calories_burned: z.coerce.number().optional().describe("Calories burned"),
    duration_minutes: z.coerce
      .number()
      .optional()
      .describe("Duration in minutes"),
    date: z.string().optional().describe("Date in YYYY-MM-DD format"),
  }),
  run: async (args) => {
    const ownerEmail = getRequestUserEmail();
    if (!ownerEmail) throw new Error("no authenticated user");

    const result = await db()
      .update(schema.exercises)
      .set({
        name: args.name,
        calories_burned: args.calories_burned ?? undefined,
        duration_minutes: args.duration_minutes ?? undefined,
        date: args.date ? String(args.date).split("T")[0] : undefined,
      })
      .where(
        and(
          eq(schema.exercises.id, args.id),
          eq(schema.exercises.owner_email, ownerEmail),
        ),
      )
      .returning();

    return result[0];
  },
});