User:Ahecht/Scripts/flag.js
//
//jshint maxerr:256
var api = new mw.Api();
var titlePrefix = "Module:Sandbox/Ahecht/flag/";
var statusText = '';
function apiGet(titles) {
var query = {
action: "query",
prop: "revisions",
rvprop: "content",
titles: titles
};
api.get( query )
.done (function (data) {
if (typeof data !== 'undefined' && data.query && data.query.pages) {
page = Object.entries(data.query.pages)[0][1];
if (typeof page.revisions !== 'undefined') {
return processTemplate(page.revisions[0]['*']);
} else {
statusText = "Error: Revisions not found!";
showStatus(statusText);
return statusText;
}
}
statusText = "Error: Data not found!";
showStatus(statusText);
return statusText;
} ).fail (function (err) {
statusText = "Error: API error:" + err;
showStatus(statusText);
return statusText;
} );
return;
}
function processTemplate(content) {
var output = {};
if(typeof content !== 'undefined' && /^\{\{\s*(?:safesubst
//Valid data template
var matches = content.replace(/
/(?<=\n)[ \t]*\|[ \t]*([^=]*?)[ \t]*=[ \t]*([^\n]*?)[ \t]*(?=\n)/g
);
for (var match of matches) {
// Don't store passthrough values
if (match[2] != "{{{" + match[1] + "|}}}") {
output[match[1]] = match[2];
}
}
}
if (typeof output.alias !== 'undefined') {
return generateLua(output, 1);
} else {
statusText = "Error processing template!";
showStatus( statusText );
return statusText;
}
return;
}
function generateLua(args, depth) {
var params = {};
var k, resolvedkey, ifeq;
//Find subparameters
for (var key in args) {
//Resolve #ifeq statement in key
ifeq = key.match(/\{\{\s*#ifeq:\s*\{\{\{altlink\}\}\}\s*\|(.*?)\|(.*?)\|.*?\}\}/);
if (ifeq !== null && ifeq[1] !== null && ifeq[2] !== null) {
resolvedkey = ifeq[2].trim() + '-' + ifeq[1].trim().split(' ')[0];
} else {
resolvedkey = key;
}
//Split by hyphens
if ( !/[{}]/.test(resolvedkey) && /[-]/.test(resolvedkey) ) {
k = resolvedkey.split("-");
} else {
k = [resolvedkey];
}
//Create subkey for prefix if it doesn't already exist
if (typeof params[k[0]] === 'undefined') {
params[k[0]] = {};
}
var value = args[key];
//Assign either suffix subarray or #default
var subkey = "#default";
if (typeof k[1] !== 'undefined') {
subkey = k.slice(1).join("-");
}
//Check if value has a switch or #ifeq statement and store
if (typeof value === 'string') {
if ( /^\{\{\s*(?:safesubst
params[k[0]][subkey] = doSwitch(value);
} else {
var parseIfeq = value.match(/\{\{\s*(?:safesubst
if (parseIfeq !== null && parseIfeq[1] !== null && parseIfeq[2] !== null) {
params[k[0]][subkey] = {};
params[k[0]][subkey][parseIfeq[1]] = parseIfeq[2];
} else {
params[k[0]][subkey] = value;
}
}
} else if(typeof value === 'undefined') {
params[k[0]][subkey] = "false";
} else {
params[k[0]][subkey] = value;
}
}
//Move parameters up a level if only #default exists
for (key in params) {
var count = 0;
for (k in params[key]) { count ++; }
if ( (count == 1) && (typeof params[key]['#default'] !== 'undefined') ) {
params[key] = params[key]['#default'];
}
}
var output = [];
//Generate lua output
for (key in params) {
k = '\t'.repeat(depth+1) + '["' + key + '"] = ';
if (typeof(params[key]) === 'string') {
var outString = k;
// Set blank to false
if (params[key] === '') {
outString += "false";
} else {
outString += '"' + params[key] + '"';
}
// Put #default at start of outString
if (key == "#default") {
output.unshift(outString);
} else {
output.push(outString);
}
} else if (typeof(params[key]) === 'object') {
//Remove redundant levels
for (var sub in params[key]) {
if (sub == '#default' && typeof(params[key][sub]) === 'object') {
for (var subsub in params[key][sub]) {
params[key][subsub] = params[key][sub][subsub];
delete params[key][sub][subsub];
}
}
}
for (sub in params[key]) {
if (sub.search(key + '-') === 0) {
params[key][sub.replace(key + '-','')] = params[key][sub];
delete params[key][sub];
} else if (sub == key) {
params[key]['#default'] = params[key][sub];
delete params[key][sub];
}
}
var subout = generateLua(params[key], depth+1);
output.push(k + '{\n' + subout.join(",\n") + '\n' + '\t'.repeat(depth+1) + '}');
}
}
if (depth != 1) {
return output;
} else {
var outtext = 'return {\n\t["' + args.alias + '"] = {\n' + output.join(',\n') + '\n\t},\n}';
showStatus(outtext);
return apiPost(args.alias, outtext);
}
return;
}
function doSwitch(value) {
var trimmedValue = value.replace(/^\{\{\s*(?:safesubst
var matches = trimmedValue.match(/(?<=\|)(.*?)(?=\|)/g);
var valueObj = {};
for (i=0;i var mi = matches[i].split("="); for (j=i;j var mj = matches[j].split("="); if ( typeof mj[1] !== 'undefined' ) { valueObj[mi[0]] = mj[1]; break; } else if ( j == (matches.length - 1) ) { if (i == j) { valueObj['#default'] = mj[0]; } else { valueObj[mi[0]] = mj[0]; } } } } return valueObj; } function apiPost(alias, text) { var params = { action: "edit" }; var type; if (text == 'documentation') { type = text; params.title = titlePrefix + alias.replace(/\s/, "_") + '/doc'; params.summary = "Creating documentation for " + titlePrefix + alias + ""; params.text = "{{:" + titlePrefix + "doc|Country data " + alias + "}}"; } else { type = 'module'; params.title = titlePrefix + alias.replace(/\s/, "_"); params.summary = "Extracting data from Template:Country data " + alias + ""; params.text = text; } api.postWithEditToken( params ).done( function ( data ) { if ( typeof data !== 'undefined' && data ) { if (data.edit && data.edit.result && data.edit.result === "Success" ) { statusText = "Edit saved successfully to https:" + mw.config.values.wgServer + mw.config.values.wgArticlePath.replace("$1", data.edit.title.replace(/\s/, "_")); } else { statusText = "Couldn't save " + type + " due to data error: " + JSON.stringify( data ); } } else { statusText = "Couldn't save " + type + " due to data error."; } showStatus(statusText); if (text != 'documentation') { apiPost( alias, 'documentation' ); } } ).fail( function ( err ) { statusText = "Couldn't save " + type + " due to post error: " + err; showStatus(statusText); } ); return; } function showStatus(statusText) { console.log(statusText); return; } if ( mw.config.values.wgPageName.search("Template:Country_data_") === 0 ) { apiGet(mw.config.values.wgPageName); } else if ( mw.config.values.wgPageName.search(titlePrefix) === 0 ) { apiGet("Template:Country_data_" + mw.config.values.wgPageName.split('/').pop()); } else { showStatus("Script must be run from Country Data template or Country-specific module"); } //