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
use crate::nodes::{ Expression, Prefix, }; use crate::lua_generator::{LuaGenerator, ToLua}; #[derive(Clone, Debug, PartialEq, Eq)] pub struct IndexExpression { prefix: Prefix, index: Expression, } impl IndexExpression { pub fn new<E: Into<Expression>>(prefix: Prefix, expression: E) -> Self { Self { prefix, index: expression.into(), } } pub fn mutate_prefix(&mut self) -> &mut Prefix { &mut self.prefix } } impl ToLua for IndexExpression { fn to_lua(&self, generator: &mut LuaGenerator) { self.prefix.to_lua(generator); generator.push_char('['); self.index.to_lua(generator); generator.push_char(']'); } } #[cfg(test)] mod test { use super::*; #[test] fn generate_identifier() { let output = IndexExpression::new( Prefix::from_name("foo"), Prefix::from_name("bar"), ).to_lua_string(); assert_eq!(output, "foo[bar]"); } }