77 lines
1.8 KiB
TypeScript
77 lines
1.8 KiB
TypeScript
import { remark } from "remark";
|
|
import html from "remark-html";
|
|
|
|
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 interface ChatMessage {
|
|
id: string;
|
|
conversationId: string;
|
|
messageType: MessageType;
|
|
messageSenderType: MessageSenderType;
|
|
senderUid: string;
|
|
senderName: string;
|
|
senderAvatarUrl: string;
|
|
htmlText?: string | undefined;
|
|
text: string | undefined;
|
|
imageUrls?: string[] | undefined;
|
|
createdAt: number;
|
|
status: MessageStatus;
|
|
}
|
|
|
|
export interface RawMessage {
|
|
id?: number;
|
|
conversation_id: number;
|
|
user?: string;
|
|
message?: string;
|
|
created_at?: string;
|
|
updated_at?: string;
|
|
}
|
|
|
|
export const toChatMessage = async (m: RawMessage): Promise<ChatMessage> => {
|
|
const createdAt = new Date(m.created_at ?? "").getTime();
|
|
const imageUrls: string[] = [];
|
|
const imageUrl = undefined;
|
|
if (imageUrl) {
|
|
imageUrls.push(imageUrl);
|
|
}
|
|
|
|
const messageType = MessageType.Text;
|
|
const messageSenderType =
|
|
m.user === "user" ? MessageSenderType.User : MessageSenderType.Ai;
|
|
|
|
const content = m.message ?? "";
|
|
const processedContent = await remark().use(html).process(content);
|
|
const contentHtml = processedContent.toString();
|
|
|
|
return {
|
|
id: (m.id ?? 0).toString(),
|
|
conversationId: (m.conversation_id ?? 0).toString(),
|
|
messageType: messageType,
|
|
messageSenderType: messageSenderType,
|
|
senderUid: m.user?.toString() || "0",
|
|
senderName: m.user === "user" ? "You" : "Assistant",
|
|
senderAvatarUrl:
|
|
m.user === "user" ? "icons/avatar.svg" : "icons/app_icon.svg",
|
|
text: content,
|
|
htmlText: contentHtml,
|
|
imageUrls: imageUrls,
|
|
createdAt: createdAt,
|
|
status: MessageStatus.Ready,
|
|
};
|
|
};
|