Trilium/apps/client/src/widgets/dialogs/include_note.tsx

import { useRef, useState } from "preact/compat";
import type { EventData } from "../../components/app_context";
import { closeActiveDialog, openDialog } from "../../services/dialog";
import { t } from "../../services/i18n";
import FormGroup from "../react/FormGroup";
import FormRadioGroup from "../react/FormRadioGroup";
import Modal from "../react/Modal";
import NoteAutocomplete from "../react/NoteAutocomplete";
import ReactBasicWidget from "../react/ReactBasicWidget";
import Button from "../react/Button";
import { Suggestion, triggerRecentNotes } from "../../services/note_autocomplete";
import tree from "../../services/tree";
import froca from "../../services/froca";
import EditableTextTypeWidget from "../type_widgets/editable_text";

interface IncludeNoteDialogProps {
    textTypeWidget?: EditableTextTypeWidget;
}

function IncludeNoteDialogComponent({ textTypeWidget }: IncludeNoteDialogProps) {
    const [suggestion, setSuggestion] = useState<Suggestion | null>(null);
    const [boxSize, setBoxSize] = useState("medium");
    const autoCompleteRef = useRef<HTMLInputElement>(null);

    return (textTypeWidget &&
        <Modal
            className="include-note-dialog"
            title={t("include_note.dialog_title")}
            size="lg"
            onShown={() => triggerRecentNotes(autoCompleteRef.current)}
            onSubmit={() => {
                if (!suggestion?.notePath) {
                    return;
                }

                closeActiveDialog();
                includeNote(suggestion.notePath, textTypeWidget);
            }}
            footer={
                <Button text={t("include_note.button_include")} keyboardShortcut="Enter" />
            }
        >
            <FormGroup label={t("include_note.label_note")}>
                <NoteAutocomplete
                    placeholder={t("include_note.placeholder_search")}
                    onChange={setSuggestion}
                    inputRef={autoCompleteRef}
                />
            </FormGroup>

            <FormGroup label={t("include_note.box_size_prompt")}>
                <FormRadioGroup name="include-note-box-size"
                    currentValue={boxSize} onChange={setBoxSize}
                    values={[
                        { label: t("include_note.box_size_small"), value: "small" },
                        { label: t("include_note.box_size_medium"), value: "medium" },
                        { label: t("include_note.box_size_full"), value: "full" },
                    ]}
                />
            </FormGroup>
        </Modal>
    )
}

export default class IncludeNoteDialog extends ReactBasicWidget {

    private props: IncludeNoteDialogProps = {};

    get component() {
        return <IncludeNoteDialogComponent {...this.props} />;
    }

    async showIncludeNoteDialogEvent({ textTypeWidget }: EventData<"showIncludeDialog">) {
        this.props = { textTypeWidget };
        this.doRender();
        openDialog(this.$widget);
    }

}

async function includeNote(notePath: string, textTypeWidget: EditableTextTypeWidget) {
    const noteId = tree.getNoteIdFromUrl(notePath);
    if (!noteId) {
        return;
    }
    const note = await froca.getNote(noteId);
    const boxSize = $("input[name='include-note-box-size']:checked").val() as string;

    if (["image", "canvas", "mermaid"].includes(note?.type ?? "")) {
        // there's no benefit to use insert note functionlity for images,
        // so we'll just add an IMG tag
        textTypeWidget.addImage(noteId);
    } else {
        textTypeWidget.addIncludeNote(noteId, boxSize);
    }
}