User:Polygnotus/Scripts/Surveys.js

Code that you insert on this page could contain malicious content capable of compromising your account. If you import a script from another page with "importScript", "mw.loader.load", "iusc", or "lusc", take note that this causes you to dynamically load a remote script, which could be changed by others. Editors are responsible for all edits and actions they perform, including by scripts. User scripts are not centrally supported and may malfunction or become inoperable due to software changes. A guide to help you find broken scripts is available. If you are unsure whether code you are adding to this page is safe, you can ask at the appropriate village pump.
This code will be executed when previewing this page.

Documentation for this user script can be added at User:Polygnotus/Scripts/Surveys.

Note: After saving, you have to bypass your browser's cache to see the changes. Google Chrome, Firefox, Microsoft Edge and Safari: Hold down the ⇧ Shift key and click the Reload toolbar button. For details and instructions about other browsers, see Wikipedia:Bypass your cache.

// Wikipedia QuickSurveys Tracker for common.js
// Fetches active surveys and posts them to your userpage

(function() {
    'use strict';
    
    // Configuration
    const QUICKSURVEYS_URL = 'https://en.wikipedia.org/w/load.php?modules=ext.quicksurveys.lib&debug=true';
    const MEDIAWICK_BASE_URL = 'https://en.wikipedia.org/wiki/MediaWiki:';
    
    // Function to extract survey data from the loaded module
    function extractSurveyData(moduleText) {
        try {
            // Look for the surveyData.json content in the module
            const jsonStart = moduleText.indexOf('"resources/ext.quicksurveys.lib/surveyData.json":');
            if (jsonStart === -1) {
                console.log('No survey data found in module');
                return [];
            }
            
            // Find the start of the array
            const arrayStart = moduleText.indexOf('[', jsonStart);
            if (arrayStart === -1) {
                console.log('No array found in survey data');
                return [];
            }
            
            // Find the matching closing bracket
            let bracketCount = 0;
            let arrayEnd = arrayStart;
            for (let i = arrayStart; i < moduleText.length; i++) {
                if (moduleText[i] === '[') bracketCount++;
                if (moduleText[i] === ']') bracketCount--;
                if (bracketCount === 0) {
                    arrayEnd = i + 1;
                    break;
                }
            }
            
            let jsonString = moduleText.substring(arrayStart, arrayEnd);
            console.log('Extracted JSON length:', jsonString.length);
            console.log('JSON preview:', jsonString.substring(0, 300) + '...');
            
            const surveyData = JSON.parse(jsonString);
            console.log('Found survey data:', surveyData);
            
            const activeSurveys = [];
            
            surveyData.forEach(survey => {
                if (survey.name && survey.coverage >= 0) { // Show all surveys (including 0% coverage)
                    const surveyInfo = {
                        name: survey.name,
                        coverage: survey.coverage,
                        type: survey.type || 'unknown',
                        links: []
                    };
                    
                    // Check for direct link property
                    if (survey.link) {
                        surveyInfo.links.push({
                            type: 'main',
                            key: survey.link,
                            url: MEDIAWICK_BASE_URL + formatLinkKey(survey.link)
                        });
                    }
                    
                    // Check questions for links
                    if (survey.questions && Array.isArray(survey.questions)) {
                        survey.questions.forEach(question => {
                            if (question.link) {
                                surveyInfo.links.push({
                                    type: 'question',
                                    key: question.link,
                                    url: MEDIAWICK_BASE_URL + formatLinkKey(question.link)
                                });
                            }
                        });
                    }
                    
                    activeSurveys.push(surveyInfo);
                }
            });
            
            return activeSurveys;
        } catch (error) {
            console.error('Error parsing survey data:', error);
            return [];
        }
    }
    
    // Format link key for MediaWiki URL (capitalize first letter of each word)
    function formatLinkKey(key) {
        return key.split('-').map(word => 
            word.charAt(0).toUpperCase() + word.slice(1)
        ).join('-');
    }
    
    // Generate wiki markup for the surveys
    function generateSurveyMarkup(surveys) {
        if (surveys.length === 0) {
            return '== Wikipedia QuickSurveys ==\n\nNo surveys found in the QuickSurveys module at this time.\n\n';
        }
        
        let markup = '== Wikipedia QuickSurveys ==\n\n';
        markup += `Last updated: ${new Date().toUTCString()}\n\n`;
        
        surveys.forEach(survey => {
            markup += `=== ${survey.name} ===\n`;
            markup += `* '''Type:''' ${survey.type}\n`;
            markup += `* '''Coverage:''' ${(survey.coverage * 100).toFixed(1)}%\n`;
            
            if (survey.links.length > 0) {
                markup += `* '''MediaWiki Links:'''\n`;
                survey.links.forEach(link => {
                    markup += `** [${link.url} ${link.key}] (${link.type})\n`;
                });
            }
            markup += '\n';
        });
        
        return markup;
    }
    
    // Update userpage with survey information
    function updateUserpage(content) {
        const username = mw.config.get('wgUserName');
        if (!username) {
            console.error('Not logged in - cannot update userpage');
            return;
        }
        
        const pageTitle = `User:${username}`;
        
        // Get current page content
        new mw.Api().get({
            action: 'query',
            prop: 'revisions',
            titles: pageTitle,
            rvprop: 'content',
            rvlimit: 1,
            formatversion: 2
        }).then(data => {
            const pages = data.query.pages;
            if (pages.length === 0) {
                console.error('Could not find userpage');
                return;
            }
            
            let currentContent = '';
            if (pages[0].revisions && pages[0].revisions[0]) {
                currentContent = pages[0].revisions[0].content;
            }
            
            // Remove existing survey section if it exists
            const sectionRegex = /== Wikipedia QuickSurveys ==[\s\S]*?(?=\n== |\n----|\n\[\[Category:|\n\{\{|\n<|\n#|$)/;
            let newContent = currentContent.replace(sectionRegex, '').trim();
            
            // Add new survey section
            if (newContent) {
                newContent += '\n\n' + content;
            } else {
                newContent = content;
            }
            
            // Save the page
            return new mw.Api().postWithToken('csrf', {
                action: 'edit',
                title: pageTitle,
                text: newContent,
                summary: 'Updated active QuickSurveys information via common.js script',
                minor: true
            });
        }).then(() => {
            console.log('Successfully updated userpage with survey information');
            mw.notify('QuickSurveys information updated on your userpage!', { type: 'success' });
        }).catch(error => {
            console.error('Error updating userpage:', error);
            mw.notify('Error updating userpage: ' + error.message, { type: 'error' });
        });
    }
    
    // Main function to fetch and process surveys
    function fetchAndProcessSurveys() {
        console.log('Fetching QuickSurveys data...');
        
        fetch(QUICKSURVEYS_URL)
            .then(response => response.text())
            .then(moduleText => {
                console.log('Successfully fetched module data');
                const surveys = extractSurveyData(moduleText);
                console.log('Extracted surveys:', surveys);
                
                const markup = generateSurveyMarkup(surveys);
                console.log('Generated markup:', markup);
                
                updateUserpage(markup);
            })
            .catch(error => {
                console.error('Error fetching survey data:', error);
                mw.notify('Error fetching QuickSurveys data: ' + error.message, { type: 'error' });
            });
    }
    
    // Add to window for manual execution
    window.updateQuickSurveys = fetchAndProcessSurveys;
    
    // Auto-run when on your userpage
    mw.hook('wikipage.content').add(function() {
        const currentPage = mw.config.get('wgPageName');
        const username = mw.config.get('wgUserName');
        
        if (username && currentPage === `User:${username}`) {
            // Only add link if it doesn't already exist
            if ($('#update-quicksurveys').length === 0) {
                // Add a small link to manually update
                const $link = $('<span style="font-size: 0.8em; margin-left: 1em;">').html(
                    '[<a href="#" id="update-quicksurveys">Update QuickSurveys</a>]'
                );
                
                $('#firstHeading').append($link);
                
                $('#update-quicksurveys').on('click', function(e) {
                    e.preventDefault();
                    fetchAndProcessSurveys();
                });
            }
        }
    });
    
    console.log('QuickSurveys tracker loaded. Use window.updateQuickSurveys() to manually update, or visit your userpage and click the update link.');
})();