Merge pull request #65 from jkyberneees/fix-method-chaining

Fix method chaining
BackendStack21 · Jan 2, 2020 · 8dc9e4b · 8dc9e4b
2 parents af231ee + 8502267
commit 8dc9e4b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/index.js b/index.js
@@ -46,7 +46,11 @@ module.exports = (options = {}) => {
       return options
     },
 
-    use: router.use,
+    use: (...args) => {
+      router.use.apply(router, args)
+
+      return app
+    },
 
     handle: (req, res) => {
       // request object population
@@ -72,7 +76,11 @@ module.exports = (options = {}) => {
   }
 
   shortcuts.forEach((method) => {
-    app[method] = router[method]
+    app[method] = (...args) => {
+      router[method].apply(router, args)
+
+      return app
+    }
   })
 
   app.callback = () => app.handle

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "restana",
-  "version": "4.0.0",
+  "version": "4.0.1",
   "description": "Super fast and minimalist web framework for building REST micro-services.",
   "main": "index.js",
   "types": "index.d.ts",