[FIX]: Module source row

tamasszoke · May 30, 2018 · a0f671e · a0f671e
1 parent 20b35d6
commit a0f671e
Showing 1 changed file with 26 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -127,9 +127,33 @@ function eyecatcher() {
 
 		const trace = new Error().stack;
 		const regexp = /\.<anonymous>(.*)\n/g;
+		const regexpModule = /\module.exports(.*)\n/g;
+		let source;
+
+		try {
+
+			source = regexpModule.exec(trace)[1];
+
+		} catch(error) {
+
+			try {
 
-		let source = regexp.exec(trace)[1];
-		source = source.replace(/^.*[\\\/]/, '').replace(/[{()}]/g, '').replace(/\n|\r/g, "");
+				source = regexp.exec(trace)[1];
+
+			} catch(error) {
+
+				source = null;
+			};
+		};
+
+		if (source) {
+
+			source = source.replace(/^.*[\\\/]/, '').replace(/[{()}]/g, '').replace(/\n|\r/g, "");
+
+		} else {
+
+			source = 'Row not found!';
+		};
 
 		return ' ' + source;
 	};