|
| 1 | +use std::{fs, env}; |
1 | 2 | use anyhow::Result;
|
2 | 3 | use axmldecoder::{Cdata, Element, Node};
|
3 | 4 |
|
4 | 5 | fn main() -> Result<()> {
|
5 |
| - let args: Vec<String> = std::env::args().collect(); |
6 |
| - let fname = args.get(1).unwrap(); |
| 6 | + let args: Vec<String> = env::args().collect(); |
7 | 7 |
|
8 |
| - let f = std::fs::read(fname)?; |
9 |
| - let xml = axmldecoder::parse(&f)?; |
| 8 | + if args.len() <= 1 { |
| 9 | + for path in fs::read_dir("../examples").unwrap() { |
| 10 | + let f = std::fs::read(path?.path())?; |
| 11 | + let xml = axmldecoder::parse(&f)?; |
| 12 | + |
| 13 | + let root = xml.get_root().as_ref().unwrap(); |
| 14 | + let mut s = String::new(); |
| 15 | + s.push_str("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"); |
| 16 | + format_xml(&root, 0_usize, &mut s); |
| 17 | + |
| 18 | + let s = s.trim().to_string(); |
| 19 | + println!("{}", s); |
| 20 | + } |
| 21 | + } else { |
| 22 | + let fname = args.get(1).unwrap(); |
10 | 23 |
|
11 |
| - let root = xml.get_root().as_ref().unwrap(); |
12 |
| - let mut s = String::new(); |
13 |
| - s.push_str("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"); |
14 |
| - format_xml(&root, 0_usize, &mut s); |
| 24 | + let f = std::fs::read(fname)?; |
| 25 | + let xml = axmldecoder::parse(&f)?; |
15 | 26 |
|
16 |
| - let s = s.trim().to_string(); |
17 |
| - println!("{}", s); |
| 27 | + let root = xml.get_root().as_ref().unwrap(); |
| 28 | + let mut s = String::new(); |
| 29 | + s.push_str("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"); |
| 30 | + format_xml(&root, 0_usize, &mut s); |
| 31 | + |
| 32 | + let s = s.trim().to_string(); |
| 33 | + println!("{}", s); |
| 34 | + } |
18 | 35 |
|
19 | 36 | Ok(())
|
20 | 37 | }
|
|
0 commit comments