良い感じに

This commit is contained in:
syuilo 2018-05-07 04:08:39 +09:00
parent a549327170
commit 6785f50a1f
2 changed files with 157 additions and 194 deletions

View file

@ -1,10 +1,10 @@
import { toUnicode, toASCII } from 'punycode'; import { toUnicode, toASCII } from 'punycode';
import User from '../models/user'; import User, { IUser } from '../models/user';
import webFinger from './webfinger'; import webFinger from './webfinger';
import config from '../config'; import config from '../config';
import { createPerson } from './activitypub/models/person'; import { createPerson } from './activitypub/models/person';
export default async (username, _host, option?) => { export default async (username, _host, option?): Promise<IUser> => {
const usernameLower = username.toLowerCase(); const usernameLower = username.toLowerCase();
const hostAscii = toASCII(_host).toLowerCase(); const hostAscii = toASCII(_host).toLowerCase();
const host = toUnicode(hostAscii); const host = toUnicode(hostAscii);

View file

@ -1,5 +1,5 @@
import Note, { pack, INote } from '../../models/note'; import Note, { pack, INote } from '../../models/note';
import User, { isLocalUser, IUser, isRemoteUser } from '../../models/user'; import User, { isLocalUser, IUser, isRemoteUser, IRemoteUser, ILocalUser } from '../../models/user';
import stream, { publishLocalTimelineStream, publishGlobalTimelineStream, publishUserListStream } from '../../publishers/stream'; import stream, { publishLocalTimelineStream, publishGlobalTimelineStream, publishUserListStream } from '../../publishers/stream';
import Following from '../../models/following'; import Following from '../../models/following';
import { deliver } from '../../queue'; import { deliver } from '../../queue';
@ -19,6 +19,61 @@ import { IApp } from '../../models/app';
import UserList from '../../models/user-list'; import UserList from '../../models/user-list';
import resolveUser from '../../remote/resolve-user'; import resolveUser from '../../remote/resolve-user';
type Reason = 'reply' | 'quote' | 'mention';
/**
* ServiceWorkerへの通知を担当
*/
class NotificationManager {
private user: IUser;
private note: any;
private list: Array<{
user: ILocalUser['_id'],
reason: Reason;
}>;
constructor(user, note) {
this.user = user;
this.note = note;
}
public push(user: ILocalUser['_id'], reason: Reason) {
// 自分自身へは通知しない
if (this.user._id.equals(user)) return;
const exist = this.list.find(x => x.user.equals(user));
if (exist) {
// 「メンションされているかつ返信されている」場合は、メンションとしての通知ではなく返信としての通知にする
if (reason != 'mention') {
exist.reason = reason;
}
} else {
this.list.push({
user, reason
});
}
}
public deliver() {
this.list.forEach(async x => {
const mentionee = x.user;
// ミュート情報を取得
const mentioneeMutes = await Mute.find({
muterId: mentionee
});
const mentioneesMutedUserIds = mentioneeMutes.map(m => m.muteeId.toString());
// 通知される側のユーザーが通知する側のユーザーをミュートしていない限りは通知する
if (!mentioneesMutedUserIds.includes(this.user._id.toString())) {
pushSw(mentionee, x.reason, this.note);
}
});
}
}
export default async (user: IUser, data: { export default async (user: IUser, data: {
createdAt?: Date; createdAt?: Date;
text?: string; text?: string;
@ -41,7 +96,7 @@ export default async (user: IUser, data: {
const tags = data.tags || []; const tags = data.tags || [];
let tokens = null; let tokens: any[] = null;
if (data.text) { if (data.text) {
// Analyze // Analyze
@ -120,6 +175,8 @@ export default async (user: IUser, data: {
// Serialize // Serialize
const noteObj = await pack(note); const noteObj = await pack(note);
const nm = new NotificationManager(user, noteObj);
const render = async () => { const render = async () => {
const content = data.renote && data.text == null const content = data.renote && data.text == null
? renderAnnounce(data.renote.uri ? data.renote.uri : await renderNote(data.renote)) ? renderAnnounce(data.renote.uri ? data.renote.uri : await renderNote(data.renote))
@ -127,123 +184,86 @@ export default async (user: IUser, data: {
return packAp(content); return packAp(content);
}; };
// タイムラインへの投稿 if (!silent) {
if (note.channelId == null) { if (isLocalUser(user)) {
if (!silent) { if (note.visibility == 'private' || note.visibility == 'followers' || note.visibility == 'specified') {
if (isLocalUser(user)) { // Publish event to myself's stream
if (note.visibility == 'private' || note.visibility == 'followers' || note.visibility == 'specified') { stream(note.userId, 'note', await pack(note, user, {
// Publish event to myself's stream detail: true
stream(note.userId, 'note', await pack(note, user, { }));
detail: true } else {
})); // Publish event to myself's stream
} else { stream(note.userId, 'note', noteObj);
// Publish event to myself's stream
stream(note.userId, 'note', noteObj);
// Publish note to local timeline stream // Publish note to local timeline stream
if (note.visibility != 'home') { if (note.visibility != 'home') {
publishLocalTimelineStream(noteObj); publishLocalTimelineStream(noteObj);
}
} }
} }
}
// Publish note to global timeline stream // Publish note to global timeline stream
publishGlobalTimelineStream(noteObj); publishGlobalTimelineStream(noteObj);
if (note.visibility == 'specified') { if (note.visibility == 'specified') {
data.visibleUsers.forEach(async u => { data.visibleUsers.forEach(async u => {
stream(u._id, 'note', await pack(note, u, { stream(u._id, 'note', await pack(note, u, {
detail: true detail: true
})); }));
}); });
} }
if (note.visibility == 'public' || note.visibility == 'home' || note.visibility == 'followers') { if (note.visibility == 'public' || note.visibility == 'home' || note.visibility == 'followers') {
// フォロワーに配信 // フォロワーに配信
Following.find({ Following.find({
followeeId: note.userId followeeId: note.userId
}).then(followers => { }).then(followers => {
followers.map(async following => { followers.map(async following => {
const follower = following._follower; const follower = following._follower;
if (isLocalUser(follower)) { if (isLocalUser(follower)) {
// ストーキングしていない場合 // ストーキングしていない場合
if (!following.stalk) { if (!following.stalk) {
// この投稿が返信ならスキップ // この投稿が返信ならスキップ
if (note.replyId && !note._reply.userId.equals(following.followerId) && !note._reply.userId.equals(note.userId)) return; if (note.replyId && !note._reply.userId.equals(following.followerId) && !note._reply.userId.equals(note.userId)) return;
}
// Publish event to followers stream
stream(following.followerId, 'note', noteObj);
} else {
//#region AP配送
// フォロワーがリモートユーザーかつ投稿者がローカルユーザーなら投稿を配信
if (isLocalUser(user)) {
deliver(user, await render(), follower.inbox);
}
//#endergion
} }
});
});
}
// リストに配信 // Publish event to followers stream
UserList.find({ stream(following.followerId, 'note', noteObj);
userIds: note.userId } else {
}).then(lists => { //#region AP配送
lists.forEach(list => { // フォロワーがリモートユーザーかつ投稿者がローカルユーザーなら投稿を配信
publishUserListStream(list._id, 'note', noteObj); if (isLocalUser(user)) {
deliver(user, await render(), follower.inbox);
}
//#endergion
}
}); });
}); });
} }
//#region リプライとAnnounceのAP配送 // リストに配信
UserList.find({
// 投稿がリプライかつ投稿者がローカルユーザーかつリプライ先の投稿の投稿者がリモートユーザーなら配送 userIds: note.userId
if (data.reply && isLocalUser(user) && isRemoteUser(data.reply._user)) { }).then(lists => {
deliver(user, await render(), data.reply._user.inbox); lists.forEach(list => {
} publishUserListStream(list._id, 'note', noteObj);
});
// 投稿がRenoteかつ投稿者がローカルユーザーかつRenote元の投稿の投稿者がリモートユーザーなら配送 });
if (data.renote && isLocalUser(user) && isRemoteUser(data.renote._user)) {
deliver(user, await render(), data.renote._user.inbox);
}
//#endergion
} }
// チャンネルへの投稿 //#region リプライとAnnounceのAP配送
/* TODO
if (note.channelId) {
promises.push(
// Increment channel index(notes count)
Channel.update({ _id: note.channelId }, {
$inc: {
index: 1
}
}),
// Publish event to channel // 投稿がリプライかつ投稿者がローカルユーザーかつリプライ先の投稿の投稿者がリモートユーザーなら配送
promisedNoteObj.then(noteObj => { if (data.reply && isLocalUser(user) && isRemoteUser(data.reply._user)) {
publishChannelStream(note.channelId, 'note', noteObj); deliver(user, await render(), data.reply._user.inbox);
}), }
Promise.all([ // 投稿がRenoteかつ投稿者がローカルユーザーかつRenote元の投稿の投稿者がリモートユーザーなら配送
promisedNoteObj, if (data.renote && isLocalUser(user) && isRemoteUser(data.renote._user)) {
deliver(user, await render(), data.renote._user.inbox);
// Get channel watchers }
ChannelWatching.find({ //#endergion
channelId: note.channelId,
// 削除されたドキュメントは除く
deletedAt: { $exists: false }
})
]).then(([noteObj, watches]) => {
// チャンネルの視聴者(のタイムライン)に配信
watches.forEach(w => {
stream(w.userId, 'note', noteObj);
});
})
);
}*/
//#region メンション //#region メンション
if (data.text) { if (data.text) {
@ -251,56 +271,44 @@ export default async (user: IUser, data: {
const mentions = tokens const mentions = tokens
.filter(t => t.type == 'mention'); .filter(t => t.type == 'mention');
mentions.forEach(async m => { let mentionedUsers = await Promise.all(mentions.map(async m => {
const u = await resolveUser(m.username, m.host); try {
return await resolveUser(m.username, m.host);
if (isLocalUser(u)) { } catch (e) {
// Fetch mentioned user return null;
const mentionee = await User
.findOne({
usernameLower: m.username.toLowerCase()
}, { _id: true });
// When mentioned user not found
if (mentionee == null) return;
// 既に言及されたユーザーに対する返信や引用renoteの場合も無視
if (data.reply && data.reply.userId.equals(mentionee._id)) return;
if (data.renote && data.renote.userId.equals(mentionee._id)) return;
// Create notification
notify(mentionee._id, user._id, 'mention', {
noteId: note._id
});
} else {
deliver(user, await render(), u.inbox);
} }
}); }));
}
//#endregion
const mentions = []; // TODO: Drop dupulicates
mentionedUsers = mentionedUsers.filter(x => x != null);
async function addMention(mentionee, reason) { mentionedUsers.filter(u => isLocalUser(u)).forEach(async u => {
// Reject if already added // 既に言及されたユーザーに対する返信や引用renoteの場合も無視
if (mentions.some(x => x.equals(mentionee))) return; if (data.reply && data.reply.userId.equals(u._id)) return;
if (data.renote && data.renote.userId.equals(u._id)) return;
// Add mention // Create notification
mentions.push(mentionee); notify(u._id, user._id, 'mention', {
noteId: note._id
// Publish event });
if (!user._id.equals(mentionee)) {
const mentioneeMutes = await Mute.find({ nm.push(u._id, 'mention');
muter_id: mentionee, });
deleted_at: { $exists: false }
mentionedUsers.filter(u => isRemoteUser(u)).forEach(async u => {
deliver(user, await render(), (u as IRemoteUser).inbox);
});
// Append mentions data
if (mentions.length > 0) {
Note.update({ _id: note._id }, {
$set: {
mentions
}
}); });
const mentioneesMutedUserIds = mentioneeMutes.map(m => m.muteeId.toString());
if (mentioneesMutedUserIds.indexOf(user._id.toString()) == -1) {
event(mentionee, reason, noteObj);
pushSw(mentionee, reason, noteObj);
}
} }
} }
//#endregion
// If has in reply to note // If has in reply to note
if (data.reply) { if (data.reply) {
@ -339,8 +347,7 @@ export default async (user: IUser, data: {
watch(user._id, data.reply); watch(user._id, data.reply);
} }
// Add mention nm.push(data.reply.userId, 'reply');
addMention(data.reply.userId, 'reply');
} }
// If it is renote // If it is renote
@ -375,7 +382,7 @@ export default async (user: IUser, data: {
// If it is quote renote // If it is quote renote
if (data.text) { if (data.text) {
// Add mention // Add mention
addMention(data.renote.userId, 'quote'); nm.push(data.renote.userId, 'quote');
} else { } else {
// Publish event // Publish event
if (!user._id.equals(data.renote.userId)) { if (!user._id.equals(data.renote.userId)) {
@ -401,48 +408,4 @@ export default async (user: IUser, data: {
}); });
} }
} }
// If has text content
if (data.text) {
// 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;
// 既に言及されたユーザーに対する返信や引用renoteの場合も無視
if (data.reply && data.reply.userId.equals(mentionee._id)) return;
if (data.renote && data.renote.userId.equals(mentionee._id)) return;
// Add mention
addMention(mentionee._id, 'mention');
// Create notification
notify(mentionee._id, user._id, 'mention', {
noteId: note._id
});
}));
}
// Append mentions data
if (mentions.length > 0) {
Note.update({ _id: note._id }, {
$set: {
mentions
}
});
}
}); });