fluentci_pdk/
pixi.rs

1use extism_pdk::*;
2use fluentci_types::{cache::Cache, file::File, pixi as types, service::Service};
3use serde::{Deserialize, Serialize};
4
5#[host_fn]
6extern "ExtismHost" {
7    fn set_runner(runner: String);
8    fn with_exec(args: Json<Vec<String>>);
9    fn with_workdir(path: String);
10    fn with_cache(cache: Json<Cache>);
11    fn with_file(file: Json<File>);
12    fn stdout() -> String;
13    fn stderr() -> String;
14    fn as_service(name: String) -> Json<Service>;
15    fn with_service(service_id: String);
16    fn set_envs(envs: Json<Vec<(String, String)>>);
17    fn wait_on(args: Json<Vec<u32>>);
18    fn with_secret_variable(params: Json<Vec<String>>);
19}
20
21#[derive(Serialize, Deserialize)]
22pub struct Pixi {
23    pub id: String,
24}
25
26impl From<types::Pixi> for Pixi {
27    fn from(pixi: types::Pixi) -> Self {
28        Pixi { id: pixi.id }
29    }
30}
31
32impl Pixi {
33    pub fn with_exec(&self, args: Vec<&str>) -> Result<Pixi, Error> {
34        unsafe { set_runner("pixi".into()) }?;
35        unsafe {
36            with_exec(Json::from(
37                args.into_iter()
38                    .map(|x| x.to_string())
39                    .collect::<Vec<String>>(),
40            ))
41        }?;
42        Ok(Pixi {
43            id: self.id.clone(),
44        })
45    }
46
47    pub fn with_workdir(&self, path: &str) -> Result<Pixi, Error> {
48        unsafe { with_workdir(path.into()) }?;
49        Ok(Pixi {
50            id: self.id.clone(),
51        })
52    }
53
54    pub fn with_cache(&self, path: &str, cache_id: &str) -> Result<Pixi, Error> {
55        unsafe {
56            with_cache(Json(Cache {
57                id: cache_id.into(),
58                path: path.into(),
59                ..Default::default()
60            }))
61        }?;
62        Ok(Pixi {
63            id: self.id.clone(),
64        })
65    }
66
67    pub fn with_file(&self, path: &str, file_id: &str) -> Result<Pixi, Error> {
68        unsafe {
69            with_file(Json(File {
70                id: file_id.into(),
71                path: path.into(),
72            }))
73        }?;
74        Ok(Pixi {
75            id: self.id.clone(),
76        })
77    }
78
79    pub fn stdout(&self) -> Result<String, Error> {
80        unsafe { stdout() }
81    }
82
83    pub fn stderr(&self) -> Result<String, Error> {
84        unsafe { stderr() }
85    }
86
87    pub fn as_service(&self, name: &str) -> Result<String, Error> {
88        let service = unsafe { as_service(name.into())? };
89        Ok(service.into_inner().id)
90    }
91
92    pub fn with_service(&self, service_id: &str) -> Result<Pixi, Error> {
93        unsafe { with_service(service_id.into())? }
94        Ok(Pixi {
95            id: self.id.clone(),
96        })
97    }
98
99    pub fn with_env_variable(&self, name: &str, value: &str) -> Result<Pixi, Error> {
100        unsafe { set_envs(Json(vec![(name.into(), value.into())]))? };
101        Ok(Pixi {
102            id: self.id.clone(),
103        })
104    }
105
106    pub fn wait_on(&self, port: u32, timeout: Option<u32>) -> Result<Pixi, Error> {
107        unsafe {
108            wait_on(Json(vec![port, timeout.unwrap_or(60)]))?;
109        }
110        Ok(Pixi {
111            id: self.id.clone(),
112        })
113    }
114
115    pub fn with_secret_variable(&self, name: &str, secret_id: &str) -> Result<Pixi, Error> {
116        unsafe {
117            with_secret_variable(Json(vec![name.into(), secret_id.into()]))?;
118        }
119        Ok(Pixi {
120            id: self.id.clone(),
121        })
122    }
123}
fluentci_pdk/pixi.rs

fluentci_pdk/
pixi.rs
