Add internals feature.

This commit is contained in:
Stephen Chung
2020-06-23 10:43:24 +08:00
parent a9b168ba99
commit 54c5c139f9
7 changed files with 46 additions and 7 deletions

View File

@@ -63,8 +63,9 @@
//! | `only_i64` | Set the system integer type to `i64` and disable all other integer types. `INT` is set to `i64`. |
//! | `no_std` | Build for `no-std`. Notice that additional dependencies will be pulled in to replace `std` features. |
//! | `sync` | Restrict all values types to those that are `Send + Sync`. Under this feature, `Engine`, `Scope` and `AST` are all `Send + Sync`. |
//! | `internals` | Expose internal data structures (beware they may be volatile from version to version). |
//!
//! See [The Rhai Book](https://schungx.github.io/rhai/) for details on the Rhai script engine and language.
//! See [The Rhai Book](https://schungx.github.io/rhai) for details on the Rhai script engine and language.
#![cfg_attr(feature = "no_std", no_std)]
@@ -125,3 +126,23 @@ pub mod module_resolvers {
#[cfg(not(feature = "no_optimize"))]
pub use optimize::OptimizationLevel;
// Expose internal data structures.
#[cfg(feature = "internals")]
pub use token::Token;
#[cfg(feature = "internals")]
pub use parser::Expr;
#[cfg(feature = "internals")]
pub use parser::Stmt;
#[cfg(feature = "internals")]
pub use module::ModuleRef;
#[cfg(feature = "internals")]
pub use utils::StaticVec;
#[cfg(feature = "internals")]
pub use parser::ReturnType;

View File

@@ -65,20 +65,34 @@ impl AST {
}
/// Get the statements.
#[cfg(not(feature = "internals"))]
pub(crate) fn statements(&self) -> &Vec<Stmt> {
&self.0
}
/// Get the statements.
#[cfg(feature = "internals")]
pub fn statements(&self) -> &Vec<Stmt> {
&self.0
}
/// Get a mutable reference to the statements.
pub(crate) fn statements_mut(&mut self) -> &mut Vec<Stmt> {
&mut self.0
}
/// Get the script-defined functions.
/// Get the internal `Module` containing all script-defined functions.
#[cfg(not(feature = "internals"))]
pub(crate) fn lib(&self) -> &Module {
&self.1
}
/// Get the internal `Module` containing all script-defined functions.
#[cfg(feature = "internals")]
pub fn lib(&self) -> &Module {
&self.1
}
/// Merge two `AST` into one. Both `AST`'s are untouched and a new, merged, version
/// is returned.
///