Merge pull request #23 from hatal175/master

Add set_title to FileDialog
PolyMeilex · Sep 12, 2021 · da92e0c · da92e0c
2 parents c1855df + ac7fe4d
commit da92e0c
Show file tree

Hide file tree

Showing 4 changed files with 63 additions and 4 deletions.
diff --git a/src/backend/gtk3/file_dialog/dialog_ffi.rs b/src/backend/gtk3/file_dialog/dialog_ffi.rs
@@ -156,7 +156,7 @@ impl GtkFileDialog {
 impl GtkFileDialog {
     pub fn build_pick_file(opt: &FileDialog) -> Self {
         let mut dialog =
-            GtkFileDialog::new("Open File", GtkFileChooserAction::Open, "Cancel", "Open");
+            GtkFileDialog::new(opt.title.as_deref().unwrap_or("Open File"), GtkFileChooserAction::Open, "Cancel", "Open");
 
         dialog.add_filters(&opt.filters);
         dialog.set_path(opt.starting_directory.as_deref());
@@ -166,7 +166,7 @@ impl GtkFileDialog {
 
     pub fn build_save_file(opt: &FileDialog) -> Self {
         let mut dialog =
-            GtkFileDialog::new("Save File", GtkFileChooserAction::Save, "Cancel", "Save");
+            GtkFileDialog::new(opt.title.as_deref().unwrap_or("Save File"), GtkFileChooserAction::Save, "Cancel", "Save");
 
         unsafe { gtk_sys::gtk_file_chooser_set_do_overwrite_confirmation(dialog.ptr as _, 1) };
 
@@ -194,7 +194,7 @@ impl GtkFileDialog {
 
     pub fn build_pick_folder(opt: &FileDialog) -> Self {
         let dialog = GtkFileDialog::new(
-            "Select Folder",
+            opt.title.as_deref().unwrap_or("Select Folder"),
             GtkFileChooserAction::SelectFolder,
             "Cancel",
             "Select",
@@ -206,7 +206,7 @@ impl GtkFileDialog {
 
     pub fn build_pick_files(opt: &FileDialog) -> Self {
         let mut dialog =
-            GtkFileDialog::new("Open File", GtkFileChooserAction::Open, "Cancel", "Open");
+            GtkFileDialog::new(opt.title.as_deref().unwrap_or("Open File"), GtkFileChooserAction::Open, "Cancel", "Open");
 
         unsafe { gtk_sys::gtk_file_chooser_set_select_multiple(dialog.ptr as _, 1) };
         dialog.add_filters(&opt.filters);

diff --git a/src/backend/macos/file_dialog/panel_ffi.rs b/src/backend/macos/file_dialog/panel_ffi.rs
@@ -116,6 +116,13 @@ impl Panel {
         }
     }
 
+    pub fn set_title(&self, title: &str) {
+        unsafe {
+            let title = make_nsstring(title);
+            let () = msg_send![self.panel, setTitle: title];
+        }
+    }
+
     pub fn get_result(&self) -> PathBuf {
         unsafe {
             let url = msg_send![self.panel, URL];
@@ -155,6 +162,10 @@ impl Panel {
             panel.set_file_name(file_name);
         }
 
+        if let Some(title) = &opt.title {
+            panel.set_title(title);
+        }
+
         panel.set_can_choose_directories(NO);
         panel.set_can_choose_files(YES);
 
@@ -176,6 +187,10 @@ impl Panel {
             panel.set_file_name(file_name);
         }
 
+        if let Some(title) = &opt.title {
+            panel.set_title(title);
+        }
+
         panel
     }
 
@@ -186,6 +201,10 @@ impl Panel {
             panel.set_path(path, opt.file_name.as_deref());
         }
 
+        if let Some(title) = &opt.title {
+            panel.set_title(title);
+        }
+
         panel.set_can_choose_directories(YES);
         panel.set_can_choose_files(NO);
 
@@ -203,6 +222,10 @@ impl Panel {
             panel.set_path(path, opt.file_name.as_deref());
         }
 
+        if let Some(title) = &opt.title {
+            panel.set_title(title);
+        }
+
         panel.set_can_choose_directories(NO);
         panel.set_can_choose_files(YES);
         panel.set_allows_multiple_selection(YES);

diff --git a/src/backend/win_cid/file_dialog/dialog_ffi.rs b/src/backend/win_cid/file_dialog/dialog_ffi.rs
@@ -174,6 +174,17 @@ impl IDialog {
         Ok(())
     }
 
+    fn set_title(&self, title: &Option<String>) -> Result<(), HRESULT> {
+        if let Some(title) = title {
+            let wide_title: Vec<u16> = OsStr::new(title).encode_wide().chain(once(0)).collect();
+
+            unsafe {
+                (*self.0).SetTitle(wide_title.as_ptr()).check()?;
+            }
+        }
+        Ok(())
+    }
+
     pub fn get_results(&self) -> Result<Vec<PathBuf>, HRESULT> {
         unsafe {
             let mut res_items: *mut IShellItemArray = ptr::null_mut();
@@ -237,6 +248,7 @@ impl IDialog {
         dialog.add_filters(&opt.filters)?;
         dialog.set_path(&opt.starting_directory)?;
         dialog.set_file_name(&opt.file_name)?;
+        dialog.set_title(&opt.title)?;
 
         Ok(dialog)
     }
@@ -247,6 +259,7 @@ impl IDialog {
         dialog.add_filters(&opt.filters)?;
         dialog.set_path(&opt.starting_directory)?;
         dialog.set_file_name(&opt.file_name)?;
+        dialog.set_title(&opt.title)?;
 
         Ok(dialog)
     }
@@ -255,6 +268,7 @@ impl IDialog {
         let dialog = IDialog::new_open_dialog(opt)?;
 
         dialog.set_path(&opt.starting_directory)?;
+        dialog.set_title(&opt.title)?;
 
         unsafe {
             dialog.SetOptions(FOS_PICKFOLDERS).check()?;
@@ -269,6 +283,7 @@ impl IDialog {
         dialog.add_filters(&opt.filters)?;
         dialog.set_path(&opt.starting_directory)?;
         dialog.set_file_name(&opt.file_name)?;
+        dialog.set_title(&opt.title)?;
 
         unsafe {
             dialog.SetOptions(FOS_ALLOWMULTISELECT).check()?;

diff --git a/src/dialog.rs b/src/dialog.rs
@@ -22,6 +22,7 @@ pub struct FileDialog {
     pub(crate) filters: Vec<Filter>,
     pub(crate) starting_directory: Option<PathBuf>,
     pub(crate) file_name: Option<String>,
+    pub(crate) title: Option<String>,
     #[cfg(feature = "parent")]
     pub(crate) parent: Option<RawWindowHandle>,
 }
@@ -73,6 +74,16 @@ impl FileDialog {
         self
     }
 
+    /// Set the title of the dialog.
+    /// #### Supported Platforms:
+    /// - Windows
+    /// - Linux
+    /// - Mac (Only below version 10.11)
+    pub fn set_title(mut self, title: &str) -> Self {
+        self.title = Some(title.into());
+        self
+    }
+
     #[cfg(feature = "parent")]
     /// Set parent windows explicitly (optional)
     /// Suported in: `macos` and `windows`
@@ -172,6 +183,16 @@ impl AsyncFileDialog {
         self
     }
 
+    /// Set the title of the dialog.
+    /// #### Supported Platforms:
+    /// - Windows
+    /// - Linux
+    /// - Mac (Only below version 10.11)
+    pub fn set_title(mut self, title: &str) -> Self {
+        self.file_dialog = self.file_dialog.set_title(title);
+        self
+    }
+
     #[cfg(feature = "parent")]
     /// Set parent windows explicitly (optional)
     /// Suported in: `macos` and `windows`