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 context = rune_modules::with_config(false)?;
//! context.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)
}