fix: Added hack for condition schema in Monaco Editor (#1051)

kestra-io · Mar 9, 2023 · 93f526c · 93f526c
1 parent 3e9486f
commit 93f526c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/core/src/main/java/io/kestra/core/docs/JsonSchemaGenerator.java b/core/src/main/java/io/kestra/core/docs/JsonSchemaGenerator.java
@@ -63,10 +63,12 @@ public <T> Map<String, Object> schemas(Class<? extends T> cls) {
             // hack
             if (cls == Flow.class) {
                 fixFlow(map);
+                fixCondition(map);
             } else if (cls == Task.class) {
                 fixTask(map);
             } else if (cls == AbstractTrigger.class) {
                 fixTrigger(map);
+                fixCondition(map);
             }
 
             return map;
@@ -128,8 +130,15 @@ private static void fixTask(Map<String, Object> map) {
     @SuppressWarnings("unchecked")
     private static void fixTrigger(Map<String, Object> map) {
         var definitions = (Map<String, Map<String, Object>>) map.get("definitions");
-        var task = definitions.get("io.kestra.core.models.triggers.AbstractTrigger-2");
-        var allOf = (List<Object>) task.get("allOf");
+        var trigger = definitions.get("io.kestra.core.models.triggers.AbstractTrigger-2");
+        var allOf = (List<Object>) trigger.get("allOf");
+        allOf.remove(1);
+    }
+
+    private static void fixCondition(Map<String, Object> map) {
+        var definitions = (Map<String, Map<String, Object>>) map.get("definitions");
+        var condition = definitions.get("io.kestra.core.models.conditions.Condition-2");
+        var allOf = (List<Object>) condition.get("allOf");
         allOf.remove(1);
     }