feat: support better reboot mechanism based on comfy-cli

glob/manager_core.py

@@ -21,7 +21,7 @@ sys.path.append(glob_path)
 import cm_global
 from manager_util import *
 
-version = [2, 21, 3]
+version = [2, 22]
 version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')
 
 comfyui_manager_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))

glob/manager_server.py

@@ -1112,6 +1112,14 @@ def restart(self):
     except Exception as e:
         pass
 
+    if '__COMFY_CLI_SESSION__' in os.environ:
+        with open(os.path.join(os.environ['__COMFY_CLI_SESSION__'], '.reboot'), 'w') as file:
+            pass
+
+        print(f"\nRestarting...\n\n")
+        exit(0)
+
+    print(f"\nRestarting... [Legacy Mode]\n\n")
     return os.execv(sys.executable, [sys.executable] + sys.argv)