feat(migration): integrate batch import rewriting into migration command

Add NAPI bindings for rewrite_imports_in_directory to expose the Rust
batch import rewriting functionality to TypeScript. The migration command
now rewrites vite/vitest imports in all TypeScript/JavaScript files,
not just config files.

Changes:
- Add BatchRewriteResult and BatchRewriteError NAPI structs
- Add rewriteImportsInDirectory NAPI function
- Call rewriteAllImports in both standalone and monorepo migrations
- Display progress showing modified files and any errors

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: fengmk2

crates/vite_migration/src/file_walker.rs

@@ -3,8 +3,9 @@ use std::path::{Path, PathBuf};
 use ignore::WalkBuilder;
 use vite_error::Error;
 
+// TODO: only support esm files for now
 /// File extensions to process for import rewriting
-const TS_JS_EXTENSIONS: &[&str] = &["ts", "tsx", "mts", "cts", "js", "jsx", "mjs", "cjs"];
+const TS_JS_EXTENSIONS: &[&str] = &["ts", "tsx", "mts", "js", "jsx", "mjs"];
 
 /// Result of walking TypeScript/JavaScript files
 #[derive(Debug)]
@@ -163,7 +164,7 @@ mod tests {
 
         let result = find_ts_files(temp.path()).unwrap();
 
-        assert_eq!(result.files.len(), 8);
+        assert_eq!(result.files.len(), 6);
     }
 
     #[test]

crates/vite_migration/src/import_rewriter.rs

@@ -116,7 +116,7 @@ fix: $NEW_IMPORT
 
 /// Result of rewriting imports in a file
 #[derive(Debug)]
