timeshifter/tacview-splitter
1
0
Fork 0
mirror of https://github.com/132nd-vWing/tacview-splitter.git synced 2026-08-26 06:48:36 +02:00
Code Issues Projects Releases Wiki Activity

refactored get_output_filenames

Browse Source
This commit is contained in:
132nd-Professor
2021-07-17 20:59:12 +02:00
parent d9695eab02
commit efa43023f4
1 changed files with 42 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main.rs
+42 -13
View File
@@ -3,7 +3,7 @@ use std::{fs, str};
use std::io::{BufRead, BufReader}; use std::io::{BufRead, BufReader};
use zip; use zip;
use tacview_splitter::lib; use tacview_splitter::lib;
use tacview_splitter::lib::Handling; use tacview_splitter::lib::{Handling, FilenamesVariant};
const COMMENT: char = '#'; const COMMENT: char = '#';
const MINUS: char = '-'; const MINUS: char = '-';
@@ -26,7 +26,7 @@ fn main() {
let (header, body) = split_into_header_and_body(lines); let (header, body) = split_into_header_and_body(lines);
let bodies_by_coalition = divide_body_by_coalition(&body); let bodies_by_coalition = divide_body_by_coalition(&body);
let output_filenames = get_output_filenames(&input_filename); let output_filenames = get_output_filenames(&input_filename, is_zip);
if is_zip { if is_zip {
let mut descriptors = lib::Descriptors::<zip::ZipWriter<fs::File>>::new(output_filenames); let mut descriptors = lib::Descriptors::<zip::ZipWriter<fs::File>>::new(output_filenames);
descriptors.write(header, bodies_by_coalition); descriptors.write(header, bodies_by_coalition);
@@ -138,23 +138,52 @@ fn determine_line_type(line: &String) -> LineType {
return line_type return line_type
} }
fn get_output_filenames(input_filename: &String) -> lib::OutputFilenames { fn get_output_filenames(input_filename: &String, is_zip: bool) -> lib::OutputFilenames {
let blue = input_filename.replace(".zip", "_blue.zip"); let output_filenames_zip: FilenamesVariant;
let red = input_filename.replace(".zip", "_red.zip"); let output_filenames_txt: FilenamesVariant;
let violet = input_filename.replace(".zip", "_violet.zip");
let output_filenames_zip = lib::FilenamesVariant{blue, red, violet};
sanity_check_output_filenames(input_filename, &output_filenames_zip);
let blue = input_filename.replace(".txt", "_blue.txt"); if is_zip {
let red = input_filename.replace(".txt", "_red.txt"); output_filenames_zip = get_output_filenames_for_extension(
let violet = input_filename.replace(".txt", "_violet.txt"); input_filename, EXTENSION_ZIP, EXTENSION_ZIP
let output_filenames_txt = lib::FilenamesVariant{blue, red, violet}; );
sanity_check_output_filenames(input_filename, &output_filenames_txt); output_filenames_txt = get_output_filenames_for_extension(
input_filename, EXTENSION_ZIP, EXTENSION_TXT
);
} else {
output_filenames_zip = get_output_filenames_dummy();
output_filenames_txt = get_output_filenames_for_extension(
input_filename, EXTENSION_TXT, EXTENSION_TXT
);
}
let output_filenames = lib::OutputFilenames{txt: output_filenames_txt, zip: output_filenames_zip}; let output_filenames = lib::OutputFilenames{txt: output_filenames_txt, zip: output_filenames_zip};
output_filenames output_filenames
} }
fn get_output_filenames_dummy() -> lib::FilenamesVariant {
let blue = "".to_string();
let red = "".to_string();
let violet = "".to_string();
let output_filenames = lib::FilenamesVariant{blue, red, violet};
output_filenames
}
fn get_output_filenames_for_extension(input_filename: &String, old_extension: &str, new_extension: &str) -> lib::FilenamesVariant {
let blue = get_output_filenames_individual(input_filename, old_extension, new_extension, "_blue");
let red = get_output_filenames_individual(input_filename, old_extension, new_extension,"_red");
let violet = get_output_filenames_individual(input_filename, old_extension, new_extension,"_violet");
let output_filenames_zip = lib::FilenamesVariant{blue, red, violet};
sanity_check_output_filenames(input_filename, &output_filenames_zip);
output_filenames_zip
}
fn get_output_filenames_individual(input_filename: &String, old_extension: &str, new_extension: &str, coalition: &str) -> String {
let mut output_extension = coalition.to_owned();
output_extension.push_str(new_extension);
let output_filename = input_filename.replace(old_extension, &output_extension);
output_filename
}
fn sanity_check_output_filenames(input_filename: &String, output_filenames: &lib::FilenamesVariant) { fn sanity_check_output_filenames(input_filename: &String, output_filenames: &lib::FilenamesVariant) {
if input_filename == &output_filenames.blue || if input_filename == &output_filenames.blue ||
input_filename == &output_filenames.red || input_filename == &output_filenames.red ||