obsidian-livesync/src/modules/essentialObsidian/ModuleCheckRemoteSize.ts

import { LOG_LEVEL_INFO, LOG_LEVEL_NOTICE, LOG_LEVEL_VERBOSE } from "octagonal-wheels/common/logger";
import { sizeToHumanReadable } from "octagonal-wheels/number";
import { $msg } from "src/lib/src/common/i18n.ts";
import type { LiveSyncCore } from "../../main.ts";
import { AbstractObsidianModule } from "../AbstractObsidianModule.ts";
import { EVENT_REQUEST_CHECK_REMOTE_SIZE, eventHub } from "@/common/events.ts";

export class ModuleCheckRemoteSize extends AbstractObsidianModule {
    checkRemoteSize(): Promise<boolean> {
        this.settings.notifyThresholdOfRemoteStorageSize = 1;
        return this._allScanStat();
    }

    private async _allScanStat(): Promise<boolean> {
        if (this.core.managers.networkManager.isOnline === false) {
            this._log("Network is offline, skipping remote size check.", LOG_LEVEL_INFO);
            return true;
        }
        this._log($msg("moduleCheckRemoteSize.logCheckingStorageSizes"), LOG_LEVEL_VERBOSE);
        if (this.settings.notifyThresholdOfRemoteStorageSize < 0) {
            const message = $msg("moduleCheckRemoteSize.msgSetDBCapacity");
            const ANSWER_0 = $msg("moduleCheckRemoteSize.optionNoWarn");
            const ANSWER_800 = $msg("moduleCheckRemoteSize.option800MB");
            const ANSWER_2000 = $msg("moduleCheckRemoteSize.option2GB");
            const ASK_ME_NEXT_TIME = $msg("moduleCheckRemoteSize.optionAskMeLater");

            const ret = await this.core.confirm.askSelectStringDialogue(
                message,
                [ANSWER_0, ANSWER_800, ANSWER_2000, ASK_ME_NEXT_TIME],
                {
                    defaultAction: ASK_ME_NEXT_TIME,
                    title: $msg("moduleCheckRemoteSize.titleDatabaseSizeNotify"),
                    timeout: 40,
                }
            );
            if (ret == ANSWER_0) {
                this.settings.notifyThresholdOfRemoteStorageSize = 0;
                await this.core.saveSettings();
            } else if (ret == ANSWER_800) {
                this.settings.notifyThresholdOfRemoteStorageSize = 800;
                await this.core.saveSettings();
            } else if (ret == ANSWER_2000) {
                this.settings.notifyThresholdOfRemoteStorageSize = 2000;
                await this.core.saveSettings();
            }
        }
        if (this.settings.notifyThresholdOfRemoteStorageSize > 0) {
            const remoteStat = await this.core.replicator?.getRemoteStatus(this.settings);
            if (remoteStat) {
                const estimatedSize = remoteStat.estimatedSize;
                if (estimatedSize) {
                    const maxSize = this.settings.notifyThresholdOfRemoteStorageSize * 1024 * 1024;
                    if (estimatedSize > maxSize) {
                        const message = $msg("moduleCheckRemoteSize.msgDatabaseGrowing", {
                            estimatedSize: sizeToHumanReadable(estimatedSize),
                            maxSize: sizeToHumanReadable(maxSize),
                        });
                        const newMax = ~~(estimatedSize / 1024 / 1024) + 100;
                        const ANSWER_ENLARGE_LIMIT = $msg("moduleCheckRemoteSize.optionIncreaseLimit", {
                            newMax: newMax.toString(),
                        });
                        const ANSWER_REBUILD = $msg("moduleCheckRemoteSize.optionRebuildAll");
                        const ANSWER_IGNORE = $msg("moduleCheckRemoteSize.optionDismiss");
                        const ret = await this.core.confirm.askSelectStringDialogue(
                            message,
                            [ANSWER_ENLARGE_LIMIT, ANSWER_REBUILD, ANSWER_IGNORE],
                            {
                                defaultAction: ANSWER_IGNORE,
                                title: $msg("moduleCheckRemoteSize.titleDatabaseSizeLimitExceeded"),
                                timeout: 60,
                            }
                        );
                        if (ret == ANSWER_REBUILD) {
                            const ret = await this.core.confirm.askYesNoDialog(
                                $msg("moduleCheckRemoteSize.msgConfirmRebuild"),
                                { defaultOption: "No" }
                            );
                            if (ret == "yes") {
                                this.core.settings.notifyThresholdOfRemoteStorageSize = -1;
                                await this.saveSettings();
                                await this.core.rebuilder.scheduleRebuild();
                            }
                        } else if (ret == ANSWER_ENLARGE_LIMIT) {
                            this.settings.notifyThresholdOfRemoteStorageSize = ~~(estimatedSize / 1024 / 1024) + 100;
                            this._log(
                                $msg("moduleCheckRemoteSize.logThresholdEnlarged", {
                                    size: this.settings.notifyThresholdOfRemoteStorageSize.toString(),
                                }),
                                LOG_LEVEL_NOTICE
                            );
                            await this.core.saveSettings();
                        } else {
                            // Dismiss or Close the dialog
                        }

                        this._log(
                            $msg("moduleCheckRemoteSize.logExceededWarning", {
                                measuredSize: sizeToHumanReadable(estimatedSize),
                                notifySize: sizeToHumanReadable(
                                    this.settings.notifyThresholdOfRemoteStorageSize * 1024 * 1024
                                ),
                            }),
                            LOG_LEVEL_INFO
                        );
                    } else {
                        this._log(
                            $msg("moduleCheckRemoteSize.logCurrentStorageSize", {
                                measuredSize: sizeToHumanReadable(estimatedSize),
                            }),
                            LOG_LEVEL_INFO
                        );
                    }
                }
            }
        }
        return true;
    }

    private _everyOnloadStart(): Promise<boolean> {
        this.addCommand({
            id: "livesync-reset-remote-size-threshold-and-check",
            name: "Reset notification threshold and check the remote database usage",
            callback: async () => {
                await this.checkRemoteSize();
            },
        });
        eventHub.onEvent(EVENT_REQUEST_CHECK_REMOTE_SIZE, () => this.checkRemoteSize());
        return Promise.resolve(true);
    }
    onBindFunction(core: LiveSyncCore, services: typeof core.services): void {
        services.appLifecycle.onScanningStartupIssues.addHandler(this._allScanStat.bind(this));
        services.appLifecycle.onInitialise.addHandler(this._everyOnloadStart.bind(this));
    }
}