import { ChatCompletionRequestMessage as OpenAIMessage } from "openai";
import { Collection, Message as DiscordMessage } from "discord.js";

import config from "./config.json";

export function formatMessage(message: DiscordMessage): string {
  let rvalue: string = message.cleanContent;

  for (const attachment of message.attachments) {
    rvalue += ` [Attachment: ${attachment[1].name}`;
    rvalue += attachment[1].description ? ` - ${attachment[1].description}` : "";
    rvalue += "]";
  }

  for (const embed of message.embeds) {
    rvalue += ` [Embed:`;
    rvalue += embed.title                 ? ` ${embed.title}` : "";
    rvalue += embed.author                ? ` by ${embed.author.name}` : "";
    rvalue += (embed.title || embed.author) && embed.description ? " -": "";
    rvalue += embed.description           ? ` ${embed.description}` : "";
    rvalue += "]";
  }

  return rvalue;
}

export default function toOpenAIMessages(messages: Collection<string, DiscordMessage>): OpenAIMessage[] {
  const rvalue: OpenAIMessage[] = [];

  rvalue.push({ role: "system", content: config.systemPrompt});

  messages
  .sort((a, b) => a.createdTimestamp - b.createdTimestamp)
  .each(message => {
    rvalue.push({
      role: message.author.id == message.client.user.id ? "assistant" : "user",
      content: formatMessage(message),
      name: message.member?.displayName,
    });
  });

  return rvalue;
}