@kamaalio/codemod-kit 0.0.21 → 0.0.23

package/dist/codemods/types.d.ts

@@ -1,4 +1,4 @@
-import type { Rule, SgNode, SgRoot } from '@ast-grep/napi';
+import type { Edit, Rule, SgNode, SgRoot } from '@ast-grep/napi';
 import type { NapiLang } from '@ast-grep/napi/types/lang.js';
 import type { Kinds, TypesMap } from '@ast-grep/napi/types/staticTypes.js';
 import type { Optional } from '../utils/type-utils.js';
@@ -19,5 +19,5 @@ export type Modifications = {
 };
 export type FindAndReplaceConfig = {
     rule: Rule<TypesMap>;
-    transformer: ((node: SgNode<TypesMap, Kinds<TypesMap>>, rule: Rule<TypesMap>) => Optional<string>) | string;
+    transformer: ((node: SgNode<TypesMap, Kinds<TypesMap>>, rule: Rule<TypesMap>) => Optional<string> | Array<Edit | string>) | string;
 };

package/dist/index.cjs

@@ -166,9 +166,57 @@ function findAndReplaceEdits(content, rule, transformer) {
     return kamaal_namespaceObject.arrays.compactMap(nodes, (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 edits = [];
+        const valuesToTransform = [];
+        if (Array.isArray(transformed)) for (const item of transformed)if ('string' == typeof item) valuesToTransform.push(item);
+        else edits.push(item);
+        const metaVariables = Object.values(extractMetaVariables(node, rule));
+        return kamaal_namespaceObject.arrays.compactMap(valuesToTransform.map((transformedValue)=>metaVariables.reduce((acc, { original, value })=>acc.replaceAll(original, value), transformedValue)), (transformedValueWithMetaVariablesReplaced)=>{
+            if (transformedValueWithMetaVariablesReplaced === node.text()) return null;
+            return node.replace(transformedValueWithMetaVariablesReplaced);
+        }).concat(edits);
+    }).flat(1);
+}
+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();

package/dist/index.js

@@ -120,9 +120,57 @@ function findAndReplaceEdits(content, rule, transformer) {
     return arrays.compactMap(nodes, (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 edits = [];
+        const valuesToTransform = [];
+        if (Array.isArray(transformed)) for (const item of transformed)if ('string' == typeof item) valuesToTransform.push(item);
+        else edits.push(item);
+        const metaVariables = Object.values(extractMetaVariables(node, rule));
+        return arrays.compactMap(valuesToTransform.map((transformedValue)=>metaVariables.reduce((acc, { original, value })=>acc.replaceAll(original, value), transformedValue)), (transformedValueWithMetaVariablesReplaced)=>{
+            if (transformedValueWithMetaVariablesReplaced === node.text()) return null;
+            return node.replace(transformedValueWithMetaVariablesReplaced);
+        }).concat(edits);
+    }).flat(1);
+}
+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();

package/package.json

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