@kamaalio/codemod-kit 0.0.20 → 0.0.22

package/dist/codemods/types.d.ts

@@ -19,5 +19,5 @@ export type Modifications = {
 };
 export type FindAndReplaceConfig = {
     rule: Rule<TypesMap>;
-    transformer: ((node: SgNode<TypesMap, Kinds<TypesMap>>) => Optional<string>) | string;
+    transformer: ((node: SgNode<TypesMap, Kinds<TypesMap>>, rule: Rule<TypesMap>) => Optional<string>) | string;
 };

package/dist/index.cjs

@@ -164,12 +164,54 @@ function findAndReplaceEdits(content, rule, transformer) {
         rule
     });
     return kamaal_namespaceObject.arrays.compactMap(nodes, (node)=>{
-        const transformed = 'string' == typeof transformer ? transformer : transformer(node);
+        const transformed = 'string' == typeof transformer ? transformer : transformer(node, rule);
         if (null == transformed) return null;
-        if (transformed === node.text()) return null;
-        return node.replace(transformed);
+        const transformedValueWithMetaVariablesReplaced = Object.values(extractMetaVariables(node, rule)).reduce((acc, { original, value })=>acc.replaceAll(original, value), transformed);
+        if (transformedValueWithMetaVariablesReplaced === node.text()) return null;
+        return node.replace(transformedValueWithMetaVariablesReplaced);
     });
 }
+function extractMetaVariables(node, rule) {
+    const pattern = rule.pattern?.toString();
+    if (null == pattern) return {};
+    const metaVarRegex = /\$(\$\$)?([A-Z]+)/g;
+    const patternMetaVars = [];
+    let match;
+    while(null !== (match = metaVarRegex.exec(pattern))){
+        const isMultiple = null != match[1];
+        const varName = match[2];
+        patternMetaVars.push({
+            name: varName,
+            fullMatch: match[0],
+            patternIndex: match.index,
+            isMultiple
+        });
+    }
+    if (0 === patternMetaVars.length) return {};
+    let regexPattern = pattern.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    for (const metaVar of patternMetaVars){
+        const escapedFullMatch = metaVar.fullMatch.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+        regexPattern = metaVar.isMultiple ? regexPattern.replace(escapedFullMatch, '(.*?)') : regexPattern.replace(escapedFullMatch, '(.+?)');
+    }
+    const nodeText = node.text();
+    const textMatch = nodeText.match(new RegExp(regexPattern));
+    if (null == textMatch) return {};
+    const metaVariables = {};
+    for(let index = 0; index < patternMetaVars.length; index += 1){
+        const metaVar = patternMetaVars[index];
+        const capturedValue = textMatch[index + 1];
+        if (!capturedValue) continue;
+        const valueStart = nodeText.indexOf(capturedValue);
+        const valueEnd = valueStart + capturedValue.length;
+        metaVariables[metaVar.name] = {
+            start: valueStart,
+            end: valueEnd,
+            value: capturedValue,
+            original: metaVar.fullMatch
+        };
+    }
+    return metaVariables;
+}
 function findAndReplace(content, rule, transformer) {
     const root = content.root();
     const edits = findAndReplaceEdits(content, rule, transformer);

package/dist/index.js

@@ -118,12 +118,54 @@ function findAndReplaceEdits(content, rule, transformer) {
         rule
     });
     return arrays.compactMap(nodes, (node)=>{
-        const transformed = 'string' == typeof transformer ? transformer : transformer(node);
+        const transformed = 'string' == typeof transformer ? transformer : transformer(node, rule);
         if (null == transformed) return null;
-        if (transformed === node.text()) return null;
-        return node.replace(transformed);
+        const transformedValueWithMetaVariablesReplaced = Object.values(extractMetaVariables(node, rule)).reduce((acc, { original, value })=>acc.replaceAll(original, value), transformed);
+        if (transformedValueWithMetaVariablesReplaced === node.text()) return null;
+        return node.replace(transformedValueWithMetaVariablesReplaced);
     });
 }
+function extractMetaVariables(node, rule) {
+    const pattern = rule.pattern?.toString();
+    if (null == pattern) return {};
+    const metaVarRegex = /\$(\$\$)?([A-Z]+)/g;
+    const patternMetaVars = [];
+    let match;
+    while(null !== (match = metaVarRegex.exec(pattern))){
+        const isMultiple = null != match[1];
+        const varName = match[2];
+        patternMetaVars.push({
+            name: varName,
+            fullMatch: match[0],
+            patternIndex: match.index,
+            isMultiple
+        });
+    }
+    if (0 === patternMetaVars.length) return {};
+    let regexPattern = pattern.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    for (const metaVar of patternMetaVars){
+        const escapedFullMatch = metaVar.fullMatch.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+        regexPattern = metaVar.isMultiple ? regexPattern.replace(escapedFullMatch, '(.*?)') : regexPattern.replace(escapedFullMatch, '(.+?)');
+    }
+    const nodeText = node.text();
+    const textMatch = nodeText.match(new RegExp(regexPattern));
+    if (null == textMatch) return {};
+    const metaVariables = {};
+    for(let index = 0; index < patternMetaVars.length; index += 1){
+        const metaVar = patternMetaVars[index];
+        const capturedValue = textMatch[index + 1];
+        if (!capturedValue) continue;
+        const valueStart = nodeText.indexOf(capturedValue);
+        const valueEnd = valueStart + capturedValue.length;
+        metaVariables[metaVar.name] = {
+            start: valueStart,
+            end: valueEnd,
+            value: capturedValue,
+            original: metaVar.fullMatch
+        };
+    }
+    return metaVariables;
+}
 function findAndReplace(content, rule, transformer) {
     const root = content.root();
     const edits = findAndReplaceEdits(content, rule, transformer);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kamaalio/codemod-kit",
-  "version": "0.0.20",
+  "version": "0.0.22",
   "type": "module",
   "author": "Kamaal Farah",
   "license": "MIT",