From e84c5fce379ef8925a749960e90754487b244f82 Mon Sep 17 00:00:00 2001
From: ByteXenon <125568681+ByteXenon@users.noreply.github.com>
Date: Mon, 4 Nov 2024 02:22:05 -0700
Subject: [PATCH] Add `--pull-request` [`-pr`] parameter

---
 sherlock_project/sherlock.py | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/sherlock_project/sherlock.py b/sherlock_project/sherlock.py
index e36ca6cc..ec5e703f 100644
--- a/sherlock_project/sherlock.py
+++ b/sherlock_project/sherlock.py
@@ -655,6 +655,14 @@ def main():
         default=None,
         help="Load data from a JSON file or an online, valid, JSON file.",
     )
+    parser.add_argument(
+        "--pull-request",
+        "-pr",
+        action="store",
+        dest="pull_request",
+        default=None,
+        help="Load data.json file from a specific pull request.",
+    )
     parser.add_argument(
         "--timeout",
         action="store",
@@ -779,12 +787,25 @@ def main():
 
     # Create object with all information about sites we are aware of.
     try:
-        if args.local:
+        if args.local and args.pull_request is None:
             sites = SitesInformation(
                 os.path.join(os.path.dirname(__file__), "resources/data.json")
             )
         else:
-            sites = SitesInformation(args.json_file)
+            # Check for conflicting --json and --pull-request arguments
+            if args.json_file is not None and args.pull_request is not None:
+                print("You cannot use both --json and --pull-request at the same time.")
+                sys.exit(1)
+
+            json_file_location = args.json_file
+            if args.pull_request is not None:
+                pull_url = f"https://api.github.com/repos/sherlock-project/sherlock/pulls/{args.pull_request}"
+                pull_request_raw = requests.get(pull_url).text
+                pull_request_json = json_loads(pull_request_raw)
+                head_commit_sha = pull_request_json["head"]["sha"]
+                json_file_location = f"https://raw.githubusercontent.com/sherlock-project/sherlock/{head_commit_sha}/sherlock_project/resources/data.json"
+
+            sites = SitesInformation(json_file_location)
     except Exception as error:
         print(f"ERROR:  {error}")
         sys.exit(1)