fix(#161): use() accepts array

hoangvvo · Apr 10, 2022 · dfd24aa · dfd24aa
1 parent 89196b1
commit dfd24aa
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/src/index.d.ts b/src/index.d.ts
@@ -33,6 +33,9 @@ declare module "next-connect" {
   interface NextConnect<Req, Res> {
     (req: Req, res: Res): Promise<void>;
 
+    use<ReqExt = {}, ResExt = {}>(
+      handlers: Middleware<Req & ReqExt, Res & ResExt>[]
+    ): this;
     use<ReqExt = {}, ResExt = {}>(
       ...handlers: Middleware<Req & ReqExt, Res & ResExt>[]
     ): this;

diff --git a/src/index.js b/src/index.js
@@ -35,6 +35,7 @@ export default function factory({
     return nc;
   }
   nc.use = function use(base, ...fns) {
+    if (Array.isArray(base)) return this.use("/", ...base);
     if (typeof base === "function") return this.use("/", base, ...fns);
     if (typeof base === "string" && base !== "/") {
       let slashAdded = false;

diff --git a/test/index.test.js b/test/index.test.js
@@ -294,6 +294,22 @@ describe("use()", () => {
     await request(app).get("/some/path").expect("no");
   });
 
+  it("use accepts an array of middleware", async () => {
+    const handler = nc();
+    handler.use([(req, res, next) => {
+      req.ok1 = "ok1";
+      next();
+    },(req, res, next) => {
+      req.ok2 = "ok2";
+      next();
+    }]);
+    handler.get((req, res) => {
+      res.end(`${req.ok1}-${req.ok2}` || "no");
+    });
+    const app = createServer(handler);
+    await request(app).get("/some/path").expect("ok1-ok2");
+  });
+
   it("mount subapp", () => {
     const handler2 = nc();
     handler2.use((req, res, next) => {