apps/api/src/ai/tools/stop-timer.ts

import type { AppContext } from "@api/ai/agents/config/shared";
import { db } from "@midday/db/client";
import { stopTimer } from "@midday/db/queries";
import { getAppUrl } from "@midday/utils/envs";
import { formatDate } from "@midday/utils/format";
import { tool } from "ai";
import { formatDistance } from "date-fns";
import { z } from "zod";

const stopTimerSchema = z.object({
  entryId: z.string().nullable().optional().describe("Timer entry ID"),
  assignedId: z.string().nullable().optional().describe("User ID"),
});

export const stopTimerTool = tool({
  description:
    "Stop the current running timer - calculates and saves the duration.",
  inputSchema: stopTimerSchema,
  execute: async function* ({ entryId, assignedId }, executionOptions) {
    const appContext = executionOptions.experimental_context as AppContext;
    const teamId = appContext.teamId as string;
    const userId = assignedId || appContext.userId || null;

    if (!teamId) {
      yield {
        text: "Unable to stop timer: Team ID not found in context.",
      };

      return;
    }

    try {
      // Stop the timer
      const result = await stopTimer(db, {
        teamId,
        entryId: entryId || undefined,
        assignedId: userId,
      });

      // Handle discarded entries (under 60 seconds)
      if (result.discarded) {
        yield {
          text: `Timer discarded - entry was under 1 minute and was not saved.\n\n**Project:** ${result.project?.name || "Unknown"}`,
          link: {
            text: "View tracker",
            url: `${getAppUrl()}/tracker`,
          },
        };
        return;
      }

      const duration = result.duration ? Number(result.duration) : 0;
      const start = new Date(0);
      const end = new Date(duration * 1000);
      const formattedDuration = formatDistance(start, end, {
        includeSeconds: false,
      });

      const startTime = result.start
        ? formatDate(result.start, "HH:mm")
        : "N/A";
      const stopTime = result.stop ? formatDate(result.stop, "HH:mm") : "N/A";

      const response = `Timer stopped successfully!\n\n**Project:** ${result.project?.name || "Unknown"}\n**Duration:** ${formattedDuration}\n**Started:** ${startTime}\n**Stopped:** ${stopTime}\n**Description:** ${result.description || "None"}`;

      yield {
        text: response,
        link: {
          text: "View tracker",
          url: `${getAppUrl()}/tracker`,
        },
      };
    } catch (error) {
      yield {
        text: `Failed to stop timer: ${error instanceof Error ? error.message : "Unknown error"}`,
      };
    }
  },
});