From 3648146f3573fc517652a13e52f88bb9f43ec9c5 Mon Sep 17 00:00:00 2001
From: Alexander Piskun <13381981+bigcat88@users.noreply.github.com>
Date: Sat, 24 Aug 2024 12:17:42 +0300
Subject: [PATCH] added `lazy_loading` to Logic nodes

Signed-off-by: Alexander Piskun <bigcat88@icloud.com>
---
 __init__.py | 40 ++++++++++++++++++++++++++++++++--------
 1 file changed, 32 insertions(+), 8 deletions(-)

diff --git a/__init__.py b/__init__.py
index 78a8db9..3031b7f 100644
--- a/__init__.py
+++ b/__init__.py
@@ -191,8 +191,8 @@ class VixUiCheckboxLogic:
                 "custom_id": ("STRING", {"default": ""}),
             },
             "optional": {
-                "input_off_state": (any_typ,),
-                "input_on_state": (any_typ,),
+                "input_off_state": (any_typ, {"lazy": True}),
+                "input_on_state": (any_typ, {"lazy": True}),
                 "hidden": ("BOOLEAN", {"default": False}),
             },
         }
@@ -208,6 +208,12 @@ class VixUiCheckboxLogic:
             return (kwargs.get("input_off_state", None),)
         return (kwargs.get("input_on_state", None),)
 
+    @staticmethod
+    def check_lazy_status(state, **kwargs):
+        if state is False:
+            return ["input_off_state"]
+        return ["input_on_state"]
+
 
 class VixUiListLogic:
     @classmethod
@@ -223,12 +229,12 @@ class VixUiListLogic:
                 "custom_id": ("STRING", {"default": ""}),
             },
             "optional": {
-                "input_first": (any_typ,),
-                "input_second": (any_typ,),
-                "input_third": (any_typ,),
-                "input_fourth": (any_typ,),
-                "input_fifth": (any_typ,),
-                "input_sixth": (any_typ,),
+                "input_first": (any_typ, {"lazy": True}),
+                "input_second": (any_typ, {"lazy": True}),
+                "input_third": (any_typ, {"lazy": True}),
+                "input_fourth": (any_typ, {"lazy": True}),
+                "input_fifth": (any_typ, {"lazy": True}),
+                "input_sixth": (any_typ, {"lazy": True}),
                 "hidden": ("BOOLEAN", {"default": False}),
             },
         }
@@ -256,6 +262,24 @@ class VixUiListLogic:
             return (kwargs["input_sixth"],)
         raise RuntimeError("Workflow logic error")
 
+    @staticmethod
+    def check_lazy_status(default_value, **kwargs):
+        list_with_values: list = json.loads(kwargs["possible_values"])
+        index_to_return = list_with_values.index(default_value)
+        if index_to_return == 0:
+            return ["input_first"]
+        if index_to_return == 1:
+            return ["input_second"]
+        if index_to_return == 2:
+            return ["input_third"]
+        if index_to_return == 3:
+            return ["input_fourth"]
+        if index_to_return == 4:
+            return ["input_fifth"]
+        if index_to_return == 5:
+            return ["input_sixth"]
+        raise RuntimeError("Workflow logic error")
+
 
 class VixUiWorkflowMetadata:
     @classmethod