Add the 'create_from_template' method to rewriting context

Author: HugoGGuerrier

lkql_jit/language/src/main/java/com/adacore/lkql_jit/built_ins/methods/RewritingContextMethods.java

@@ -6,6 +6,7 @@
 package com.adacore.lkql_jit.built_ins.methods;
 
 import com.adacore.libadalang.Libadalang;
+import com.adacore.libadalang.Libadalang.GrammarRule;
 import com.adacore.libadalang.Libadalang.MemberReference;
 import com.adacore.libadalang.Libadalang.RewritingContext;
 import com.adacore.libadalang.Libadalang.RewritingNode;
@@ -15,7 +16,9 @@
 import com.adacore.lkql_jit.exception.LKQLRuntimeException;
 import com.adacore.lkql_jit.nodes.utils.RewritingNodeConverter;
 import com.adacore.lkql_jit.nodes.utils.RewritingNodeConverterNodeGen;
+import com.adacore.lkql_jit.runtime.values.lists.LKQLList;
 import com.adacore.lkql_jit.utils.LKQLTypesHelper;
+import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import com.oracle.truffle.api.dsl.Specialization;
 import com.oracle.truffle.api.frame.VirtualFrame;
 
@@ -181,4 +184,39 @@ public Object executeGeneric(VirtualFrame frame, RewritingContext ctx, Object ob
             return ctx;
         }
     }
+
+    @BuiltInMethod(
+        name = "create_from_template",
+        doc = "Create a new node from the provided template, filling '{}' with provided" +
+        " argument, and parsing the template with the specified grammar rule"
+    )
+    public abstract static class CreateFromTemplateExpr extends BaseRewritingContextExpr {
+
+        @TruffleBoundary
+        private static GrammarRule ruleFromString(String ruleName) {
+            return GrammarRule.valueOf(ruleName.toUpperCase());
+        }
+
+        @Specialization
+        public RewritingNode doGeneric(
+            VirtualFrame frame,
+            RewritingContext ctx,
+            String template,
+            String grammarRule,
+            LKQLList arguments
+        ) {
+            // Translate the provided LKQL list into a rewriting node list
+            final var args = new RewritingNode[(int) arguments.size()];
+            for (int i = 0; i < args.length; i++) {
+                args[i] = convert(frame, arguments.get(i), true);
+            }
+
+            // Then call the internal function to process the template
+            try {
+                return ctx.createFromTemplate(template, ruleFromString(grammarRule), args);
+            } catch (Exception e) {
+                throw LKQLRuntimeException.fromJavaException(e, this.callNode);
+            }
+        }
+    }
 }

testsuite/tests/rewriting/rewriting_api/create_from_template/custom.lkql

@@ -0,0 +1,8 @@
+@check(
+    message="Custom rule violation",
+    auto_fix=(n, ctx) => ctx.replace(
+        n,
+        ctx.create_from_template("Call_Something ({});", "call_stmt_rule", [n.f_label_name])
+    )
+)
+fun custom(node) = node is GotoStmt

testsuite/tests/rewriting/rewriting_api/create_from_template/main.adb

@@ -0,0 +1,4 @@
+procedure Main is
+begin
+   goto lbl;   --  FLAG
+end Main;

testsuite/tests/rewriting/rewriting_api/create_from_template/test.out

@@ -0,0 +1,11 @@
+main.adb:3:4: rule violation: Custom rule violation
+3 |    goto lbl;   --  FLAG
+  |    ^^^^^^^^^
+
+Patched "main.adb":
+===================
+
+procedure Main is
+begin
+   Call_Something(lbl);end Main;
+

testsuite/tests/rewriting/rewriting_api/create_from_template/test.yaml

@@ -0,0 +1,5 @@
+driver: checker
+input_sources:
+  - main.adb
+rule_name: custom
+auto_fix: True