fixes method chaining for use and routes

BackendStack21 · Jan 2, 2020 · 8502267 · 8502267
1 parent b647012
commit 8502267
Showing 1 changed file with 10 additions and 2 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