良い感じに
This commit is contained in:
syuilo
parent
a549327170
commit
6785f50a1f
|
|
@ -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);
|
||||||
|
|
|
||||||
|
|
@ -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
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
});
|
});
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue