From a6a30ca42b89378c93c406a8f42ebb421c749da9 Mon Sep 17 00:00:00 2001
From: "kyj@bowong.ai" <kyj@bowong.ai>
Date: Thu, 20 Mar 2025 16:40:35 +0800
Subject: [PATCH] =?UTF-8?q?ADD=20=E5=A2=9E=E5=8A=A0=E8=87=AA=E5=AE=9A?=
 =?UTF-8?q?=E4=B9=89=E7=BC=96=E7=A0=81=E8=AF=BB=E5=8F=96=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 __init__.py              |   5 +-
 nodes/text.py            | 107 +++++++++++++++++++++++++++++++++++++++
 user/text_file_dirs.json |   5 ++
 3 files changed, 116 insertions(+), 1 deletion(-)
 create mode 100644 nodes/text.py
 create mode 100644 user/text_file_dirs.json

diff --git a/__init__.py b/__init__.py
index 545fd1c..f4b0af7 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,3 +1,4 @@
+from .nodes.text import LoadText
 from .nodes.traverse_folder import TraverseFolder
 from .nodes.unload_all_models import UnloadAllModels
 from .nodes.string_empty_judgement import StringEmptyJudgement
@@ -23,6 +24,7 @@ NODE_CLASS_MAPPINGS = {
     "StringEmptyJudgement": StringEmptyJudgement,
     "unloadAllModels": UnloadAllModels,
     "TraverseFolder": TraverseFolder,
+    "LoadTextCustom": LoadText
 }
 
 # A dictionary that contains the friendly/humanly readable titles for the nodes
@@ -37,5 +39,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "VideoPointCompute": "视频帧位计算",
     "StringEmptyJudgement": "字符串是否为空",
     "unloadAllModels": "卸载所有已加载模型",
-    "TraverseFolder": "遍历文件夹"
+    "TraverseFolder": "遍历文件夹",
+    "LoadTextCustom": "读取文本文件"
 }
diff --git a/nodes/text.py b/nodes/text.py
new file mode 100644
index 0000000..1facc7a
--- /dev/null
+++ b/nodes/text.py
@@ -0,0 +1,107 @@
+import json
+import os
+import folder_paths
+
+
+def get_allowed_dirs():
+    dir = os.path.abspath(os.path.join(__file__, "../../user"))
+    file = os.path.join(dir, "text_file_dirs.json")
+    with open(file, "r") as f:
+        return json.loads(f.read())
+
+
+def get_valid_dirs():
+    return get_allowed_dirs().keys()
+
+
+def get_dir_from_name(name):
+    dirs = get_allowed_dirs()
+    if name not in dirs:
+        raise KeyError(name + " dir not found")
+
+    path = dirs[name]
+    path = path.replace("$input", folder_paths.get_input_directory())
+    path = path.replace("$output", folder_paths.get_output_directory())
+    path = path.replace("$temp", folder_paths.get_temp_directory())
+    return path
+
+
+def is_child_dir(parent_path, child_path):
+    parent_path = os.path.abspath(parent_path)
+    child_path = os.path.abspath(child_path)
+    return os.path.commonpath([parent_path]) == os.path.commonpath([parent_path, child_path])
+
+
+def get_real_path(dir):
+    dir = dir.replace("/**/", "/")
+    dir = os.path.abspath(dir)
+    dir = os.path.split(dir)[0]
+    return dir
+
+
+def get_file(root_dir, file):
+    if file == "[none]" or not file or not file.strip():
+        raise ValueError("No file")
+
+    root_dir = get_dir_from_name(root_dir)
+    root_dir = get_real_path(root_dir)
+    if not os.path.exists(root_dir):
+        os.mkdir(root_dir)
+    full_path = os.path.join(root_dir, file)
+
+    if not is_child_dir(root_dir, full_path):
+        raise ReferenceError()
+
+    return full_path
+
+
+class LoadText:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "root_dir": (list(get_valid_dirs()), {}),
+                "file": (["[none]"], {
+                    "pysssss.binding": [{
+                        "source": "root_dir",
+                        "callback": [{
+                            "type": "set",
+                            "target": "$this.disabled",
+                            "value": True
+                        }, {
+                            "type": "fetch",
+                            "url": "/pysssss/text-file/{$source.value}",
+                            "then": [{
+                                "type": "set",
+                                "target": "$this.options.values",
+                                "value": "$result"
+                            }, {
+                                "type": "validate-combo"
+                            }, {
+                                "type": "set",
+                                "target": "$this.disabled",
+                                "value": False
+                            }]
+                        }],
+                    }]
+                }),
+                "encoding": ("STRING", {"default": "utf-8"}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING",)
+
+    FUNCTION = "load"
+
+    CATEGORY = "不忘科技-自定义节点🚩"
+
+    @classmethod
+    def VALIDATE_INPUTS(self, root_dir, file, **kwargs):
+        if file == "[none]" or not file or not file.strip():
+            return True
+        get_file(root_dir, file)
+        return True
+
+    def load(self, root_dir, file, encoding):
+        with open(get_file(root_dir,file), "r", encoding=encoding) as f:
+            return (f.read(),)
diff --git a/user/text_file_dirs.json b/user/text_file_dirs.json
new file mode 100644
index 0000000..15adc22
--- /dev/null
+++ b/user/text_file_dirs.json
@@ -0,0 +1,5 @@
+{
+	"input": "$input/**/*.txt",
+	"output": "$output/**/*.txt",
+	"temp": "$temp/**/*.txt"
+}