@thinksharpe/react-compiler-unmemo 0.5.3 → 0.5.4

package/README.md

@@ -3,7 +3,7 @@
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg?style=flat-square)](https://opensource.org/licenses/MIT)
 [![Node.js](https://img.shields.io/badge/node-%3E%3D18-brightgreen?style=flat-square)](https://nodejs.org)
-[![Tests](https://img.shields.io/badge/tests-55%20passing-brightgreen?style=flat-square)](./tests)
+[![Tests](https://img.shields.io/badge/tests-62%20passing-brightgreen?style=flat-square)](./tests)
 
 A codemod that removes `useMemo` and `useCallback` from your React codebase so you can adopt [React Compiler](https://react.dev/learn/react-compiler) without the manual cleanup.
 

package/helpers/remove-hooks.mjs

@@ -436,31 +436,33 @@ export function processFile(filePath, { dryRun = false } = {}) {
 
   // Phase 3: Clean up imports
   if (changed) {
-    // Handle: import React, { useMemo, useCallback, ... } from "react"; (single or double quotes)
+    // Handle: import React, { useMemo, useCallback, ... } from "react"; (single or double quotes, semicolon optional)
     content = content.replace(
-      /import React, \{([^}]*)\} from (["'])react\2\s*;/g,
+      /import React, \{([^}]*)\} from (["'])react\2[^\S\n]*;?/g,
       (match, imports, quote) => {
+        const semi = match.trimEnd().endsWith(";") ? ";" : "";
         const cleaned = imports
           .split(",")
           .map((s) => s.trim())
           .filter((s) => s && s !== "useMemo" && s !== "useCallback")
           .join(", ");
-        if (!cleaned) return `import React from ${quote}react${quote};`;
-        return `import React, { ${cleaned} } from ${quote}react${quote};`;
+        if (!cleaned) return `import React from ${quote}react${quote}${semi}`;
+        return `import React, { ${cleaned} } from ${quote}react${quote}${semi}`;
       },
     );
 
-    // Handle: import { useMemo, useCallback, ... } from "react"; (single or double quotes)
+    // Handle: import { useMemo, useCallback, ... } from "react"; (single or double quotes, semicolon optional)
     content = content.replace(
-      /import \{([^}]*)\} from (["'])react\2\s*;/g,
+      /import \{([^}]*)\} from (["'])react\2[^\S\n]*;?/g,
       (match, imports, quote) => {
+        const semi = match.trimEnd().endsWith(";") ? ";" : "";
         const cleaned = imports
           .split(",")
           .map((s) => s.trim())
           .filter((s) => s && s !== "useMemo" && s !== "useCallback")
           .join(", ");
         if (!cleaned) return ""; // Remove empty import entirely
-        return `import { ${cleaned} } from ${quote}react${quote};`;
+        return `import { ${cleaned} } from ${quote}react${quote}${semi}`;
       },
     );
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@thinksharpe/react-compiler-unmemo",
-  "version": "0.5.3",
+  "version": "0.5.4",
   "description": "Remove useMemo and useCallback hooks from React codebases to leverage React Compiler automatic optimization",
   "license": "MIT",
   "type": "module",