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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
//! The native `toml` module for the [Rune Language].
//!
//! [Rune Language]: https://github.com/rune-rs/rune
//!
//! ## Usage
//!
//! Add the following to your `Cargo.toml`:
//!
//! ```toml
//! rune-modules = {version = "0.6.14", features = ["toml"]}
//! ```
//!
//! Install it into your context:
//!
//! ```rust
//! # fn main() -> runestick::Result<()> {
//! let mut context = runestick::Context::with_default_modules()?;
//! context.install(&rune_modules::toml::module()?)?;
//! # Ok(())
//! # }
//! ```
//!
//! Use it in Rune:
//!
//! ```rust,ignore
//! use toml;
//!
//! fn main() {
//!     let data = toml::from_string("[hello]\nworld = 42");
//!     dbg(data);
//! }
//! ```

use runestick::{Bytes, ContextError, Module, Value};

/// Construct the `toml` module.
pub fn module() -> Result<Module, ContextError> {
    let mut module = Module::new(&["toml"]);
    module.function(&["from_bytes"], from_bytes)?;
    module.function(&["from_string"], from_string)?;
    module.function(&["to_string"], to_string)?;
    module.function(&["to_bytes"], to_bytes)?;
    Ok(module)
}

fn from_bytes(bytes: &[u8]) -> runestick::Result<Value> {
    Ok(toml::from_slice(&bytes)?)
}

/// Get value from toml string.
fn from_string(string: &str) -> runestick::Result<Value> {
    Ok(toml::from_str(string)?)
}

/// Convert any value to a toml string.
fn to_string(value: Value) -> runestick::Result<String> {
    Ok(toml::to_string(&value)?)
}

/// Convert any value to toml bytes.
fn to_bytes(value: Value) -> runestick::Result<Bytes> {
    let bytes = toml::to_vec(&value)?;
    Ok(Bytes::from_vec(bytes))
}