MediaWiki:Common.js: Difference between revisions

Revision as of 20:46, 17 May 2026

/* ================================================================
   CSV Export Button — injects "⬇ Export CSV" only above wikitables
   that have a <caption> element (data tables with |+ in wikitext).
   These are the numeric/calculable tables (costs, times, power…).
   Descriptive tables (Building/Function, Strengths/Weaknesses…)
   have no caption and are left untouched.
   ================================================================ */
mw.hook('wikipage.content').add(function ($content) {
    var pageName = mw.config.get('wgTitle') || 'export';
    var safeTitle = pageName.replace(/ /g, '_');
    var exportIndex = 0;

    $content.find('table.wikitable').each(function () {
        var $table = $(this);

        // Only add button if the table has a <caption>
        if ($table.find('caption').length === 0) return;

        exportIndex++;
        var $btn = $('<button>')
            .text('⬇ Export CSV')
            .attr('title', 'Download this table as a CSV file')
            .css({
                cursor: 'pointer',
                background: '#1e1e1e',
                color: '#e07000',
                border: '1px solid #e07000',
                borderRadius: '4px',
                padding: '3px 12px',
                fontSize: '0.82em',
                fontWeight: 'bold',
                marginBottom: '4px',
                display: 'inline-block'
            })
            .on('click', function () {
                var rows = $table.find('tr');
                var csv = [];
                rows.each(function () {
                    var cells = $(this).find('th, td');
                    var line = [];
                    cells.each(function () {
                        var txt = $(this).text().replace(/\s+/g, ' ').trim();
                        if (txt.indexOf(',') !== -1 || txt.indexOf('"') !== -1) {
                            txt = '"' + txt.replace(/"/g, '""') + '"';
                        }
                        line.push(txt);
                    });
                    if (line.length > 0) csv.push(line.join(','));
                });

                var blob = new Blob([csv.join('\n')], { type: 'text/csv;charset=utf-8;' });
                var url = URL.createObjectURL(blob);
                var suffix = exportIndex > 1 ? '_' + exportIndex : '';
                var $a = $('<a>').attr('href', url).attr('download', safeTitle + suffix + '.csv').css('display','none');
                $('body').append($a);
                $a[0].click();
                $a.remove();
                setTimeout(function () { URL.revokeObjectURL(url); }, 1000);
            });

        $table.before($('<div>').css('margin', '0 0 2px 0').append($btn));
    });
});