fix: furthur improve iife matching

pionxzh · May 1, 2024 · 8546f81 · 8546f81
1 parent a1b56f1
commit 8546f81
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/packages/ast-utils/src/matchers/isIIFE.ts b/packages/ast-utils/src/matchers/isIIFE.ts
@@ -6,10 +6,7 @@ import type { ASTNode, ASTPath, CallExpression, Collection, ExpressionStatement,
  * ```js
  * (() => { ... })(...)
  * (function() { ... })(...)
- * ```
- *
- * @example
- * ```js
+ * !(() => { ... })(...)
  * !function() { ... }(...)
  * ```
  */
@@ -25,10 +22,7 @@ export function isStatementIIFE(j: JSCodeshift, node: Statement): node is Expres
  * ```js
  * (() => { ... })(...)
  * (function() { ... })(...)
- * ```
- *
- * @example
- * ```js
+ * !(() => { ... })(...)
  * !function() { ... }(...)
  * ```
  */
@@ -73,6 +67,12 @@ export function findIIFEs(
                 operator: '!',
                 argument: {
                     type: 'CallExpression',
+                    callee: {
+                        type: (type: string) => {
+                            return type === 'FunctionExpression'
+                                || type === 'ArrowFunctionExpression'
+                        },
+                    },
                 },
             },
         })