From 82466ee6fcd6f592fd8ca93d15adcc1a75efa1f4 Mon Sep 17 00:00:00 2001
From: Chris Cochrun <chris@cochrun.xyz>
Date: Wed, 17 May 2023 16:38:22 -0500
Subject: [PATCH] bug: fixing some adding and updating issues

---
 src/qml/presenter/ImageEditor.qml |  2 +-
 src/rust/image_model.rs           | 22 ++++++++++++++++++----
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/src/qml/presenter/ImageEditor.qml b/src/qml/presenter/ImageEditor.qml
index e49552e..d02390e 100644
--- a/src/qml/presenter/ImageEditor.qml
+++ b/src/qml/presenter/ImageEditor.qml
@@ -120,7 +120,7 @@ Item {
     }
 
     function changeImage(index) {
-        let img = imageProxyModel.getImage(index);
+        let img = imageProxyModel.imageModel.getItem(index);
         root.image = img;
         console.log(img.filePath.toString());
     }
diff --git a/src/rust/image_model.rs b/src/rust/image_model.rs
index 2df2f1b..94181ac 100644
--- a/src/rust/image_model.rs
+++ b/src/rust/image_model.rs
@@ -220,8 +220,15 @@ mod image_model {
                 .execute(db);
             match result {
                 Ok(_i) => {
-                    let image = self.as_mut().images_mut().get_mut(index as usize).unwrap();
-                    image.title = updated_title;
+                    for image in self
+                        .as_mut()
+                        .images_mut()
+                        .iter_mut()
+                        .filter(|x| x.id == index)
+                    {
+                        image.title = updated_title.clone();
+                        println!("rust-title: {:?}", image.title);
+                    }
                     self.as_mut()
                         .emit_data_changed(model_index, model_index, &vector_roles);
                     true
@@ -246,8 +253,15 @@ mod image_model {
                 .execute(db);
             match result {
                 Ok(_i) => {
-                    let image = self.as_mut().images_mut().get_mut(index as usize).unwrap();
-                    image.path = updated_file_path;
+                    for image in self
+                        .as_mut()
+                        .images_mut()
+                        .iter_mut()
+                        .filter(|x| x.id == index)
+                    {
+                        image.path = updated_file_path.clone();
+                        println!("rust-title: {:?}", image.path);
+                    }
                     self.as_mut()
                         .emit_data_changed(model_index, model_index, &vector_roles);
                     true