Add localization

2 files changed, 8 insertions, 5 deletions

diff --git a/src/handlers/export.rs b/src/handlers/export.rs
index 0b64a73..a0a4451 100644
--- a/src/handlers/export.rs
+++ b/src/handlers/export.rs
@@ -18,24 +18,26 @@ use std::path::PathBuf;
 use crate::AppModel;
 use crate::actions::ImageFormat;
 use crate::dialogs;
+use crate::tr;
+use crate::stages::LocalizedStageType;
 use wmap_renderer::{render_to_png, render_to_svg};
 
 /// Exports the map to an image file in the specified format.
 pub fn export_to_file(model: &AppModel, path: PathBuf, format: ImageFormat) {
     let result: Result<(), String> = match format {
         ImageFormat::Png => {
-            render_to_png(&model.map, &model.stage_type, &model.render_configuration)
+            render_to_png(&model.map, &model.stage_type.to_localized(), &model.render_configuration)
                 .map_err(|e| e.to_string())
                 .and_then(|data| std::fs::write(&path, data).map_err(|e| e.to_string()))
         }
         ImageFormat::Svg => {
-            render_to_svg(&model.map, &model.stage_type, &model.render_configuration)
+            render_to_svg(&model.map, &model.stage_type.to_localized(), &model.render_configuration)
                 .map_err(|e| e.to_string())
                 .and_then(|data| std::fs::write(&path, data).map_err(|e| e.to_string()))
         }
     };
 
     if let Err(error) = result {
-        dialogs::show_error(&model.window, "Error Exporting Image", &error);
+        dialogs::show_error(&model.window, &tr!("error.file.export"), &error);
     }
 }
diff --git a/src/handlers/file.rs b/src/handlers/file.rs
index 615a1f7..bd501e8 100644
--- a/src/handlers/file.rs
+++ b/src/handlers/file.rs
@@ -23,6 +23,7 @@ use crate::actions::Action;
 use crate::constants;
 use crate::dialogs;
 use crate::file_registry;
+use crate::tr;
 
 /// Loads a file into the current document, updating the file registry.
 pub fn load_file(model: &mut AppModel, path: PathBuf) {
@@ -35,7 +36,7 @@ pub fn load_file(model: &mut AppModel, path: PathBuf) {
             model.modified = false;
         }
         Err(error) => {
-            dialogs::show_error(&model.window, "Error Opening File", &error.to_string());
+            dialogs::show_error(&model.window, &tr!("errors.file.open"), &error.to_string());
         }
     }
 }
@@ -58,7 +59,7 @@ pub fn save_to_file(model: &mut AppModel, path: PathBuf, close_after: bool) {
             }
         }
         Err(error) => {
-            dialogs::show_error(&model.window, "Error Saving File", &error.to_string());
+            dialogs::show_error(&model.window, &tr!("errors.file.save"), &error.to_string());
         }
     }
 }