Wužiwar:J budissin/shortdescs-in-category.js

/**
 * Shows button on category pages to see short description of all pages in the category.
 * 
 * Look for a button saying "Show SDs" towards the right, above the list of pages.
 * 
 */

$.ready.then(function() {

	if (mw.config.get('wgNamespaceNumber') !== 14) {
		return;
	}
	
	$('#mw-pages > p').append(
		$('<button>').text('Show SDs').attr('id', 'showSD-button').addClass('mw-ui-button').css('float', 'right').click(showsSDs)
	);

	if (location.href.indexOf('&showSD=y') !== -1) {
		showsSDs();
	}

});

function fixLinks(showingSDs) {
	if (showingSDs) {
		$('#mw-pages > a').each(function() {
			if (this.href.indexOf('#') !== -1) {
				this.href = this.href.replace(/#/, '&showSD=y#');
			} else {
				this.href += '&showSD=y';
			}
		});
	} else {
		$('#mw-pages > a').each(function() {
			this.href = this.href.replace(/&showSD=y/g, '');
		});
	}
}

function hideSDs(e) {
	if (e) e.preventDefault();
	
	fixLinks(false);
	$('#showSD-button').text('Show SDs').off('click').on('click', showsSDs);
	$('.cat-shortdesc-none, .cat-shortdesc-local, .cat-shortdesc-wikidata').remove();
	
}

function showsSDs(e) {
	if (e) e.preventDefault();

	fixLinks(true);
	$('#showSD-button').text('Hide SDs').off('click').on('click', hideSDs);	
	
	mw.util.addCSS('.cat-shortdesc-none { color: red } .cat-shortdesc-wikidata { color: MediumVioletRed }');

	var titles = $('#mw-pages li').get().map(function(e) {
		var a = e.querySelector('a');
		return mw.Title.newFromText(a.textContent).getSubjectPage().toText();
	});

	var titleSets = arrayChunk(titles, 50);
	var api = new mw.Api();
	var wdApi = new mw.ForeignApi('https://www.wikidata.org/w/api.php', { anonymous: true });

	var promiseArr = titleSets.map(function(set) {
		return api.get({
			"action": "query",
			"format": "json",
			"prop": "description",
			"titles": set,
			"formatversion": "2",
		});
	});
	var promiseArrWd = titleSets.map(function(set) {
		return wdApi.get({
			"action": "wbgetentities",
			"sites": "enwiki",
			"titles": set,
			"props": "descriptions|sitelinks",
			"languages": "en",
			"sitefilter": "enwiki"
		});
	});

	$.when.apply($, promiseArr.concat(promiseArrWd)).then(function() {
		var descs = {};
		Array.prototype.slice.call(arguments).forEach(function(response) {
			var json = response[0];
			if (json.query) { // enwiki API output
				json.query.pages.forEach(function(pg) {
					descs[pg.title] = {
						value: pg.description
					};
				});		
			} else if (json.entities) { // wikidata API output
				$.each(json.entities, function(id, data) {
					var pagename = data.sitelinks && data.sitelinks.enwiki && data.sitelinks.enwiki.title;
					if (pagename) {
						var entry = descs[pagename];
						if (entry && !entry.value) {
							entry.value = data.descriptions && data.descriptions.en && data.descriptions.en.value;
							entry.wd = true;
						} 
					}
				});
			}
		});	
		attachDescriptions(descs);
	});
}

function attachDescriptions(descs) {
	$('#mw-pages li').get().forEach(function(e) {
		var a = e.querySelector('a');
		var title = mw.Title.newFromText(a.textContent).getSubjectPage().toText();
		if (!descs[title] || !descs[title].value) {
			$(e).append(
				$('<span>').text(' – no shortdesc')
					.addClass('cat-shortdesc-none')
			);
		} else {
			$(e).append(
				$('<span>').text(' – ' + descs[title].value)
					.addClass(descs[title].wd ? 'cat-shortdesc-wikidata' : 'cat-shortdesc-local')
			);
		}
	});
}

function arrayChunk(arr, size) {
	var numChunks = Math.ceil(arr.length / size);
	var result = new Array(numChunks);
	for (var i = 0; i < numChunks; i++) {
		result[i] = arr.slice(i * size, (i + 1) * size);
	}
	return result;
}