diff options
Diffstat (limited to 'ext/io/lib.rs')
-rw-r--r-- | ext/io/lib.rs | 55 |
1 files changed, 37 insertions, 18 deletions
diff --git a/ext/io/lib.rs b/ext/io/lib.rs index e0d649e0a..8d80eec25 100644 --- a/ext/io/lib.rs +++ b/ext/io/lib.rs @@ -110,36 +110,36 @@ deno_core::extension!(deno_io, let t = &mut state.resource_table; let rid = t.add(fs::FileResource::new( - Rc::new(match stdio.stdin { - StdioPipe::Inherit => StdFileResourceInner::new( + Rc::new(match stdio.stdin.pipe { + StdioPipeInner::Inherit => StdFileResourceInner::new( StdFileResourceKind::Stdin, STDIN_HANDLE.try_clone().unwrap(), ), - StdioPipe::File(pipe) => StdFileResourceInner::file(pipe), + StdioPipeInner::File(pipe) => StdFileResourceInner::file(pipe), }), "stdin".to_string(), )); assert_eq!(rid, 0, "stdin must have ResourceId 0"); let rid = t.add(FileResource::new( - Rc::new(match stdio.stdout { - StdioPipe::Inherit => StdFileResourceInner::new( + Rc::new(match stdio.stdout.pipe { + StdioPipeInner::Inherit => StdFileResourceInner::new( StdFileResourceKind::Stdout, STDOUT_HANDLE.try_clone().unwrap(), ), - StdioPipe::File(pipe) => StdFileResourceInner::file(pipe), + StdioPipeInner::File(pipe) => StdFileResourceInner::file(pipe), }), "stdout".to_string(), )); assert_eq!(rid, 1, "stdout must have ResourceId 1"); let rid = t.add(FileResource::new( - Rc::new(match stdio.stderr { - StdioPipe::Inherit => StdFileResourceInner::new( + Rc::new(match stdio.stderr.pipe { + StdioPipeInner::Inherit => StdFileResourceInner::new( StdFileResourceKind::Stderr, STDERR_HANDLE.try_clone().unwrap(), ), - StdioPipe::File(pipe) => StdFileResourceInner::file(pipe), + StdioPipeInner::File(pipe) => StdFileResourceInner::file(pipe), }), "stderr".to_string(), )); @@ -148,22 +148,41 @@ deno_core::extension!(deno_io, }, ); -pub enum StdioPipe { - Inherit, - File(StdFile), +#[derive(Default)] +pub struct StdioPipe { + pipe: StdioPipeInner, } -impl Default for StdioPipe { - fn default() -> Self { - Self::Inherit +impl StdioPipe { + pub const fn inherit() -> Self { + StdioPipe { + pipe: StdioPipeInner::Inherit, + } + } + + pub fn file(f: impl Into<StdFile>) -> Self { + StdioPipe { + pipe: StdioPipeInner::File(f.into()), + } } } +#[derive(Default)] +enum StdioPipeInner { + #[default] + Inherit, + File(StdFile), +} + impl Clone for StdioPipe { fn clone(&self) -> Self { - match self { - StdioPipe::Inherit => StdioPipe::Inherit, - StdioPipe::File(pipe) => StdioPipe::File(pipe.try_clone().unwrap()), + match &self.pipe { + StdioPipeInner::Inherit => Self { + pipe: StdioPipeInner::Inherit, + }, + StdioPipeInner::File(pipe) => Self { + pipe: StdioPipeInner::File(pipe.try_clone().unwrap()), + }, } } } |