From 93f526c96d6757e678576b605db3c4a24a23da69 Mon Sep 17 00:00:00 2001
From: YannC <37600690+Skraye@users.noreply.github.com>
Date: Thu, 9 Mar 2023 15:50:38 +0100
Subject: [PATCH] fix: Added hack for condition schema in Monaco Editor (#1051)

---
 .../io/kestra/core/docs/JsonSchemaGenerator.java    | 13 +++++++++++--
 1 file changed, 11 insertions(+), 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
index 996aedf5cfc..ed2df4dabed 100644
--- 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);
     }