feat: getEnhancerByName accept generic parameter

Signed-off-by: Douglas McConnachie <[email protected]>
loopbackio · Apr 18, 2020 · 1d46514 · 1d46514
1 parent fbcedbc
commit 1d46514
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/openapi-v3/src/enhancers/spec-enhancer.service.ts b/packages/openapi-v3/src/enhancers/spec-enhancer.service.ts
@@ -71,10 +71,12 @@ export class OASEnhancerService {
    * Find an enhancer by its name
    * @param name The name of the enhancer you want to find
    */
-  async getEnhancerByName(name: string): Promise<OASEnhancer | undefined> {
+  async getEnhancerByName<T extends OASEnhancer>(
+    name: string,
+  ): Promise<T | undefined> {
     // Get the latest list of enhancers
     const enhancers = await this.getEnhancers();
-    return enhancers.find(e => e.name === name);
+    return enhancers.find(e => e.name === name) as T | undefined;
   }
 
   /**