lifegpc/msg-tool
1
0
Fork 0
mirror of https://github.com/lifegpc/msg-tool.git synced 2026-06-18 00:45:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added a new command option archive-encoding

Browse Source
This commit is contained in:
lifegpc
2025-06-03 10:24:23 +08:00
parent 3e1c7c28fb
commit 9b4f6e4e6e
7 changed files with 98 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
src/main.rs
View File

@@ -34,6 +34,33 @@ fn get_encoding(
builder.default_encoding()
}
fn get_archived_encoding(
arg: &args::Arg,
builder: &Box<dyn scripts::ScriptBuilder + Send + Sync>,
encoding: types::Encoding,
) -> types::Encoding {
match &arg.encoding {
Some(enc) => {
return match enc {
&types::TextEncoding::Default => builder.default_encoding(),
&types::TextEncoding::Auto => types::Encoding::Auto,
&types::TextEncoding::Cp932 => types::Encoding::Cp932,
&types::TextEncoding::Utf8 => types::Encoding::Utf8,
&types::TextEncoding::Gb2312 => types::Encoding::Gb2312,
};
}
None => {}
}
#[cfg(windows)]
match &arg.code_page {
Some(code_page) => {
return types::Encoding::CodePage(*code_page);
}
None => {}
}
builder.default_archive_encoding().unwrap_or(encoding)
}
fn get_output_encoding(arg: &args::Arg) -> types::Encoding {
match &arg.output_encoding {
Some(enc) => {
@@ -96,8 +123,14 @@ pub fn parse_script(
for builder in scripts::BUILDER.iter() {
if typ == builder.script_type() {
let encoding = get_encoding(arg, builder);
let archive_encoding = get_archived_encoding(arg, builder, encoding);
return Ok((
builder.build_script_from_file(filename, encoding, config)?,
builder.build_script_from_file(
filename,
encoding,
archive_encoding,
config,
)?,
builder,
));
}
@@ -123,8 +156,9 @@ pub fn parse_script(
if exts_builder.len() == 1 {
let builder = exts_builder.first().unwrap();
let encoding = get_encoding(arg, builder);
let archive_encoding = get_archived_encoding(arg, builder, encoding);
return Ok((
builder.build_script_from_file(filename, encoding, config)?,
builder.build_script_from_file(filename, encoding, archive_encoding, config)?,
builder,
));
}
@@ -153,8 +187,9 @@ pub fn parse_script(
if best_builders.len() == 1 {
let builder = best_builders.first().unwrap();
let encoding = get_encoding(arg, builder);
let archive_encoding = get_archived_encoding(arg, builder, encoding);
return Ok((
builder.build_script_from_file(filename, encoding, config)?,
builder.build_script_from_file(filename, encoding, archive_encoding, config)?,
builder,
));
}
@@ -194,8 +229,9 @@ pub fn parse_script_from_archive(
if exts_builder.len() == 1 {
let builder = exts_builder.first().unwrap();
let encoding = get_encoding(arg, builder);
let archive_encoding = get_archived_encoding(arg, builder, encoding);
return Ok((
builder.build_script(file.data().to_vec(), encoding, config)?,
builder.build_script(file.data().to_vec(), encoding, archive_encoding, config)?,
builder,
));
}
@@ -218,8 +254,9 @@ pub fn parse_script_from_archive(
if best_builders.len() == 1 {
let builder = best_builders.first().unwrap();
let encoding = get_encoding(arg, builder);
let archive_encoding = get_archived_encoding(arg, builder, encoding);
return Ok((
builder.build_script(file.data().to_vec(), encoding, config)?,
builder.build_script(file.data().to_vec(), encoding, archive_encoding, config)?,
builder,
));
}