misskey/src/services/note/create.ts

378 lines
9.2 KiB
TypeScript
Raw Normal View History

2018-04-18 07:53:17 +02:00
import { MongoError } from 'mongodb';
2018-04-07 19:30:37 +02:00
import Note, { pack, INote } from '../../models/note';
2018-04-05 21:04:50 +02:00
import User, { isLocalUser, IUser, isRemoteUser } from '../../models/user';
import stream, { publishLocalTimelineStream, publishGlobalTimelineStream } from '../../publishers/stream';
2018-04-04 17:40:34 +02:00
import Following from '../../models/following';
2018-04-05 16:24:51 +02:00
import { deliver } from '../../queue';
2018-04-04 17:40:34 +02:00
import renderNote from '../../remote/activitypub/renderer/note';
import renderCreate from '../../remote/activitypub/renderer/create';
2018-04-07 23:55:26 +02:00
import renderAnnounce from '../../remote/activitypub/renderer/announce';
2018-04-13 07:39:08 +02:00
import packAp from '../../remote/activitypub/renderer';
2018-04-04 17:40:34 +02:00
import { IDriveFile } from '../../models/drive-file';
import notify from '../../publishers/notify';
2018-04-07 19:30:37 +02:00
import NoteWatching from '../../models/note-watching';
2018-04-04 17:40:34 +02:00
import watch from './watch';
import Mute from '../../models/mute';
import pushSw from '../../publishers/push-sw';
import event from '../../publishers/stream';
import parse from '../../text/parse';
2018-04-04 17:50:57 +02:00
import html from '../../text/html';
import { IApp } from '../../models/app';
2018-04-02 10:11:14 +02:00
2018-04-05 21:04:50 +02:00
export default async (user: IUser, data: {
2018-04-05 11:43:06 +02:00
createdAt?: Date;
text?: string;
2018-04-07 19:30:37 +02:00
reply?: INote;
renote?: INote;
2018-04-05 11:43:06 +02:00
media?: IDriveFile[];
geo?: any;
2018-04-04 20:21:11 +02:00
poll?: any;
2018-04-05 11:43:06 +02:00
viaMobile?: boolean;
2018-04-04 20:21:11 +02:00
tags?: string[];
cw?: string;
visibility?: string;
2018-04-04 17:50:57 +02:00
uri?: string;
app?: IApp;
2018-04-07 19:30:37 +02:00
}, silent = false) => new Promise<INote>(async (res, rej) => {
2018-04-05 21:04:50 +02:00
if (data.createdAt == null) data.createdAt = new Date();
if (data.visibility == null) data.visibility = 'public';
2018-04-07 23:55:26 +02:00
if (data.viaMobile == null) data.viaMobile = false;
2018-04-04 20:21:11 +02:00
2018-04-05 21:04:50 +02:00
const tags = data.tags || [];
2018-04-02 10:11:14 +02:00
2018-04-04 17:40:34 +02:00
let tokens = null;
2018-04-02 10:11:14 +02:00
2018-04-05 21:04:50 +02:00
if (data.text) {
2018-04-04 17:40:34 +02:00
// Analyze
2018-04-05 21:04:50 +02:00
tokens = parse(data.text);
2018-04-02 10:11:14 +02:00
2018-04-04 17:40:34 +02:00
// Extract hashtags
const hashtags = tokens
.filter(t => t.type == 'hashtag')
.map(t => t.hashtag);
hashtags.forEach(tag => {
if (tags.indexOf(tag) == -1) {
tags.push(tag);
}
});
2018-04-02 10:11:14 +02:00
}
2018-04-05 21:04:50 +02:00
const insert: any = {
createdAt: data.createdAt,
mediaIds: data.media ? data.media.map(file => file._id) : [],
replyId: data.reply ? data.reply._id : null,
2018-04-07 19:30:37 +02:00
renoteId: data.renote ? data.renote._id : null,
2018-04-05 21:04:50 +02:00
text: data.text,
2018-04-04 17:50:57 +02:00
textHtml: tokens === null ? null : html(tokens),
2018-04-05 21:04:50 +02:00
poll: data.poll,
cw: data.cw,
2018-04-04 17:40:34 +02:00
tags,
userId: user._id,
2018-04-05 21:04:50 +02:00
viaMobile: data.viaMobile,
geo: data.geo || null,
appId: data.app ? data.app._id : null,
visibility: data.visibility,
2018-04-02 10:11:14 +02:00
2018-04-04 17:40:34 +02:00
// 以下非正規化データ
2018-04-05 21:04:50 +02:00
_reply: data.reply ? { userId: data.reply.userId } : null,
2018-04-07 19:30:37 +02:00
_renote: data.renote ? { userId: data.renote.userId } : null,
2018-04-05 21:04:50 +02:00
_user: {
host: user.host,
2018-04-07 23:55:26 +02:00
inbox: isRemoteUser(user) ? user.inbox : undefined
2018-04-05 21:04:50 +02:00
}
2018-04-05 08:50:52 +02:00
};
2018-04-05 21:04:50 +02:00
if (data.uri != null) insert.uri = data.uri;
2018-04-05 08:50:52 +02:00
// 投稿を作成
2018-04-18 07:53:17 +02:00
let note: INote;
try {
note = await Note.insert(insert);
} catch (e) {
// duplicate key error
if (e instanceof MongoError && e.code === 11000) {
return res(null);
}
console.error(e);
return rej('something happened');
}
2018-04-02 10:11:14 +02:00
2018-04-07 19:30:37 +02:00
res(note);
2018-04-04 16:12:35 +02:00
2018-04-17 15:31:52 +02:00
// Increment notes count
2018-04-04 16:12:35 +02:00
User.update({ _id: user._id }, {
$inc: {
2018-04-07 19:30:37 +02:00
notesCount: 1
2018-04-04 16:12:35 +02:00
}
});
2018-04-04 17:40:34 +02:00
// Serialize
2018-04-07 19:30:37 +02:00
const noteObj = await pack(note);
2018-04-04 16:12:35 +02:00
// タイムラインへの投稿
2018-04-07 19:30:37 +02:00
if (note.channelId == null) {
2018-04-04 17:40:34 +02:00
if (isLocalUser(user)) {
// Publish event to myself's stream
2018-04-07 19:30:37 +02:00
stream(note.userId, 'note', noteObj);
// Publish note to local timeline stream
publishLocalTimelineStream(noteObj);
2018-04-04 17:40:34 +02:00
}
2018-04-04 16:12:35 +02:00
// Publish note to global timeline stream
publishGlobalTimelineStream(noteObj);
2018-04-04 16:12:35 +02:00
// Fetch all followers
const followers = await Following.aggregate([{
$lookup: {
from: 'users',
localField: 'followerId',
foreignField: '_id',
2018-04-05 20:42:55 +02:00
as: 'user'
2018-04-04 16:12:35 +02:00
}
}, {
$match: {
2018-04-07 19:30:37 +02:00
followeeId: note.userId
2018-04-04 16:12:35 +02:00
}
}], {
_id: false
});
if (!silent) {
2018-04-07 23:55:26 +02:00
const render = async () => {
const content = data.renote && data.text == null
? renderAnnounce(data.renote.uri ? data.renote.uri : await renderNote(data.renote))
: renderCreate(await renderNote(note));
2018-04-13 07:39:08 +02:00
return packAp(content);
2018-04-07 23:55:26 +02:00
};
2018-04-05 16:19:47 +02:00
2018-04-05 21:04:50 +02:00
// 投稿がリプライかつ投稿者がローカルユーザーかつリプライ先の投稿の投稿者がリモートユーザーなら配送
if (data.reply && isLocalUser(user) && isRemoteUser(data.reply._user)) {
2018-04-15 10:53:25 +02:00
deliver(user, await render(), data.reply._user.inbox);
2018-04-07 23:55:26 +02:00
}
// 投稿がRenoteかつ投稿者がローカルユーザーかつRenote元の投稿の投稿者がリモートユーザーなら配送
if (data.renote && isLocalUser(user) && isRemoteUser(data.renote._user)) {
2018-04-15 10:53:25 +02:00
deliver(user, await render(), data.renote._user.inbox);
2018-04-05 21:04:50 +02:00
}
2018-04-07 23:55:26 +02:00
Promise.all(followers.map(async follower => {
2018-04-05 20:42:55 +02:00
follower = follower.user[0];
2018-04-05 16:19:47 +02:00
if (isLocalUser(follower)) {
// Publish event to followers stream
2018-04-07 19:30:37 +02:00
stream(follower._id, 'note', noteObj);
2018-04-05 16:19:47 +02:00
} else {
// フォロワーがリモートユーザーかつ投稿者がローカルユーザーなら投稿を配信
if (isLocalUser(user)) {
2018-04-15 10:53:25 +02:00
deliver(user, await render(), follower.inbox);
2018-04-05 16:19:47 +02:00
}
2018-04-04 16:12:35 +02:00
}
2018-04-05 16:19:47 +02:00
}));
}
2018-04-04 16:12:35 +02:00
}
// チャンネルへの投稿
/* TODO
2018-04-07 19:30:37 +02:00
if (note.channelId) {
2018-04-04 16:12:35 +02:00
promises.push(
2018-04-07 19:30:37 +02:00
// Increment channel index(notes count)
Channel.update({ _id: note.channelId }, {
2018-04-04 16:12:35 +02:00
$inc: {
index: 1
}
}),
// Publish event to channel
2018-04-07 19:30:37 +02:00
promisedNoteObj.then(noteObj => {
publishChannelStream(note.channelId, 'note', noteObj);
2018-04-04 16:12:35 +02:00
}),
Promise.all([
2018-04-07 19:30:37 +02:00
promisedNoteObj,
2018-04-04 16:12:35 +02:00
// Get channel watchers
ChannelWatching.find({
2018-04-07 19:30:37 +02:00
channelId: note.channelId,
2018-04-04 16:12:35 +02:00
// 削除されたドキュメントは除く
deletedAt: { $exists: false }
})
2018-04-07 19:30:37 +02:00
]).then(([noteObj, watches]) => {
2018-04-04 16:12:35 +02:00
// チャンネルの視聴者(のタイムライン)に配信
watches.forEach(w => {
2018-04-07 19:30:37 +02:00
stream(w.userId, 'note', noteObj);
2018-04-04 16:12:35 +02:00
});
})
);
}*/
2018-04-04 17:40:34 +02:00
const mentions = [];
async function addMention(mentionee, reason) {
// Reject if already added
if (mentions.some(x => x.equals(mentionee))) return;
// Add mention
mentions.push(mentionee);
// Publish event
if (!user._id.equals(mentionee)) {
const mentioneeMutes = await Mute.find({
muter_id: mentionee,
deleted_at: { $exists: false }
});
const mentioneesMutedUserIds = mentioneeMutes.map(m => m.muteeId.toString());
if (mentioneesMutedUserIds.indexOf(user._id.toString()) == -1) {
2018-04-07 19:30:37 +02:00
event(mentionee, reason, noteObj);
pushSw(mentionee, reason, noteObj);
2018-04-04 17:40:34 +02:00
}
}
}
2018-04-07 19:30:37 +02:00
// If has in reply to note
2018-04-05 21:04:50 +02:00
if (data.reply) {
2018-04-04 17:40:34 +02:00
// Increment replies count
2018-04-07 19:30:37 +02:00
Note.update({ _id: data.reply._id }, {
2018-04-04 17:40:34 +02:00
$inc: {
repliesCount: 1
}
});
// (自分自身へのリプライでない限りは)通知を作成
2018-04-05 21:04:50 +02:00
notify(data.reply.userId, user._id, 'reply', {
2018-04-07 19:30:37 +02:00
noteId: note._id
2018-04-04 17:40:34 +02:00
});
// Fetch watchers
2018-04-07 19:30:37 +02:00
NoteWatching.find({
noteId: data.reply._id,
2018-04-04 17:40:34 +02:00
userId: { $ne: user._id },
// 削除されたドキュメントは除く
deletedAt: { $exists: false }
}, {
fields: {
userId: true
}
}).then(watchers => {
watchers.forEach(watcher => {
notify(watcher.userId, user._id, 'reply', {
2018-04-07 19:30:37 +02:00
noteId: note._id
2018-04-04 17:40:34 +02:00
});
});
});
// この投稿をWatchする
2018-04-07 20:58:11 +02:00
if (isLocalUser(user) && user.settings.autoWatch !== false) {
2018-04-05 21:04:50 +02:00
watch(user._id, data.reply);
2018-04-04 17:40:34 +02:00
}
// Add mention
2018-04-05 21:04:50 +02:00
addMention(data.reply.userId, 'reply');
2018-04-04 17:40:34 +02:00
}
2018-04-07 19:30:37 +02:00
// If it is renote
if (data.renote) {
2018-04-04 17:40:34 +02:00
// Notify
2018-04-07 19:30:37 +02:00
const type = data.text ? 'quote' : 'renote';
notify(data.renote.userId, user._id, type, {
2018-04-07 23:55:26 +02:00
noteId: note._id
2018-04-04 17:40:34 +02:00
});
// Fetch watchers
2018-04-07 19:30:37 +02:00
NoteWatching.find({
noteId: data.renote._id,
2018-04-07 23:55:26 +02:00
userId: { $ne: user._id }
2018-04-04 17:40:34 +02:00
}, {
fields: {
userId: true
}
}).then(watchers => {
watchers.forEach(watcher => {
notify(watcher.userId, user._id, type, {
2018-04-07 19:30:37 +02:00
noteId: note._id
2018-04-04 17:40:34 +02:00
});
});
});
// この投稿をWatchする
2018-04-07 20:58:11 +02:00
if (isLocalUser(user) && user.settings.autoWatch !== false) {
2018-04-07 19:30:37 +02:00
watch(user._id, data.renote);
2018-04-04 17:40:34 +02:00
}
2018-04-07 19:30:37 +02:00
// If it is quote renote
2018-04-05 21:04:50 +02:00
if (data.text) {
2018-04-04 17:40:34 +02:00
// Add mention
2018-04-07 19:30:37 +02:00
addMention(data.renote.userId, 'quote');
2018-04-04 17:40:34 +02:00
} else {
// Publish event
2018-04-07 19:30:37 +02:00
if (!user._id.equals(data.renote.userId)) {
event(data.renote.userId, 'renote', noteObj);
2018-04-04 17:40:34 +02:00
}
}
2018-04-07 19:30:37 +02:00
// 今までで同じ投稿をRenoteしているか
const existRenote = await Note.findOne({
2018-04-04 17:40:34 +02:00
userId: user._id,
2018-04-07 19:30:37 +02:00
renoteId: data.renote._id,
2018-04-04 17:40:34 +02:00
_id: {
2018-04-07 19:30:37 +02:00
$ne: note._id
2018-04-04 17:40:34 +02:00
}
});
2018-04-04 16:12:35 +02:00
2018-04-07 19:30:37 +02:00
if (!existRenote) {
// Update renoteee status
Note.update({ _id: data.renote._id }, {
2018-04-04 17:40:34 +02:00
$inc: {
2018-04-07 19:30:37 +02:00
renoteCount: 1
2018-04-04 17:40:34 +02:00
}
});
}
}
// If has text content
2018-04-05 21:04:50 +02:00
if (data.text) {
2018-04-04 17:40:34 +02:00
// Extract an '@' mentions
const atMentions = tokens
.filter(t => t.type == 'mention')
.map(m => m.username)
// Drop dupulicates
.filter((v, i, s) => s.indexOf(v) == i);
// Resolve all mentions
await Promise.all(atMentions.map(async mention => {
// Fetch mentioned user
// SELECT _id
const mentionee = await User
.findOne({
usernameLower: mention.toLowerCase()
}, { _id: true });
// When mentioned user not found
if (mentionee == null) return;
2018-04-07 19:30:37 +02:00
// 既に言及されたユーザーに対する返信や引用renoteの場合も無視
2018-04-05 21:04:50 +02:00
if (data.reply && data.reply.userId.equals(mentionee._id)) return;
2018-04-07 19:30:37 +02:00
if (data.renote && data.renote.userId.equals(mentionee._id)) return;
2018-04-04 17:40:34 +02:00
// Add mention
addMention(mentionee._id, 'mention');
// Create notification
notify(mentionee._id, user._id, 'mention', {
2018-04-08 11:52:42 +02:00
noteId: note._id
2018-04-04 17:40:34 +02:00
});
}));
}
// Append mentions data
if (mentions.length > 0) {
2018-04-07 19:30:37 +02:00
Note.update({ _id: note._id }, {
2018-04-04 17:40:34 +02:00
$set: {
mentions
}
});
}
});