doukutsu-rs/src/i18n.rs

103 lines
3.1 KiB
Rust

use std::collections::HashMap;
use crate::framework::context::Context;
use crate::framework::filesystem;
use crate::game::scripting::tsc::text_script::TextScriptEncoding;
use crate::game::shared_game_state::FontData;
#[derive(Debug, Clone)]
pub struct Locale {
pub code: String,
pub name: String,
pub font: FontData,
pub encoding: Option<TextScriptEncoding>,
pub stage_encoding: Option<TextScriptEncoding>,
strings: HashMap<String, String>,
}
impl Default for Locale {
fn default() -> Self {
Locale {
code: "en".to_owned(),
name: "English".to_owned(),
font: FontData { path: String::new(), scale: 1.0, space_offset: 0.0 },
encoding: None,
stage_encoding: None,
strings: HashMap::new(),
}
}
}
impl Locale {
pub fn new(ctx: &mut Context, base_paths: &Vec<String>, code: &str) -> Locale {
let file = filesystem::open_find(ctx, base_paths, &format!("locale/{code}.json")).unwrap();
let json: serde_json::Value = serde_json::from_reader(file).unwrap();
let strings = Locale::flatten(&json);
let name = strings["name"].clone();
let font_name = strings["font"].clone();
let font_scale = strings["font_scale"].parse::<f32>().unwrap_or(1.0);
let font = FontData::new(font_name, font_scale, 0.0);
let encoding = if let Some(enc) = strings.get("encoding").clone() {
Some(TextScriptEncoding::from(enc.as_str()))
} else {
None
};
let stage_encoding = if let Some(enc) = strings.get("stage_encoding").clone() {
Some(TextScriptEncoding::from(enc.as_str()))
} else {
None
};
Locale { code: code.to_string(), name, font, encoding, stage_encoding, strings }
}
fn flatten(json: &serde_json::Value) -> HashMap<String, String> {
let mut strings = HashMap::new();
for (key, value) in json.as_object().unwrap() {
match value {
serde_json::Value::String(string) => {
strings.insert(key.to_owned(), string.to_owned());
}
serde_json::Value::Object(_) => {
let substrings = Locale::flatten(value);
for (sub_key, sub_value) in substrings.iter() {
strings.insert(format!("{}.{}", key, sub_key), sub_value.to_owned());
}
}
_ => {}
}
}
strings
}
/// if the key does not exists, return the origin key instead
pub fn t<'a: 'b, 'b>(&'a self, key: &'b str) -> &'b str {
if let Some(str) = self.strings.get(key) {
str
} else {
key
}
}
pub fn tt(&self, key: &str, args: &[(&str, &str)]) -> String {
let mut string = self.t(key).to_owned();
for (key, value) in args.iter() {
string = string.replace(&format!("{{{}}}", key), &value);
}
string
}
pub fn set_font(&mut self, font: FontData) {
self.font = font;
}
}