From 081a0408c8babc343a23872af69fdc8db4aa3e91 Mon Sep 17 00:00:00 2001
From: nkchocoai <nkai3359@gmail.com>
Date: Mon, 22 Jan 2024 16:21:17 +0900
Subject: [PATCH] Add ConstString node

---
 __init__.py |  4 ++++
 py/node.py  | 44 +++++++++++++++++++++++++++++++++++---------
 2 files changed, 39 insertions(+), 9 deletions(-)

diff --git a/__init__.py b/__init__.py
index 9f2f1c5..2931ca5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -4,12 +4,16 @@ NODE_CLASS_MAPPINGS = {
     "PromptUtilitiesFormatString": PromptUtilitiesFormatString,
     "PromptUtilitiesJoinStringList": PromptUtilitiesJoinStringList,
     "PromptUtilitiesLoadPreset": PromptUtilitiesLoadPreset,
+    "PromptUtilitiesConstString": PromptUtilitiesConstString,
+    "PromptUtilitiesConstStringMultiLine": PromptUtilitiesConstStringMultiLine,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "PromptUtilitiesFormatString": "Format String",
     "PromptUtilitiesJoinStringList": "Join String List",
     "PromptUtilitiesLoadPreset": "Load Preset",
+    "PromptUtilitiesConstString": "Const String",
+    "PromptUtilitiesConstStringMultiLine": "Const String(multi line)",
 }
 
 WEB_DIRECTORY = "./js"
diff --git a/py/node.py b/py/node.py
index a3669a1..3d37494 100644
--- a/py/node.py
+++ b/py/node.py
@@ -2,7 +2,39 @@ import folder_paths
 
 from .preset import PresetManager
 
-class PromptUtilitiesFormatString:
+class BaseNode:
+    CATEGORY = "PromptUtilities"
+
+
+class PromptUtilitiesConstStringBase(BaseNode):
+    RETURN_TYPES = ("STRING",)
+    FUNCTION = "get_string"
+
+    def get_string(self, string):
+        return (string,)
+
+
+class PromptUtilitiesConstString(PromptUtilitiesConstStringBase):
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "string": ("STRING", {"default": "", "multiline": False})
+            }
+        }
+
+
+class PromptUtilitiesConstStringMultiLine(PromptUtilitiesConstStringBase):
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "string": ("STRING", {"default": "", "multiline": True})
+            }
+        }
+
+
+class PromptUtilitiesFormatString(BaseNode):
     @classmethod
     def INPUT_TYPES(s):
         input_types = {
@@ -22,8 +54,6 @@ class PromptUtilitiesFormatString:
     RETURN_TYPES = ("STRING",)
     FUNCTION = "format"
 
-    CATEGORY = "PromptUtilities"
-
     def format(self, prompt, **kwargs):
         print(kwargs)
         result = prompt
@@ -32,7 +62,7 @@ class PromptUtilitiesFormatString:
         return (result,)
 
 
-class PromptUtilitiesJoinStringList:
+class PromptUtilitiesJoinStringList(BaseNode):
     @classmethod
     def INPUT_TYPES(s):
         input_types = {
@@ -49,8 +79,6 @@ class PromptUtilitiesJoinStringList:
     RETURN_TYPES = ("STRING",)
     FUNCTION = "join"
 
-    CATEGORY = "PromptUtilities"
-
     def join(self, separator, **kwargs):
         PresetManager.load_presets()
         print(kwargs)
@@ -59,7 +87,7 @@ class PromptUtilitiesJoinStringList:
         return (result,)
 
 
-class PromptUtilitiesLoadPreset:
+class PromptUtilitiesLoadPreset(BaseNode):
     @classmethod
     def INPUT_TYPES(s):
         return {
@@ -71,8 +99,6 @@ class PromptUtilitiesLoadPreset:
     RETURN_TYPES = ("STRING",)
     FUNCTION = "load_preset"
 
-    CATEGORY = "PromptUtilities"
-
     def load_preset(self, preset):
         prompt = PresetManager.get_prompt(preset)
         return (prompt,)
\ No newline at end of file