From 4ab45d72a8892a9e2bf755c26e250c1ee338767c Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Wed, 26 Jan 2022 14:44:04 +0100
Subject: [PATCH] litex_setup: Add --status argument to display Git status of
 repositories.

---
 litex_setup.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/litex_setup.py b/litex_setup.py
index 192e7433e..a29897b3b 100755
--- a/litex_setup.py
+++ b/litex_setup.py
@@ -229,6 +229,17 @@ def litex_setup_install_repos(config="standard", user_mode=False):
             print_status("Make sure that ~/.local/bin is in your PATH")
             print_status("export PATH=$PATH:~/.local/bin")
 
+# Git repositories status --------------------------------------------------------------------------
+
+def litex_setup_status_repos(config="standard"):
+    print_status("Getting status of Git repositories...", underline=True)
+    for name in install_configs[config]:
+        repo = git_repos[name]
+        os.chdir(os.path.join(current_path, name))
+        git_sha1_cmd = ["git", "rev-parse", "--short=7", "HEAD"]
+        git_sha1     = subprocess.check_output(git_sha1_cmd).decode("UTF-8")
+        print(f"{name}: sha1=0x{git_sha1}", end="")
+
 # GCC toolchains download --------------------------------------------------------------------------
 
 def gcc_toolchain_download(url, filename):
@@ -310,6 +321,7 @@ def main():
     parser.add_argument("--install",   action="store_true", help="Install Git repositories.")
     parser.add_argument("--user",      action="store_true", help="Install in User-Mode.")
     parser.add_argument("--config",    default="standard",  help="Install config (minimal, standard, full).")
+    parser.add_argument("--status",    action="store_true", help="Display Git status of repositories.")
 
     # GCC toolchains.
     parser.add_argument("--gcc", default=None, help="Download/Extract GCC Toolchain (riscv, powerpc, openrisc or lm32).")
@@ -346,6 +358,10 @@ def main():
     if args.install:
         litex_setup_install_repos(config=args.config, user_mode=args.user)
 
+    # Status.
+    if args.status:
+        litex_setup_status_repos(config=args.config)
+
     # GCC.
     os.chdir(os.path.join(current_path))
     if args.gcc == "riscv":