RocketChat/Rocket.Chat

View on GitHub
apps/meteor/server/methods/resetAvatar.ts

Summary

Maintainability
A
2 hrs
Test Coverage
import { api } from '@rocket.chat/core-services';
import type { IUser } from '@rocket.chat/core-typings';
import type { ServerMethods } from '@rocket.chat/ddp-client';
import { Users } from '@rocket.chat/models';
import { DDPRateLimiter } from 'meteor/ddp-rate-limiter';
import { Meteor } from 'meteor/meteor';

import { hasPermissionAsync } from '../../app/authorization/server/functions/hasPermission';
import { FileUpload } from '../../app/file-upload/server';
import { settings } from '../../app/settings/server';

declare module '@rocket.chat/ddp-client' {
    // eslint-disable-next-line @typescript-eslint/naming-convention
    interface ServerMethods {
        resetAvatar(userId: IUser['_id']): void;
    }
}

Meteor.methods<ServerMethods>({
    async resetAvatar(userId) {
        const uid = Meteor.userId();
        if (!uid) {
            throw new Meteor.Error('error-invalid-user', 'Invalid user', {
                method: 'resetAvatar',
            });
        }
        const canEditOtherUserAvatar = await hasPermissionAsync(uid, 'edit-other-user-avatar');

        if (!settings.get('Accounts_AllowUserAvatarChange') && !canEditOtherUserAvatar) {
            throw new Meteor.Error('error-not-allowed', 'Not allowed', {
                method: 'resetAvatar',
            });
        }

        let user;

        if (userId && userId !== uid) {
            if (!canEditOtherUserAvatar) {
                throw new Meteor.Error('error-unauthorized', 'Unauthorized', {
                    method: 'resetAvatar',
                });
            }

            user = await Users.findOneById(userId, { projection: { _id: 1, username: 1 } });
        } else {
            user = await Meteor.userAsync();
        }

        if (!user?.username) {
            throw new Meteor.Error('error-invalid-desired-user', 'Invalid desired user', {
                method: 'resetAvatar',
            });
        }

        await FileUpload.getStore('Avatars').deleteByName(user.username);
        await Users.unsetAvatarData(user._id);
        void api.broadcast('user.avatarUpdate', { username: user.username, avatarETag: undefined });
    },
});

DDPRateLimiter.addRule(
    {
        type: 'method',
        name: 'resetAvatar',
        userId() {
            return true;
        },
    },
    1,
    60000,
);