From f9cf4cc884ec84be0ac65d76be008c53f69bad11 Mon Sep 17 00:00:00 2001
From: lingo <shufflellg@gmail.com>
Date: Fri, 15 Jan 2021 13:54:14 +0800
Subject: [PATCH] fix match

---
 src/MethodExtractor.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/MethodExtractor.ts b/src/MethodExtractor.ts
index 48955e8..f6562c3 100644
--- a/src/MethodExtractor.ts
+++ b/src/MethodExtractor.ts
@@ -83,6 +83,13 @@ export class MethodExtractor {
 				return;
 			}
 
+			let className = clause.types[0].getText();
+			const idxOfT = className.indexOf('<');
+			if (idxOfT != -1) {
+				className = className.substring(0, idxOfT).trim();
+			}
+
+			return className;
 			return clause.types[0].getText();
 		}
 	}
@@ -90,8 +97,9 @@ export class MethodExtractor {
 	getImportDeclaration(sourceDescriptor: SourceFileDescriptor, className: string): ImportDeclaration {
 		const extendsSource = sourceDescriptor.importClause.find(imp => {
 			const x: ImportDeclaration = <ImportDeclaration>imp;
-			const bindings = <NamedImports>x.importClause.namedBindings;
+			return x.importClause.getText() === className;
 
+			const bindings = <NamedImports>x.importClause.namedBindings;
 			return bindings.elements.some(x => x.name.getText() === className);
 		});