prelim script

requirements.txt

@@ -0,0 +1,5 @@
+distro
+pre-commit
+probe-rs
+docker
+docker-compose

setup.py

@@ -0,0 +1,132 @@
+import os
+import platform
+import subprocess
+import sys
+import venv
+
+try:
+    import distro  # for Linux distribution detection
+except ImportError:
+    distro = None
+
+def run_command(command, check=True, shell=False):
+    try:
+        result = subprocess.run(command, check=check, shell=shell, text=True, capture_output=True)
+        print(result.stdout)
+        if result.stderr:
+            print(result.stderr, file=sys.stderr)
+    except subprocess.CalledProcessError as e:
+        print(f"Command failed: {e.cmd}\nReturn code: {e.returncode}\nOutput: {e.output}\nError: {e.stderr}", file=sys.stderr)
+        sys.exit(1)
+
+def check_docker_and_rust():
+    print("This script requires Docker and Rust to be installed.")
+    answer = input("Do you have Docker and Rust installed? (yes/no): ").strip().lower()
+    if answer != 'yes':
+        print("Please install Docker and Rust before running this script.")
+        sys.exit(1)
+
+def create_venv(venv_path):
+    try:
+        venv.EnvBuilder(with_pip=True).create(venv_path)
+        print(f"Virtual environment created at {venv_path}")
+    except Exception as e:
+        print(f"Failed to create virtual environment: {e}", file=sys.stderr)
+        sys.exit(1)
+
+def modify_activation_scripts(venv_path):
+    os_type = platform.system()
+    activate_path = os.path.join(venv_path, 'Scripts', 'activate') if os_type == 'Windows' else os.path.join(venv_path, 'bin', 'activate')
+    deactivate_path = os.path.join(venv_path, 'Scripts', 'deactivate') if os_type == 'Windows' else os.path.join(venv_path, 'bin', 'deactivate')
+
+    # Modify the activate script to start Docker container
+    try:
+        with open(activate_path, 'a') as activate_file:
+            activate_file.write("\n# Start Docker container when venv is activated\n")
+            activate_file.write("docker-compose up -d\n")
+
+	    # aliases
+
+        print("Activation script modified to start Docker container.")
+    except Exception as e:
+        print(f"Failed to modify activation script: {e}", file=sys.stderr)
+        sys.exit(1)
+
+    # Modify the deactivate script to stop Docker container
+    try:
+        with open(deactivate_path, 'a') as deactivate_file:
+            deactivate_file.write("\n# Stop Docker container when venv is deactivated\n")
+            deactivate_file.write("docker-compose down\n")
+        print("Deactivation script modified to stop Docker container.")
+    except Exception as e:
+        print(f"Failed to modify deactivation script: {e}", file=sys.stderr)
+        sys.exit(1)
+
+def install_requirements(venv_python):
+    try:
+        run_command([venv_python, '-m', 'pip', 'install', '-r', 'requirements.txt'])
+    except Exception as e:
+        print(f"Failed to install requirements: {e}", file=sys.stderr)
+        sys.exit(1)
+
+def install_precommit(venv_python):
+    try:
+        run_command([venv_python, '-m', 'pre_commit', 'install'])
+    except Exception as e:
+        print(f"Failed to install pre-commit: {e}", file=sys.stderr)
+        sys.exit(1)
+
+def install_probe_rs(venv_python):
+    try:
+        run_command([venv_python, '-m', 'pip', 'install', 'probe-rs'])
+    except Exception as e:
+        print(f"Failed to install probe-rs: {e}", file=sys.stderr)
+        sys.exit(1)
+
+def install_usbip():
+    if distro:
+        distro_name = distro.id()
+        if distro_name in ["ubuntu", "debian"]:
+            run_command(["sudo", "apt-get", "install", "-y", "usbip"])
+        elif distro_name == "fedora":
+            run_command(["sudo", "dnf", "install", "-y", "usbip"])
+        elif distro_name == "arch":
+            run_command(["sudo", "pacman", "-S", "--noconfirm", "usbip"])
+        else:
+            print(f"USBIP installation not supported for {distro_name}", file=sys.stderr)
+    else:
+        print("distro module not found, cannot determine Linux distribution", file=sys.stderr)
+
+def main():
+    # Step 0: Check for Docker and Rust
+    check_docker_and_rust()
+
+    os_type = platform.system()
+    venv_path = ".venv"
+
+    # Step 1: Create the Python virtual environment
+    create_venv(venv_path)
+
+    # Step 2: Modify activation and deactivation scripts
+    modify_activation_scripts(venv_path)
+
+    # Use the venv's Python
+    venv_python = os.path.join(venv_path, 'Scripts', 'python') if os_type == "Windows" else os.path.join(venv_path, 'bin', 'python')
+
+    # Step 3: Install all Python packages from requirements.txt
+    install_requirements(venv_python)
+
+    # Step 4: Run pre-commit install
+    install_precommit(venv_python)
+
+    # Step 5: Install probe-rs
+    install_probe_rs(venv_python)
+
+    # Step 6: Install usbip if on Linux
+    if os_type == "Linux":
+        install_usbip()
+
+    print("Setup completed successfully!")
+
+if __name__ == "__main__":
+    main()