User:Caorongjin/wordcount.js
//
// This code is based on https://en.wikipedia.org/wiki/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(pList[i])) {
pList[i].style.cssText = oldStyle;
}
}
}
var listTypes = ["li", "dd"];
for (var j = 0; j < listTypes.length; j++) {
var liList = bodyContent.getElementsByTagName(listTypes[j]);
if (liList) {
for (i=0; i < liList.length; i++) {
liList[i].style.cssText = oldStyle;
}
}
}
var hList = bodyContent.getElementsByClassName("mw-headline");
if (hList) {
for (i=0; i < hList.length; i++) {
hList[i].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 = "
Word counts (doc):";
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 = pList[i];
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(listTypes[j]);
if (liList) {
for (i=0; i < liList.length; i++) {
var li = liList[i];
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 = hList[i];
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 );
}
});
});
//