From 180102bf176596e99c4dd79a40918e1a911114a8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8D=82=EF=BE=8C=E5=8D=82=E3=84=9A=20=D2=9C=E3=84=96?=
 =?UTF-8?q?=E5=8D=A9=E5=8D=A9=E5=8D=82=D2=9C=E5=8D=82?=
 <koppakaajay123@gmail.com>
Date: Sun, 16 Jun 2024 16:17:16 +0530
Subject: [PATCH] Auto-update dependencies after a git pull (#2203)

* Added a file to mark dev installation and also modified gdbinit.py to calculate and verify the hash of poetry.lock

---------

Co-authored-by: B1N4RY-P4R45173 <kopakaajay123@gmail.com>
Co-authored-by: Disconnect3d <dominik.b.czarnota@gmail.com>
---
 gdbinit.py   | 57 ++++++++++++++++++++++++++++++++++++++++++++++++----
 setup-dev.sh |  5 +++++
 2 files changed, 58 insertions(+), 4 deletions(-)

diff --git a/gdbinit.py b/gdbinit.py
index ca277046d..16d14fc70 100644
--- a/gdbinit.py
+++ b/gdbinit.py
@@ -1,8 +1,10 @@
 from __future__ import annotations
 
 import cProfile
+import hashlib
 import os
 import site
+import subprocess
 import sys
 import time
 from glob import glob
@@ -18,9 +20,55 @@ if environ.get("PWNDBG_PROFILE") == "1":
 
 # Get virtualenv's site-packages path
 venv_path = os.environ.get("PWNDBG_VENV_PATH")
-if venv_path == "PWNDBG_PLEASE_SKIP_VENV" or path.exists(path.dirname(__file__) + "/.skip-venv"):
-    pass
-else:
+
+
+def calculate_hash(file_path):
+    with open(file_path, "rb") as f:
+        file_hash = hashlib.sha256()
+        while True:
+            chunk = f.read(8192)
+            if not chunk:
+                break
+            file_hash.update(chunk)
+    return file_hash.hexdigest()
+
+
+def run_poetry_install(dev=False):
+    command = ["poetry", "install"]
+    if dev:
+        command.extend(("--with", "dev"))
+    result = subprocess.run(command, capture_output=True, text=True)
+    return result.stdout.strip(), result.stderr.strip(), result.returncode
+
+
+def update_deps(file_path):
+    poetry_lock_path = os.path.join(os.path.dirname(file_path), "poetry.lock")
+    poetry_lock_hash_path = os.path.join(venv_path, "poetry.lock.hash")
+    dev_marker_path = os.path.join(venv_path, "dev.marker")
+
+    current_hash = calculate_hash(poetry_lock_path)
+    stored_hash = None
+    if os.path.exists(poetry_lock_hash_path):
+        with open(poetry_lock_hash_path, "r") as f:
+            stored_hash = f.read().strip()
+
+    # checks if dev.marker exists
+    dev_mode = os.path.exists(dev_marker_path)
+
+    # if hashes don't match, run the appropriate command based on dev.marker file
+    if current_hash != stored_hash:
+        stdout, stderr, returncode = run_poetry_install(dev=dev_mode)
+        with open(poetry_lock_hash_path, "w") as f:
+            f.write(current_hash)
+        if returncode == 0 and ("No dependencies to install or update" not in stdout):
+            print(stdout)
+        elif returncode != 0:
+            print(stderr, file=sys.stderr)
+
+
+if venv_path != "PWNDBG_PLEASE_SKIP_VENV" and not path.exists(
+    path.dirname(__file__) + "/.skip-venv"
+):
     directory, file = path.split(__file__)
     directory = path.expanduser(directory)
     directory = path.abspath(directory)
@@ -32,6 +80,8 @@ else:
         print(f"Cannot find Pwndbg virtualenv directory: {venv_path}: please re-run setup.sh")
         sys.exit(1)
 
+    update_deps(__file__)
+
     site_pkgs_path = glob(os.path.join(venv_path, "lib/*/site-packages"))[0]
 
     # add virtualenv's site-packages to sys.path and run .pth files
@@ -55,7 +105,6 @@ else:
 
 # Force UTF-8 encoding (to_string=True to skip output appearing to the user)
 gdb.execute("set charset UTF-8", to_string=True)
-
 environ["PWNLIB_NOTERM"] = "1"
 
 import pwndbg  # noqa: F401
diff --git a/setup-dev.sh b/setup-dev.sh
index be02cfc38..8007ba0fc 100755
--- a/setup-dev.sh
+++ b/setup-dev.sh
@@ -223,4 +223,9 @@ if linux; then
 
     source "${PWNDBG_VENV_PATH}/bin/activate"
     ~/.local/bin/poetry install --with dev
+
+    # Create a developer marker file
+    DEV_MARKER_PATH="${PWNDBG_VENV_PATH}/dev.marker"
+    touch "${DEV_MARKER_PATH}"
+    echo "Developer marker created at ${DEV_MARKER_PATH}"
 fi