Directive injection and XML utilities

crates/parser/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "as3_parser"
-version = "1.0.15"
+version = "1.0.16"
 edition = "2021"
 authors = ["hydroper <[email protected]>"]
 repository = "https://github.com/hydroper/as3parser"

crates/parser/tree.rs

@@ -122,6 +122,8 @@ mod normal_configuration_directive;
 pub use normal_configuration_directive::*;
 mod package_concat_directive;
 pub use package_concat_directive::*;
+mod directive_injection_node;
+pub use directive_injection_node::*;
 
 // Miscellaneous
 mod attributes;

crates/parser/tree/directive.rs

@@ -30,6 +30,7 @@ pub enum Directive {
     IncludeDirective(IncludeDirective),
     NormalConfigurationDirective(NormalConfigurationDirective),
     PackageConcatDirective(PackageConcatDirective),
+    DirectiveInjection(DirectiveInjectionNode),
     VariableDefinition(VariableDefinition),
     FunctionDefinition(FunctionDefinition),
     ClassDefinition(ClassDefinition),
@@ -68,6 +69,7 @@ impl Directive {
             Self::IncludeDirective(d) => d.location.clone(),
             Self::NormalConfigurationDirective(d) => d.location.clone(),
             Self::PackageConcatDirective(d) => d.location.clone(),
+            Self::DirectiveInjection(d) => d.location.clone(),
             Self::VariableDefinition(d) => d.location.clone(),
             Self::FunctionDefinition(d) => d.location.clone(),
             Self::ClassDefinition(d) => d.location.clone(),

crates/parser/tree/directive_injection_node.rs

@@ -0,0 +1,9 @@
+use crate::ns::*;
+use serde::{Serialize, Deserialize};
+
+/// Node that allows modification to the directive sequence.
+#[derive(Clone, Serialize, Deserialize)]
+pub struct DirectiveInjectionNode {
+    pub location: Location,
+    pub directives: RefCell<Vec<Rc<Directive>>>,
+}

crates/parser/tree/mxml.rs

@@ -30,6 +30,18 @@ pub struct MxmlElement {
     pub closing_name: Option<MxmlName>,
 }
 
+impl MxmlElement {
+    pub fn inner_text(&self) -> String {
+        let mut j = String::new();
+        if let Some(c) = self.content.as_ref() {
+            for c1 in c.iter() {
+                j.push_str(&c1.inner_text());
+            }
+        }
+        j
+    }
+}
+
 #[derive(Clone, Serialize, Deserialize)]
 pub struct MxmlAttribute {
     pub location: Location,
@@ -120,6 +132,16 @@ impl MxmlContent {
             Self::Element(e) => e.location.clone(),
         }
     }
+
+    pub fn inner_text(&self) -> String {
+        match self {
+            Self::Characters((data, _)) => data.clone(),
+            Self::CData((data, _)) => data["<![CDATA[".len()..(data.len() - 3)].to_owned(),
+            Self::Comment(_) => String::new(),
+            Self::ProcessingInstruction { .. } => String::new(),
+            Self::Element(e) => e.inner_text(),
+        }
+    }
 }
 
 /// Mapping of namespace prefixes.