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) => {
	// Fetch relation to other users who the I follows
	const followings = await Following
		.find({
			followerId: me
		});

	// 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;
};