From 58c37ea7858777220af52a21d512a8c75050223d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Emilio=20L=C3=B3pez?= <emilio.lopez@trailofbits.com>
Date: Tue, 27 Sep 2022 19:03:54 -0300
Subject: [PATCH] Version check

---
 slither/tools/doctor/__main__.py | 45 ++++++++++++++++++++++++++++----
 slither/tools/doctor/packages.py | 24 +++++++++++++++++
 2 files changed, 64 insertions(+), 5 deletions(-)
 create mode 100644 slither/tools/doctor/packages.py

diff --git a/slither/tools/doctor/__main__.py b/slither/tools/doctor/__main__.py
index 74f854672..d1d73df2f 100644
--- a/slither/tools/doctor/__main__.py
+++ b/slither/tools/doctor/__main__.py
@@ -4,6 +4,8 @@ from pathlib import Path
 
 from crytic_compile import cryticparser
 import crytic_compile.crytic_compile as crytic_compile
+
+from slither.tools.doctor.packages import get_installed_version, get_github_version
 from slither.utils.colors import red, yellow, green
 
 
@@ -25,9 +27,43 @@ def parse_args():
     return parser.parse_args()
 
 
+def show_versions():
+    print("## Software versions", end="\n\n")
+    versions = {
+        "Slither": (get_installed_version("slither-analyzer"), get_github_version("slither")),
+        "crytic-compile": (
+            get_installed_version("crytic-compile"),
+            get_github_version("crytic-compile"),
+        ),
+        "solc-select": (get_installed_version("solc-select"), get_github_version("solc-select")),
+    }
+
+    outdated = {
+        name
+        for name, (installed, latest) in versions.items()
+        if not installed or not latest or latest > installed
+    }
+
+    for name, (installed, latest) in versions.items():
+        color = yellow if name in outdated else green
+        print(f"{name + ':':<16}{color(installed or 'N/A'):<16} (latest is {latest or 'Unknown'})")
+
+    if len(outdated) > 0:
+        print()
+        print(
+            yellow(
+                f"Please update {', '.join(outdated)} to the latest release before creating a bug report."
+            )
+        )
+    else:
+        print()
+        print(green("Your tools are up to date."))
+
+    print(end="\n\n")
+
+
 def detect_platform(project: str, **kwargs):
-    print("## Project platform")
-    print()
+    print("## Project platform", end="\n\n")
 
     path = Path(project)
     if path.is_file():
@@ -73,8 +109,7 @@ def detect_platform(project: str, **kwargs):
 
 
 def compile_project(project: str, **kwargs):
-    print("## Project compilation")
-    print()
+    print("## Project compilation", end="\n\n")
 
     print("Invoking crytic-compile on the project, please wait...")
 
@@ -91,9 +126,9 @@ def compile_project(project: str, **kwargs):
 
 def main():
     args = parse_args()
-
     kwargs = vars(args)
 
+    show_versions()
     detect_platform(**kwargs)
     compile_project(**kwargs)
 
diff --git a/slither/tools/doctor/packages.py b/slither/tools/doctor/packages.py
new file mode 100644
index 000000000..6cbf48f89
--- /dev/null
+++ b/slither/tools/doctor/packages.py
@@ -0,0 +1,24 @@
+from importlib import metadata
+import json
+from typing import Optional
+from packaging.version import parse, LegacyVersion, Version
+import urllib
+
+
+def get_installed_version(name: str) -> Optional[LegacyVersion | Version]:
+    try:
+        return parse(metadata.version(name))
+    except metadata.PackageNotFoundError:
+        return None
+
+
+def get_github_version(name: str) -> Optional[LegacyVersion | Version]:
+    try:
+        with urllib.request.urlopen(
+            f"https://api.github.com/repos/crytic/{name}/releases/latest"
+        ) as response:
+            text = response.read()
+            data = json.loads(text)
+            return parse(data["tag_name"])
+    except:
+        return None