RocketChat/Rocket.Chat

import type { IOEmbedCache, RocketChatRecordDeleted } from '@rocket.chat/core-typings';
import type { IOEmbedCacheModel } from '@rocket.chat/model-typings';
import type { Collection, Db, DeleteResult, IndexDescription } from 'mongodb';

import { BaseRaw } from './BaseRaw';

export class OEmbedCacheRaw extends BaseRaw<IOEmbedCache> implements IOEmbedCacheModel {
    constructor(db: Db, trash?: Collection<RocketChatRecordDeleted<IOEmbedCache>>) {
        super(db, 'oembed_cache', trash);
    }

    protected modelIndexes(): IndexDescription[] {
        return [{ key: { updatedAt: 1 } }];
    }

    async createWithIdAndData(_id: string, data: any): Promise<IOEmbedCache> {
        const record = {
            _id,
            data,
            updatedAt: new Date(),
        };
        record._id = (await this.insertOne(record)).insertedId;
        return record;
    }

    removeBeforeDate(date: Date): Promise<DeleteResult> {
        const query = {
            updatedAt: {
                $lte: date,
            },
        };
        return this.deleteMany(query);
    }
}