import { Grid } from "@chakra-ui/react";
import { forwardRef, useColorMode } from "@chakra-ui/react";
import { useMemo } from "react";
import { Message } from "src/types/Conversation";

import { FlaggableElement } from "./FlaggableElement";

export interface ValidLabel {
  name: string;
  display_text: string;
  help_text: string;
}

export const Messages = ({
  messages,
  post_id,
  valid_labels,
}: {
  messages: Message[];
  post_id: string;
  valid_labels: ValidLabel[];
}) => {
  const items = messages.map((messageProps: Message, i: number) => {
    const { message_id, text } = messageProps;
    return (
      <FlaggableElement
        text={text}
        post_id={post_id}
        message_id={message_id}
        key={i + text}
        flaggable_labels={valid_labels}
      >
        <MessageView {...messageProps} />
      </FlaggableElement>
    );
  });
  // Maybe also show a legend of the colors?
  return <Grid gap={2}>{items}</Grid>;
};

export const MessageView = forwardRef<Message, "div">(({ is_assistant, text }: Message, ref) => {
  const { colorMode } = useColorMode();

  const bgColor = useMemo(() => {
    if (colorMode === "light") {
      return is_assistant ? "bg-slate-800" : "bg-sky-900";
    } else {
      return is_assistant ? "bg-black" : "bg-sky-900";
    }
  }, [colorMode, is_assistant]);

  return (
    <div ref={ref} className={`${bgColor} p-4 rounded-md text-white whitespace-pre-wrap`}>
      {text}
    </div>
  );
});

MessageView.displayName = "MessageView";