find_code 插件 react_vite
const fs = require("fs");
const path = require("path");
const parser = require("@babel/parser");
const traverse = require("@babel/traverse").default;
const generate = require("@babel/generator").default;// 讀取文件內容
const filePath = path.join(__dirname, "index.tsx");
const code = fs.readFileSync(filePath, "utf8");// 解析代碼生成 AST
const ast = parser.parse(code, {sourceType: "module",plugins: ["jsx"],
});// 遍歷 AST
// 遍歷 AST
traverse(ast, {JSXOpeningElement(path) {const line = path.node.loc.start.line;const pathAttribute = {type: "JSXAttribute",name: { type: "JSXIdentifier", name: "path" },value: {type: "StringLiteral",value: `${filePath}:${line}`,},};// 檢查是否已經存在 path 屬性，如果不存在則添加const existingPathAttribute = path.node.attributes.find((attr) => {return (attr.name &&attr.name.type === "JSXIdentifier" &&attr.name.name === "path");});if (!existingPathAttribute) {path.node.attributes.push(pathAttribute);}},
});// 生成新代碼，設置 retainLines 為 true 避免生成不必要的轉義序列
const { code: newCode } = generate(ast, {retainLines: true,jsescOption: {minimal: true,},
});// 寫入文件
fs.writeFileSync(filePath, newCode, "utf8");console.log("代碼修改完成");
// module
import fs from "fs/promises";
import path from "path";
import parser from "@babel/parser";
import traverse from "@babel/traverse";
import generate from "@babel/generator";
// 定義處理文件的異步函數
async function processFile(filePath) {try {// 讀取文件內容const code = await fs.readFile(filePath, "utf8");// 解析代碼生成 ASTconst ast = parser.parse(code, {sourceType: "module",plugins: ["jsx"],});// 遍歷 ASTtraverse.default(ast, {JSXOpeningElement(path) {const line = path?.node?.loc?.start?.line;const pathAttribute = {type: "JSXAttribute",name: { type: "JSXIdentifier", name: "path" },value: {type: "StringLiteral",value: `${filePath}:${line}`,},};// 檢查是否已經存在 path 屬性，如果不存在則添加const existingPathAttribute = path.node.attributes.find((attr) => {return (attr?.name &&attr?.name.type === "JSXIdentifier" &&attr?.name.name === "path");});if (!existingPathAttribute) {path.node.attributes.push(pathAttribute);}},});// 生成新代碼，設置 retainLines 為 true 避免生成不必要的轉義序列const { code: newCode } = generate.default(ast, {retainLines: true,jsescOption: {minimal: true,},});// 寫入文件await fs.writeFile(filePath, newCode, "utf8");console.log("代碼修改完成");} catch (error) {console.error("處理文件時出錯:", error);}
}// 獲取要處理的文件路徑
const filePath = path.join(process.cwd(), "node/index.tsx");
// 調用處理函數
processFile(filePath);
代碼修改完成效果
import React from "react";
const Test = () => {return (<div path="/Users/guojie/跳槽學習文檔——速成/vite/node/index.tsx:4">我是根目錄<div path="/Users/guojie/跳槽學習文檔——速成/vite/node/index.tsx:6">我是子目錄</div><span path="/Users/guojie/跳槽學習文檔——速成/vite/node/index.tsx:7">我是孫目錄</span></div>);
};
export default Test;
)
