@rocket.chat/forked-matrix-bot-sdk/src/e2ee/decorators.ts

TypeScript/JavaScript SDK for Matrix bots and appservices

import { MatrixClient } from "../MatrixClient";
import { CryptoClient } from "./CryptoClient";

/**
 * Flags a MatrixClient function as needing end-to-end encryption enabled.
 * @category Encryption
 */
export function requiresCrypto() {
    return function (target: any, propertyKey: string, descriptor: PropertyDescriptor) {
        const originalMethod = descriptor.value;
        descriptor.value = function (...args: any[]) {
            const client: MatrixClient = this;
            if (!client.crypto) {
                throw new Error("End-to-end encryption is not enabled");
            }

            return originalMethod.apply(this, args);
        };
    };
}

/**
 * Flags a CryptoClient function as needing the CryptoClient to be ready.
 * @category Encryption
 */
export function requiresReady() {
    return function (target: any, propertyKey: string, descriptor: PropertyDescriptor) {
        const originalMethod = descriptor.value;
        descriptor.value = function (...args: any[]) {
            const crypto: CryptoClient = this;
            if (!crypto.isReady) {
                throw new Error("End-to-end encryption has not initialized");
            }

            return originalMethod.apply(this, args);
        };
    };
}