import { names, uniqueNamesGenerator } from "unique-names-generator";
import fact from "./evaluators/fact";
import goal from "./evaluators/goal";
import { ActionExample, type Evaluator } from "./types";

export const defaultEvaluators: Evaluator[] = [fact, goal];

/**
 * Template used for the evaluation completion.
 */
export const evaluationTemplate = `TASK: Based on the conversation and conditions, determine which evaluation functions are appropriate to call.
Examples:
{{evaluatorExamples}}

INSTRUCTIONS: You are helping me to decide which appropriate functions to call based on the conversation between {{senderName}} and {{agentName}}.

{{recentMessages}}

Evaluator Functions:
{{evaluators}}

Evaluator Conditions:
{{evaluatorConditions}}

TASK: Based on the most recent conversation, determine which evaluators functions are appropriate to call to call.
Include the name of evaluators that are relevant and should be called in the array
Available evaluator names to include are {{evaluatorNames}}
Respond with a JSON array containing a field for description in a JSON block formatted for markdown with this structure:
\`\`\`json
[
  'evaluatorName',
  'evaluatorName'
]
\`\`\`

Your response must include the JSON block.`;

/**
 * Formats the names of evaluators into a comma-separated list, each enclosed in single quotes.
 * @param evaluators - An array of evaluator objects.
 * @returns A string that concatenates the names of all evaluators, each enclosed in single quotes and separated by commas.
 */
export function formatEvaluatorNames(evaluators: Evaluator[]) {
  return evaluators
    .map((evaluator: Evaluator) => `'${evaluator.name}'`)
    .join(",\n");
}

/**
 * Formats evaluator details into a string, including both the name and description of each evaluator.
 * @param evaluators - An array of evaluator objects.
 * @returns A string that concatenates the name and description of each evaluator, separated by a colon and a newline character.
 */
export function formatEvaluators(evaluators: Evaluator[]) {
  return evaluators
    .map(
      (evaluator: Evaluator) => `'${evaluator.name}: ${evaluator.description}'`,
    )
    .join(",\n");
}

/**
 * Formats the conditions under which each evaluator is relevant into a readable string.
 * @param evaluators - An array of evaluator objects.
 * @returns A string that concatenates the name and condition of each evaluator, separated by a colon and a newline character.
 */
export function formatEvaluatorConditions(evaluators: Evaluator[]) {
  return evaluators
    .map(
      (evaluator: Evaluator) => `'${evaluator.name}: ${evaluator.condition}'`,
    )
    .join(",\n");
}

/**
 * Formats evaluator examples into a readable string, replacing placeholders with generated names.
 * @param evaluators - An array of evaluator objects, each containing examples to format.
 * @returns A string that presents each evaluator example in a structured format, including context, messages, and outcomes, with placeholders replaced by generated names.
 */
export function formatEvaluatorExamples(evaluators: Evaluator[]) {
  return evaluators
    .map((evaluator) => {
      return evaluator.examples
        .map((example) => {
          const exampleNames = Array.from({ length: 5 }, () =>
            uniqueNamesGenerator({ dictionaries: [names] }),
          );

          let formattedContext = example.context;
          let formattedOutcome = example.outcome;

          exampleNames.forEach((name, index) => {
            const placeholder = `{{user${index + 1}}}`;
            formattedContext = formattedContext.replaceAll(placeholder, name);
            formattedOutcome = formattedOutcome.replaceAll(placeholder, name);
          });

          const formattedMessages = example.messages
            .map((message: ActionExample) => {
              let messageString = `${message.user}: ${message.content.content}`;
              exampleNames.forEach((name, index) => {
                const placeholder = `{{user${index + 1}}}`;
                messageString = messageString.replaceAll(placeholder, name);
              });
              return (
                messageString +
                (message.content.action ? ` (${message.content.action})` : "")
              );
            })
            .join("\n");

          return `Context:\n${formattedContext}\n\nMessages:\n${formattedMessages}\n\nOutcome:\n${formattedOutcome}`;
        })
        .join("\n\n");
    })
    .join("\n\n");
}

/**
 * Generates a string describing the conditions under which each evaluator example is relevant.
 * @param evaluators - An array of evaluator objects, each containing examples.
 * @returns A string that describes the conditions for each evaluator example, formatted with the evaluator name, example number, and condition.
 */
export function formatEvaluatorExampleConditions(evaluators: Evaluator[]) {
  return evaluators
    .map((evaluator) =>
      evaluator.examples
        .map(
          (_example, index) =>
            `${evaluator.name} Example ${index + 1}: ${evaluator.condition}`,
        )
        .join("\n"),
    )
    .join("\n\n");
}

/**
 * Generates a string summarizing the descriptions of each evaluator example.
 * @param evaluators - An array of evaluator objects, each containing examples.
 * @returns A string that summarizes the descriptions for each evaluator example, formatted with the evaluator name, example number, and description.
 */
export function formatEvaluatorExampleDescriptions(evaluators: Evaluator[]) {
  return evaluators
    .map((evaluator) =>
      evaluator.examples
        .map(
          (_example, index) =>
            `${evaluator.name} Example ${index + 1}: ${evaluator.description}`,
        )
        .join("\n"),
    )
    .join("\n\n");
}