dalbit_core/modifiers/
mod.rs

1use std::str::FromStr;
2
3use anyhow::{anyhow, Result};
4use darklua_core::rules::Rule;
5use full_moon::{ast::Ast, visitors::VisitorMut};
6
7pub(crate) mod ast_util;
8mod convert_bit32;
9mod optimize_table_initializers;
10mod remove_generalized_iteration;
11mod remove_number_literals;
12mod remove_redeclared_keys;
13mod runtime_identifier;
14
15pub use convert_bit32::*;
16pub use optimize_table_initializers::*;
17pub use remove_generalized_iteration::*;
18pub use remove_number_literals::*;
19pub use remove_redeclared_keys::*;
20pub use runtime_identifier::*;
21
22pub trait VisitorMutWrapper {
23    fn visit_ast_boxed(&mut self, ast: Ast) -> Ast;
24}
25
26impl<T: VisitorMut> VisitorMutWrapper for T {
27    fn visit_ast_boxed(&mut self, ast: Ast) -> Ast {
28        self.visit_ast(ast)
29    }
30}
31
32pub enum Modifier {
33    DarkluaRule(Box<dyn Rule>),
34    FullMoonVisitor(Box<dyn VisitorMutWrapper>),
35}
36
37impl FromStr for Modifier {
38    type Err = anyhow::Error;
39
40    fn from_str(s: &str) -> Result<Self> {
41        let modifier = match s {
42            REMOVE_GENERALIZED_ITERATION_MODIFIER_NAME => {
43                Modifier::DarkluaRule(Box::<RemoveGeneralizedIteration>::default())
44            }
45            REMOVE_NUMBER_LITERALS_MODIFIER_NAME => {
46                Modifier::DarkluaRule(Box::<RemoveNumberLiterals>::default())
47            }
48            OPTIMIZE_TABLE_INITIALIZERS_MODIFIER_NAME => {
49                Modifier::DarkluaRule(Box::<OptimizeTableInitializers>::default())
50            }
51            CONVERT_BIT32_MODIFIER_NAME => Modifier::FullMoonVisitor(Box::new(
52                ConvertBit32::default(),
53            )
54                as Box<dyn VisitorMutWrapper>),
55            REMOVE_REDECLARED_KEYS_RULE_NAME => {
56                Modifier::DarkluaRule(Box::<RemoveRedeclaredKeys>::default())
57            }
58            _ => Modifier::DarkluaRule(s.parse::<Box<dyn Rule>>().map_err(|err| anyhow!(err))?),
59        };
60
61        Ok(modifier)
62    }
63}
dalbit_core/modifiers/mod.rs

dalbit_core/modifiers/
mod.rs
