From 31ae2b258dcd3d6218cb7947f3e8a7a362d43b88 Mon Sep 17 00:00:00 2001
From: Lam Wei Li <peteriman@mail.com>
Date: Sun, 2 Oct 2022 00:54:50 +0800
Subject: [PATCH 1/3] refactor(LoggingEvent): loop through location keys
 instead of hard-coding one-by-one

---
 lib/LoggingEvent.js | 55 +++++++++++++++++++++------------------------
 1 file changed, 25 insertions(+), 30 deletions(-)

diff --git a/lib/LoggingEvent.js b/lib/LoggingEvent.js
index e6a7ceb6..f39383ab 100644
--- a/lib/LoggingEvent.js
+++ b/lib/LoggingEvent.js
@@ -1,4 +1,5 @@
 /* eslint max-classes-per-file: ["error", 2] */
+/* eslint no-underscore-dangle: ["error", { "allow": ["_getLocationKeys"] }] */
 
 const flatted = require('flatted');
 const levels = require('./levels');
@@ -63,18 +64,27 @@ class LoggingEvent {
     this.pid = process.pid;
     this.error = error;
 
-    if (location) {
-      this.fileName = location.fileName;
-      this.lineNumber = location.lineNumber;
-      this.columnNumber = location.columnNumber;
-      this.callStack = location.callStack;
-      this.className = location.className;
-      this.functionName = location.functionName;
-      this.functionAlias = location.functionAlias;
-      this.callerName = location.callerName;
+    if (typeof location !== 'undefined') {
+      this.constructor._getLocationKeys().forEach((key) => {
+        if (typeof location[key] !== 'undefined') this[key] = location[key];
+      });
     }
   }
 
+  /** @private */
+  static _getLocationKeys() {
+    return [
+      'fileName',
+      'lineNumber',
+      'columnNumber',
+      'callStack',
+      'className',
+      'functionName',
+      'functionAlias',
+      'callerName',
+    ];
+  }
+
   serialise() {
     return flatted.stringify(this, (key, value) => {
       // JSON.stringify(new Error('test')) returns {}, which is not really useful for us.
@@ -107,27 +117,12 @@ class LoggingEvent {
         }
         return serde.deserialise(value);
       });
-      if (
-        rehydratedEvent.fileName ||
-        rehydratedEvent.lineNumber ||
-        rehydratedEvent.columnNumber ||
-        rehydratedEvent.callStack ||
-        rehydratedEvent.className ||
-        rehydratedEvent.functionName ||
-        rehydratedEvent.functionAlias ||
-        rehydratedEvent.callerName
-      ) {
-        rehydratedEvent.location = {
-          fileName: rehydratedEvent.fileName,
-          lineNumber: rehydratedEvent.lineNumber,
-          columnNumber: rehydratedEvent.columnNumber,
-          callStack: rehydratedEvent.callStack,
-          className: rehydratedEvent.className,
-          functionName: rehydratedEvent.functionName,
-          functionAlias: rehydratedEvent.functionAlias,
-          callerName: rehydratedEvent.callerName,
-        };
-      }
+      this._getLocationKeys().forEach((key) => {
+        if (typeof rehydratedEvent[key] !== 'undefined') {
+          if (!rehydratedEvent.location) rehydratedEvent.location = {};
+          rehydratedEvent.location[key] = rehydratedEvent[key];
+        }
+      });
       event = new LoggingEvent(
         rehydratedEvent.categoryName,
         levels.getLevel(rehydratedEvent.level.levelStr),

From 6f697ad317bf2709fe5f0b607661440d34504e16 Mon Sep 17 00:00:00 2001
From: Lam Wei Li <peteriman@mail.com>
Date: Sun, 2 Oct 2022 00:57:43 +0800
Subject: [PATCH 2/3] fix(LoggingEvent): throws error for invalid location
 passed when creating a new LoggingEvent

---
 lib/LoggingEvent.js           |  5 +++++
 test/tap/LoggingEvent-test.js | 15 +++++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/lib/LoggingEvent.js b/lib/LoggingEvent.js
index f39383ab..db561cd2 100644
--- a/lib/LoggingEvent.js
+++ b/lib/LoggingEvent.js
@@ -65,6 +65,11 @@ class LoggingEvent {
     this.error = error;
 
     if (typeof location !== 'undefined') {
+      if (!location || typeof location !== 'object' || Array.isArray(location))
+        throw new TypeError(
+          'Invalid location type passed to LoggingEvent constructor'
+        );
+
       this.constructor._getLocationKeys().forEach((key) => {
         if (typeof location[key] !== 'undefined') this[key] = location[key];
       });
diff --git a/test/tap/LoggingEvent-test.js b/test/tap/LoggingEvent-test.js
index 72c6c7a2..012d4c24 100644
--- a/test/tap/LoggingEvent-test.js
+++ b/test/tap/LoggingEvent-test.js
@@ -4,6 +4,21 @@ const LoggingEvent = require('../../lib/LoggingEvent');
 const levels = require('../../lib/levels');
 
 test('LoggingEvent', (batch) => {
+  batch.test('should throw error for invalid location', (t) => {
+    t.throws(
+      () =>
+        new LoggingEvent(
+          'cheese',
+          levels.DEBUG,
+          ['log message'],
+          undefined,
+          []
+        ),
+      'Invalid location type passed to LoggingEvent constructor'
+    );
+    t.end();
+  });
+
   batch.test('should serialise to flatted', (t) => {
     const event = new LoggingEvent(
       'cheese',

From 6a8b706590f374a01a5089a567f7f11cab57e7ff Mon Sep 17 00:00:00 2001
From: Lam Wei Li <peteriman@mail.com>
Date: Sun, 2 Oct 2022 00:59:04 +0800
Subject: [PATCH 3/3] test(Logger): apart from asserting a throw, also assert
 the thrown error message

---
 test/tap/logger-test.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/tap/logger-test.js b/test/tap/logger-test.js
index cde4ae6d..3320bfb9 100644
--- a/test/tap/logger-test.js
+++ b/test/tap/logger-test.js
@@ -371,7 +371,10 @@ test('../../lib/logger', (batch) => {
     t.equal(events[0].fileName, initialEvent.fileName);
     t.equal(events[0].columnNumber, initialEvent.columnNumber);
 
-    t.throws(() => logger.setParseCallStackFunction('not a function'));
+    t.throws(
+      () => logger.setParseCallStackFunction('not a function'),
+      'Invalid type passed to setParseCallStackFunction'
+    );
 
     t.end();
   });