forked from FoundKeyGang/FoundKey
wip
This commit is contained in:
syuilo
parent
2df02a9d70
commit
ed48349e39
1 changed files with 137 additions and 133 deletions
|
|
@ -23,6 +23,7 @@ import config from '../../config';
|
||||||
import registerHashtag from '../register-hashtag';
|
import registerHashtag from '../register-hashtag';
|
||||||
import isQuote from '../../misc/is-quote';
|
import isQuote from '../../misc/is-quote';
|
||||||
import { TextElementMention } from '../../mfm/parse/elements/mention';
|
import { TextElementMention } from '../../mfm/parse/elements/mention';
|
||||||
|
import { TextElementHashtag } from '../../mfm/parse/elements/hashtag';
|
||||||
|
|
||||||
type Type = 'reply' | 'renote' | 'quote' | 'mention';
|
type Type = 'reply' | 'renote' | 'quote' | 'mention';
|
||||||
|
|
||||||
|
|
@ -58,7 +59,7 @@ class NotificationManager {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export default async (user: IUser, data: {
|
type Option = {
|
||||||
createdAt?: Date;
|
createdAt?: Date;
|
||||||
text?: string;
|
text?: string;
|
||||||
reply?: INote;
|
reply?: INote;
|
||||||
|
|
@ -72,98 +73,32 @@ export default async (user: IUser, data: {
|
||||||
visibleUsers?: IUser[];
|
visibleUsers?: IUser[];
|
||||||
uri?: string;
|
uri?: string;
|
||||||
app?: IApp;
|
app?: IApp;
|
||||||
}, silent = false) => new Promise<INote>(async (res, rej) => {
|
};
|
||||||
|
|
||||||
|
export default async (user: IUser, data: Option, silent = false) => new Promise<INote>(async (res, rej) => {
|
||||||
if (data.createdAt == null) data.createdAt = new Date();
|
if (data.createdAt == null) data.createdAt = new Date();
|
||||||
if (data.visibility == null) data.visibility = 'public';
|
if (data.visibility == null) data.visibility = 'public';
|
||||||
if (data.viaMobile == null) data.viaMobile = false;
|
if (data.viaMobile == null) data.viaMobile = false;
|
||||||
|
|
||||||
let tags: string[] = [];
|
|
||||||
|
|
||||||
let tokens: any[] = null;
|
|
||||||
|
|
||||||
if (data.text) {
|
|
||||||
// Analyze
|
|
||||||
tokens = parse(data.text);
|
|
||||||
|
|
||||||
// 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);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
tags = tags.filter(tag => tag.length <= 100);
|
|
||||||
|
|
||||||
if (data.visibleUsers) {
|
if (data.visibleUsers) {
|
||||||
data.visibleUsers = data.visibleUsers.filter(x => x != null);
|
data.visibleUsers = data.visibleUsers.filter(x => x != null);
|
||||||
}
|
}
|
||||||
|
|
||||||
const insert: any = {
|
// Parse MFM
|
||||||
createdAt: data.createdAt,
|
const tokens = data.text ? parse(data.text) : [];
|
||||||
mediaIds: data.media ? data.media.map(file => file._id) : [],
|
|
||||||
replyId: data.reply ? data.reply._id : null,
|
|
||||||
renoteId: data.renote ? data.renote._id : null,
|
|
||||||
text: data.text,
|
|
||||||
poll: data.poll,
|
|
||||||
cw: data.cw == null ? null : data.cw,
|
|
||||||
tags,
|
|
||||||
tagsLower: tags.map(tag => tag.toLowerCase()),
|
|
||||||
userId: user._id,
|
|
||||||
viaMobile: data.viaMobile,
|
|
||||||
geo: data.geo || null,
|
|
||||||
appId: data.app ? data.app._id : null,
|
|
||||||
visibility: data.visibility,
|
|
||||||
visibleUserIds: data.visibility == 'specified'
|
|
||||||
? data.visibleUsers
|
|
||||||
? data.visibleUsers.map(u => u._id)
|
|
||||||
: []
|
|
||||||
: [],
|
|
||||||
|
|
||||||
// 以下非正規化データ
|
const tags = extractHashtags(tokens);
|
||||||
_reply: data.reply ? { userId: data.reply.userId } : null,
|
|
||||||
_renote: data.renote ? { userId: data.renote.userId } : null,
|
|
||||||
_user: {
|
|
||||||
host: user.host,
|
|
||||||
inbox: isRemoteUser(user) ? user.inbox : undefined
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if (data.uri != null) insert.uri = data.uri;
|
|
||||||
|
|
||||||
// メンション
|
|
||||||
const mentionedUsers = await extractMentionedUsers(tokens);
|
const mentionedUsers = await extractMentionedUsers(tokens);
|
||||||
|
|
||||||
// Append mentions data
|
const note = await insertNote(user, data, tokens, tags, mentionedUsers);
|
||||||
if (mentionedUsers.length > 0) {
|
|
||||||
insert.mentions = mentionedUsers.map(u => u._id);
|
|
||||||
insert.mentionedRemoteUsers = mentionedUsers.filter(u => isRemoteUser(u)).map(u => ({
|
|
||||||
uri: (u as IRemoteUser).uri,
|
|
||||||
username: u.username,
|
|
||||||
host: u.host
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
|
|
||||||
// 投稿を作成
|
|
||||||
let note: INote;
|
|
||||||
try {
|
|
||||||
note = await Note.insert(insert);
|
|
||||||
} catch (e) {
|
|
||||||
// duplicate key error
|
|
||||||
if (e.code === 11000) {
|
|
||||||
return res(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
console.error(e);
|
|
||||||
return rej('something happened');
|
|
||||||
}
|
|
||||||
|
|
||||||
res(note);
|
res(note);
|
||||||
|
|
||||||
|
if (note == null) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
// ハッシュタグ登録
|
// ハッシュタグ登録
|
||||||
tags.map(tag => registerHashtag(user, tag));
|
tags.map(tag => registerHashtag(user, tag));
|
||||||
|
|
||||||
|
|
@ -200,62 +135,8 @@ export default async (user: IUser, data: {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!silent) {
|
if (!silent) {
|
||||||
if (isLocalUser(user)) {
|
publish(user, note, noteObj, data.reply, data.renote, data.visibleUsers, noteActivity);
|
||||||
if (note.visibility == 'private' || note.visibility == 'followers' || note.visibility == 'specified') {
|
|
||||||
// Publish event to myself's stream
|
|
||||||
stream(note.userId, 'note', await pack(note, user, {
|
|
||||||
detail: true
|
|
||||||
}));
|
|
||||||
} else {
|
|
||||||
// Publish event to myself's stream
|
|
||||||
stream(note.userId, 'note', noteObj);
|
|
||||||
|
|
||||||
// Publish note to local and hybrid timeline stream
|
|
||||||
if (note.visibility != 'home') {
|
|
||||||
publishLocalTimelineStream(noteObj);
|
|
||||||
}
|
}
|
||||||
if (note.visibility == 'public') {
|
|
||||||
publishHybridTimelineStream(null, noteObj);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Publish note to global timeline stream
|
|
||||||
if (note.visibility == 'public' && note.replyId == null) {
|
|
||||||
publishGlobalTimelineStream(noteObj);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (note.visibility == 'specified') {
|
|
||||||
data.visibleUsers.forEach(async u => {
|
|
||||||
const n = await pack(note, u, {
|
|
||||||
detail: true
|
|
||||||
});
|
|
||||||
stream(u._id, 'note', n);
|
|
||||||
publishHybridTimelineStream(u._id, n);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
if (note.visibility == 'public' || note.visibility == 'home' || note.visibility == 'followers') {
|
|
||||||
// フォロワーに配信
|
|
||||||
publishToFollowers(note, noteObj, user, noteActivity);
|
|
||||||
}
|
|
||||||
|
|
||||||
// リストに配信
|
|
||||||
publishToUserLists(note, noteObj);
|
|
||||||
}
|
|
||||||
|
|
||||||
//#region リプライとAnnounceのAP配送
|
|
||||||
|
|
||||||
// 投稿がリプライかつ投稿者がローカルユーザーかつリプライ先の投稿の投稿者がリモートユーザーなら配送
|
|
||||||
if (data.reply && isLocalUser(user) && isRemoteUser(data.reply._user)) {
|
|
||||||
deliver(user, noteActivity, data.reply._user.inbox);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 投稿がRenoteかつ投稿者がローカルユーザーかつRenote元の投稿の投稿者がリモートユーザーなら配送
|
|
||||||
if (data.renote && isLocalUser(user) && isRemoteUser(data.renote._user)) {
|
|
||||||
deliver(user, noteActivity, data.renote._user.inbox);
|
|
||||||
}
|
|
||||||
//#endergion
|
|
||||||
|
|
||||||
// If has in reply to note
|
// If has in reply to note
|
||||||
if (data.reply) {
|
if (data.reply) {
|
||||||
|
|
@ -308,6 +189,129 @@ export default async (user: IUser, data: {
|
||||||
index(note);
|
index(note);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
async function publish(user: IUser, note: INote, noteObj: any, reply: INote, renote: INote, visibleUsers: IUser[], noteActivity: any) {
|
||||||
|
if (isLocalUser(user)) {
|
||||||
|
// 投稿がリプライかつ投稿者がローカルユーザーかつリプライ先の投稿の投稿者がリモートユーザーなら配送
|
||||||
|
if (reply && isRemoteUser(reply._user)) {
|
||||||
|
deliver(user, noteActivity, reply._user.inbox);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 投稿がRenoteかつ投稿者がローカルユーザーかつRenote元の投稿の投稿者がリモートユーザーなら配送
|
||||||
|
if (renote && isRemoteUser(renote._user)) {
|
||||||
|
deliver(user, noteActivity, renote._user.inbox);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (['private', 'followers', 'specified'].includes(note.visibility)) {
|
||||||
|
// Publish event to myself's stream
|
||||||
|
stream(note.userId, 'note', await pack(note, user, {
|
||||||
|
detail: true
|
||||||
|
}));
|
||||||
|
} else {
|
||||||
|
// Publish event to myself's stream
|
||||||
|
stream(note.userId, 'note', noteObj);
|
||||||
|
|
||||||
|
// Publish note to local and hybrid timeline stream
|
||||||
|
if (note.visibility != 'home') {
|
||||||
|
publishLocalTimelineStream(noteObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (note.visibility == 'public') {
|
||||||
|
publishHybridTimelineStream(null, noteObj);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Publish note to global timeline stream
|
||||||
|
if (note.visibility == 'public' && note.replyId == null) {
|
||||||
|
publishGlobalTimelineStream(noteObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (note.visibility == 'specified') {
|
||||||
|
visibleUsers.forEach(async (u) => {
|
||||||
|
const n = await pack(note, u, {
|
||||||
|
detail: true
|
||||||
|
});
|
||||||
|
stream(u._id, 'note', n);
|
||||||
|
publishHybridTimelineStream(u._id, n);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (['public', 'home', 'followers'].includes(note.visibility)) {
|
||||||
|
// フォロワーに配信
|
||||||
|
publishToFollowers(note, noteObj, user, noteActivity);
|
||||||
|
}
|
||||||
|
|
||||||
|
// リストに配信
|
||||||
|
publishToUserLists(note, noteObj);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function insertNote(user: IUser, data: Option, tokens: ReturnType<typeof parse>, tags: string[], mentionedUsers: IUser[]) {
|
||||||
|
const insert: any = {
|
||||||
|
createdAt: data.createdAt,
|
||||||
|
mediaIds: data.media ? data.media.map(file => file._id) : [],
|
||||||
|
replyId: data.reply ? data.reply._id : null,
|
||||||
|
renoteId: data.renote ? data.renote._id : null,
|
||||||
|
text: data.text,
|
||||||
|
poll: data.poll,
|
||||||
|
cw: data.cw == null ? null : data.cw,
|
||||||
|
tags,
|
||||||
|
tagsLower: tags.map(tag => tag.toLowerCase()),
|
||||||
|
userId: user._id,
|
||||||
|
viaMobile: data.viaMobile,
|
||||||
|
geo: data.geo || null,
|
||||||
|
appId: data.app ? data.app._id : null,
|
||||||
|
visibility: data.visibility,
|
||||||
|
visibleUserIds: data.visibility == 'specified'
|
||||||
|
? data.visibleUsers
|
||||||
|
? data.visibleUsers.map(u => u._id)
|
||||||
|
: []
|
||||||
|
: [],
|
||||||
|
|
||||||
|
// 以下非正規化データ
|
||||||
|
_reply: data.reply ? { userId: data.reply.userId } : null,
|
||||||
|
_renote: data.renote ? { userId: data.renote.userId } : null,
|
||||||
|
_user: {
|
||||||
|
host: user.host,
|
||||||
|
inbox: isRemoteUser(user) ? user.inbox : undefined
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if (data.uri != null) insert.uri = data.uri;
|
||||||
|
|
||||||
|
// Append mentions data
|
||||||
|
if (mentionedUsers.length > 0) {
|
||||||
|
insert.mentions = mentionedUsers.map(u => u._id);
|
||||||
|
insert.mentionedRemoteUsers = mentionedUsers.filter(u => isRemoteUser(u)).map(u => ({
|
||||||
|
uri: (u as IRemoteUser).uri,
|
||||||
|
username: u.username,
|
||||||
|
host: u.host
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
// 投稿を作成
|
||||||
|
try {
|
||||||
|
return await Note.insert(insert);
|
||||||
|
} catch (e) {
|
||||||
|
// duplicate key error
|
||||||
|
if (e.code === 11000) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.error(e);
|
||||||
|
throw 'something happened';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function extractHashtags(tokens: ReturnType<typeof parse>): string[] {
|
||||||
|
// Extract hashtags
|
||||||
|
const hashtags = tokens
|
||||||
|
.filter(t => t.type == 'hashtag')
|
||||||
|
.map(t => (t as TextElementHashtag).hashtag)
|
||||||
|
.filter(tag => tag.length <= 100);
|
||||||
|
|
||||||
|
return [...new Set(hashtags)];
|
||||||
|
}
|
||||||
|
|
||||||
function index(note: INote) {
|
function index(note: INote) {
|
||||||
if (note.text == null || config.elasticsearch == null) return;
|
if (note.text == null || config.elasticsearch == null) return;
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue