1use crate::extract::{extract_xfa_from_bytes, XfaPackets};
24
25#[derive(Debug, Clone, Copy, PartialEq, Eq)]
27pub enum XfaType {
28 None,
30 Static,
33 Dynamic,
35}
36
37impl std::fmt::Display for XfaType {
38 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
39 match self {
40 XfaType::None => write!(f, "None"),
41 XfaType::Static => write!(f, "Static"),
42 XfaType::Dynamic => write!(f, "Dynamic"),
43 }
44 }
45}
46
47pub fn detect_xfa_type_from_packets(packets: &XfaPackets) -> XfaType {
51 if packets.packets.is_empty() && packets.full_xml.is_none() {
53 return XfaType::None;
54 }
55
56 let template_xml: Option<&str> = packets.template();
59 let full_xml: Option<&str> = packets.full_xml.as_deref();
60
61 let search_text: &str = template_xml.or(full_xml).unwrap_or("");
62
63 if search_text.contains(r#"baseProfile="interactiveForms""#) {
64 XfaType::Static
65 } else if !search_text.is_empty() || packets.packets.iter().any(|(name, _)| name == "template")
66 {
67 XfaType::Dynamic
68 } else {
69 XfaType::Dynamic
72 }
73}
74
75pub fn detect_xfa_type(pdf_bytes: &[u8]) -> XfaType {
80 match extract_xfa_from_bytes(pdf_bytes.to_vec()) {
82 Ok(packets) => detect_xfa_type_from_packets(&packets),
83 Err(_) => XfaType::None,
84 }
85}
86
87#[cfg(test)]
90mod tests {
91 use super::*;
92 use crate::extract::XfaPackets;
93
94 fn packets_from_xml(xml: &str) -> XfaPackets {
95 let mut p = XfaPackets::default();
98 p.full_xml = Some(xml.to_string());
99 if xml.contains("<template") {
101 let start = xml.find("<template").unwrap();
102 let end = xml
104 .find("</template>")
105 .map(|i| i + "</template>".len())
106 .or_else(|| {
107 Some(xml.len())
109 })
110 .unwrap();
111 p.packets
112 .push(("template".to_string(), xml[start..end].to_string()));
113 }
114 p
115 }
116
117 #[test]
118 fn empty_packets_returns_none() {
119 let p = XfaPackets::default();
120 assert_eq!(detect_xfa_type_from_packets(&p), XfaType::None);
121 }
122
123 #[test]
124 fn static_form_detected_via_base_profile() {
125 let xml = r#"<xdp:xdp xmlns:xdp="http://ns.adobe.com/xdp/"><template xmlns="http://www.xfa.org/schema/xfa-template/3.3/" baseProfile="interactiveForms"><subform name="root"/></template></xdp:xdp>"#;
126 let p = packets_from_xml(xml);
127 assert_eq!(detect_xfa_type_from_packets(&p), XfaType::Static);
128 }
129
130 #[test]
131 fn dynamic_form_detected_when_no_base_profile() {
132 let xml = r#"<xdp:xdp xmlns:xdp="http://ns.adobe.com/xdp/"><template xmlns="http://www.xfa.org/schema/xfa-template/3.3/"><subform name="root"><occur min="0" max="-1"/></subform></template></xdp:xdp>"#;
133 let p = packets_from_xml(xml);
134 assert_eq!(detect_xfa_type_from_packets(&p), XfaType::Dynamic);
135 }
136
137 #[test]
138 fn xfa_type_none_on_empty_pdf_bytes() {
139 assert_eq!(detect_xfa_type(&[]), XfaType::None);
141 }
142
143 #[test]
144 fn xfa_type_display() {
145 assert_eq!(XfaType::None.to_string(), "None");
146 assert_eq!(XfaType::Static.to_string(), "Static");
147 assert_eq!(XfaType::Dynamic.to_string(), "Dynamic");
148 }
149
150 #[test]
151 fn packets_with_only_full_xml_static() {
152 let xml = r#"<?xml version="1.0"?><xdp:xdp xmlns:xdp="http://ns.adobe.com/xdp/"><template baseProfile="interactiveForms"><subform/></template></xdp:xdp>"#;
154 let mut p = XfaPackets::default();
155 p.full_xml = Some(xml.to_string());
156 assert_eq!(detect_xfa_type_from_packets(&p), XfaType::Static);
158 }
159
160 #[test]
161 fn datasets_only_packet_treated_as_dynamic() {
162 let mut p = XfaPackets::default();
163 p.packets
164 .push(("datasets".to_string(), "<xfa:datasets/>".to_string()));
165 assert_eq!(detect_xfa_type_from_packets(&p), XfaType::Dynamic);
167 }
168}