Support folder selection in FilePickerTextBox

Author: saddam213

OnnxStack.UI/UserControls/FilePickerTextBox.xaml

@@ -40,17 +40,40 @@
                             <MultiDataTrigger.Conditions>
                                 <Condition Binding="{Binding IsEnabled, ElementName=UI}" Value="True"/>
                                 <Condition Binding="{Binding IsRequired}" Value="True"/>
+                                <Condition Binding="{Binding IsFolderPicker}" Value="False"/>
                             </MultiDataTrigger.Conditions>
                             <Setter Property="Text">
                                 <Setter.Value>
                                     <Binding Path="FileName" UpdateSourceTrigger="PropertyChanged" NotifyOnValidationError="True">
                                         <Binding.ValidationRules>
-                                            <local:FileExistsValidationRule IsRequired="True" ValidatesOnTargetUpdated="True" />
+                                            <local:FileExistsValidationRule IsRequired="True" IsFolder="False" ValidatesOnTargetUpdated="True" />
                                         </Binding.ValidationRules>
                                     </Binding>
                                 </Setter.Value>
                             </Setter>
                         </MultiDataTrigger>
+                        <MultiDataTrigger>
+                            <MultiDataTrigger.Conditions>
+                                <Condition Binding="{Binding IsEnabled, ElementName=UI}" Value="True"/>
+                                <Condition Binding="{Binding IsRequired}" Value="True"/>
+                                <Condition Binding="{Binding IsFolderPicker}" Value="True"/>
+                            </MultiDataTrigger.Conditions>
+                            <Setter Property="Text">
+                                <Setter.Value>
+                                    <Binding Path="FileName" UpdateSourceTrigger="PropertyChanged" NotifyOnValidationError="True">
+                                        <Binding.ValidationRules>
+                                            <local:FileExistsValidationRule IsRequired="True" IsFolder="True" ValidatesOnTargetUpdated="True" />
+                                        </Binding.ValidationRules>
+                                    </Binding>
+                                </Setter.Value>
+                            </Setter>
+                        </MultiDataTrigger>
+                        <MultiDataTrigger>
+                            <MultiDataTrigger.Conditions>
+                                <Condition Binding="{Binding IsRequired}" Value="False"/>
+                            </MultiDataTrigger.Conditions>
+                            <Setter Property="Validation.ErrorTemplate" Value="{x:Null}"/>
+                        </MultiDataTrigger>
                     </Style.Triggers>
                 </Style>
             </TextBox.Style>

OnnxStack.UI/UserControls/FilePickerTextBox.xaml.cs

@@ -15,7 +15,7 @@ public partial class FilePickerTextBox : UserControl
         public FilePickerTextBox()
         {
             ClearFileCommand = new RelayCommand(ClearFile, CanExecuteClearFile);
-            OpenFileDialogCommand = new RelayCommand(OpenFilePicker, CanExecuteOpenFilePicker);
+            OpenFileDialogCommand = new RelayCommand(OpenPicker, CanExecuteOpenPicker);
             InitializeComponent();
         }
 
@@ -37,6 +37,9 @@ public FilePickerTextBox()
         public static readonly DependencyProperty IsRequiredProperty =
             DependencyProperty.Register("IsRequired", typeof(bool), typeof(FilePickerTextBox));
 
+        public static readonly DependencyProperty IsFolderPickerProperty =
+            DependencyProperty.Register("IsFolderPicker", typeof(bool), typeof(FilePickerTextBox));
+
 
         /// <summary>
         /// Gets or sets the clear file command.
@@ -115,6 +118,46 @@ public bool IsRequired
         }
 
 
+        /// <summary>
+        /// Gets or sets a value indicating whether this instance is folder picker.
+        /// </summary>
+        /// <value>
+        ///   <c>true</c> if this instance is folder picker; otherwise, <c>false</c>.
+        /// </value>
+        public bool IsFolderPicker
+        {
+            get { return (bool)GetValue(IsFolderPickerProperty); }
+            set { SetValue(IsFolderPickerProperty, value); }
+        }
+
+
+        /// <summary>
+        /// Opens the picker.
+        /// </summary>
+        private void OpenPicker()
+        {
+            if (IsFolderPicker)
+            {
+                OpenFolderPicker();
+                return;
+            }
+
+            OpenFilePicker();
+        }
+
+
+        /// <summary>
+        /// Determines whether this instance can execute OpenPicker.
+        /// </summary>
+        /// <returns>
+        ///   <c>true</c> if this instance can execute OpenPicker; otherwise, <c>false</c>.
+        /// </returns>
+        private bool CanExecuteOpenPicker()
+        {
+            return true;
+        }
+
+
         /// <summary>
         /// Opens the file picker.
         /// </summary>
@@ -135,14 +178,19 @@ private void OpenFilePicker()
 
 
         /// <summary>
-        /// Determines whether this instance can execute OpenFilePicker.
+        /// Opens the folder picker.
         /// </summary>
-        /// <returns>
-        ///   <c>true</c> if this instance can execute OpenFilePicker; otherwise, <c>false</c>.
-        /// </returns>
-        private bool CanExecuteOpenFilePicker()
+        private void OpenFolderPicker()
         {
-            return true;
+            var folderBrowserDialog = new System.Windows.Forms.FolderBrowserDialog
+            {
+                Description = Title,
+                InitialDirectory = InitialDirectory,
+                UseDescriptionForTitle = true
+            };
+            var dialogResult = folderBrowserDialog.ShowDialog();
+            if (dialogResult == System.Windows.Forms.DialogResult.OK)
+                FileName = folderBrowserDialog.SelectedPath;
         }
 
 
@@ -174,6 +222,7 @@ private bool CanExecuteClearFile()
     /// <seealso cref="System.Windows.Controls.ValidationRule" />
     public class FileExistsValidationRule : ValidationRule
     {
+        public bool IsFolder { get; set; }
         public bool IsRequired { get; set; }
 
         public override ValidationResult Validate(object value, CultureInfo cultureInfo)
@@ -182,9 +231,12 @@ public override ValidationResult Validate(object value, CultureInfo cultureInfo)
             if (!IsRequired && string.IsNullOrEmpty(filename))
                 return ValidationResult.ValidResult;
 
-            if (!File.Exists(filename))
+            if (!IsFolder && !File.Exists(filename))
                 return new ValidationResult(false, $"File does not exist");
 
+            if (IsFolder && !Directory.Exists(filename))
+                return new ValidationResult(false, $"Directory does not exist");
+
             return ValidationResult.ValidResult;
         }
     }