User:MusikAnimal/importWatchlist.js

From Wikipedia, the free encyclopedia
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.
/* 

INSTRUCTIONS:

Add the following line to your common.js:
 
  importScript('User:MusikAnimal/importWatchlist.js'); // Linkback: [[User:MusikAnimal/importWatchlist.js]]
 
Please keep the linkback comment so I can see how many people use this)

There will be a new box at the top of your watchlist. Enter the username of the account whose watchlist you wish to import.
You will also need their watchlist token, which can be found in their Preferences under the Watchlist tab. [[Special:Preferences#mw-prefsection-watchlist]]

*/

if($("#mw-watchlist-resetbutton").is(":visible")) {
	$("#mw-watchlist-resetbutton").after(
		"<form id='import_watchlist_form'><fieldset style='margin:20px 0px'>" +
			"<legend>Import watchlist</legend>" +
			"<div><label for='import_watchlist_username'>Source watchlist username:</label><input type='text' id='import_watchlist_username' /></div>" +
			"<div><label for='import_watchlist_token'>Source watchlist token:</label><input type='text' id='import_watchlist_token' />&nbsp;(can be found at <a href='https://en.wikipedia.org/wiki/Special:Preferences#mw-prefsection-watchlist'>Special:Preferences#mw-prefsection-watchlist</a>)</div>" +
			"<div id='import_watchlist_progress'></div>" +
			"<button id='import_watchlist_submit'>Import</button>" +
		"</fieldset></form>"
	);
	
	$("#import_watchlist_form").submit(function(e) {
		$("#import_watchlist_form").find("input,button").prop("disabled",true);
		$("#import_watchlist_progress").html("Fetching watchlist...")
		
		importWatchlistArray = [];
		// do this craziness until Promises are more well supported
		tempCount = 0;
		tempFn = function() {
			delete tempFn; // no dup calls
			$("#import_watchlist_progress").html(importWatchlistArray.length + " watches queued for import");
			for(var i=0; i<importWatchlistArray.length; i+=50) {
				importWatches(importWatchlistArray.slice(i,i+49), function(ret) {
					tempCount += 50;
					if(tempCount < importWatchlistArray.length) {
						$("#import_watchlist_progress").html("Working... " + tempCount + " of " + importWatchlistArray.length + " imported");
					} else {
						// done
						$("#import_watchlist_progress").html("<div style='font-weight:bold;color:#006400'>Complete! " + importWatchlistArray.length + " pages imported to watchlist :)</div>");
						delete tempCount; // why not
					}
					if(!ret) {
						$("#import_watchlist_progress").after("<div style='color:red'>Error while importing batch " + parseInt(i/50) + ". Not all pages may have imported.</div>")
					}
				});
			}
		}
		getWatchlist(null, tempFn);
	
		e.preventDefault();
	});
}

function importWatches(watches, fn) {
	data = {
		action : "watch",
		format : "json",
		titles : watches.join("|"),
		token : mw.user.tokens.get('watchToken')
	}
	$.ajax({
		type : "POST",
		dataType : "json",
		url : mw.util.wikiScript('api'),
		data : data,
		success : function(data) {
			return fn(true);
		},
		error : function(data) {
			return fn(false);
		}
	});
}

function getWatchlist(wrcontinue, fn) {
	$.getJSON(mw.util.wikiScript('api') +
		"?action=query&list=watchlistraw&wrowner="+$('#import_watchlist_username').val() +
		"&wrtoken="+$('#import_watchlist_token').val() +
		(wrcontinue ? "&wrcontinue="+wrcontinue : "") +
		"&wrlimit=500&rawcontinue=&format=json", null, function(data) {
			if(!data.watchlistraw) {
				// error
				$("#import_watchlist_progress").html("<div style='color:red'>Error fetching watchlist from user " + $("#import_watchlist_username").val() + ". Possible username/token mismatch.");
				$("#import_watchlist_form").find("input,button").prop("disabled",false);
				return false;
			}
			
			// first push to array
			for(var i=0; i<data.watchlistraw.length; i++) {
				importWatchlistArray.push(data.watchlistraw[i].title);
			}
			
			if(data.watchlistraw.length === 500) {
				// may still be more to export
				if(data['query-continue'] && data['query-continue'].watchlistraw && data['query-continue'].watchlistraw.wrcontinue) {
					getWatchlist(data['query-continue'].watchlistraw.wrcontinue,function() {
						if(data.watchlistraw.length < 500) {
							// we're at the end
							tempFn();
						}
					});
				}
			} else {
				tempFn();
			}
	});
}

function getAjax(url) {
  // Return a new promise.
  return new Promise(function(resolve, reject) {
    // Do the usual XHR stuff
    var req = new XMLHttpRequest();
    req.open('GET', url);

    req.onload = function() {
      // This is called even on 404 etc
      // so check the status
      if (req.status == 200) {
        // Resolve the promise with the response text
        resolve(req.response);
      }
      else {
        // Otherwise reject with the status text
        // which will hopefully be a meaningful error
        reject(Error(req.statusText));
      }
    };

    // Handle network errors
    req.onerror = function() {
      reject(Error("Network Error"));
    };

    // Make the request
    req.send();
  });
}