MediaWiki:Common.js

mw.loader.using('ext.visualEditor.desktopArticleTarget.init').then(function () {
    mw.hook('ve.activationComplete').add(function () {

        // --- Define the command ---
        function doHighlight(surface) {
            var fragment = surface.getModel().getFragment();
            var selectedText = fragment.getText();
            if (!selectedText) return;

            // Replace selected text with the template call
            fragment.insertContent('{{Highlight|' + selectedText + '|yellow}}');
        }

        var commandName = 'insertHighlight';

        ve.ui.commandRegistry.register(
            new ve.ui.Command(
                commandName,
                'content',
                'exec',
                { args: [doHighlight] }
            )
        );

        // --- Define the tool ---
        function HighlightTool(toolGroup, config) {
            ve.ui.Tool.call(this, toolGroup, config);
        }
        OO.inheritClass(HighlightTool, ve.ui.Tool);

        HighlightTool.static.name = 'highlight';
        HighlightTool.static.group = 'insert';
        HighlightTool.static.icon = 'highlight'; // Built-in OOUI icon
        HighlightTool.static.title = 'Highlight text';
        HighlightTool.static.commandName = commandName;

        ve.ui.toolFactory.register(HighlightTool);

        // --- Add to toolbar ---
        ve.ui.targetToolbarGroups.content.push({
            name: 'highlightGroup',
            type: 'bar',
            include: [ 'highlight' ]
        });
    });
});