import { types } from "mobx-state-tree";
import { withSetPropAction } from "../_helpers/withSetPropAction";

export enum MessageType {
  Text = "Text",
  Image = "Image",
  ImageWithText = "ImageWithText",
  Error = "Error",
}

export enum MessageSenderType {
  Ai = "Ai",
  User = "User",
}

export enum MessageStatus {
  Ready = "ready",
  Pending = "pending",
}

export const ChatMessage = types
  .model("ChatMessage", {
    id: types.string,
    conversationId: types.string,
    messageType: types.enumeration(Object.values(MessageType)),
    messageSenderType: types.enumeration(Object.values(MessageSenderType)),
    senderUid: types.string,
    senderName: types.string,
    senderAvatarUrl: types.maybeNull(types.string),
    text: types.maybe(types.string),
    imageUrls: types.maybe(types.array(types.string)),
    createdAt: types.number,
    status: types.enumeration(Object.values(MessageStatus)),
  })
  .actions(withSetPropAction);