RocketChat/Rocket.Chat

import type { IInvite, RocketChatRecordDeleted } from '@rocket.chat/core-typings';
import type { IInvitesModel } from '@rocket.chat/model-typings';
import type { Collection, Db, UpdateResult } from 'mongodb';

import { BaseRaw } from './BaseRaw';

export class InvitesRaw extends BaseRaw<IInvite> implements IInvitesModel {
    constructor(db: Db, trash?: Collection<RocketChatRecordDeleted<IInvite>>) {
        super(db, 'invites', trash);
    }

    findOneByUserRoomMaxUsesAndExpiration(userId: string, rid: string, maxUses: number, daysToExpire: number): Promise<IInvite | null> {
        return this.findOne({
            rid,
            userId,
            days: daysToExpire,
            maxUses,
            ...(daysToExpire > 0 ? { expires: { $gt: new Date() } } : {}),
            ...(maxUses > 0 ? { uses: { $lt: maxUses } } : {}),
        });
    }

    increaseUsageById(_id: string, uses = 1): Promise<UpdateResult> {
        return this.updateOne(
            { _id },
            {
                $inc: {
                    uses,
                },
            },
        );
    }

    async countUses(): Promise<number> {
        const [result] = await this.col.aggregate<{ totalUses: number }>([{ $group: { _id: null, totalUses: { $sum: '$uses' } } }]).toArray();

        return result?.totalUses || 0;
    }
}