fixup! Improve ImportDeclaration

EduardSergeev · Jun 20, 2024 · eda221a · eda221a
1 parent 5f5d298
commit eda221a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/input/before/UnusedImportProvider.hs b/input/before/UnusedImportProvider.hs
@@ -1,6 +1,6 @@
 import Data.List (sort, tails)
 import Data.Maybe
-import Data.Monoid (All(..), Any(getAny), Product(getProduct), Sum(..))
+import Data.Monoid (All(..), Any(getAny), Product(getProduct), Sum(..), Any(..), First(..))
 
 foo :: Ord a => [a] -> [a]
 foo xs =

diff --git a/src/features/importProvider/importDeclaration.ts b/src/features/importProvider/importDeclaration.ts
@@ -78,10 +78,9 @@ export default class ImportDeclaration {
   }
 
   public removeElement(elem: string) {
-    const regex = new RegExp(`^${elem}(\(.+\))?$`);
     const before = this.importElements;
 
-    const index = this._importElements.findIndex(elem => regex.test(elem));
+    const index = this._importElements.findIndex(oldElem => oldElem === elem || oldElem == `${elem}(..)`);
     if (index !== -1) {
       if (this._importElements.length > 1) {
         if (index === this._importElements.length - 1) {