firefish/packages/backend/src/services/messages/create.ts

import type { CacheableUser, User } from "@/models/entities/user.js";
import type { UserGroup } from "@/models/entities/user-group.js";
import type { DriveFile } from "@/models/entities/drive-file.js";
import {
	MessagingMessages,
	UserGroupJoinings,
	Mutings,
	Users,
} from "@/models/index.js";
import { genId, publishToChatStream, toPuny, ChatEvent } from "backend-rs";
import type { MessagingMessage } from "@/models/entities/messaging-message.js";
import {
	publishMessagingIndexStream,
	publishMainStream,
	publishGroupMessagingStream,
} from "@/services/stream.js";
import { pushNotification } from "@/services/push-notification.js";
import { Not } from "typeorm";
import type { Note } from "@/models/entities/note.js";
import renderNote from "@/remote/activitypub/renderer/note.js";
import renderCreate from "@/remote/activitypub/renderer/create.js";
import { renderActivity } from "@/remote/activitypub/renderer/index.js";
import { deliver } from "@/queue/index.js";
import { Instances } from "@/models/index.js";

export async function createMessage(
	user: { id: User["id"]; host: User["host"] },
	recipientUser: CacheableUser | undefined,
	recipientGroup: UserGroup | undefined,
	text: string | null | undefined,
	file: DriveFile | null,
	uri?: string,
) {
	const message = {
		id: genId(),
		createdAt: new Date(),
		fileId: file ? file.id : null,
		recipientId: recipientUser ? recipientUser.id : null,
		groupId: recipientGroup ? recipientGroup.id : null,
		text: text ? text.trim() : null,
		userId: user.id,
		isRead: false,
		reads: [] as any[],
		uri,
	} as MessagingMessage;

	await MessagingMessages.insert(message);

	const messageObj = await MessagingMessages.pack(message);

	if (recipientUser) {
		if (Users.isLocalUser(user)) {
			// 自分のストリーム
			publishToChatStream(
				message.userId,
				recipientUser.id,
				ChatEvent.Message,
				messageObj,
			);
			publishMessagingIndexStream(message.userId, "message", messageObj);
			publishMainStream(message.userId, "messagingMessage", messageObj);
		}

		if (Users.isLocalUser(recipientUser)) {
			// 相手のストリーム
			publishToChatStream(
				recipientUser.id,
				message.userId,
				ChatEvent.Message,
				messageObj,
			);
			publishMessagingIndexStream(recipientUser.id, "message", messageObj);
			publishMainStream(recipientUser.id, "messagingMessage", messageObj);
		}
	} else if (recipientGroup) {
		// グループのストリーム
		publishGroupMessagingStream(recipientGroup.id, "message", messageObj);

		// メンバーのストリーム
		const joinings = await UserGroupJoinings.findBy({
			userGroupId: recipientGroup.id,
		});
		for (const joining of joinings) {
			publishMessagingIndexStream(joining.userId, "message", messageObj);
			publishMainStream(joining.userId, "messagingMessage", messageObj);
		}
	}

	// 2秒経っても(今回作成した)メッセージが既読にならなかったら「未読のメッセージがありますよ」イベントを発行する
	setTimeout(async () => {
		const freshMessage = await MessagingMessages.findOneBy({ id: message.id });
		if (freshMessage == null) return; // メッセージが削除されている場合もある

		if (recipientUser && Users.isLocalUser(recipientUser)) {
			if (freshMessage.isRead) return; // 既読

			//#region ただしミュートされているなら発行しない
			const mute = await Mutings.findBy({
				muterId: recipientUser.id,
			});
			if (mute.map((m) => m.muteeId).includes(user.id)) return;
			//#endregion

			publishMainStream(recipientUser.id, "unreadMessagingMessage", messageObj);
			pushNotification(recipientUser.id, "unreadMessagingMessage", messageObj);
		} else if (recipientGroup) {
			const joinings = await UserGroupJoinings.findBy({
				userGroupId: recipientGroup.id,
				userId: Not(user.id),
			});
			for (const joining of joinings) {
				if (freshMessage.reads.includes(joining.userId)) return; // 既読
				publishMainStream(joining.userId, "unreadMessagingMessage", messageObj);
				pushNotification(joining.userId, "unreadMessagingMessage", messageObj);
			}
		}
	}, 2000);

	if (
		recipientUser &&
		Users.isLocalUser(user) &&
		Users.isRemoteUser(recipientUser)
	) {
		const instance = await Instances.findOneBy({
			host: toPuny(recipientUser.host),
		});
		const note = {
			id: message.id,
			createdAt: message.createdAt,
			fileIds: message.fileId ? [message.fileId] : [],
			text: message.text,
			userId: message.userId,
			visibility: "specified",
			mentions: [recipientUser].map((u) => u.id),
			mentionedRemoteUsers: JSON.stringify(
				[recipientUser].map((u) => ({
					uri: u.uri,
					username: u.username,
					host: u.host,
				})),
			),
		} as Note;

		let renderedNote: Record<string, unknown> = await renderNote(
			note,
			false,
			true,
		);

		// TODO: For pleroma and its fork instances, the actor will have a boolean "capabilities": { acceptsChatMessages: boolean } property. May use that instead of checking instance.softwareName. https://kazv.moe/objects/ca5c0b88-88ce-48a7-bf88-54d45f6ce781
		// ChatMessage document from Pleroma: https://docs.pleroma.social/backend/development/ap_extensions/#chatmessages
		// Note: LitePub has been stalled since 2019-06-29 and is incomplete as a specification
		if (
			instance?.softwareName &&
			["akkoma", "pleroma", "lemmy"].includes(
				instance.softwareName.toLowerCase(),
			)
		) {
			const tmp_note = renderedNote;
			renderedNote = {
				type: "ChatMessage",
				attributedTo: tmp_note.attributedTo,
				content: tmp_note.content,
				id: tmp_note.id,
				published: tmp_note.published,
				to: tmp_note.to,
				tag: tmp_note.tag,
				cc: [],
			};
			// A recently fixed bug, empty arrays will be rejected by pleroma
			if (
				Array.isArray(tmp_note.attachment) &&
				tmp_note.attachment.length !== 0
			) {
				renderedNote.attachment = tmp_note.attachment;
			}
		}

		const activity = renderActivity(renderCreate(renderedNote, note));

		deliver(user, activity, recipientUser.inbox);
	}
	return messageObj;
}