forked from FoundKeyGang/FoundKey
wip
This commit is contained in:
syuilo
parent
0ac695edbd
commit
0009a371c8
2 changed files with 235 additions and 245 deletions
|
|
@ -1,76 +1,90 @@
|
||||||
import parseAcct from '../acct/parse';
|
import parseAcct from '../../acct/parse';
|
||||||
import Post, { pack } from '../models/post';
|
import Post, { pack, IPost } from '../../models/post';
|
||||||
import User, { isLocalUser, isRemoteUser, IUser } from '../models/user';
|
import User, { isLocalUser, isRemoteUser, IUser } from '../../models/user';
|
||||||
import stream from '../publishers/stream';
|
import stream from '../../publishers/stream';
|
||||||
import Following from '../models/following';
|
import Following from '../../models/following';
|
||||||
import { createHttp } from '../queue';
|
import { createHttp } from '../../queue';
|
||||||
import renderNote from '../remote/activitypub/renderer/note';
|
import renderNote from '../../remote/activitypub/renderer/note';
|
||||||
import renderCreate from '../remote/activitypub/renderer/create';
|
import renderCreate from '../../remote/activitypub/renderer/create';
|
||||||
import context from '../remote/activitypub/renderer/context';
|
import context from '../../remote/activitypub/renderer/context';
|
||||||
|
import { IDriveFile } from '../../models/drive-file';
|
||||||
|
import notify from '../../publishers/notify';
|
||||||
|
import PostWatching from '../../models/post-watching';
|
||||||
|
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';
|
||||||
|
|
||||||
export default async (user: IUser, post, reply, repost, atMentions) => {
|
export default async (user: IUser, content: {
|
||||||
post.mentions = [];
|
createdAt: Date;
|
||||||
|
text: string;
|
||||||
|
reply: IPost;
|
||||||
|
repost: IPost;
|
||||||
|
media: IDriveFile[];
|
||||||
|
geo: any;
|
||||||
|
viaMobile: boolean;
|
||||||
|
tags: string[];
|
||||||
|
}) => new Promise(async (res, rej) => {
|
||||||
|
const tags = content.tags || [];
|
||||||
|
|
||||||
function addMention(mentionee) {
|
let tokens = null;
|
||||||
// Reject if already added
|
|
||||||
if (post.mentions.some(x => x.equals(mentionee))) return;
|
|
||||||
|
|
||||||
// Add mention
|
if (content.text) {
|
||||||
post.mentions.push(mentionee);
|
// Analyze
|
||||||
|
tokens = parse(content.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);
|
||||||
|
}
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
if (reply) {
|
// 投稿を作成
|
||||||
// Add mention
|
const post = await Post.insert({
|
||||||
addMention(reply.userId);
|
createdAt: content.createdAt,
|
||||||
post.replyId = reply._id;
|
mediaIds: content.media ? content.media.map(file => file._id) : [],
|
||||||
post._reply = { userId: reply.userId };
|
replyId: content.reply ? content.reply._id : null,
|
||||||
} else {
|
repostId: content.repost ? content.repost._id : null,
|
||||||
post.replyId = null;
|
text: content.text,
|
||||||
post._reply = null;
|
tags,
|
||||||
}
|
userId: user._id,
|
||||||
|
viaMobile: content.viaMobile,
|
||||||
|
geo: content.geo || null,
|
||||||
|
|
||||||
if (repost) {
|
// 以下非正規化データ
|
||||||
if (post.text) {
|
_reply: content.reply ? { userId: content.reply.userId } : null,
|
||||||
// Add mention
|
_repost: content.repost ? { userId: content.repost.userId } : null,
|
||||||
addMention(repost.userId);
|
});
|
||||||
}
|
|
||||||
|
|
||||||
post.repostId = repost._id;
|
res(post);
|
||||||
post._repost = { userId: repost.userId };
|
|
||||||
} else {
|
|
||||||
post.repostId = null;
|
|
||||||
post._repost = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
await Promise.all(atMentions.map(async mention => {
|
|
||||||
// Fetch mentioned user
|
|
||||||
// SELECT _id
|
|
||||||
const { _id } = await User
|
|
||||||
.findOne(parseAcct(mention), { _id: true });
|
|
||||||
|
|
||||||
// Add mention
|
|
||||||
addMention(_id);
|
|
||||||
}));
|
|
||||||
|
|
||||||
const inserted = await Post.insert(post);
|
|
||||||
|
|
||||||
User.update({ _id: user._id }, {
|
User.update({ _id: user._id }, {
|
||||||
// Increment my posts count
|
// Increment posts count
|
||||||
$inc: {
|
$inc: {
|
||||||
postsCount: 1
|
postsCount: 1
|
||||||
},
|
},
|
||||||
|
// Update latest post
|
||||||
$set: {
|
$set: {
|
||||||
latestPost: post._id
|
latestPost: post
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
const postObj = await pack(inserted);
|
// Serialize
|
||||||
|
const postObj = await pack(post);
|
||||||
|
|
||||||
// タイムラインへの投稿
|
// タイムラインへの投稿
|
||||||
if (!post.channelId) {
|
if (!post.channelId) {
|
||||||
// Publish event to myself's stream
|
// Publish event to myself's stream
|
||||||
stream(post.userId, 'post', postObj);
|
if (isLocalUser(user)) {
|
||||||
|
stream(post.userId, 'post', postObj);
|
||||||
|
}
|
||||||
|
|
||||||
// Fetch all followers
|
// Fetch all followers
|
||||||
const followers = await Following.aggregate([{
|
const followers = await Following.aggregate([{
|
||||||
|
|
@ -144,6 +158,172 @@ export default async (user: IUser, post, reply, repost, atMentions) => {
|
||||||
);
|
);
|
||||||
}*/
|
}*/
|
||||||
|
|
||||||
return Promise.all(promises);
|
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) {
|
||||||
|
event(mentionee, reason, postObj);
|
||||||
|
pushSw(mentionee, reason, postObj);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If has in reply to post
|
||||||
|
if (content.reply) {
|
||||||
|
// Increment replies count
|
||||||
|
Post.update({ _id: content.reply._id }, {
|
||||||
|
$inc: {
|
||||||
|
repliesCount: 1
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// (自分自身へのリプライでない限りは)通知を作成
|
||||||
|
notify(content.reply.userId, user._id, 'reply', {
|
||||||
|
postId: post._id
|
||||||
|
});
|
||||||
|
|
||||||
|
// Fetch watchers
|
||||||
|
PostWatching.find({
|
||||||
|
postId: content.reply._id,
|
||||||
|
userId: { $ne: user._id },
|
||||||
|
// 削除されたドキュメントは除く
|
||||||
|
deletedAt: { $exists: false }
|
||||||
|
}, {
|
||||||
|
fields: {
|
||||||
|
userId: true
|
||||||
|
}
|
||||||
|
}).then(watchers => {
|
||||||
|
watchers.forEach(watcher => {
|
||||||
|
notify(watcher.userId, user._id, 'reply', {
|
||||||
|
postId: post._id
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
// この投稿をWatchする
|
||||||
|
if (isLocalUser(user) && user.account.settings.autoWatch !== false) {
|
||||||
|
watch(user._id, content.reply);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add mention
|
||||||
|
addMention(content.reply.userId, 'reply');
|
||||||
|
}
|
||||||
|
|
||||||
|
// If it is repost
|
||||||
|
if (content.repost) {
|
||||||
|
// Notify
|
||||||
|
const type = content.text ? 'quote' : 'repost';
|
||||||
|
notify(content.repost.userId, user._id, type, {
|
||||||
|
post_id: post._id
|
||||||
|
});
|
||||||
|
|
||||||
|
// Fetch watchers
|
||||||
|
PostWatching.find({
|
||||||
|
postId: content.repost._id,
|
||||||
|
userId: { $ne: user._id },
|
||||||
|
// 削除されたドキュメントは除く
|
||||||
|
deletedAt: { $exists: false }
|
||||||
|
}, {
|
||||||
|
fields: {
|
||||||
|
userId: true
|
||||||
|
}
|
||||||
|
}).then(watchers => {
|
||||||
|
watchers.forEach(watcher => {
|
||||||
|
notify(watcher.userId, user._id, type, {
|
||||||
|
postId: post._id
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
// この投稿をWatchする
|
||||||
|
if (isLocalUser(user) && user.account.settings.autoWatch !== false) {
|
||||||
|
watch(user._id, content.repost);
|
||||||
|
}
|
||||||
|
|
||||||
|
// If it is quote repost
|
||||||
|
if (content.text) {
|
||||||
|
// Add mention
|
||||||
|
addMention(content.repost.userId, 'quote');
|
||||||
|
} else {
|
||||||
|
// Publish event
|
||||||
|
if (!user._id.equals(content.repost.userId)) {
|
||||||
|
event(content.repost.userId, 'repost', postObj);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 今までで同じ投稿をRepostしているか
|
||||||
|
const existRepost = await Post.findOne({
|
||||||
|
userId: user._id,
|
||||||
|
repostId: content.repost._id,
|
||||||
|
_id: {
|
||||||
|
$ne: post._id
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!existRepost) {
|
||||||
|
// Update repostee status
|
||||||
|
Post.update({ _id: content.repost._id }, {
|
||||||
|
$inc: {
|
||||||
|
repostCount: 1
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If has text content
|
||||||
|
if (content.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;
|
||||||
|
|
||||||
|
// 既に言及されたユーザーに対する返信や引用repostの場合も無視
|
||||||
|
if (content.reply && content.reply.userId.equals(mentionee._id)) return;
|
||||||
|
if (content.repost && content.repost.userId.equals(mentionee._id)) return;
|
||||||
|
|
||||||
|
// Add mention
|
||||||
|
addMention(mentionee._id, 'mention');
|
||||||
|
|
||||||
|
// Create notification
|
||||||
|
notify(mentionee._id, user._id, 'mention', {
|
||||||
|
post_id: post._id
|
||||||
|
});
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Append mentions data
|
||||||
|
if (mentions.length > 0) {
|
||||||
|
Post.update({ _id: post._id }, {
|
||||||
|
$set: {
|
||||||
|
mentions
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
|
||||||
|
|
@ -1,190 +0,0 @@
|
||||||
import Mute from '../models/mute';
|
|
||||||
import Post, { pack } from '../models/post';
|
|
||||||
import Watching from '../models/post-watching';
|
|
||||||
import User from '../models/user';
|
|
||||||
import stream from '../publishers/stream';
|
|
||||||
import notify from '../publishers/notify';
|
|
||||||
import pushSw from '../publishers/push-sw';
|
|
||||||
import queue from '../queue';
|
|
||||||
import watch from './watch';
|
|
||||||
|
|
||||||
export default async (user, mentions, post) => {
|
|
||||||
const promisedPostObj = pack(post);
|
|
||||||
const promises = [
|
|
||||||
User.update({ _id: user._id }, {
|
|
||||||
// Increment my posts count
|
|
||||||
$inc: {
|
|
||||||
postsCount: 1
|
|
||||||
},
|
|
||||||
|
|
||||||
$set: {
|
|
||||||
latestPost: post._id
|
|
||||||
}
|
|
||||||
}),
|
|
||||||
new Promise((resolve, reject) => queue.create('http', {
|
|
||||||
type: 'deliverPost',
|
|
||||||
id: post._id,
|
|
||||||
}).save(error => error ? reject(error) : resolve())),
|
|
||||||
] as Array<Promise<any>>;
|
|
||||||
|
|
||||||
function addMention(promisedMentionee, reason) {
|
|
||||||
// Publish event
|
|
||||||
promises.push(promisedMentionee.then(mentionee => {
|
|
||||||
if (user._id.equals(mentionee)) {
|
|
||||||
return Promise.resolve();
|
|
||||||
}
|
|
||||||
|
|
||||||
return Promise.all([
|
|
||||||
promisedPostObj,
|
|
||||||
Mute.find({
|
|
||||||
muterId: mentionee,
|
|
||||||
deletedAt: { $exists: false }
|
|
||||||
})
|
|
||||||
]).then(([postObj, mentioneeMutes]) => {
|
|
||||||
const mentioneesMutedUserIds = mentioneeMutes.map(m => m.muteeId.toString());
|
|
||||||
if (mentioneesMutedUserIds.indexOf(user._id.toString()) == -1) {
|
|
||||||
stream(mentionee, reason, postObj);
|
|
||||||
pushSw(mentionee, reason, postObj);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
|
|
||||||
// If has in reply to post
|
|
||||||
if (post.replyId) {
|
|
||||||
promises.push(
|
|
||||||
// Increment replies count
|
|
||||||
Post.update({ _id: post.replyId }, {
|
|
||||||
$inc: {
|
|
||||||
repliesCount: 1
|
|
||||||
}
|
|
||||||
}),
|
|
||||||
|
|
||||||
// 自分自身へのリプライでない限りは通知を作成
|
|
||||||
promisedPostObj.then(({ reply }) => {
|
|
||||||
return notify(reply.userId, user._id, 'reply', {
|
|
||||||
postId: post._id
|
|
||||||
});
|
|
||||||
}),
|
|
||||||
|
|
||||||
// Fetch watchers
|
|
||||||
Watching
|
|
||||||
.find({
|
|
||||||
postId: post.replyId,
|
|
||||||
userId: { $ne: user._id },
|
|
||||||
// 削除されたドキュメントは除く
|
|
||||||
deletedAt: { $exists: false }
|
|
||||||
}, {
|
|
||||||
fields: {
|
|
||||||
userId: true
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.then(watchers => {
|
|
||||||
watchers.forEach(watcher => {
|
|
||||||
notify(watcher.userId, user._id, 'reply', {
|
|
||||||
postId: post._id
|
|
||||||
});
|
|
||||||
});
|
|
||||||
})
|
|
||||||
);
|
|
||||||
|
|
||||||
// Add mention
|
|
||||||
addMention(promisedPostObj.then(({ reply }) => reply.userId), 'reply');
|
|
||||||
|
|
||||||
// この投稿をWatchする
|
|
||||||
if (user.account.settings.autoWatch !== false) {
|
|
||||||
promises.push(promisedPostObj.then(({ reply }) => {
|
|
||||||
return watch(user._id, reply);
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// If it is repost
|
|
||||||
if (post.repostId) {
|
|
||||||
const type = post.text ? 'quote' : 'repost';
|
|
||||||
|
|
||||||
promises.push(
|
|
||||||
promisedPostObj.then(({ repost }) => Promise.all([
|
|
||||||
// Notify
|
|
||||||
notify(repost.userId, user._id, type, {
|
|
||||||
postId: post._id
|
|
||||||
}),
|
|
||||||
|
|
||||||
// この投稿をWatchする
|
|
||||||
// TODO: ユーザーが「Repostしたときに自動でWatchする」設定を
|
|
||||||
// オフにしていた場合はしない
|
|
||||||
watch(user._id, repost)
|
|
||||||
])),
|
|
||||||
|
|
||||||
// Fetch watchers
|
|
||||||
Watching
|
|
||||||
.find({
|
|
||||||
postId: post.repostId,
|
|
||||||
userId: { $ne: user._id },
|
|
||||||
// 削除されたドキュメントは除く
|
|
||||||
deletedAt: { $exists: false }
|
|
||||||
}, {
|
|
||||||
fields: {
|
|
||||||
userId: true
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.then(watchers => {
|
|
||||||
watchers.forEach(watcher => {
|
|
||||||
notify(watcher.userId, user._id, type, {
|
|
||||||
postId: post._id
|
|
||||||
});
|
|
||||||
});
|
|
||||||
})
|
|
||||||
);
|
|
||||||
|
|
||||||
// If it is quote repost
|
|
||||||
if (post.text) {
|
|
||||||
// Add mention
|
|
||||||
addMention(promisedPostObj.then(({ repost }) => repost.userId), 'quote');
|
|
||||||
} else {
|
|
||||||
promises.push(promisedPostObj.then(postObj => {
|
|
||||||
// Publish event
|
|
||||||
if (!user._id.equals(postObj.repost.userId)) {
|
|
||||||
stream(postObj.repost.userId, 'repost', postObj);
|
|
||||||
}
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
|
|
||||||
// 今までで同じ投稿をRepostしているか
|
|
||||||
const existRepost = await Post.findOne({
|
|
||||||
userId: user._id,
|
|
||||||
repostId: post.repostId,
|
|
||||||
_id: {
|
|
||||||
$ne: post._id
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!existRepost) {
|
|
||||||
// Update repostee status
|
|
||||||
promises.push(Post.update({ _id: post.repostId }, {
|
|
||||||
$inc: {
|
|
||||||
repostCount: 1
|
|
||||||
}
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Resolve all mentions
|
|
||||||
await promisedPostObj.then(({ reply, repost }) => Promise.all(mentions.map(async mention => {
|
|
||||||
// 既に言及されたユーザーに対する返信や引用repostの場合も無視
|
|
||||||
if (reply && reply.userId.equals(mention)) return;
|
|
||||||
if (repost && repost.userId.equals(mention)) return;
|
|
||||||
|
|
||||||
// Add mention
|
|
||||||
addMention(mention, 'mention');
|
|
||||||
|
|
||||||
// Create notification
|
|
||||||
await notify(mention, user._id, 'mention', {
|
|
||||||
postId: post._id
|
|
||||||
});
|
|
||||||
})));
|
|
||||||
|
|
||||||
await Promise.all(promises);
|
|
||||||
|
|
||||||
return promisedPostObj;
|
|
||||||
};
|
|
||||||
Loading…
Reference in a new issue