scoop-uv 0.11.0

//! pyenv environment discovery

use std::fs;
use std::path::{Path, PathBuf};

use crate::error::{Result, ScoopError};

use super::common;
use super::source::{EnvironmentSource, EnvironmentStatus, SourceEnvironment, SourceType};

/// Discovers pyenv-virtualenv environments
#[derive(Debug)]
pub struct PyenvDiscovery {
    /// Root path of pyenv (typically ~/.pyenv)
    root: PathBuf,
}

impl PyenvDiscovery {
    /// Creates a new discovery instance for the given pyenv root.
    pub fn new(root: PathBuf) -> Self {
        Self { root }
    }

    /// Creates a discovery instance using the default pyenv root.
    ///
    /// Uses `$PYENV_ROOT` if set, otherwise `~/.pyenv`.
    pub fn default_root() -> Option<Self> {
        let root = std::env::var("PYENV_ROOT")
            .map(PathBuf::from)
            .ok()
            .or_else(|| dirs::home_dir().map(|h| h.join(".pyenv")))?;

        if root.exists() {
            Some(Self::new(root))
        } else {
            None
        }
    }

    /// Get the versions directory path
    fn versions_dir(&self) -> PathBuf {
        self.root.join("versions")
    }

    /// Parse pyvenv.cfg to extract Python version
    fn parse_pyvenv_cfg(path: &Path) -> Option<String> {
        let cfg_path = path.join("pyvenv.cfg");
        let content = fs::read_to_string(&cfg_path).ok()?;

        // Reads both `version` (stdlib venv) and `version_info` (uv) keys.
        if let Some(version) = content
            .lines()
            .find_map(crate::core::pyvenv_version_from_line)
        {
            return Some(version);
        }

        // Fallback: try to extract from home path
        for line in content.lines() {
            let line = line.trim();
            if let Some(home) = line.strip_prefix("home") {
                let home = home.trim_start_matches([' ', '=']).trim();
                // Extract version from path like /Users/user/.pyenv/versions/3.11.0/bin
                if let Some(versions_idx) = home.find("versions/") {
                    let after_versions = &home[versions_idx + 9..];
                    if let Some(slash_idx) = after_versions.find('/') {
                        return Some(after_versions[..slash_idx].to_string());
                    }
                }
            }
        }

        None
    }

    /// Parse a single environment directory into SourceEnvironment
    fn parse_environment(
        &self,
        env_path: &Path,
        fallback_version: Option<&str>,
    ) -> Option<SourceEnvironment> {
        let name = env_path.file_name()?.to_str()?.to_string();

        // Parse Python version from pyvenv.cfg
        let python_version = Self::parse_pyvenv_cfg(env_path)
            .or_else(|| fallback_version.map(|s| s.to_string()))
            .unwrap_or_else(|| "unknown".to_string());

        // Validate environment (check for bin/python)
        let python_bin = env_path.join("bin").join("python");
        if !python_bin.exists() {
            return Some(SourceEnvironment {
                name,
                python_version,
                path: env_path.to_path_buf(),
                source_type: SourceType::Pyenv,
                size_bytes: None, // Lazy: calculated only when needed
                status: EnvironmentStatus::Corrupted {
                    reason: "Python binary not found".to_string(),
                },
            });
        }

        // Determine status (no dir_size calculation here - lazy loading)
        let status = common::determine_status(&name, &python_version);

        Some(SourceEnvironment {
            name,
            python_version,
            path: env_path.to_path_buf(),
            source_type: SourceType::Pyenv,
            size_bytes: None, // Lazy: calculated only when needed
            status,
        })
    }
}

impl EnvironmentSource for PyenvDiscovery {
    fn source_type(&self) -> SourceType {
        SourceType::Pyenv
    }

