Skip to content

Commit

Permalink
Stop returning defaults when using getPublicUser()
Browse files Browse the repository at this point in the history
  • Loading branch information
DEVTomatoCake committed Sep 1, 2024
1 parent 143cbf2 commit 269d4ee
Show file tree
Hide file tree
Showing 5 changed files with 16 additions and 13 deletions.
8 changes: 4 additions & 4 deletions src/api/routes/channels/#channel_id/invites.ts
Original file line number Diff line number Diff line change
@@ -1,17 +1,17 @@
/*
Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
Copyright (C) 2023 Spacebar and Spacebar Contributors
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
Expand Down Expand Up @@ -84,7 +84,7 @@ router.post(
}).save();

const data = invite.toJSON();
data.inviter = (await User.getPublicUser(req.user_id)).toPublicUser();
data.inviter = await User.getPublicUser(req.user_id);
data.guild = await Guild.findOne({ where: { id: guild_id } });
data.channel = channel;

Expand Down
5 changes: 4 additions & 1 deletion src/api/routes/users/#id/profile.ts
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,10 @@ router.get(

const { guild_id, with_mutual_guilds } = req.query;

const user = await User.getPublicUser(req.params.id, {
const user = await User.findOneOrFail({
where: {
id: req.params.id,
},
relations: ["connected_accounts"],
});

Expand Down
4 changes: 3 additions & 1 deletion src/api/util/handlers/Message.ts
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,9 @@ export async function handleMessage(opts: MessageOptions): Promise<Message> {
}

if (opts.author_id) {
message.author = await User.getPublicUser(opts.author_id);
message.author = await User.findOneOrFail({
where: { id: opts.author_id },
});
const rights = await getRights(opts.author_id);
rights.hasThrow("SEND_MESSAGES");
}
Expand Down
2 changes: 1 addition & 1 deletion src/gateway/opcodes/LazyRequest.ts
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ export async function onLazyRequest(this: WebSocket, { d }: Payload) {
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
// @ts-ignore
if (session?.status == "unknown") session.status = "online";
const user = (await User.getPublicUser(x)).toPublicUser(); // why is this needed?
const user = await User.getPublicUser(x);

return Send(this, {
op: OPCODES.Dispatch,
Expand Down
10 changes: 4 additions & 6 deletions src/util/entities/User.ts
Original file line number Diff line number Diff line change
Expand Up @@ -279,14 +279,12 @@ export class User extends BaseClass {
return user as UserPrivate;
}

static async getPublicUser(user_id: string, opts?: FindOneOptions<User>) {
return await User.findOneOrFail({
static async getPublicUser(user_id: string): Promise<PublicUser> {
const user = await User.findOneOrFail({
where: { id: user_id },
...opts,
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
//@ts-ignore
select: [...PublicUserProjection, ...(opts?.select || [])], // TODO: fix
select: PublicUserProjection,
});
return user.toPublicUser();
}

public static async generateDiscriminator(
Expand Down

0 comments on commit 269d4ee

Please sign in to comment.