From 558f24db40f2f28c6400c70d1b9e03287b1566f7 Mon Sep 17 00:00:00 2001
From: Pierre Boyer <pierre.s.boyer@gmail.com>
Date: Mon, 11 Apr 2016 11:29:53 +0200
Subject: [PATCH] Improve 'Edit Original' button, fix manifest / typos /
 formatting

---
 manifest.json  |  4 ++--
 src/content.js | 54 ++++++++++++++++++++++++--------------------------
 2 files changed, 28 insertions(+), 30 deletions(-)

diff --git a/manifest.json b/manifest.json
index 7062412..52d9630 100755
--- a/manifest.json
+++ b/manifest.json
@@ -1,7 +1,7 @@
 {
 	"name": "MDN Automatic Translation",
-	"version": "0.0.1",
-	"manifest_version": 2,
+	"version": "0.2.0",
+	"manifest_version": 3,
 	"description": "Adds an \"Auto-translate\" button on the MDN docs translation page, that replaces common patterns (such as section titles) automatically.",
 	"homepage_url": "https://github.com/npny/mdn-automatic-translation",
 	"applications": {
diff --git a/src/content.js b/src/content.js
index 0f6c6f8..f003c0b 100755
--- a/src/content.js
+++ b/src/content.js
@@ -8,11 +8,10 @@ function start() {
 	rules = JSON.parse(window.localStorage["rules"]);
 	domain = window.location.pathname.substring(6); // Substring 6 turns (developer.mozilla.org)/en-US/docs/* into simply /docs/*, for scoping
 
-	// The local is either the hidden input in the for (for a new article), or the
-	// language selector
+	// The locale can be determined from either a hidden input on the page or from the language selector
 	locale = document.querySelector("[name=tolocale]") ?
-		document.querySelector("[name=tolocale]").value :
-		document.querySelector("#language").value;
+	         document.querySelector("[name=tolocale]").value :
+	         document.querySelector("#language").value;
 
 	root = document.querySelector(".cke_wysiwyg_frame").contentWindow.document;
 
@@ -21,34 +20,30 @@ function start() {
 	addTagsArrows();
 }
 
+
 function addEditOriginalButton(){
-	// Original url
-	const currentUrl = document.location.href;
-	var redirectUrl;
-
-	// Editing a new translated document, after clicking on "add a tranlation",
-	// the url will reference the en-US page
-	if(currentUrl.split('/')[3] == 'en-US'){
-		// change the translate & following by edit
-		redirectUrl = currentUrl.split('$')[0] + '$edit';
-	}
-	// Editing an existing article along with the original one
- 	else {
-		// change the local by en-US
-		redirectUrl = currentUrl.replace(locale, 'en-US');
-	}
-
-	// Fetch the original article url and add a button to quickly go there
-	const button = document.createElement("button");
+
+	// Coming from either :
+	//  /en-US/    ... $translate?tolocale=<locale>
+	//  /<locale>/ ... $edit
+	//
+	// We want to end up with :
+	//  /en-US/    ... $edit
+
+	const components = document.location.href.replace(/\$translate.*/, "$edit").split("/");
+	components[3] = "en-US";
+	const href = components.join("/");
+
+	const button = document.createElement("a");
+	button.className = "button";
 	button.innerText = "Edit original";
-	button.addEventListener("click", (e) => {
-		e.preventDefault();
-		window.location.href = redirectUrl;
-	});
+	button.href = href;
 
 	document.querySelector(".translate-buttons").appendChild(button);
+
 }
 
+
 function addTranslateButton() {
 
 	const button = document.createElement("button");
@@ -123,6 +118,7 @@ function runTranslation() {
 
 }
 
+
 // This is a setInterval() loop that checks periodically for a condition,
 // and runs a callback once the condition is true, if the page is not excluded.
 function waitUntil(condition, exclude, interval, callback) {
@@ -146,8 +142,10 @@ function waitUntil(condition, exclude, interval, callback) {
 
 
 waitUntil(
-		() => document.querySelector(".cke_wysiwyg_frame") &&
-					document.querySelector(".cke_wysiwyg_frame").contentWindow.document.querySelector(".cke_editable"),
+		() =>
+			document.querySelector(".cke_wysiwyg_frame") &&
+			document.querySelector(".cke_wysiwyg_frame").contentWindow.document.querySelector(".cke_editable")
+		,
 		() => document.querySelector("body.edit"),
 		1000,
 		start