logos-execution-zone/keycard_wallet/src/python_path.rs

use std::{env, path::PathBuf};

use pyo3::{prelude::*, types::PyList};

/// Adds the project's `python/` directory and venv site-packages to Python's sys.path.
pub fn add_python_path(py: Python<'_>) -> PyResult<()> {
    let current_dir = env::current_dir().expect("Failed to get current working directory");

    let python_base = env::var("VIRTUAL_ENV")
        .ok()
        .and_then(|v| PathBuf::from(v).parent().map(PathBuf::from))
        .unwrap_or_else(|| current_dir.clone());

    let mut paths_to_add: Vec<PathBuf> = vec![
        python_base.join("keycard_wallet").join("python"),
        python_base
            .join("keycard_wallet")
            .join("python")
            .join("keycard-py"),
    ];

    // If a virtualenv is active, add its site-packages so that dependencies
    // installed in the venv (e.g. smartcard, ecdsa) are importable by the
    // pyo3 embedded interpreter, which does not inherit sys.path from the
    // shell's `python3` executable.
    if let Ok(venv) = env::var("VIRTUAL_ENV") {
        let lib = PathBuf::from(&venv).join("lib");
        if let Ok(entries) = std::fs::read_dir(&lib) {
            for entry in entries.flatten() {
                let site_packages = entry.path().join("site-packages");
                if site_packages.exists() {
                    paths_to_add.push(site_packages);
                }
            }
        }
    }

    // Sanity check — warns early if a path doesn't exist
    for path in &paths_to_add {
        if !path.exists() {
            log::info!("Warning: Python path does not exist: {}", path.display());
        }
    }

    let sys = PyModule::import(py, "sys")?;
    let binding = sys.getattr("path")?;
    let sys_path = binding.downcast::<PyList>()?;

    for path in &paths_to_add {
        let path_str = path.to_str().expect("Invalid path");

        // Avoid duplicating the path
        let already_present = sys_path
            .iter()
            .any(|p| p.extract::<&str>().map(|s| s == path_str).unwrap_or(false));

        if !already_present {
            sys_path.insert(0, path_str)?;
        }
    }

    Ok(())
}