-pub struct RewriteResult {
+struct RewriteResult {
     /// The updated file content
     pub content: String,
     /// Whether any changes were made
@@ -134,40 +134,6 @@ pub struct BatchRewriteResult {
     pub errors: Vec<(PathBuf, String)>,
 }
 
-/// Rewrite imports in a TypeScript/JavaScript file from vite/vitest to @voidzero-dev/vite-plus
-///
-/// This function reads a file and rewrites the import statements
-/// to use '@voidzero-dev/vite-plus' instead of 'vite', 'vitest', or '@vitest/*'.
-///
-/// # Arguments
-///
-/// * `file_path` - Path to the TypeScript/JavaScript file
-///
-/// # Returns
-///
-/// Returns a `RewriteResult` containing:
-/// - `content`: The updated file content
-/// - `updated`: Whether any changes were made
-///
-/// # Example
-///
-/// ```ignore
-/// use std::path::Path;
-/// use vite_migration::rewrite_import;
-///
-/// let result = rewrite_import(Path::new("src/app.ts"))?;
-/// if result.updated {
-///     std::fs::write("src/app.ts", &result.content)?;
-/// }
-/// ```
-pub fn rewrite_import(file_path: &Path) -> Result<RewriteResult, Error> {
-    // Read the file
-    let content = std::fs::read_to_string(file_path)?;
-
-    // Rewrite the imports
-    rewrite_import_content(&content)
-}
-
 /// Rewrite imports in all TypeScript/JavaScript files under a directory
 ///
 /// This function finds all TypeScript and JavaScript files in the specified directory
@@ -229,6 +195,40 @@ pub fn rewrite_imports_in_directory(root: &Path) -> Result<BatchRewriteResult, E
     Ok(result)
 }
 
+/// Rewrite imports in a TypeScript/JavaScript file from vite/vitest to @voidzero-dev/vite-plus
+///
+/// This function reads a file and rewrites the import statements
+/// to use '@voidzero-dev/vite-plus' instead of 'vite', 'vitest', or '@vitest/*'.
+///
+/// # Arguments
+///
+/// * `file_path` - Path to the TypeScript/JavaScript file
+///
+/// # Returns
+///
+/// Returns a `RewriteResult` containing:
+/// - `content`: The updated file content
+/// - `updated`: Whether any changes were made
+///
+/// # Example
+///
+/// ```ignore
+/// use std::path::Path;
+/// use vite_migration::rewrite_import;
+///
+/// let result = rewrite_import(Path::new("src/app.ts"))?;
+/// if result.updated {
+///     std::fs::write("src/app.ts", &result.content)?;
+/// }
+/// ```
+fn rewrite_import(file_path: &Path) -> Result<RewriteResult, Error> {
+    // Read the file
+    let content = std::fs::read_to_string(file_path)?;
+
+    // Rewrite the imports
+    rewrite_import_content(&content)
+}
+
 /// Rewrite imports in content from vite/vitest to @voidzero-dev/vite-plus
 ///
 /// This is the internal function that performs the actual rewrite using ast-grep.

crates/vite_migration/src/lib.rs

@@ -5,8 +5,6 @@ mod package;
 mod vite_config;
 
 pub use file_walker::{WalkResult, find_ts_files};
-pub use import_rewriter::{
-    BatchRewriteResult, RewriteResult, rewrite_import, rewrite_imports_in_directory,
-};
+pub use import_rewriter::{BatchRewriteResult, rewrite_imports_in_directory};
 pub use package::rewrite_scripts;
 pub use vite_config::{MergeResult, merge_json_config};

packages/global/binding/index.d.ts

@@ -1,5 +1,21 @@
 /* auto-generated by NAPI-RS */
 /* eslint-disable */
+/** Error from batch import rewriting */
+export interface BatchRewriteError {
+  /** The file path that had an error */
+  path: string;
+  /** The error message */
+  message: string;
+}
+
+/** Result of rewriting imports in multiple files */
+export interface BatchRewriteResult {
+  /** Files that were modified */
+  modifiedFiles: Array<string>;
+  /** Files that had errors */
+  errors: Array<BatchRewriteError>;
+}
+
 /**
  * Configuration options passed from JavaScript to Rust.
  *
@@ -154,36 +170,33 @@ export interface PathAccess {
 }
 
 /**
- * Rewrite imports in vite config from 'vite' or 'vitest/config' to '@voidzero-dev/vite-plus'
+ * Rewrite imports in all TypeScript/JavaScript files under a directory
+ *
+ * This function finds all TypeScript and JavaScript files in the specified directory
+ * (respecting `.gitignore` rules), applies the import rewrite rules to each file,
+ * and writes the modified content back to disk.
  *
  * # Arguments
  *
- * * `vite_config_path` - Path to the vite.config.ts or vite.config.js file
+ * * `root` - The root directory to search for files
  *
  * # Returns
  *
- * Returns a `RewriteResult` containing:
- * - `content`: The updated vite config content
- * - `updated`: Whether any changes were made
+ * Returns a `BatchRewriteResult` containing:
+ * - `modifiedFiles`: Files that were changed
+ * - `errors`: Files that had errors during processing
  *
  * # Example
  *
  * ```javascript
- * const result = rewriteImport('vite.config.ts');
- * if (result.updated) {
- *     fs.writeFileSync('vite.config.ts', result.content);
+ * const result = rewriteImportsInDirectory('./src');
+ * console.log(`Modified ${result.modifiedFiles.length} files`);
+ * for (const file of result.modifiedFiles) {
+ *     console.log(`  ${file}`);
  * }
  * ```
  */
-export declare function rewriteImport(viteConfigPath: string): RewriteResult;
-
-/** Result of rewriting imports in vite config */
-export interface RewriteResult {
-  /** The updated vite config content */
-  content: string;
-  /** Whether any changes were made */
-  updated: boolean;
-}
+export declare function rewriteImportsInDirectory(root: string): BatchRewriteResult;
 
 /**
  * Rewrite scripts json content using rules from rules_yaml

packages/global/binding/index.js

@@ -766,15 +766,15 @@ const {
   detectWorkspace,
   downloadPackageManager,
   mergeJsonConfig,
-  rewriteImport,
+  rewriteImportsInDirectory,
   rewriteScripts,
   run,
   runCommand,
 } = nativeBinding;
 export { detectWorkspace };
 export { downloadPackageManager };
 export { mergeJsonConfig };
-export { rewriteImport };
+export { rewriteImportsInDirectory };
 export { rewriteScripts };
 export { run };
 export { runCommand };

packages/global/binding/src/lib.rs

@@ -15,7 +15,7 @@ use vite_path::current_dir;
 
 use crate::cli::Args;
 pub use crate::{
-    migration::{merge_json_config, rewrite_import, rewrite_scripts},
+    migration::{merge_json_config, rewrite_imports_in_directory, rewrite_scripts},
     package_manager::{detect_workspace, download_package_manager},
 };
 

packages/global/binding/src/migration.rs

@@ -84,38 +84,67 @@ pub fn merge_json_config(
     })
 }
 
-/// Result of rewriting imports in vite config
+/// Error from batch import rewriting
 #[napi(object)]
-pub struct RewriteResult {
-    /// The updated vite config content
-    pub content: String,
-    /// Whether any changes were made
-    pub updated: bool,
+pub struct BatchRewriteError {
+    /// The file path that had an error
+    pub path: String,
+    /// The error message
+    pub message: String,
 }
 
-/// Rewrite imports in vite config from 'vite' or 'vitest/config' to '@voidzero-dev/vite-plus'
+/// Result of rewriting imports in multiple files
+#[napi(object)]
+pub struct BatchRewriteResult {
+    /// Files that were modified
+    pub modified_files: Vec<String>,
+    /// Files that had errors
+    pub errors: Vec<BatchRewriteError>,
+}
+
+/// Rewrite imports in all TypeScript/JavaScript files under a directory
+///
+/// This function finds all TypeScript and JavaScript files in the specified directory
+/// (respecting `.gitignore` rules), applies the import rewrite rules to each file,
+/// and writes the modified content back to disk.
 ///
 /// # Arguments
 ///
-/// * `vite_config_path` - Path to the vite.config.ts or vite.config.js file
+/// * `root` - The root directory to search for files
 ///
 /// # Returns
 ///
-/// Returns a `RewriteResult` containing:
-/// - `content`: The updated vite config content
-/// - `updated`: Whether any changes were made
+/// Returns a `BatchRewriteResult` containing:
+/// - `modifiedFiles`: Files that were changed
+/// - `errors`: Files that had errors during processing
 ///
 /// # Example
 ///
 /// ```javascript
-/// const result = rewriteImport('vite.config.ts');
-/// if (result.updated) {
-///     fs.writeFileSync('vite.config.ts', result.content);
+/// const result = rewriteImportsInDirectory('./src');
+/// console.log(`Modified ${result.modifiedFiles.length} files`);
+/// for (const file of result.modifiedFiles) {
+///     console.log(`  ${file}`);
 /// }
 /// ```
 #[napi]
-pub fn rewrite_import(vite_config_path: String) -> Result<RewriteResult> {
-    let result = vite_migration::rewrite_import(Path::new(&vite_config_path))
+pub fn rewrite_imports_in_directory(root: String) -> Result<BatchRewriteResult> {
+    let result = vite_migration::rewrite_imports_in_directory(Path::new(&root))
         .map_err(anyhow::Error::from)?;
-    Ok(RewriteResult { content: result.content, updated: result.updated })
+
+    Ok(BatchRewriteResult {
+        modified_files: result
+            .modified_files
+            .iter()
+            .map(|p| p.to_string_lossy().to_string())
+            .collect(),
+        errors: result
+            .errors
+            .iter()
+            .map(|(p, m)| BatchRewriteError {
+                path: p.to_string_lossy().to_string(),
+                message: m.clone(),
+            })
+            .collect(),
+    })
 }

packages/global/snap-tests/migration-from-vitest/snap.txt

@@ -7,7 +7,10 @@
 │
 ●  pnpm@<semver> installed
 │
-◆  ✅ Rewrote import in vitest.config.ts
+◆  ✅ Rewrote imports in 2 file(s)
+│
+●    test/hello.ts
+│    vitest.config.ts
 │
 └  ✨ Migration completed!
 
@@ -41,8 +44,6 @@ export default defineConfig({
   "devDependencies": {
     "vite": "npm:@voidzero-dev/vite-plus-core@latest",
     "vitest": "npm:@voidzero-dev/vite-plus-test@latest",
-    "@vitest/browser": "^4.0.0",
-    "@vitest/browser-playwright": "^4.0.0",
     "@voidzero-dev/vite-plus": "latest"
   },
   "scripts": {
@@ -62,8 +63,8 @@ export default defineConfig({
 }
 
 > cat test/hello.ts # check test/hello.ts
-import { server } from '@vitest/browser-playwright/context';
-import { test, describe, expect, it } from 'vitest';
+import { server } from '@voidzero-dev/vite-plus/test/browser-playwright/context';
+import { test, describe, expect, it } from '@voidzero-dev/vite-plus/test';
 
 const { readFile } = server.commands;
 

packages/global/snap-tests/migration-merge-vite-config-ts/snap.txt

@@ -7,14 +7,15 @@
 │
 ●  pnpm@<semver> installed
 │
-◆  ✅ Rewrote import in vite.config.ts
-│
-◆  ✅ Rewrote import in vitest.config.ts
-│
 ◆  ✅ Merged .oxlintrc.json into vite.config.ts
 │
 ◆  ✅ Merged .oxfmtrc.json into vite.config.ts
 │
+◆  ✅ Rewrote imports in 2 file(s)
+│
+●    vite.config.ts
+│    vitest.config.ts
+│
 └  ✨ Migration completed!
 
 

packages/global/snap-tests/migration-monorepo-pnpm-overrides-dependency-selector/snap.txt

@@ -5,7 +5,9 @@
 │
 ●  pnpm@<semver> installed
 │
-◆  ✅ Rewrote import in vite.config.ts
+◆  ✅ Rewrote imports in 1 file(s)
+│
+●    vite.config.ts
 │
 └  ✨ Migration completed!