@vector-im/matrix-bot-sdk/src/storage/MemoryStorageProvider.ts

TypeScript/JavaScript SDK for Matrix bots and appservices

import { IStorageProvider } from "./IStorageProvider";
import { IFilterInfo } from "../IFilter";
import { IAppserviceStorageProvider } from "./IAppserviceStorageProvider";

/**
 * A storage provider that persists no information by keeping it all in memory.
 * @category Storage providers
 */
export class MemoryStorageProvider implements IStorageProvider, IAppserviceStorageProvider {
    private namespaced = new Map<string, MemoryStorageProvider>();
    private syncToken: string;
    private filter: IFilterInfo;
    private appserviceUsers: { [userId: string]: { registered: boolean } } = {};
    private appserviceTransactions: { [txnId: string]: boolean } = {};
    private kvStore: { [key: string]: string } = {};

    setSyncToken(token: string | null): void {
        this.syncToken = token;
    }

    getSyncToken(): string | null {
        return this.syncToken;
    }

    setFilter(filter: IFilterInfo): void {
        this.filter = filter;
    }

    getFilter(): IFilterInfo {
        return this.filter;
    }

    addRegisteredUser(userId: string) {
        this.appserviceUsers[userId] = {
            registered: true,
        };
    }

    isUserRegistered(userId: string): boolean {
        return this.appserviceUsers[userId] && this.appserviceUsers[userId].registered;
    }

    isTransactionCompleted(transactionId: string): boolean {
        return !!this.appserviceTransactions[transactionId];
    }

    setTransactionCompleted(transactionId: string) {
        this.appserviceTransactions[transactionId] = true;
    }

    readValue(key: string): string | null | undefined {
        return this.kvStore[key];
    }

    storeValue(key: string, value: string): void {
        this.kvStore[key] = value;
    }

    storageForUser(userId: string): IStorageProvider {
        if (!this.namespaced.has(userId)) {
            this.namespaced.set(userId, new MemoryStorageProvider());
        }
        return this.namespaced.get(userId);
    }
}