User:Caorongjin/wordcount.js References

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.

This user script seems to have a documentation page at User:Caorongjin/wordcount.

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.


//<nowiki>

// This code is based on /info/en/?search=User:Dr_pda/prosesize.js

// but adds CJK support (http://stackoverflow.com/questions/2315488) and support

// for references and other lists.

//

function getWordCount(html) {

	var str = html.innerHTML.replace(/(<([^>]+)>)/ig,"").trim();



	var wordCount = 0;



	var arr = str.match(/[\u3040-\u309F]|[\u30A0-\u30FF]|[\u4E00-\u9FFF\uF900-\uFAFF\u3400-\u4DBF]|\S+/g);



	if (arr) {

		wordCount = arr.length;

	}



	return wordCount;

}



function getContentDiv() {

	var contentDiv;



	if (mw.config.get('wgAction') == 'edit') {

		contentDiv = document.getElementById('wikiPreview');

	}

	else {

		contentDiv = document.getElementById('mw-content-text');

	}



	return contentDiv;

}



function isValidListNode(node) {

	if (node.parentElement.id == "word-count-stats") {

		return false;

	}



	if (node.className.indexOf("toclevel-") > -1 || 

		node.parentElement.parentElement.id == "toc") {



		return false;

	}



	if (node.parentElement.parentElement.parentElement.className == "catlinks") {

		return false;

	}



	var bodyContent = getContentDiv();

	var curNode = node.parentElement.parentElement;



	while (curNode && (curNode != bodyContent)) {

		if (curNode.className.indexOf("infobox") > -1) {

			return false;

		}

		else if (curNode.className.indexOf("metadata") > -1) {

			return false;

		}

		else if (curNode.className.indexOf("navbox") > -1) {

			return false;

		}

		else {

			curNode = curNode.parentElement;

		}

	}



	return true;

}



function isValidParagraphNode(node) {

	if (node.parentNode.className.indexOf("mw-parser-output") > -1 ||

		node.parentNode.parentNode.className.indexOf("mw-parser-output") > -1 ||

		node.parentNode.nodeName == "BLOCKQUOTE") {



		return true;

	}

	else {

		return false;

	}

}



function isValidReferenceNode(node) {

	var bodyContent = getContentDiv();

	var curNode = node.parentElement;



	while (curNode && (curNode != bodyContent)) {

		if (curNode.classList.contains("references") ||

			curNode.classList.contains("reflist") ||

			curNode.classList.contains("refbegin")) {



			return true;

		}



		curNode = curNode.parentElement;

	}



	return false;

}



function toggleWordCount() {

	if (mw.config.get('wgAction') == 'edit') {

		var wikiPreview = document.getElementById('wikiPreview');



		var wikiPreviewStyle = window.getComputedStyle(wikiPreview);



		if (wikiPreviewStyle.display === 'none') {

			alert("You need to preview the text for the word count script to work in edit mode.");



			return;

		}

	}



	var bodyContent = getContentDiv();



	var output = document.getElementById("word-count-stats");



	if (output) {

		var oldStyle = output.className;



		var i = 0;



		// Cleanup background color

		var pList = bodyContent.getElementsByTagName("p");



		if (pList) {

			for (i=0; i < pList.length; i++){

				if (isValidParagraphNode(pListi])) {

					pListi].style.cssText = oldStyle;

				}

			}

		}



		var listTypes = "li", "dd"];

	

		for (var j = 0; j < listTypes.length; j++) {

			var liList = bodyContent.getElementsByTagName(listTypesj]);



			if (liList) {

				for (i=0; i < liList.length; i++) {

					liListi].style.cssText = oldStyle;

				}

			}

		}



		var hList = bodyContent.getElementsByClassName("mw-headline");



		if (hList) {

			for (i=0; i < hList.length; i++) {

				hListi].style.cssText = oldStyle;

			}

		}



		// Remove nodes

		output.parentNode.removeChild(output);



		var header = document.getElementById("word-count-header");



		header.parentNode.removeChild(header);

	}

	else {

		getStatistics(bodyContent);

	}

}



//

// Main counting function

//

function getStatistics(bodyContent) {



	// Statistics

	var output = document.createElement("ul");

	output.id = "word-count-stats";



	var main_body_value = document.createElement("li");

	main_body_value.id = "main-body-stat";

	output.appendChild(main_body_value);

	output.className = bodyContent.getElementsByTagName("p").item(0).style.cssText;



	var ref_value = document.createElement("li");

	ref_value.id = "ref-stat";

	output.appendChild(ref_value);



	var total_value = document.createElement("li");

	total_value.id = "total-stat";

	output.appendChild(total_value);



	bodyContent.insertBefore(output, bodyContent.firstChild);



	// Header

	var header = document.createElement("span");

	header.id = "word-count-header";

	header.innerHTML = "<br/><b>Word counts</b> (<a href='/info/en/?search=User:Caorongjin/wordcount'>doc</a>)<b>:</b>";

	bodyContent.insertBefore(header,output);



	// Create counters

	var main_body_count = 0;

	var ref_count = 0;



	var i = 0;



	// Count within paragraphs

	var pList = bodyContent.getElementsByTagName("p");



	if (pList) {

		for (i=0; i < pList.length; i++) {

			var para = pListi];



			if (isValidParagraphNode(para)) {

				var paraCount = getWordCount(para);



				if (paraCount > 0) {

					main_body_count += paraCount;

					para.style.cssText = "background-color:yellow";

				}

			}

		}

	}



	// Count within lists

	var listTypes = "li", "dd"];



	for (var j = 0; j < listTypes.length; j++) {

		var liList = bodyContent.getElementsByTagName(listTypesj]);



		if (liList) {

			for (i=0; i < liList.length; i++) {

				var li = liListi];



				if (isValidReferenceNode(li)) {

					ref_count += getWordCount(li);

					li.style.cssText = "background-color:cyan";

				}

				else if (isValidListNode(li)) {

					main_body_count += getWordCount(li);

					li.style.cssText = "background-color:yellow";

				}

			}

		}

	}



	// Count within headings

	var hList = bodyContent.getElementsByClassName("mw-headline");



	if (hList) {

		for (i=0; i < hList.length; i++) {

			var h = hListi];



			if (h.id == "Contents") {

				continue;

			}



			main_body_count += getWordCount(h);

			h.style.cssText = "background-color:yellow";

		}

	}



	main_body_value.innerHTML = "Main body: " + main_body_count + " words";

	ref_value.innerHTML = "References: " + ref_count + " words";

	total_value.innerHTML = "Total: " + (main_body_count + ref_count) + " words";

}



jQuery(function () {

	mw.loader.using( 'mediawiki.util'], function () {

		if($.inArray(mw.config.get('wgAction'), 'edit', 'view' , 'submit' , 'historysubmit' , 'purge']) !== -1) {

			$( mw.util.addPortletLink('p-tb', '#', 'Word count', 't-word-count', 'Calculate word count') )

			.click( toggleWordCount );

		}

	});

});



//</nowiki>