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

//! I/O module ignoring everything written to output.
//!
//! ```
//! use rune::{Context, ContextError};
//! use rune_modules::disable_io;
//!
//! # fn main() -> Result<(), ContextError> {
//! let mut c = rune_modules::with_config(false)?;
//! c.install(&disable_io::module()?)?;
//! # Ok(()) }
//! ```

use rune::runtime::Stack;
use rune::{ContextError, Module};

/// Provide a bunch of `std::io` functions which will cause any output to be ignored.
pub fn module() -> Result<Module, ContextError> {
    let mut module = Module::with_crate_item("std", &["io"]);

    module.function(&["print"], move |_: &str| {})?;

    module.function(&["println"], move |_: &str| {})?;

    module.raw_fn(&["dbg"], move |stack: &mut Stack, args: usize| {
        // NB: still need to maintain the stack.
        drop(stack.drain(args)?);
        stack.push(());
        Ok(())
    })?;

    Ok(module)
}