Implement fetching mutual friends

This commit is contained in:
Rory& 2025-09-30 19:24:10 +02:00
parent d647f882b1
commit 6a3155adba

View File

@ -20,158 +20,165 @@ import { route } from "@spacebar/api";
import { import {
Badge, Badge,
Config, Config,
emitEvent,
FieldErrors, FieldErrors,
handleFile,
Member, Member,
PrivateUserProjection, PrivateUserProjection,
PublicUser,
PublicUserProjection,
Relationship,
RelationshipType,
User, User,
UserProfileModifySchema, UserProfileModifySchema,
UserUpdateEvent, UserUpdateEvent,
emitEvent,
handleFile,
} from "@spacebar/util"; } from "@spacebar/util";
import { Request, Response, Router } from "express"; import { Request, Response, Router } from "express";
import { In } from "typeorm";
const router: Router = Router({ mergeParams: true }); const router: Router = Router({ mergeParams: true });
router.get( router.get("/", route({ responses: { 200: { body: "UserProfileResponse" } } }), async (req: Request, res: Response) => {
"/", if (req.params.id === "@me") req.params.id = req.user_id;
route({ responses: { 200: { body: "UserProfileResponse" } } }),
async (req: Request, res: Response) => {
if (req.params.id === "@me") req.params.id = req.user_id;
const { guild_id, with_mutual_guilds } = req.query; const { guild_id, with_mutual_guilds, with_mutual_friends, with_mutual_friends_count } = req.query;
const user = await User.getPublicUser(req.params.id, { const user = await User.getPublicUser(req.params.id, {
relations: ["connected_accounts"], relations: ["connected_accounts"],
});
const mutual_guilds: object[] = [];
let premium_guild_since;
if (with_mutual_guilds == "true") {
const requested_member = await Member.find({
where: { id: req.params.id },
});
const self_member = await Member.find({
where: { id: req.user_id },
}); });
const mutual_guilds: object[] = []; for (const rmem of requested_member) {
let premium_guild_since; if (rmem.premium_since) {
if (premium_guild_since) {
if (with_mutual_guilds == "true") { if (premium_guild_since > rmem.premium_since) {
const requested_member = await Member.find({
where: { id: req.params.id },
});
const self_member = await Member.find({
where: { id: req.user_id },
});
for (const rmem of requested_member) {
if (rmem.premium_since) {
if (premium_guild_since) {
if (premium_guild_since > rmem.premium_since) {
premium_guild_since = rmem.premium_since;
}
} else {
premium_guild_since = rmem.premium_since; premium_guild_since = rmem.premium_since;
} }
} else {
premium_guild_since = rmem.premium_since;
} }
for (const smem of self_member) { }
if (smem.guild_id === rmem.guild_id) { for (const smem of self_member) {
mutual_guilds.push({ if (smem.guild_id === rmem.guild_id) {
id: rmem.guild_id, mutual_guilds.push({
nick: rmem.nick, id: rmem.guild_id,
}); nick: rmem.nick,
} });
} }
} }
} }
}
const guild_member = const guild_member =
guild_id && typeof guild_id == "string" guild_id && typeof guild_id == "string"
? await Member.findOneOrFail({ ? await Member.findOneOrFail({
where: { id: req.params.id, guild_id: guild_id }, where: { id: req.params.id, guild_id: guild_id },
relations: ["roles"], relations: ["roles"],
}) })
: undefined; : undefined;
// TODO: make proper DTO's in util? // TODO: make proper DTO's in util?
const userProfile = { const userProfile = {
bio: req.user_bot ? null : user.bio, bio: req.user_bot ? null : user.bio,
accent_color: user.accent_color, accent_color: user.accent_color,
banner: user.banner, banner: user.banner,
pronouns: user.pronouns, pronouns: user.pronouns,
theme_colors: user.theme_colors?.map((t) => Number(t)), // these are strings for some reason, they should be numbers theme_colors: user.theme_colors?.map((t) => Number(t)), // these are strings for some reason, they should be numbers
}; };
const guildMemberProfile = { const guildMemberProfile = {
accent_color: null, accent_color: null,
banner: guild_member?.banner || null, banner: guild_member?.banner || null,
bio: guild_member?.bio || "", bio: guild_member?.bio || "",
guild_id, guild_id,
}; };
const badges = await Badge.find(); const badges = await Badge.find();
res.json({ let mutual_friends: PublicUser[] = [];
connected_accounts: user.connected_accounts.filter( let mutual_friends_count = 0;
(x) => x.visibility != 0,
),
premium_guild_since: premium_guild_since, // TODO
premium_since: user.premium_since, // TODO
mutual_guilds: mutual_guilds, // TODO {id: "", nick: null} when ?with_mutual_guilds=true
user: user.toPublicUser(),
premium_type: user.premium_type,
profile_themes_experiment_bucket: 4, // TODO: This doesn't make it available, for some reason?
user_profile: userProfile,
guild_member: guild_member?.toPublicMember(),
guild_member_profile: guild_id && guildMemberProfile,
badges: badges.filter((x) => user.badge_ids?.includes(x.id)),
});
},
);
router.patch( if (with_mutual_friends == "true" || with_mutual_friends_count == "true") {
"/", const relationshipsSelf = await Relationship.find({ where: { from_id: req.user_id, type: RelationshipType.friends } });
route({ requestBody: "UserProfileModifySchema" }), const relationshipsUser = await Relationship.find({ where: { from_id: req.params.id, type: RelationshipType.friends } });
async (req: Request, res: Response) => { const relationshipsIntersection = relationshipsSelf.filter((r1) => relationshipsUser.some((r2) => r2.to_id === r1.to_id));
const body = req.body as UserProfileModifySchema; if (with_mutual_friends_count) mutual_friends_count = relationshipsIntersection.length;
if (with_mutual_friends) {
if (body.banner) const users = await User.find({ where: { id: In(relationshipsIntersection.map((r) => r.to_id)) }, select: PublicUserProjection });
body.banner = await handleFile( mutual_friends = users.map((u) => u.toPublicUser());
`/banners/${req.user_id}`,
body.banner as string,
);
const user = await User.findOneOrFail({
where: { id: req.user_id },
select: [...PrivateUserProjection, "data"],
});
if (body.bio) {
const { maxBio } = Config.get().limits.user;
if (body.bio.length > maxBio) {
throw FieldErrors({
bio: {
code: "BIO_INVALID",
message: `Bio must be less than ${maxBio} in length`,
},
});
}
} }
}
user.assign(body); res.json({
await user.save(); connected_accounts: user.connected_accounts.filter((x) => x.visibility != 0),
premium_guild_since: premium_guild_since, // TODO
premium_since: user.premium_since, // TODO
mutual_guilds: with_mutual_guilds ? mutual_guilds : undefined, // TODO {id: "", nick: null} when ?with_mutual_guilds=true
mutual_friends: with_mutual_friends ? mutual_friends : undefined,
mutual_friends_count: with_mutual_friends_count ? mutual_friends_count : undefined,
user: user.toPublicUser(),
premium_type: user.premium_type,
profile_themes_experiment_bucket: 4, // TODO: This doesn't make it available, for some reason?
user_profile: userProfile,
guild_member: guild_member?.toPublicMember(),
guild_member_profile: guild_id && guildMemberProfile,
badges: badges.filter((x) => user.badge_ids?.includes(x.id)),
});
});
// eslint-disable-next-line @typescript-eslint/ban-ts-comment router.patch("/", route({ requestBody: "UserProfileModifySchema" }), async (req: Request, res: Response) => {
// @ts-ignore const body = req.body as UserProfileModifySchema;
delete user.data;
// TODO: send update member list event in gateway if (body.banner) body.banner = await handleFile(`/banners/${req.user_id}`, body.banner as string);
await emitEvent({ const user = await User.findOneOrFail({
event: "USER_UPDATE", where: { id: req.user_id },
user_id: req.user_id, select: [...PrivateUserProjection, "data"],
data: user, });
} as UserUpdateEvent);
res.json({ if (body.bio) {
accent_color: user.accent_color, const { maxBio } = Config.get().limits.user;
bio: user.bio, if (body.bio.length > maxBio) {
banner: user.banner, throw FieldErrors({
theme_colors: user.theme_colors, bio: {
pronouns: user.pronouns, code: "BIO_INVALID",
}); message: `Bio must be less than ${maxBio} in length`,
}, },
); });
}
}
user.assign(body);
await user.save();
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
// @ts-ignore
delete user.data;
// TODO: send update member list event in gateway
await emitEvent({
event: "USER_UPDATE",
user_id: req.user_id,
data: user,
} as UserUpdateEvent);
res.json({
accent_color: user.accent_color,
bio: user.bio,
banner: user.banner,
theme_colors: user.theme_colors,
pronouns: user.pronouns,
});
});
export default router; export default router;