move wasi-c2 into wasi-common

crates/wasi-common/cap-std-sync/src/lib.rs

@@ -0,0 +1,61 @@
+pub mod clocks;
+pub mod dir;
+pub mod file;
+pub mod sched;
+pub mod stdio;
+
+use cap_rand::RngCore;
+use std::cell::RefCell;
+use std::path::Path;
+use std::rc::Rc;
+use wasi_c2::{table::Table, Error, WasiCtx, WasiFile};
+
+pub struct WasiCtxBuilder(wasi_c2::WasiCtxBuilder);
+
+impl WasiCtxBuilder {
+    pub fn new() -> Self {
+        WasiCtxBuilder(WasiCtx::builder(
+            random(),
+            clocks::clocks(),
+            Box::new(sched::SyncSched::new()),
+            Rc::new(RefCell::new(Table::new())),
+        ))
+    }
+    pub fn env(self, var: &str, value: &str) -> Result<Self, wasi_c2::StringArrayError> {
+        let s = self.0.env(var, value)?;
+        Ok(WasiCtxBuilder(s))
+    }
+    pub fn arg(self, arg: &str) -> Result<Self, wasi_c2::StringArrayError> {
+        let s = self.0.arg(arg)?;
+        Ok(WasiCtxBuilder(s))
+    }
+    pub fn stdin(self, f: Box<dyn WasiFile>) -> Self {
+        WasiCtxBuilder(self.0.stdin(f))
+    }
+    pub fn stdout(self, f: Box<dyn WasiFile>) -> Self {
+        WasiCtxBuilder(self.0.stdout(f))
+    }
+    pub fn stderr(self, f: Box<dyn WasiFile>) -> Self {
+        WasiCtxBuilder(self.0.stderr(f))
+    }
+    pub fn inherit_stdio(self) -> Self {
+        self.stdin(Box::new(crate::stdio::stdin()))
+            .stdout(Box::new(crate::stdio::stdout()))
+            .stderr(Box::new(crate::stdio::stderr()))
+    }
+    pub fn preopened_dir(
+        self,
+        dir: cap_std::fs::Dir,
+        path: impl AsRef<Path>,
+    ) -> Result<Self, Error> {
+        let dir = Box::new(crate::dir::Dir::from_cap_std(dir));
+        Ok(WasiCtxBuilder(self.0.preopened_dir(dir, path)?))
+    }
+    pub fn build(self) -> Result<WasiCtx, Error> {
+        self.0.build()
+    }
+}
+
+pub fn random() -> RefCell<Box<dyn RngCore>> {
+    RefCell::new(Box::new(unsafe { cap_rand::rngs::OsRng::default() }))
+}