Execute run_antlr4 same as ya make

Execute run_antlr4 same as ya make
commit_hash:069135ba18f9d8584552ddfcd755d6f7edb9c0be

2 files changed, 10 insertions, 1 deletions

diff --git a/build/export_generators/ide-gradle/codegen_run_program.jinja b/build/export_generators/ide-gradle/codegen_run_program.jinja
index 112391cd870..64f2f399517 100644
--- a/build/export_generators/ide-gradle/codegen_run_program.jinja
+++ b/build/export_generators/ide-gradle/codegen_run_program.jinja
@@ -10,14 +10,22 @@ val {{ varprefix }}{{ custom_run['_object_index'] }} = tasks.register<Exec>("{{
     workingDir({{ PatchRoots(custom_run.cwd, false, custom_run['_object_index']) }})
 {%-        else %}
     workingDir({{ PatchRoots(codegenBindir, false, custom_run['_object_index']) }})
+{%-        endif -%}
+
+{%-        set args = custom_run.command -%}
+{%-        if args|first == "run_antlr4" -%}
+{%-            set args = [ jdk_path + "/bin/java", "-jar", custom_run.jar ] + args|reject('eq', "run_antlr4") -%}
 {%-        endif %}
 
     commandLine("bash", "-c", listOf(
-{%-        for arg in custom_run.command %}
+{%-        for arg in args %}
         {{ PatchArg(arg, custom_run['_object_index'], custom_run.depends, custom_run.outputs) }},
 {%-        endfor %}
     ).joinToString(" "))
 
+{%-        if custom_run.command|first == "run_antlr4" and custom_run.jar %}
+    inputs.files({{ PatchRoots(custom_run.jar, true) }})
+{%-        endif -%}
 {%-        if custom_run.depends|length -%}
 {%-            for depend in custom_run.depends|unique %}
     inputs.files({{ PatchRoots(depend, true) }})
diff --git a/build/export_generators/ide-gradle/generator.toml b/build/export_generators/ide-gradle/generator.toml
index f398fe85fae..a049d21df3e 100644
--- a/build/export_generators/ide-gradle/generator.toml
+++ b/build/export_generators/ide-gradle/generator.toml
@@ -109,6 +109,7 @@ custom_runs-cmake_packages="list"
 custom_runs-cmake_packages-ITEM="dict"
 custom_runs-cmake_packages-name="str"
 custom_runs-cmake_packages-components="list"
+custom_runs-jar="str"
 
 javac-flags="list"
 kotlinc-flags="list"