import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
import { Text } from "@earendil-works/pi-tui";

interface TimestampData {
  timestamp: number;
}

export default function (pi: ExtensionAPI) {
  pi.registerEntryRenderer<TimestampData>(
    "assistant-timestamp",
    (entry, _options, theme) => {
      const timestamp = entry.data?.timestamp;
      if (typeof timestamp !== "number") return;

      return new Text(
        theme.fg("dim", new Date(timestamp).toLocaleString()),
        1,
        0,
      );
    },
  );

  pi.on("turn_end", (event) => {
    if (event.message.role !== "assistant") return;

    pi.appendEntry<TimestampData>("assistant-timestamp", {
      timestamp: event.message.timestamp,
    });
  });
}