    fn scan_environments(&self) -> Result<Vec<SourceEnvironment>> {
        let mut environments = Vec::new();
        let versions_dir = self.versions_dir();

        if !versions_dir.exists() {
            return Ok(environments);
        }

        // Scan ~/.pyenv/versions/*/envs/*
        let entries = fs::read_dir(&versions_dir).map_err(ScoopError::Io)?;

        for entry in entries.flatten() {
            let python_version_path = entry.path();

            // Skip symlinks (pyenv creates symlinks for virtualenvs at top level)
            if python_version_path.is_symlink() {
                continue;
            }

            // Look for envs subdirectory
            let envs_dir = python_version_path.join("envs");
            if !envs_dir.exists() || !envs_dir.is_dir() {
                continue;
            }

            // Get fallback version from directory name
            let fallback_version = python_version_path
                .file_name()
                .and_then(|n| n.to_str())
                .map(|s| s.to_string());

            // Scan environments in this Python version's envs directory
            let env_entries = match fs::read_dir(&envs_dir) {
                Ok(entries) => entries,
                Err(_) => continue,
            };

            for env_entry in env_entries.flatten() {
                let env_path = env_entry.path();

                // Skip symlinks and non-directories
                if env_path.is_symlink() || !env_path.is_dir() {
                    continue;
                }

                if let Some(env) = self.parse_environment(&env_path, fallback_version.as_deref()) {
                    environments.push(env);
                }
            }
        }

        // Sort by name
        environments.sort_by(|a, b| a.name.cmp(&b.name));

        Ok(environments)
    }

