Use icon provider in file column builder

Author: pcastornii

shellfx-dialogs/src/main/java/com/techsenger/shellfx/dialogs/file/FileChooserDialogFxView.java

@@ -27,6 +27,7 @@
 import com.techsenger.shellfx.dialogs.style.DialogIcons;
 import com.techsenger.shellfx.material.button.ResultButton;
 import com.techsenger.shellfx.material.icon.FontIconView;
+import com.techsenger.shellfx.material.icon.GenericFontIcon;
 import com.techsenger.shellfx.material.list.TextFieldColumnListCell;
 import com.techsenger.shellfx.material.style.Spacing;
 import com.techsenger.shellfx.material.style.StyleClasses;
@@ -42,6 +43,7 @@
 import java.util.Comparator;
 import java.util.List;
 import java.util.Map;
+import java.util.function.Function;
 import javafx.collections.FXCollections;
 import javafx.collections.ObservableList;
 import javafx.geometry.Pos;
@@ -219,6 +221,9 @@ protected void updateItem(T item, boolean empty) {
 
     private AppearanceSettings settings;
 
+    private Function<T, GenericFontIcon<?>> iconProvider =
+            (f) -> f.isDirectory() ? DialogIcons.DIRECTORY : DialogIcons.FILE;
+
     public FileChooserDialogFxView() {
         super();
     }
@@ -348,6 +353,14 @@ public void setExtensionFilter(ExtensionFilter filter) {
         this.filterComboBox.getSelectionModel().select(filter);
     }
 
+    public Function<T, GenericFontIcon<?>> getIconProvider() {
+        return iconProvider;
+    }
+
+    public void setIconProvider(Function<T, GenericFontIcon<?>> iconProvider) {
+        this.iconProvider = iconProvider;
+    }
+
     @Override
     protected Composer createComposer() {
         return new FileChooserDialogFxView.Composer();
@@ -372,7 +385,7 @@ protected void build() {
 
         var columnBuilder = new FileColumnBuilder(settings.getRegularFont());
         this.fileColumnManager.registerColumnFactory(FileColumns.NAME, () -> {
-            var column = columnBuilder.buildNameColumn(DialogIcons.DIRECTORY, DialogIcons.FILE);
+            var column = columnBuilder.buildNameColumn(iconProvider);
             column.setEditable(false);
             column.setOnEditCancel(e -> {
                 var file = (T) e.getOldValue();

shellfx-storage/src/main/java/com/techsenger/shellfx/storage/FileColumnBuilder.java

@@ -26,6 +26,7 @@
 import java.time.ZoneId;
 import java.time.format.DateTimeFormatter;
 import java.util.Comparator;
+import java.util.function.Function;
 import javafx.beans.property.ReadOnlyObjectWrapper;
 import javafx.scene.Node;
 import javafx.scene.control.TableCell;
@@ -49,8 +50,8 @@ public FileColumnBuilder(Font font) {
      *
      * @return
      */
-    public <F extends GenericFile> NamedTableColumn<F, F> buildNameColumn(GenericFontIcon<?> dirIcon,
-            GenericFontIcon fileIcon) {
+    public <F extends GenericFile> NamedTableColumn<F, F> buildNameColumn(Function<F, GenericFontIcon<?>>
+            iconProvider) {
         var nameColumn = new NamedTableColumn<F, F>(FileColumns.NAME, "Name");
         nameColumn.setCellValueFactory(data -> new ReadOnlyObjectWrapper(data.getValue()));
         var converter = new FileStringConverter<F>();
@@ -87,11 +88,11 @@ protected void updateDisplay() {
                 }
             }
 
-            private Node buildIcon(GenericFile file) {
+            private Node buildIcon(F file) {
                 if (file.getEntryType() == null) {
                     return null;
                 }
-                return new FontIconView(file.isDirectory() ? dirIcon : fileIcon);
+                return new FontIconView(iconProvider.apply(file));
             }
         });
         nameColumn.setComparator(Comparator.comparing(GenericFile::getName, String.CASE_INSENSITIVE_ORDER));