misskey/src/services/update-hashtag.ts

import { IUser, isLocalUser, isRemoteUser } from '../models/user';
import Hashtag from '../models/hashtag';
import hashtagChart from './chart/hashtag';

export async function updateHashtag(user: IUser, tag: string, isUserAttached = false, inc = true) {
	tag = tag.toLowerCase();

	const index = await Hashtag.findOne({ tag });

	if (index == null && !inc) return;

	if (index != null) {
		const $push = {} as any;
		const $pull = {} as any;
		const $inc = {} as any;

		if (isUserAttached) {
			if (inc) {
				// 自分が初めてこのタグを使ったなら
				if (!index.attachedUserIds.some(id => id.equals(user._id))) {
					$push.attachedUserIds = user._id;
					$inc.attachedUsersCount = 1;
				}
				// 自分が(ローカル内で)初めてこのタグを使ったなら
				if (isLocalUser(user) && !index.attachedLocalUserIds.some(id => id.equals(user._id))) {
					$push.attachedLocalUserIds = user._id;
					$inc.attachedLocalUsersCount = 1;
				}
				// 自分が(リモートで)初めてこのタグを使ったなら
				if (isRemoteUser(user) && !index.attachedRemoteUserIds.some(id => id.equals(user._id))) {
					$push.attachedRemoteUserIds = user._id;
					$inc.attachedRemoteUsersCount = 1;
				}
			} else {
				$pull.attachedUserIds = user._id;
				$inc.attachedUsersCount = -1;
				if (isLocalUser(user)) {
					$pull.attachedLocalUserIds = user._id;
					$inc.attachedLocalUsersCount = -1;
				} else {
					$pull.attachedRemoteUserIds = user._id;
					$inc.attachedRemoteUsersCount = -1;
				}
			}
		} else {
			// 自分が初めてこのタグを使ったなら
			if (!index.mentionedUserIds.some(id => id.equals(user._id))) {
				$push.mentionedUserIds = user._id;
				$inc.mentionedUsersCount = 1;
			}
			// 自分が(ローカル内で)初めてこのタグを使ったなら
			if (isLocalUser(user) && !index.mentionedLocalUserIds.some(id => id.equals(user._id))) {
				$push.mentionedLocalUserIds = user._id;
				$inc.mentionedLocalUsersCount = 1;
			}
			// 自分が(リモートで)初めてこのタグを使ったなら
			if (isRemoteUser(user) && !index.mentionedRemoteUserIds.some(id => id.equals(user._id))) {
				$push.mentionedRemoteUserIds = user._id;
				$inc.mentionedRemoteUsersCount = 1;
			}
		}

		const q = {} as any;
		if (Object.keys($push).length > 0) q.$push = $push;
		if (Object.keys($pull).length > 0) q.$pull = $pull;
		if (Object.keys($inc).length > 0) q.$inc = $inc;
		if (Object.keys(q).length > 0) Hashtag.update({ tag }, q);
	} else {
		if (isUserAttached) {
			Hashtag.insert({
				tag,
				mentionedUserIds: [],
				mentionedUsersCount: 0,
				mentionedLocalUserIds: [],
				mentionedLocalUsersCount: 0,
				mentionedRemoteUserIds: [],
				mentionedRemoteUsersCount: 0,
				attachedUserIds: [user._id],
				attachedUsersCount: 1,
				attachedLocalUserIds: isLocalUser(user) ? [user._id] : [],
				attachedLocalUsersCount: isLocalUser(user) ? 1 : 0,
				attachedRemoteUserIds: isRemoteUser(user) ? [user._id] : [],
				attachedRemoteUsersCount: isRemoteUser(user) ? 1 : 0,
			});
		} else {
			Hashtag.insert({
				tag,
				mentionedUserIds: [user._id],
				mentionedUsersCount: 1,
				mentionedLocalUserIds: isLocalUser(user) ? [user._id] : [],
				mentionedLocalUsersCount: isLocalUser(user) ? 1 : 0,
				mentionedRemoteUserIds: isRemoteUser(user) ? [user._id] : [],
				mentionedRemoteUsersCount: isRemoteUser(user) ? 1 : 0,
				attachedUserIds: [],
				attachedUsersCount: 0,
				attachedLocalUserIds: [],
				attachedLocalUsersCount: 0,
				attachedRemoteUserIds: [],
				attachedRemoteUsersCount: 0,
			});
		}
	}

	if (!isUserAttached) {
		hashtagChart.update(tag, user);
	}
}