import { BgentRuntime } from "./runtime";
import { type Actor, type Content, type Memory, type UUID } from "./types";

/**
 * Get details for a list of actors.
 */
export async function getActorDetails({
  runtime,
  room_id,
}: {
  runtime: BgentRuntime;
  room_id: UUID;
}) {
  const actors = await runtime.databaseAdapter.getActorDetails({ room_id });
  return actors as Actor[];
}

/**
 * Format actors into a string
 * @param actors - list of actors
 * @returns string
 */
export function formatActors({ actors }: { actors: Actor[] }) {
  const actorStrings = actors.map((actor: Actor) => {
    const header = `${actor.name}${actor.details?.tagline ? ": " + actor.details?.tagline : ""}${actor.details?.summary ? "\n" + actor.details?.summary : ""}`;
    return header;
  });
  const finalActorStrings = actorStrings.join("\n");
  return finalActorStrings;
}

/**
 * Format messages into a string
 * @param messages - list of messages
 * @param actors - list of actors
 * @returns string
 */
export const formatMessages = ({
  messages,
  actors,
}: {
  messages: Memory[];
  actors: Actor[];
}) => {
  const messageStrings = messages
    .reverse()
    .filter((message: Memory) => message.user_id)
    .map((message: Memory) => {
      let messageContent = (message.content as Content).content;
      const messageAction = (message.content as Content).action;
      const sender = actors.find(
        (actor: Actor) => actor.id === message.user_id,
      )!;
      if (messageAction === "IGNORE") {
        messageContent = "*Ignored*";
      }
      return `${sender?.name}: ${messageContent} ${messageAction && messageAction !== "null" ? `(${messageAction})` : ""}`;
    })
    .join("\n");
  return messageStrings;
};