trilium/apps/client/src/widgets/collections/view_mode_storage.ts

import type FNote from "../../entities/fnote";
import server from "../../services/server";
import { ViewTypeOptions } from "../collections/interface";

const ATTACHMENT_ROLE = "viewConfig";

export type ViewModeStorageType = ViewTypeOptions | "pdfHistory";

export default class ViewModeStorage<T extends object> {

    private note: FNote;
    private attachmentName: string;

    constructor(note: FNote, viewType: ViewModeStorageType) {
        this.note = note;
        this.attachmentName = `${viewType}.json`;
    }

    async store(data: T) {
        const payload = {
            role: ATTACHMENT_ROLE,
            title: this.attachmentName,
            mime: "application/json",
            content: JSON.stringify(data),
            position: 0
        };
        await server.post(`notes/${this.note.noteId}/attachments?matchBy=title`, payload);
    }

    async restore() {
        const existingAttachments = (await this.note.getAttachmentsByRole(ATTACHMENT_ROLE))
            .filter(a => a.title === this.attachmentName);
        if (existingAttachments.length === 0) {
            return undefined;
        }

        if (existingAttachments.length > 1) {
            // Clean up duplicates.
            await Promise.all(existingAttachments.slice(1).map(async a => await server.remove(`attachments/${a.attachmentId}`)));
        }

        const attachment = existingAttachments[0];
        const attachmentData = await server.get<{ content: string } | null>(`attachments/${attachment.attachmentId}/blob`);
        return JSON.parse(attachmentData?.content ?? "{}") as T;
    }
}