1 files changed, 51 insertions, 24 deletions
diff --git a/src/lang.js b/src/lang.js
index b3cbb43..f2fab3a 100644
--- a/src/lang.js
+++ b/src/lang.js
@@ -1,35 +1,62 @@
 const fs = require("fs");
 
-var lang = "en"; // Default language
-
-// Loads fallback/default language strings
-var langDef = JSON.parse(fs.readFileSync(__dirname + `/lang/en.json`, "utf8"));
-
-// If settins are set it'll try to set the language to that instead of
-// the default, however if it can't find it, it'll still fallback to the
-// default language. This might happen as the default language is
-// retrieved from the renderer's navigator.language, which may have
-// languages we don't support yet.
-if (fs.existsSync("viper.json")) {
-	lang = JSON.parse(fs.readFileSync("viper.json", "utf8")).lang;
-	if (! lang) {lang = "en"} // Uses fallback, if language isn't set
-	if (! fs.existsSync(__dirname + `/lang/${lang}.json`)) {
-		if (fs.existsSync(__dirname + `/lang/${lang.replace(/-.*$/, "")}.json`)) {
-			lang = lang.replace(/-.*$/, "");
-		} else {
-			lang = "en"; // Uses fallback if language doesn't exist
+const enLang = JSON.parse(fs.readFileSync(__dirname + `/lang/en.json`, "utf8"));
+let lang = "";
+var langObj = {};
+
+
+function _loadTranslation(forcedlang) {
+	if (fs.existsSync("viper.json")) {
+		// Validate viper.json
+		let opts = {
+			lang: "en",
+			autolang: true,
 		}
+
+		try {
+			opts = JSON.parse(fs.readFileSync("viper.json", "utf8"));
+		}catch (e) {}
+
+		lang = opts.lang;
+
+		if (! lang) {lang = "en"}
+
+		if (forcedlang) {lang = forcedlang}
+
+		if (opts.autolang == false) {
+			lang = opts.forcedlang;
+			if (! lang) {lang = "en"}
+		}
+
+		if (! fs.existsSync(__dirname + `/lang/${lang}.json`)) {
+			if (fs.existsSync(__dirname + `/lang/${lang.replace(/-.*$/, "")}.json`)) {
+				lang = lang.replace(/-.*$/, "");
+			} else {
+				lang = "en";
+			}
+		}
+	} else {
+		lang = "en";
 	}
+
+	langObj = JSON.parse(fs.readFileSync(__dirname + `/lang/${lang}.json`, "utf8"));
 }
 
-var langObj = JSON.parse(fs.readFileSync(__dirname + `/lang/${lang}.json`, "utf8"));
 
-module.exports = (string) => {
-	if (langObj[string]) { // Returns string from language
+module.exports = (string, forcedlang) => {
+	if (lang === "") {
+		_loadTranslation();
+	}
+	 
+	if (forcedlang) {
+		_loadTranslation(forcedlang);
+	}
+
+	if (langObj[string]) {
 		return langObj[string];
-	} else { // If string doesn't exist
-		if (langDef[string]) { // Retrieves from default lang instead
-			return langDef[string];
+	} else {
+		if (enLang[string]) {
+			return enLang[string];
 		} else {
 			// If it's not in the default lang either, it returns the
 			// string, this is absolute fallback.