    /// Find a specific environment by name using O(1) direct path access.
    ///
    /// Instead of scanning all environments, this method directly checks
    /// each Python version's envs directory for the target environment.
    fn find_environment(&self, name: &str) -> Result<SourceEnvironment> {
        let versions_dir = self.versions_dir();

        if !versions_dir.exists() {
            return Err(ScoopError::PyenvEnvNotFound {
                name: name.to_string(),
            });
        }

        // Directly search in each version's envs directory
        let entries = fs::read_dir(&versions_dir).map_err(ScoopError::Io)?;

        for entry in entries.flatten() {
            let python_version_path = entry.path();

            // Skip symlinks
            if python_version_path.is_symlink() {
                continue;
            }

            // Check if envs/<name> exists
            let env_path = python_version_path.join("envs").join(name);
            if env_path.exists() && env_path.is_dir() {
                let fallback_version = python_version_path.file_name().and_then(|n| n.to_str());

                if let Some(env) = self.parse_environment(&env_path, fallback_version) {
                    return Ok(env);
                }
            }
        }

        Err(ScoopError::PyenvEnvNotFound {
            name: name.to_string(),
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use serial_test::serial;
    use std::io::Write;
    use tempfile::TempDir;

    // =========================================================================
    // Test Helpers
    // =========================================================================

    /// Creates a mock pyenv root with Python versions and their `envs/` subdirectories.
    ///
    /// Structure created:
    /// ```text
    /// <temp>/versions/
    ///   ├── 3.11.0/
    ///   │   └── bin/python (mock)
    ///   ├── 3.12.0/
    ///   │   └── bin/python (mock)
    ///   └── ...
    /// ```
    fn setup_mock_pyenv_root(versions: &[&str]) -> TempDir {
        let temp = TempDir::new().expect("Failed to create temp dir");
        let versions_dir = temp.path().join("versions");
        fs::create_dir_all(&versions_dir).expect("Failed to create versions dir");

        for ver in versions {
            let bin_dir = versions_dir.join(ver).join("bin");
            fs::create_dir_all(&bin_dir).expect("Failed to create bin dir");
            // Create a mock python binary (just needs to exist for tests)
            fs::write(bin_dir.join("python"), "#!/bin/sh\necho mock")
                .expect("Failed to write mock python");
        }
        temp
    }

    /// Creates a mock pyenv-virtualenv environment within a Python version.
    ///
    /// Structure created:
    /// ```text
    /// <pyenv_root>/versions/<python_version>/envs/<env_name>/
    ///   ├── bin/python
    ///   └── pyvenv.cfg
    /// ```
    fn create_mock_virtualenv(
        pyenv_root: &Path,
        python_version: &str,
        env_name: &str,
        create_python_bin: bool,
    ) {
        let env_path = pyenv_root
            .join("versions")
            .join(python_version)
            .join("envs")
            .join(env_name);
        let bin_dir = env_path.join("bin");
        fs::create_dir_all(&bin_dir).expect("Failed to create env bin dir");

        if create_python_bin {
            fs::write(bin_dir.join("python"), "#!/bin/sh\necho mock")
                .expect("Failed to write mock python");
        }

        // Create pyvenv.cfg
        let cfg_path = env_path.join("pyvenv.cfg");
        let cfg_content = format!(
            "home = {}/versions/{}/bin\n\
             include-system-site-packages = false\n\
             version = {}\n",
            pyenv_root.display(),
            python_version,
            python_version
        );
        fs::write(cfg_path, cfg_content).expect("Failed to write pyvenv.cfg");
    }

    // =========================================================================
    // Unit Tests: PyenvDiscovery::new()
    // =========================================================================

    #[test]
    fn test_new_creates_instance_with_given_root() {
        let path = PathBuf::from("/some/path/.pyenv");
        let discovery = PyenvDiscovery::new(path.clone());
        assert_eq!(discovery.root, path);
    }

    // =========================================================================
    // Unit Tests: PyenvDiscovery::default_root()
    // =========================================================================

    #[test]
    #[serial]
    fn test_default_root_uses_pyenv_root_env_var() {
        let temp = TempDir::new().unwrap();
        let pyenv_path = temp.path().to_path_buf();

        // SAFETY: Test is serialized, no concurrent access to env vars
        unsafe {
            std::env::set_var("PYENV_ROOT", &pyenv_path);
        }

        let discovery = PyenvDiscovery::default_root();

        // SAFETY: Test cleanup
        unsafe {
            std::env::remove_var("PYENV_ROOT");
        }

        assert!(discovery.is_some());
        assert_eq!(discovery.unwrap().root, pyenv_path);
    }

    #[test]
    #[serial]
    fn test_default_root_returns_none_for_nonexistent_path() {
        // SAFETY: Test is serialized, no concurrent access to env vars
        unsafe {
            std::env::set_var("PYENV_ROOT", "/nonexistent/path/that/does/not/exist");
        }

        let discovery = PyenvDiscovery::default_root();

        // SAFETY: Test cleanup
        unsafe {
            std::env::remove_var("PYENV_ROOT");
        }

        assert!(discovery.is_none());
    }

    // =========================================================================
    // Unit Tests: PyenvDiscovery::parse_pyvenv_cfg()
    // =========================================================================

    #[test]
    fn test_parse_pyvenv_cfg_extracts_version() {
        let temp_dir = TempDir::new().unwrap();
        let cfg_path = temp_dir.path().join("pyvenv.cfg");

        let mut file = fs::File::create(&cfg_path).unwrap();
        writeln!(file, "home = /Users/test/.pyenv/versions/3.11.0/bin").unwrap();
        writeln!(file, "include-system-site-packages = false").unwrap();
        writeln!(file, "version = 3.11.0").unwrap();

        let version = PyenvDiscovery::parse_pyvenv_cfg(temp_dir.path());
        assert_eq!(version, Some("3.11.0".to_string()));
    }

    #[test]
    fn test_parse_pyvenv_cfg_handles_no_spaces() {
        let temp_dir = TempDir::new().unwrap();
        let cfg_path = temp_dir.path().join("pyvenv.cfg");

        // Some tools write without spaces around '='
        fs::write(&cfg_path, "version=3.10.5\nhome=/usr/bin\n").unwrap();

        let version = PyenvDiscovery::parse_pyvenv_cfg(temp_dir.path());
        assert_eq!(version, Some("3.10.5".to_string()));
    }

    #[test]
    fn test_parse_pyvenv_cfg_fallback_to_home_path() {
        let temp_dir = TempDir::new().unwrap();
        let cfg_path = temp_dir.path().join("pyvenv.cfg");

        // No version key, should fallback to parsing home path
        let mut file = fs::File::create(&cfg_path).unwrap();
        writeln!(file, "home = /Users/test/.pyenv/versions/3.9.7/bin").unwrap();
        writeln!(file, "include-system-site-packages = false").unwrap();

        let version = PyenvDiscovery::parse_pyvenv_cfg(temp_dir.path());
        assert_eq!(version, Some("3.9.7".to_string()));
    }

    #[test]
    fn test_parse_pyvenv_cfg_returns_none_for_missing_file() {
        let temp_dir = TempDir::new().unwrap();
        // No pyvenv.cfg created
        let version = PyenvDiscovery::parse_pyvenv_cfg(temp_dir.path());
        assert!(version.is_none());
    }

    #[test]
    fn test_parse_pyvenv_cfg_returns_none_for_malformed_content() {
        let temp_dir = TempDir::new().unwrap();
        let cfg_path = temp_dir.path().join("pyvenv.cfg");

        fs::write(&cfg_path, "garbage content without version info").unwrap();

        let version = PyenvDiscovery::parse_pyvenv_cfg(temp_dir.path());
        assert!(version.is_none());
    }

    // =========================================================================
    // Integration Tests: scan_environments()
    // =========================================================================

    #[test]
    fn test_scan_empty_pyenv_returns_empty() {
        let temp = TempDir::new().unwrap();
        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());

        let result = discovery.scan_environments();

        assert!(result.is_ok());
        assert!(result.unwrap().is_empty());
    }

    #[test]
    fn test_scan_pyenv_with_versions_but_no_envs() {
        // pyenv versions (3.11.0, 3.12.0) are NOT virtualenvs
        // They are base Python installations without envs/ subdirectory
        let temp = setup_mock_pyenv_root(&["3.11.0", "3.12.0"]);
        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());

        let result = discovery.scan_environments();

        assert!(result.is_ok());
        // No envs/ subdirectory means no virtualenvs discovered
        assert!(result.unwrap().is_empty());
    }

    #[test]
    fn test_scan_discovers_virtualenvs_in_envs_directory() {
        let temp = setup_mock_pyenv_root(&["3.11.0", "3.12.0"]);

        // Create virtualenvs in envs/ subdirectories
        create_mock_virtualenv(temp.path(), "3.11.0", "project-a", true);
        create_mock_virtualenv(temp.path(), "3.11.0", "project-b", true);
        create_mock_virtualenv(temp.path(), "3.12.0", "project-c", true);

        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());
        let result = discovery.scan_environments().unwrap();

        assert_eq!(result.len(), 3);

        // Results are sorted by name
        assert_eq!(result[0].name, "project-a");
        assert_eq!(result[0].python_version, "3.11.0");
        assert!(matches!(result[0].source_type, SourceType::Pyenv));

        assert_eq!(result[1].name, "project-b");
        assert_eq!(result[1].python_version, "3.11.0");

        assert_eq!(result[2].name, "project-c");
        assert_eq!(result[2].python_version, "3.12.0");
    }

