fix: options as function type

fastify · Feb 2, 2024 · 08b86b8 · 08b86b8
1 parent 4f2e04b
commit 08b86b8
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/index.d.ts b/index.d.ts
@@ -65,7 +65,7 @@ declare namespace avvio {
   }
 
   interface Use<I, C = context<I>> {
-    <O>(fn: avvio.Plugin<O, I>, options?: O): C;
+    <O>(fn: avvio.Plugin<O, I>, options?: O | ((server: C) => O)): C;
   }
 
   interface Ready<I, C = context<I>> {

diff --git a/test/types/index.ts b/test/types/index.ts
@@ -35,6 +35,18 @@ import * as avvio from "../../";
     server.close;
   });
 
+  app.use(async (server, options) => {},
+    (server) => {
+      server.use;
+      server.after;
+      server.ready;
+      server.on;
+      server.start;
+      server.override;
+      server.onClose;
+      server.close;
+  });
+
   app.after(err => {
     if (err) throw err;
   });
@@ -156,6 +168,18 @@ import * as avvio from "../../";
     server.close;
   });
 
+  app.use(async (server, options) => {},
+    (server) => {
+      server.use;
+      server.after;
+      server.ready;
+      server.on;
+      server.start;
+      server.override;
+      server.onClose;
+      server.close;
+    });
+
   app.after(err => {
     if (err) throw err;
   });
@@ -270,6 +294,14 @@ import * as avvio from "../../";
     server.typescriptIs;
   });
 
+  app.use(async (server, options) => {},
+   ((server) => {
+      server.use;
+      server.after;
+      server.ready;
+      server.typescriptIs;
+  }));
+
   app.after(err => {
     if (err) throw err;
   });