Notes/src/public/app/widgets/containers/launcher.ts

import CalendarWidget from "../buttons/calendar.js";
import SpacerWidget from "../spacer.js";
import BookmarkButtons from "../bookmark_buttons.js";
import ProtectedSessionStatusWidget from "../buttons/protected_session_status.js";
import SyncStatusWidget from "../sync_status.js";
import BasicWidget from "../basic_widget.js";
import NoteLauncher from "../buttons/launcher/note_launcher.js";
import ScriptLauncher from "../buttons/launcher/script_launcher.js";
import CommandButtonWidget from "../buttons/command_button.js";
import utils from "../../services/utils.js";
import TodayLauncher from "../buttons/launcher/today_launcher.js";
import HistoryNavigationButton from "../buttons/history_navigation.js";
import QuickSearchLauncherWidget from "../quick_search_launcher.js";
import FNote from "../../entities/fnote.js";
import type { CommandNames } from "../../components/app_context.js";

interface InnerWidget extends BasicWidget {
    settings?: {
        titlePlacement: "bottom";
    };
}

export default class LauncherWidget extends BasicWidget {
    private innerWidget!: InnerWidget;
    private isHorizontalLayout: boolean;

    constructor(isHorizontalLayout: boolean) {
        super();

        this.isHorizontalLayout = isHorizontalLayout;
    }

    isEnabled() {
        return this.innerWidget.isEnabled();
    }

    doRender() {
        this.$widget = this.innerWidget.render();
    }

    async initLauncher(note: FNote) {
        if (note.type !== "launcher") {
            throw new Error(`Note '${note.noteId}' '${note.title}' is not a launcher even though it's in the launcher subtree`);
        }

        if (!utils.isDesktop() && note.isLabelTruthy("desktopOnly")) {
            return false;
        }

        const launcherType = note.getLabelValue("launcherType");

        if (glob.TRILIUM_SAFE_MODE && launcherType === "customWidget") {
            return false;
        }

        let widget: BasicWidget;
        if (launcherType === "command") {
            widget = this.initCommandLauncherWidget(note).class("launcher-button");
        } else if (launcherType === "note") {
            widget = new NoteLauncher(note).class("launcher-button");
        } else if (launcherType === "script") {
            widget = new ScriptLauncher(note).class("launcher-button");
        } else if (launcherType === "customWidget") {
            widget = await this.initCustomWidget(note);
        } else if (launcherType === "builtinWidget") {
            widget = this.initBuiltinWidget(note);
        } else {
            throw new Error(`Unrecognized launcher type '${launcherType}' for launcher '${note.noteId}' title '${note.title}'`);
        }

        if (!widget) {
            throw new Error(`Unknown initialization error for note '${note.noteId}', title '${note.title}'`);
        }

        this.child(widget);
        this.innerWidget = widget as InnerWidget;
        if (this.isHorizontalLayout && this.innerWidget.settings) {
            this.innerWidget.settings.titlePlacement = "bottom";
        }

        return true;
    }

    initCommandLauncherWidget(note: FNote) {
        return new CommandButtonWidget()
            .title(() => note.title)
            .icon(() => note.getIcon())
            .command(() => note.getLabelValue("command") as CommandNames);
    }

    async initCustomWidget(note: FNote) {
        const widget = await note.getRelationTarget("widget");

        if (widget) {
            return await widget.executeScript();
        } else {
            throw new Error(`Custom widget of launcher '${note.noteId}' '${note.title}' is not defined.`);
        }
    }

    initBuiltinWidget(note: FNote) {
        const builtinWidget = note.getLabelValue("builtinWidget");
        switch (builtinWidget) {
            case "calendar":
                return new CalendarWidget(note.title, note.getIcon());
            case "spacer":
                // || has to be inside since 0 is a valid value
                const baseSize = parseInt(note.getLabelValue("baseSize") || "40");
                const growthFactor = parseInt(note.getLabelValue("growthFactor") || "100");

                return new SpacerWidget(baseSize, growthFactor);
            case "bookmarks":
                return new BookmarkButtons(this.isHorizontalLayout);
            case "protectedSession":
                return new ProtectedSessionStatusWidget();
            case "syncStatus":
                return new SyncStatusWidget();
            case "backInHistoryButton":
                return new HistoryNavigationButton(note, "backInNoteHistory");
            case "forwardInHistoryButton":
                return new HistoryNavigationButton(note, "forwardInNoteHistory");
            case "todayInJournal":
                return new TodayLauncher(note);
            case "quickSearch":
                return new QuickSearchLauncherWidget(this.isHorizontalLayout);
            default:
                throw new Error(`Unrecognized builtin widget ${builtinWidget} for launcher ${note.noteId} "${note.title}"`);
        }
    }
}