fix: fixes options (closes #44 and closes #45)

koajs · Jan 13, 2019 · 2603f32 · 2603f32
1 parent 6fc2c39
commit 2603f32
Showing 1 changed file with 10 additions and 18 deletions.
diff --git a/src/index.js b/src/index.js
@@ -2,24 +2,16 @@ const csrf = require('csrf');
 const autoBind = require('auto-bind');
 
 class CSRF {
-  constructor(opts) {
-    this.opts = opts || {};
-
-    if (!this.opts.invalidTokenMessage) {
-      this.opts.invalidTokenMessage = 'Invalid CSRF token';
-    }
-
-    if (!this.opts.invalidTokenStatusCode) {
-      this.opts.invalidTokenStatusCode = 403;
-    }
-
-    if (!this.opts.excludedMethods) {
-      this.opts.excludedMethods = ['GET', 'HEAD', 'OPTIONS'];
-    }
-
-    if (typeof this.opts.disableQuery !== 'boolean') {
-      this.opts.disableQuery = false;
-    }
+  constructor(opts = {}) {
+    this.opts = Object.assign(
+      {
+        invalidTokenMessage: 'Invalid CSRF token',
+        invalidTokenStatusCode: 403,
+        excludedMethods: ['GET', 'HEAD', 'OPTIONS'],
+        disableQuery: false
+      },
+      opts
+    );
 
     this.tokens = csrf(opts);