From 09c73710c9145afdd22bcdb6da68db8e346e35b6 Mon Sep 17 00:00:00 2001
From: vladlearns <vladlearns@gmail.com>
Date: Wed, 8 Mar 2023 23:00:55 +0200
Subject: [PATCH 1/3] chore: auto update all extensions using scripts

---
 extensions/update-all.bat | 1 +
 extensions/update-all.sh  | 3 +++
 2 files changed, 4 insertions(+)
 create mode 100644 extensions/update-all.bat
 create mode 100644 extensions/update-all.sh

diff --git a/extensions/update-all.bat b/extensions/update-all.bat
new file mode 100644
index 000000000..75a17cbe7
--- /dev/null
+++ b/extensions/update-all.bat
@@ -0,0 +1 @@
+for /d %%i in (*) do @if exist "%%i\.git" (echo Pulling updates for %%i... & git -C "%%i" pull)
\ No newline at end of file
diff --git a/extensions/update-all.sh b/extensions/update-all.sh
new file mode 100644
index 000000000..b00de9278
--- /dev/null
+++ b/extensions/update-all.sh
@@ -0,0 +1,3 @@
+ls | while read dir; do if [ -d "$dir/.git" ]; 
+then echo "Pulling updates for $dir..."; 
+git -C "$dir" pull; fi; done
\ No newline at end of file

From b07b7057f0636aa142471ec27841a8001a85f98b Mon Sep 17 00:00:00 2001
From: vladlearns <vladlearns@gmail.com>
Date: Thu, 9 Mar 2023 16:25:18 +0200
Subject: [PATCH 2/3] chore: removed scripts and added a flag to launch.py

---
 extensions/update-all.bat |  1 -
 extensions/update-all.sh  |  3 ---
 launch.py                 | 14 +++++++++++++-
 3 files changed, 13 insertions(+), 5 deletions(-)
 delete mode 100644 extensions/update-all.bat
 delete mode 100644 extensions/update-all.sh

diff --git a/extensions/update-all.bat b/extensions/update-all.bat
deleted file mode 100644
index 75a17cbe7..000000000
--- a/extensions/update-all.bat
+++ /dev/null
@@ -1 +0,0 @@
-for /d %%i in (*) do @if exist "%%i\.git" (echo Pulling updates for %%i... & git -C "%%i" pull)
\ No newline at end of file
diff --git a/extensions/update-all.sh b/extensions/update-all.sh
deleted file mode 100644
index b00de9278..000000000
--- a/extensions/update-all.sh
+++ /dev/null
@@ -1,3 +0,0 @@
-ls | while read dir; do if [ -d "$dir/.git" ]; 
-then echo "Pulling updates for $dir..."; 
-git -C "$dir" pull; fi; done
\ No newline at end of file
diff --git a/launch.py b/launch.py
index a68bb3a91..ba3067915 100644
--- a/launch.py
+++ b/launch.py
@@ -161,7 +161,15 @@ def git_clone(url, dir, name, commithash=None):
     if commithash is not None:
         run(f'"{git}" -C "{dir}" checkout {commithash}', None, "Couldn't checkout {name}'s hash: {commithash}")
 
-        
+def git_pull_recursive(dir):
+    for subdir, _, _ in os.walk(dir):
+        if os.path.exists(os.path.join(subdir, '.git')):
+            try:
+                output = subprocess.check_output(['git', '-C', subdir, 'pull'])
+                print(f"Pulled changes for repository in '{subdir}':\n{output.decode('utf-8').strip()}\n")
+            except subprocess.CalledProcessError as e:
+                print(f"Couldn't perform 'git pull' on repository in '{subdir}':\n{e.output.decode('utf-8').strip()}\n")
+
 def version_check(commit):
     try:
         import requests
@@ -247,6 +255,7 @@ def prepare_environment():
     args, _ = parser.parse_known_args(sys.argv)
 
     sys.argv, _ = extract_arg(sys.argv, '-f')
+    sys.argv, update_all_extensions = extract_arg(sys.argv, '--update-all-extensions')
     sys.argv, skip_torch_cuda_test = extract_arg(sys.argv, '--skip-torch-cuda-test')
     sys.argv, skip_python_version_check = extract_arg(sys.argv, '--skip-python-version-check')
     sys.argv, reinstall_xformers = extract_arg(sys.argv, '--reinstall-xformers')
@@ -312,6 +321,9 @@ def prepare_environment():
 
     if update_check:
         version_check(commit)
+
+    if update_all_extensions:
+        git_pull_recursive(dir_extensions)
     
     if "--exit" in sys.argv:
         print("Exiting because of --exit argument")

From 13081dd45ece33457f6cb2cad3a8e7840a0a6eaf Mon Sep 17 00:00:00 2001
From: vladlearns <vladlearns@gmail.com>
Date: Thu, 9 Mar 2023 16:56:06 +0200
Subject: [PATCH 3/3] chore: added autostash flag to pull

---
 launch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/launch.py b/launch.py
index ba3067915..8cbf1ca5a 100644
--- a/launch.py
+++ b/launch.py
@@ -165,7 +165,7 @@ def git_pull_recursive(dir):
     for subdir, _, _ in os.walk(dir):
         if os.path.exists(os.path.join(subdir, '.git')):
             try:
-                output = subprocess.check_output(['git', '-C', subdir, 'pull'])
+                output = subprocess.check_output(['git', '-C', subdir, 'pull', '--autostash'])
                 print(f"Pulled changes for repository in '{subdir}':\n{output.decode('utf-8').strip()}\n")
             except subprocess.CalledProcessError as e:
                 print(f"Couldn't perform 'git pull' on repository in '{subdir}':\n{e.output.decode('utf-8').strip()}\n")