r2x 0.1.13

The r2x-cli plugin manager for model interoperability.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use crate::plugins::error::PluginError;
use r2x_config::Config;
use r2x_manifest::package_discovery::PackageLocator;
use r2x_manifest::types::Manifest;
use r2x_python::utils::resolve_site_package_path;
use std::path::PathBuf;

pub struct PluginContext {
    pub(crate) config: Config,
    pub(crate) manifest: Manifest,
    pub(crate) uv_path: String,
    pub(crate) venv_path: String,
    pub(crate) python_path: String,
    pub(crate) locator: PackageLocator,
}

impl PluginContext {
    pub fn load() -> Result<Self, PluginError> {
        let mut config = Config::load()
            .map_err(|e| PluginError::Config(format!("Failed to load config: {e}")))?;

        config
            .ensure_uv_path()
            .map_err(|e| PluginError::Config(format!("Failed to setup uv: {e}")))?;
        config
            .ensure_cache_path()
            .map_err(|e| PluginError::Config(format!("Failed to setup cache: {e}")))?;
        config
            .ensure_venv_path()
            .map_err(|e| PluginError::Config(format!("Failed to setup venv: {e}")))?;

        let uv_path = config
            .uv_path
            .clone()
            .ok_or_else(|| PluginError::Config("uv path not configured".to_string()))?;

        let venv_path = config.get_venv_path();
        let python_path = config.get_venv_python_path();

        let site_packages = resolve_site_package_path(&PathBuf::from(&venv_path))
            .map_err(|e| PluginError::Python(format!("Failed to resolve site-packages: {e}")))?;

        let locator = PackageLocator::new(site_packages).map_err(|e| {
            PluginError::Locator(format!("Failed to initialize package locator: {e}"))
        })?;

        let manifest = Manifest::load()?;

        Ok(PluginContext {
            config,
            manifest,
            uv_path,
            venv_path,
            python_path,
            locator,
        })
    }

    pub(crate) fn refresh_locator(&mut self) -> Result<(), PluginError> {
        self.locator
            .refresh()
            .map_err(|e| PluginError::Locator(format!("Failed to refresh package locator: {e}")))
    }
}