trilium/src/services/spaced_update.ts

type Updater = () => void;

class SpacedUpdate {

    private updater: Updater;
    private lastUpdated: number;
    private changed: boolean;
    private updateInterval: number;
    private changeForbidden: boolean;

    constructor(updater: Updater, updateInterval = 1000) {
        this.updater = updater;
        this.lastUpdated = Date.now();
        this.changed = false;
        this.updateInterval = updateInterval;
        this.changeForbidden = false;
    }

    scheduleUpdate() {
        if (!this.changeForbidden) {
            this.changed = true;
            setTimeout(() => this.triggerUpdate());
        }
    }

    async updateNowIfNecessary() {
        if (this.changed) {
            this.changed = false; // optimistic...

            try {
                await this.updater();
            }
            catch (e) {
                this.changed = true;

                throw e;
            }
        }
    }

    isAllSavedAndTriggerUpdate() {
        const allSaved = !this.changed;

        this.updateNowIfNecessary();

        return allSaved;
    }

    triggerUpdate() {
        if (!this.changed) {
            return;
        }

        if (Date.now() - this.lastUpdated > this.updateInterval) {
            this.updater();
            this.lastUpdated = Date.now();
            this.changed = false;
        }
        else {
            // update isn't triggered but changes are still pending, so we need to schedule another check
            this.scheduleUpdate();
        }
    }

    async allowUpdateWithoutChange(callback: () => void) {
        this.changeForbidden = true;

        try {
            await callback();
        }
        finally {
            this.changeForbidden = false;
        }
    }
}

export default SpacedUpdate;