feat: add dev environment setup check script

Author: therealharshit

scripts/check_env.py

@@ -5,60 +5,47 @@
 from __future__ import annotations
 
 import subprocess
+import sys
+import re
 
 PIXI_VERSION = "0.39.0"
 CARGO_VERSION = "1.81.0"
 RUST_VERSION = "1.81.0"
 
-
-def check_version(cmd: str, expected: str, update: str, install: str) -> bool:
+def get_version_output(cmd: str) -> str | None:
     try:
-        output = subprocess.check_output([cmd, "--version"])
-
-        version = output.strip().decode("utf-8").split(" ")[1]
+        output = subprocess.check_output([cmd, "--version"], stderr=subprocess.STDOUT)
+        return output.strip().decode("utf-8")
+    except (FileNotFoundError, subprocess.CalledProcessError):
+        return None
 
-        if version != expected:
-            print(f"Expected {cmd} version {expected}, got {version}")
-            print(f"Please run `{update}`")
-            return False
-        else:
-            print(f"{cmd} version {version} is correct")
-            return True
+def extract_version(output: str) -> str | None:
+    match = re.search(r"\d+\.\d+\.\d+", output)
+    return match.group(0) if match else None
 
-    except FileNotFoundError:
+def check_version(cmd: str, expected: str, update: str, install: str) -> bool:
+    output = get_version_output(cmd)
+    if output is None:
         print(f"{cmd} not found in PATH. Please install via {install}")
         return False
-
+    
+    version = extract_version(output)
+    if version != expected:
+        print(f"Expected {cmd} version {expected}, got {version}")
+        print(f"Please run `{update}`")
+        return False
+    
+    print(f"{cmd} version {version} is correct")
+    return True
 
 def main() -> int:
-    success = True
-
-    success &= check_version(
-        "pixi",
-        PIXI_VERSION,
-        f"pixi self-update --version {PIXI_VERSION}",
-        "https://pixi.sh/latest/",
-    )
-
-    success &= check_version(
-        "cargo",
-        CARGO_VERSION,
-        f"rustup install {CARGO_VERSION}",
-        "https://rustup.rs/",
-    )
-
-    success &= check_version(
-        "rustc",
-        RUST_VERSION,
-        f"rustup install {RUST_VERSION}",
-        "https://rustup.rs/",
-    )
-
-    if success:
-        exit(0)
-    else:
-        exit(1)
-
+    success = all([
+        check_version("pixi", PIXI_VERSION, f"pixi self-update --version {PIXI_VERSION}", "https://pixi.sh/latest/"),
+        check_version("cargo", CARGO_VERSION, f"rustup install {CARGO_VERSION}", "https://rustup.rs/"),
+        check_version("rustc", RUST_VERSION, f"rustup install {RUST_VERSION}", "https://rustup.rs/"),
+    ])
+    
+    sys.exit(0 if success else 1)
 
 if __name__ == "__main__":
     main()