osc-variant/src/main.rs

/*
 * MIT License
 *
 * Copyright (c) 2023 Comprehensive Cancer Center Mainfranken
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

use std::fs;
use std::fs::OpenOptions;
use std::io::Write;
use std::ops::Add;

use clap::Parser;
use quick_xml::de::from_str;
use quick_xml::se::Serializer;
use serde::Serialize;

use crate::model::onkostar_editor::OnkostarEditor;

mod model;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
#[command(propagate_version = true, arg_required_else_help(true))]
struct Cli {
    #[arg(
        long = "input",
        help = "Eingabedatei"
    )]
    input: String,
    #[arg(
        long = "output",
        help = "Ausgabedatei (Optional)"
    )]
    output: Option<String>
}

fn main() {
    let cli = Cli::parse();

    let contents = fs::read_to_string(cli.input)
        .expect("Should have been able to read the file");

    let xml: OnkostarEditor = from_str(contents.as_str()).unwrap();

    let mut buf = String::new();

    let mut serializier = Serializer::new(&mut buf);
    serializier.indent(' ', 2);

    xml.serialize(serializier).expect("Generated XML");

    let output = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n".to_string()
        .add(buf
            // Replace &apos; and &quot; as used in original file
            .replace("&apos;", "'")
            .replace("&quot;", "\"")
            .as_str());

    match cli.output {
        Some(filename) => {
            let mut file = OpenOptions::new()
                .read(false)
                .write(true)
                .create(true)
                .truncate(true)
                .open(filename).unwrap();
            file.write_all(output.as_bytes()).expect("Should have written output file");
        },
        None => {
            println!("{}", output)
        }
    }
}