    #[test]
    fn test_scan_detects_corrupted_envs_without_python_binary() {
        let temp = setup_mock_pyenv_root(&["3.11.0"]);

        // Create env WITHOUT python binary
        create_mock_virtualenv(temp.path(), "3.11.0", "broken-env", false);

        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());
        let result = discovery.scan_environments().unwrap();

        assert_eq!(result.len(), 1);
        assert_eq!(result[0].name, "broken-env");
        assert!(matches!(
            result[0].status,
            EnvironmentStatus::Corrupted { .. }
        ));
    }

    #[test]
    fn test_scan_skips_symlinks_in_versions_directory() {
        let temp = setup_mock_pyenv_root(&["3.11.0"]);
        create_mock_virtualenv(temp.path(), "3.11.0", "real-env", true);

        // pyenv creates symlinks for virtualenvs at top level of versions/
        // e.g., versions/real-env -> versions/3.11.0/envs/real-env
        let symlink_path = temp.path().join("versions").join("real-env");
        let target = temp
            .path()
            .join("versions")
            .join("3.11.0")
            .join("envs")
            .join("real-env");

        #[cfg(unix)]
        std::os::unix::fs::symlink(&target, &symlink_path).unwrap();

        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());
        let result = discovery.scan_environments().unwrap();

        // Should only find 1 env, not 2 (symlink should be skipped)
        assert_eq!(result.len(), 1);
        assert_eq!(result[0].name, "real-env");
    }

    // =========================================================================
    // Integration Tests: find_environment()
    // =========================================================================

    #[test]
    fn test_find_environment_returns_correct_env() {
        let temp = setup_mock_pyenv_root(&["3.11.0", "3.12.0"]);
        create_mock_virtualenv(temp.path(), "3.11.0", "target-env", true);
        create_mock_virtualenv(temp.path(), "3.12.0", "other-env", true);

        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());
        let result = discovery.find_environment("target-env");

        assert!(result.is_ok());
        let env = result.unwrap();
        assert_eq!(env.name, "target-env");
        assert_eq!(env.python_version, "3.11.0");
    }

    #[test]
    fn test_find_environment_returns_error_for_nonexistent() {
        let temp = setup_mock_pyenv_root(&["3.11.0"]);
        create_mock_virtualenv(temp.path(), "3.11.0", "existing-env", true);

        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());
        let result = discovery.find_environment("nonexistent");

        assert!(result.is_err());
    }

    #[test]
    fn test_find_environment_returns_error_when_versions_dir_missing() {
        let temp = TempDir::new().unwrap();
        // No versions/ directory created

        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());
        let result = discovery.find_environment("any-env");

        assert!(result.is_err());
    }

    // =========================================================================
    // Unit Tests: determine_status (via common module)
    // =========================================================================

    #[test]
    fn test_determine_status_eol_python() {
        // Use a unique name that definitely won't exist in scoop
        let status = common::determine_status("nonexistent_eol_test_env_xyz", "3.7.0");
        assert!(matches!(status, EnvironmentStatus::PythonEol { .. }));

        let status = common::determine_status("nonexistent_eol_test_env_xyz", "2.7.18");
        assert!(matches!(status, EnvironmentStatus::PythonEol { .. }));
    }

    #[test]
    fn test_determine_status_ready() {
        let status = common::determine_status("nonexistent_env_name_12345", "3.12.0");
        assert!(matches!(status, EnvironmentStatus::Ready));
    }

    #[test]
    fn test_determine_status_eol_boundary() {
        // 3.8 is the EOL boundary
        let status = common::determine_status("boundary_test_env", "3.8.19");
        assert!(matches!(status, EnvironmentStatus::PythonEol { .. }));

        // 3.9 is supported
        let status = common::determine_status("boundary_test_env", "3.9.0");
        assert!(matches!(status, EnvironmentStatus::Ready));
    }

    // =========================================================================
    // Edge Cases
    // =========================================================================

    #[test]
    fn test_scan_handles_permission_errors_gracefully() {
        // Just test that we don't panic on unreadable directories
        // Actual permission testing is platform-specific
        let temp = TempDir::new().unwrap();
        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());

        // Should not panic, just return empty
        let result = discovery.scan_environments();
        assert!(result.is_ok());
    }

    #[test]
    fn test_source_type_returns_pyenv() {
        let temp = TempDir::new().unwrap();
        let discovery = PyenvDiscovery::new(temp.path().to_path_buf());

        assert!(matches!(discovery.source_type(), SourceType::Pyenv));
    }
}