From 2e90b86a596d8548a2fad1aa6d3b70ae00c8f451 Mon Sep 17 00:00:00 2001
From: Raqueebuddin Aziz <raqueebuddin.aziz.314@gmail.com>
Date: Fri, 1 Dec 2023 07:33:50 +0530
Subject: [PATCH] add lifecycle scripts

---
 __init__.py  | 26 ++++++++------------------
 disable.py   | 14 ++++++++++++++
 enable.py    | 16 ++++++++++++++++
 install.py   | 16 ++++++++++++++++
 uninstall.py | 14 ++++++++++++++
 5 files changed, 68 insertions(+), 18 deletions(-)
 create mode 100644 disable.py
 create mode 100644 enable.py
 create mode 100644 install.py
 create mode 100644 uninstall.py

diff --git a/__init__.py b/__init__.py
index 42d2afd..ebb1123 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,22 +1,12 @@
-from .wildcard import *
+"""
+@author: VanillaCode314
+@title: Simple Wildcard
+@nickname: Simple Wildcard
+@description: A simple wildcard node for ComfyUI. Can also be used a style prompt node.
+"""
 
-try:
-    import os, shutil
-    import folder_paths
-
-    module_js_directory = os.path.join(
-        os.path.dirname(os.path.realpath(__file__)), "js"
-    )
-    application_root_directory = os.path.dirname(folder_paths.__file__)
-    application_web_extensions_directory = os.path.join(
-        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
-    )
-    shutil.copytree(
-        module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
-    )
-except:
-    pass
+from .wildcard import Wildcard
 
 NODE_CLASS_MAPPINGS = {"SimpleWildcard": Wildcard}
-NODE_DISPLAY_NAME_MAPPINGS = {"SimpleWildcard": "Wildcard"}
+NODE_DISPLAY_NAME_MAPPINGS = {"SimpleWildcard": "SimpleWildcard"}
 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
diff --git a/disable.py b/disable.py
new file mode 100644
index 0000000..6d12d72
--- /dev/null
+++ b/disable.py
@@ -0,0 +1,14 @@
+try:
+    import os, shutil
+    import folder_paths
+
+    module_js_directory = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)), "js"
+    )
+    application_root_directory = os.path.dirname(folder_paths.__file__)
+    application_web_extensions_directory = os.path.join(
+        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
+    )
+    shutil.rmtree(application_web_extensions_directory)
+except:
+    pass
diff --git a/enable.py b/enable.py
new file mode 100644
index 0000000..00ac25a
--- /dev/null
+++ b/enable.py
@@ -0,0 +1,16 @@
+try:
+    import os, shutil
+    import folder_paths
+
+    module_js_directory = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)), "js"
+    )
+    application_root_directory = os.path.dirname(folder_paths.__file__)
+    application_web_extensions_directory = os.path.join(
+        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
+    )
+    shutil.copytree(
+        module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
+    )
+except:
+    pass
diff --git a/install.py b/install.py
new file mode 100644
index 0000000..00ac25a
--- /dev/null
+++ b/install.py
@@ -0,0 +1,16 @@
+try:
+    import os, shutil
+    import folder_paths
+
+    module_js_directory = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)), "js"
+    )
+    application_root_directory = os.path.dirname(folder_paths.__file__)
+    application_web_extensions_directory = os.path.join(
+        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
+    )
+    shutil.copytree(
+        module_js_directory, application_web_extensions_directory, dirs_exist_ok=True
+    )
+except:
+    pass
diff --git a/uninstall.py b/uninstall.py
new file mode 100644
index 0000000..6d12d72
--- /dev/null
+++ b/uninstall.py
@@ -0,0 +1,14 @@
+try:
+    import os, shutil
+    import folder_paths
+
+    module_js_directory = os.path.join(
+        os.path.dirname(os.path.realpath(__file__)), "js"
+    )
+    application_root_directory = os.path.dirname(folder_paths.__file__)
+    application_web_extensions_directory = os.path.join(
+        application_root_directory, "web", "extensions", "vanilla.simple.wildcard"
+    )
+    shutil.rmtree(application_web_extensions_directory)
+except:
+    pass