misskey/src/server/api/common/get-friends.ts
2018-07-11 14:03:21 +09:00

52 lines
1.1 KiB
TypeScript

import * as mongodb from 'mongodb';
import Following from '../../../models/following';
export const getFriendIds = async (me: mongodb.ObjectID, includeMe = true) => {
// Fetch relation to other users who the I follows
// SELECT followee
const followings = await Following
.find({
followerId: me
}, {
fields: {
followeeId: true
}
});
// ID list of other users who the I follows
const myfollowingIds = followings.map(following => following.followeeId);
if (includeMe) {
myfollowingIds.push(me);
}
return myfollowingIds;
};
export const getFriends = async (me: mongodb.ObjectID, includeMe = true, remoteOnly = false) => {
const q: any = remoteOnly ? {
followerId: me,
'_followee.host': { $ne: null }
} : {
followerId: me
};
// Fetch relation to other users who the I follows
const followings = await Following
.find(q);
// ID list of other users who the I follows
const myfollowings = followings.map(following => ({
id: following.followeeId,
stalk: following.stalk
}));
if (includeMe) {
myfollowings.push({
id: me,
stalk: true
});
}
return myfollowings;
};