/**
 * @file /src/utils/clipboard.ts
 * @name Clipboard
 * @description Utility functions for clipboard operations.
 */

import { DOMSerializer, Fragment, Node as PMNode, Schema } from "@tiptap/pm/model";
import { Plugin, PluginKey } from "@tiptap/pm/state";

/**
 * Constructs a clipboard serialiser that serialises only the children of the nodes of the specified schema.
 *
 * @param schema - The schema to use for serialisation.
 * @returns {DOMSerializer} The constructed clipboard serialiser.
 */
const constructChildOnlyClipboardSerialiser = (schema: Schema, isNode: (node: PMNode) => boolean): DOMSerializer => {
    // Extend DOMSerializer to override serializeFragment
    const clipboardSerialiser = Object.create(DOMSerializer.fromSchema(schema));

    // Override serializeFragment
    clipboardSerialiser.serializeFragment = (fragment: Fragment, options = {}, target = document.createDocumentFragment()) => {
        const serializer = DOMSerializer.fromSchema(schema);

        fragment.forEach((node) => {
            if (isNode(node)) {
                // Serialize only the children of the page node
                serializer.serializeFragment(node.content, options, target);
            } else {
                // Serialize non-page nodes directly
                serializer.serializeNode(node, options);
            }
        });

        return target;
    };

    return clipboardSerialiser;
};

/**
 * Constructs a clipboard plugin with the specified key and serialiser.
 *
 * @param name - The plugin key.
 * @param serialiser - The serialiser to use for clipboard operations.
 * @returns {Plugin} The constructed clipboard plugin.
 */
const constructClipboardPlugin = (name: string, serialiser: DOMSerializer): Plugin => {
    return new Plugin({
        key: new PluginKey(name),
        props: {
            clipboardSerializer: serialiser,
        },
    });
};

/**
 * Constructs a clipboard plugin that serialises only the children of the nodes of the specified schema.
 *
 * @param name - The plugin key.
 * @param schema - The schema to use for serialisation.
 * @param isNode - A function that returns true if the node is a page node.
 * @returns {Plugin} The constructed clipboard plugin.
 */
export const constructChildOnlyClipboardPlugin = (name: string, schema: Schema, isNode: (node: PMNode) => boolean): Plugin => {
    const clipboardSerialiser = constructChildOnlyClipboardSerialiser(schema, isNode);
    return constructClipboardPlugin(name, clipboardSerialiser);
};