mirror of
https://github.com/lifegpc/msg-tool.git
synced 2026-06-09 06:18:46 +08:00
Add name table support
This commit is contained in:
@@ -3,3 +3,4 @@ pub mod encoding;
|
||||
#[cfg(windows)]
|
||||
mod encoding_win;
|
||||
pub mod files;
|
||||
pub mod name_replacement;
|
||||
|
||||
28
src/utils/name_replacement.rs
Normal file
28
src/utils/name_replacement.rs
Normal file
@@ -0,0 +1,28 @@
|
||||
use crate::types::*;
|
||||
use anyhow::Result;
|
||||
use std::collections::HashMap;
|
||||
|
||||
pub fn read_csv(path: &str) -> Result<HashMap<String, String>> {
|
||||
let mut reader = csv::ReaderBuilder::new()
|
||||
.has_headers(true)
|
||||
.from_path(path)?;
|
||||
let mut map = HashMap::new();
|
||||
for result in reader.deserialize() {
|
||||
let record: NameTableCell = result?;
|
||||
if record.jp_name.is_empty() || record.cn_name.is_empty() {
|
||||
continue;
|
||||
}
|
||||
map.insert(record.jp_name, record.cn_name);
|
||||
}
|
||||
Ok(map)
|
||||
}
|
||||
|
||||
pub fn replace_message(mes: &mut Vec<Message>, name_table: &HashMap<String, String>) {
|
||||
for message in mes.iter_mut() {
|
||||
if let Some(name) = &message.name {
|
||||
if let Some(replace_name) = name_table.get(name) {
|
||||
message.name = Some(replace_name.clone());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user