import "mdast-util-to-hast";
import "mdast-util-directive";

import { h } from "hastscript";
import { Root } from "mdast";
import { visit, SKIP } from "unist-util-visit";

/**
 * MDAST plugin to transform `ak-preview` directives into preview badges.
 */
function remarkPreviewDirective() {
    return function (tree: Root) {
        visit(tree, "textDirective", function (node) {
            if (node.name !== "ak-preview") return SKIP;

            const data = node.data || (node.data = {});

            const hast = h("span", {
                ...node.attributes,
                className: "badge badge--preview",
                title: `This feature is in preview and may change in the future.`,
                "aria-description": "Preview badge",
                role: "img",
            });

            data.hName = hast.tagName;
            data.hProperties = hast.properties;

            data.hChildren = [
                {
                    type: "text",
                    value: "Preview",
                },
            ];

            node.children = [];

            return SKIP;
        });
    };
}

export default remarkPreviewDirective;