refactor(formatLocation): upgrade to ES6

willmendesneto · willmendesneto · commit c826edde1be8 · 2017-04-17T22:00:47.000+10:00
diff --git a/lib/formatLocation.js b/lib/formatLocation.js
@@ -2,36 +2,53 @@
 	MIT License http://www.opensource.org/licenses/mit-license.php
 	Author Tobias Koppers @sokra
 */
-module.exports = function formatLocation(loc) {
-	if(typeof loc === "string")
-		return loc;
-	if(typeof loc === "number")
-		return loc + "";
-	if(loc && typeof loc === "object") {
-		if(loc.start && loc.end) {
-			if(typeof loc.start.line === "number" && typeof loc.end.line === "number" && typeof loc.end.column === "number" && loc.start.line === loc.end.line)
-				return formatPosition(loc.start) + "-" + loc.end.column;
-			return formatPosition(loc.start) + "-" + formatPosition(loc.end);
-		}
-		if(loc.start)
-			return formatPosition(loc.start);
-		return formatPosition(loc);
-	}
-	return "";
 
-	function formatPosition(pos) {
-		if(typeof pos === "string")
+"use strict";
+
+const formatPosition = (pos) => {
+	if(pos === null)
+		return "";
+	const typeOfPos = typeof pos;
+	switch(typeOfPos) {
+		case "string":
 			return pos;
-		if(typeof pos === "number")
-			return pos + "";
-		if(pos && typeof pos === "object") {
+		case "number":
+			return `${pos}`;
+		case "object":
 			if(typeof pos.line === "number" && typeof pos.column === "number")
-				return pos.line + ":" + pos.column;
-			if(typeof pos.line === "number")
-				return pos.line + ":?";
-			if(typeof pos.index === "number")
-				return "+" + pos.index;
-		}
+				return `${pos.line}:${pos.column}`;
+			else if(typeof pos.line === "number")
+				return `${pos.line}:?`;
+			else if(typeof pos.index === "number")
+				return `+${pos.index}`;
+			else
+				return "";
+		default:
+			return "";
+	}
+};
+
+const formatLocation = (loc) => {
+	if(loc === null)
 		return "";
+	const typeOfLoc = typeof loc;
+	switch(typeOfLoc) {
+		case "string":
+			return loc;
+		case "number":
+			return `${loc}`;
+		case "object":
+			if(loc.start && loc.end) {
+				if(typeof loc.start.line === "number" && typeof loc.end.line === "number" && typeof loc.end.column === "number" && loc.start.line === loc.end.line)
+					return `${formatPosition(loc.start)}-${loc.end.column}`;
+				return `${formatPosition(loc.start)}-${formatPosition(loc.end)}`;
+			}
+			if(loc.start)
+				return formatPosition(loc.start);
+			return formatPosition(loc);
+		default:
+			return "";
 	}
 };
+
+module.exports = formatLocation;
