feat: support custom web middlewares (#1563)

config/index.js

@@ -62,6 +62,7 @@ var config = {
   viewDir: path.join(root, 'view', 'web'),
 
   customRegistryMiddlewares: [],
+  customWebMiddlewares: [],
 
   // config for koa-limit middleware
   // for limit download rates
@@ -302,6 +303,13 @@ if (process.env.NODE_ENV === 'test') {
       yield next;
     };
   });
+
+  config.customWebMiddlewares.push(() => {
+    return function* (next) {
+      this.set('x-custom-web-middleware', 'true');
+      yield next;
+    };
+  });
 }
 
 if (process.env.NODE_ENV !== 'test') {

servers/web.js

@@ -42,6 +42,10 @@ if (config.pagemock) {
   }));
 }
 
+for (const mw of config.customWebMiddlewares) {
+  app.use(mw(app));
+}
+
 app.use(opensearch);
 app.keys = ['todokey', config.sessionSecret];
 app.proxy = true;

test/controllers/web/user/show.test.js

@@ -81,6 +81,7 @@ describe('controllers/web/user/show.test.js', function () {
 
           // he.enclde('fengmk2@gmail.com') ↓
           assert(res.text.includes('fengmk2@gmail.com'));
+          assert(res.headers['x-custom-web-middleware'] === 'true');
           done()
         });
     });