fix(types): Add plugin function definition

Signed-off-by: Roberto Bianchi <[email protected]>
mcollina · Apr 22, 2024 · 4b3967b · 4b3967b
1 parent 53456d1
commit 4b3967b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/index.d.ts b/index.d.ts
@@ -3,6 +3,7 @@ import {
   type FastifyBaseLogger,
   type FastifyReply,
   type FastifyRequest,
+  type FastifyPluginCallback,
 } from "fastify";
 
 declare namespace fastifyasyncforge {
@@ -12,4 +13,5 @@ declare namespace fastifyasyncforge {
   export function logger<T extends FastifyBaseLogger>(): T;
 }
 
+declare function fastifyasyncforge(): FastifyPluginCallback;
 export = fastifyasyncforge;
diff --git a/index.test-d.ts b/index.test-d.ts
@@ -1,15 +1,18 @@
 import { expectAssignable, expectError, expectType } from "tsd";
-import { app, logger, reply, request } from ".";
+import fastifyasyncforge, { app, logger, reply, request } from ".";
 import fastify, {
   type FastifyInstance,
   type FastifyBaseLogger,
   type FastifyRequest,
   type FastifyReply,
   type RawServerDefault,
   type RouteGenericInterface,
+  type FastifyPluginCallback,
 } from "fastify";
 
 const fastifyInstance = fastify();
+expectAssignable<FastifyInstance>(fastifyInstance.register(fastifyasyncforge));
+expectAssignable<FastifyPluginCallback>(fastifyasyncforge);
 
 // app
 expectAssignable<FastifyInstance>(fastifyInstance);