Feat: add async-busboy for multipart/formdata enctype

index.js

@@ -16,10 +16,13 @@
 
 const debug = require('debug')('method-override')
 const methods = require('methods')
+const asyncBusboy = require('async-busboy')
 
 const ALLOWED_METHODS = 'POST'
 const HTTP_METHOD_OVERRIDE_HEADER = "X-HTTP-Method-Override"
 
+let asyncBusboyBody = null
+
 /**
  * Method Override:
  *
@@ -57,7 +60,7 @@ function methodOverride(getter, options) {
     ? ALLOWED_METHODS.split(' ')
     : options.methods
 
-  return (ctx, next) => {
+  return async (ctx, next) => {
     const req = ctx.request
     let method
     let val
@@ -69,7 +72,10 @@ function methodOverride(getter, options) {
       return next()
     }
 
-    val = get(req, ctx.response)
+    asyncBusboyBody = null
+    ctx.req.getAsyncBusboyBody = async () => (asyncBusboyBody || await asyncBusboy(ctx.req))
+
+    val = await get(req, ctx.response, ctx.req)
     method = Array.isArray(val) ? val[0] : val
 
     // replace
@@ -102,8 +108,15 @@ function createGetter(str) {
 function createQueryOrBodyGetter(key) {
   return queryOrBodyGetter
 
-  function queryOrBodyGetter(req) {
-    return req.query[key] || (req.body && req.body[key])
+  async function queryOrBodyGetter(req, ...args) {
+    const method = req.query[key] || (req.body && req.body[key])
+    if (method) {
+      return method
+    } else {
+      const koaReq = args[1]
+      asyncBusboyBody = await asyncBusboy(koaReq)
+      return asyncBusboyBody.fields[key]
+    }
   }
 }
 

package.json

@@ -18,6 +18,7 @@
     "middleware"
   ],
   "dependencies": {
+    "async-busboy": "^0.3.3",
     "debug": "*",
     "methods": "*"
   },