Skip processing if di already processed (#95)

src/babel/__tests__/unit.test.js

@@ -546,6 +546,28 @@ describe('babel plugin', () => {
     );
   });
 
+  it('should skip processing the file if di() already processed', () => {
+    const input = `
+      import { di as _di } from "react-magnetic-di";
+      import React from 'react';
+      import Modal from 'modal';
+
+      function MyComponent() {
+        const [_Modal] = _di([Modal], MyComponent);
+        return <_Modal />;
+      }
+    `;
+    expect(babel(input)).toMatchInlineSnapshot(`
+      "import { di as _di } from "react-magnetic-di";
+      import React from 'react';
+      import Modal from 'modal';
+      function MyComponent() {
+        const [_Modal] = _di([Modal], MyComponent);
+        return __jsx(_Modal, null);
+      }"
+    `);
+  });
+
   it('should add di to all top level functions', () => {
     const input = `
       import React, { memo, forwardRef } from 'react';

src/babel/index.js

@@ -129,16 +129,22 @@ module.exports = function (babel) {
 
         state.findDiIndentifier(t, path.node.body, path.scope);
 
-        collectDiReferencePaths(t, state.diIdentifier, path.scope).forEach(
-          (p, i, arr) => {
-            const hasMulti =
-              p.getFunctionParent() === arr[i + 1]?.getFunctionParent();
-            if (isEnabled && !hasMulti) state.addDi(p);
-            else p.parentPath.remove();
-          }
+        const diRefPaths = collectDiReferencePaths(
+          t,
+          state.diIdentifier,
+          path.scope
         );
-
-        if (!isEnabled) return;
+        diRefPaths.forEach((p, i, arr) => {
+          const hasMulti =
+            p.getFunctionParent() === arr[i + 1]?.getFunctionParent();
+          if (isEnabled && !hasMulti) state.addDi(p);
+          else p.parentPath.remove();
+        });
+
+        const alreadyProcessed = diRefPaths.some((p) =>
+          p.parentPath?.parentPath?.isVariableDeclarator()
+        );
+        if (!isEnabled || alreadyProcessed) return;
 
         collectDepsReferencePaths(t, path.get('body')).forEach((p) =>
           state.addDependency(p)