From 4c87c35af4e4b731e9679c15a10f79394b2a5b74 Mon Sep 17 00:00:00 2001
From: ltdrdata <dr.lt.data@gmail.com>
Date: Sun, 21 May 2023 08:45:54 +0900
Subject: [PATCH] missing file fix

---
 __init__.py   |  2 +-
 git_helper.py | 54 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+), 1 deletion(-)
 create mode 100644 git_helper.py

diff --git a/__init__.py b/__init__.py
index c85336d..1f5b8aa 100644
--- a/__init__.py
+++ b/__init__.py
@@ -16,7 +16,7 @@ sys.path.append('../..')
 from torchvision.datasets.utils import download_url
 
 # ensure .js
-print("### Loading: ComfyUI-Manager (V0.5)")
+print("### Loading: ComfyUI-Manager (V0.5.1)")
 
 comfy_path = os.path.dirname(folder_paths.__file__)
 custom_nodes_path = os.path.join(comfy_path, 'custom_nodes')
diff --git a/git_helper.py b/git_helper.py
new file mode 100644
index 0000000..d5bd794
--- /dev/null
+++ b/git_helper.py
@@ -0,0 +1,54 @@
+import sys
+import os
+import git
+
+def gitclone(custom_nodes_path, url):
+    repo_name = os.path.splitext(os.path.basename(url))[0]
+    repo_path = os.path.join(custom_nodes_path, repo_name)
+
+    # Clone the repository from the remote URL
+    repo = git.Repo.clone_from(url, repo_path)
+    repo.git.clear_cache()
+    repo.close()
+
+def gitcheck(path):
+    # Fetch the latest commits from the remote repository
+    repo = git.Repo(path)
+    remote_name = 'origin'
+    remote = repo.remote(name=remote_name)
+    remote.fetch()
+
+    # Get the current commit hash and the commit hash of the remote branch
+    commit_hash = repo.head.commit.hexsha
+    remote_commit_hash = repo.refs[f'{remote_name}/HEAD'].object.hexsha
+
+    # Compare the commit hashes to determine if the local repository is behind the remote repository
+    if commit_hash != remote_commit_hash:
+        print("CUSTOM NODE CHECK: True")
+    else:
+        print("CUSTOM NODE CHECK: False")
+
+def gitpull(path):
+    # Check if the path is a git repository
+    if not os.path.exists(os.path.join(path, '.git')):
+        raise ValueError('Not a git repository')
+
+    # Pull the latest changes from the remote repository
+    repo = git.Repo(path)
+    origin = repo.remote(name='origin')
+    origin.pull()
+
+    repo.close()
+
+try:
+    if sys.argv[1] == "--clone":
+        gitclone(sys.argv[2], sys.argv[3])
+    elif sys.argv[1] == "--check":
+        gitcheck(sys.argv[2])
+    elif sys.argv[1] == "--pull":
+        gitpull(sys.argv[2])
+    exit(0)
+except:
+    exit(-1)
+    
+