From bdba41afe5191c2a65030066308071d09ab3748e Mon Sep 17 00:00:00 2001
From: Christopher Sardegna <github@reagentx.net>
Date: Sat, 29 Jul 2023 12:33:09 -0700
Subject: [PATCH] Use correct path for extension

---
 imessage-exporter/src/app/attachment_manager.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/imessage-exporter/src/app/attachment_manager.rs b/imessage-exporter/src/app/attachment_manager.rs
index 3ffa89b2..25d14b1b 100644
--- a/imessage-exporter/src/app/attachment_manager.rs
+++ b/imessage-exporter/src/app/attachment_manager.rs
@@ -5,7 +5,10 @@ use std::{
 };
 
 use filetime::{set_file_times, FileTime};
-use imessage_database::tables::{attachment::Attachment, messages::Message};
+use imessage_database::{
+    tables::{attachment::Attachment, messages::Message},
+    util::platform::Platform,
+};
 use uuid::Uuid;
 
 use crate::app::{
@@ -56,8 +59,6 @@ impl AttachmentManager {
                 return None;
             }
 
-            let ext = from.extension()?;
-
             // Create a path to copy the file to
             let mut to = config.attachment_path();
 
@@ -68,7 +69,8 @@ impl AttachmentManager {
             // Add a random filename
             to.push(Uuid::new_v4().to_string());
 
-            to.set_extension(ext);
+            // Set the new file's extension to the original one
+            to.set_extension(attachment.extension()?);
 
             match self {
                 AttachmentManager::Compatible => match &config.converter {