feat: support customized middlewares (#1436)

cnpm · Feb 19, 2019 · 99c4c3f · 99c4c3f
1 parent 444b5d1
commit 99c4c3f
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/config/index.js b/config/index.js
@@ -50,6 +50,8 @@ var config = {
   // view files directory
   viewDir: path.join(root, 'view', 'web'),
 
+  customRegistryMiddlewares: [],
+
   // config for koa-limit middleware
   // for limit download rates
   limit: {
@@ -268,6 +270,12 @@ var config = {
 
 if (process.env.NODE_ENV === 'test') {
   config.enableAbbreviatedMetadata = true;
+  config.customRegistryMiddlewares.push(() => {
+    return function* (next) {
+      this.set('x-custom-middleware', 'true');
+      yield next;
+    };
+  });
 }
 
 if (process.env.NODE_ENV !== 'test') {

diff --git a/servers/registry.js b/servers/registry.js
@@ -23,7 +23,7 @@ var maxrequests = require('koa-maxrequests');
 var proxy = require('koa-proxy');
 app.use(proxy({
   host:  'https://registry.npmjs.org',
-  match: /^\/\-\/npm\/v1\/security\/audits/       
+  match: /^\/\-\/npm\/v1\/security\/audits/
 }));
 
 
@@ -51,6 +51,10 @@ app.use(notFound);
 app.use(conditional());
 app.use(etag());
 
+for (const middleware of config.customRegistryMiddlewares) {
+  app.use(middleware(app));
+}
+
 /**
  * Routes
  */

diff --git a/test/controllers/registry/package/list.test.js b/test/controllers/registry/package/list.test.js
@@ -41,6 +41,18 @@ describe('test/controllers/registry/package/list.test.js', () => {
     });
   });
 
+  it('should use costomized registry middleware', done => {
+    request(app)
+    .get('/@cnpmtest/testmodule-list-1')
+    .expect(200, function (err, res) {
+      should.not.exist(err);
+      var data = res.body;
+      data.name.should.equal('@cnpmtest/testmodule-list-1');
+      assert(res.headers['x-custom-middleware'] === 'true');
+      done();
+    });
+  });
+
   it('should return all versions', done => {
     request(app)
     .get('/@cnpmtest/testmodule-list-1')