MediaWiki:Common.js

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

        console.log('VisualEditor activated, adding Highlight tool...');

        // --- Command ---
        function doHighlight(surface) {
            var fragment = surface.getModel().getFragment();
            var selectedText = fragment.getText();
            if (!selectedText) {
                return;
            }
            fragment.insertContent('{{Highlight|' + selectedText + '|yellow}}');
        }

        var commandName = 'insertHighlight';
        if (!ve.ui.commandRegistry.lookup(commandName)) {
            ve.ui.commandRegistry.register(
                new ve.ui.Command(
                    commandName,
                    'content',
                    'exec',
                    { args: [doHighlight] }
                )
            );
        }

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

        HighlightTool.static.name = 'highlight';
        HighlightTool.static.icon = 'highlight';
        HighlightTool.static.title = 'Highlight text';
        HighlightTool.static.commandName = commandName;

        if (!ve.ui.toolFactory.lookup('highlight')) {
            ve.ui.toolFactory.register(HighlightTool);
            console.log('Highlight tool registered');
        }

        // --- Force add group to toolbar ---
        var toolbar = target.getToolbar();
        if (toolbar) {
            console.log('Adding Highlight button to toolbar...');
            toolbar.setup([
                {
                    include: [ 'bold', 'italic', 'highlight' ] // put it next to Bold/Italic
                }
            ]);
            toolbar.emit('updateState'); // refresh toolbar state
        }
    });
});