fix: support ts generic

unjs · Apr 20, 2022 · c085469 · c085469
1 parent f618c68
commit c085469
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/src/utils.ts b/src/utils.ts
@@ -39,7 +39,7 @@ export function stripCommentsAndStrings (code: string) {
 }
 
 export function makeMatchRegex (imports: Import[]): RegExp {
-  return new RegExp(`(?:\\b|^)(${imports.map(i => escapeRE(i.as)).join('|')})\\s*(?:[.(\\)\\[\\];+\\-*&\\|\`])`, 'g')
+  return new RegExp(`(?:\\b|^)(${imports.map(i => escapeRE(i.as)).join('|')})\\s*(?:[.(\\)\\[\\];+\\-*&\\|\`<>])`, 'g')
 }
 
 export function toImports (imports: Import[], isCJS = false) {

diff --git a/test/__snapshots__/fixtures.test.ts.snap b/test/__snapshots__/fixtures.test.ts.snap
@@ -90,6 +90,11 @@ exports[`fixtures > template-tag.ts 1`] = `
 "
 `;
 
+exports[`fixtures > ts-generic.ts 1`] = `
+"import { useState } from 'react';useState<number>()
+"
+`;
+
 exports[`fixtures > vue.js 1`] = `
 "import { ref } from 'vue';const a = ref(0)
 "

diff --git a/test/fixtures/ts-generic.ts b/test/fixtures/ts-generic.ts
@@ -0,0 +1 @@
+useState<number>()