Fix follow duplicate (#3548)

* フォローとリクエスト両方存在しても解除する

* 既にフォローしてても承認できるように
This commit is contained in:
MeiMei 2018-12-08 18:55:00 +09:00 committed by syuilo
parent 3b7be81491
commit 07218ce03c
2 changed files with 21 additions and 4 deletions

View File

@ -5,6 +5,7 @@ import unfollow from '../../../../services/following/delete';
import cancelRequest from '../../../../services/following/requests/cancel'; import cancelRequest from '../../../../services/following/requests/cancel';
import { IFollow } from '../../type'; import { IFollow } from '../../type';
import FollowRequest from '../../../../models/follow-request'; import FollowRequest from '../../../../models/follow-request';
import Following from '../../../../models/following';
export default async (actor: IRemoteUser, activity: IFollow): Promise<void> => { export default async (actor: IRemoteUser, activity: IFollow): Promise<void> => {
const id = typeof activity.object == 'string' ? activity.object : activity.object.id; const id = typeof activity.object == 'string' ? activity.object : activity.object.id;
@ -30,9 +31,16 @@ export default async (actor: IRemoteUser, activity: IFollow): Promise<void> => {
followeeId: followee._id followeeId: followee._id
}); });
const following = await Following.findOne({
followerId: actor._id,
followeeId: followee._id
});
if (req) { if (req) {
await cancelRequest(followee, actor); await cancelRequest(followee, actor);
} else { }
if (following) {
await unfollow(actor, followee); await unfollow(actor, followee);
} }
}; };

View File

@ -1,4 +1,4 @@
import User, { IUser, isRemoteUser, ILocalUser, pack as packUser } from '../../../models/user'; import User, { IUser, isRemoteUser, ILocalUser, pack as packUser, isLocalUser } from '../../../models/user';
import FollowRequest from '../../../models/follow-request'; import FollowRequest from '../../../models/follow-request';
import pack from '../../../remote/activitypub/renderer'; import pack from '../../../remote/activitypub/renderer';
import renderFollow from '../../../remote/activitypub/renderer/follow'; import renderFollow from '../../../remote/activitypub/renderer/follow';
@ -9,6 +9,8 @@ import { publishMainStream } from '../../../stream';
import perUserFollowingChart from '../../../chart/per-user-following'; import perUserFollowingChart from '../../../chart/per-user-following';
export default async function(followee: IUser, follower: IUser) { export default async function(followee: IUser, follower: IUser) {
let incremented = 1;
await Following.insert({ await Following.insert({
createdAt: new Date(), createdAt: new Date(),
followerId: follower._id, followerId: follower._id,
@ -25,6 +27,13 @@ export default async function(followee: IUser, follower: IUser) {
inbox: isRemoteUser(followee) ? followee.inbox : undefined, inbox: isRemoteUser(followee) ? followee.inbox : undefined,
sharedInbox: isRemoteUser(followee) ? followee.sharedInbox : undefined sharedInbox: isRemoteUser(followee) ? followee.sharedInbox : undefined
} }
}).catch(e => {
if (e.code === 11000 && isRemoteUser(follower) && isLocalUser(followee)) {
console.log(`Accept => Insert duplicated ignore. ${follower._id} => ${followee._id}`);
incremented = 0;
} else {
throw e;
}
}); });
if (isRemoteUser(follower)) { if (isRemoteUser(follower)) {
@ -45,7 +54,7 @@ export default async function(followee: IUser, follower: IUser) {
//#region Increment following count //#region Increment following count
await User.update({ _id: follower._id }, { await User.update({ _id: follower._id }, {
$inc: { $inc: {
followingCount: 1 followingCount: incremented
} }
}); });
//#endregion //#endregion
@ -53,7 +62,7 @@ export default async function(followee: IUser, follower: IUser) {
//#region Increment followers count //#region Increment followers count
await User.update({ _id: followee._id }, { await User.update({ _id: followee._id }, {
$inc: { $inc: {
followersCount: 1 followersCount: incremented
} }
}); });
//#endregion //#endregion