From 6ae8cde4a92b1316a4463f30ae0b5053720aa083 Mon Sep 17 00:00:00 2001
From: Douglas McConnachie <dougal83+git@gmail.com>
Date: Tue, 7 Apr 2020 16:25:19 +0100
Subject: [PATCH] feat: getEnhancerByName accept generic parameter

Signed-off-by: Douglas McConnachie <dougal83+git@gmail.com>
---
 packages/openapi-v3/src/enhancers/spec-enhancer.service.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/openapi-v3/src/enhancers/spec-enhancer.service.ts b/packages/openapi-v3/src/enhancers/spec-enhancer.service.ts
index c39a9f883a01..f196f797d545 100644
--- 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;
   }
 
   /**