Add mimetype based icons

src/main/java/app/service/FiletreeOperations.java

@@ -17,6 +17,8 @@ import javafx.scene.image.ImageView;
 
 public class FiletreeOperations {
 
+  private static int iconSize = 20;
+
   // FIXME: File specific icons not working properly
   // TODO: Clean up code that is not in use
   // TODO: Error check for recursiveness, and files without icons
@@ -33,7 +35,11 @@ public class FiletreeOperations {
     Image folder = new Image(FiletreeOperations.class.getResourceAsStream("/graphics/folder.png"));
 
     if (file.isDirectory()) {
-      CheckBoxTreeItem<String> element = new CheckBoxTreeItem<>(file.getName(), new ImageView(folder));
+      ImageView icon = new ImageView(folder);
+      icon.setFitHeight(iconSize);
+      icon.setFitWidth(iconSize);
+
+      CheckBoxTreeItem<String> element = new CheckBoxTreeItem<>(file.getName(), icon);
       parent.getChildren().add(element);
 
       List<File> dirList = new ArrayList<>();
@@ -48,10 +54,15 @@ public class FiletreeOperations {
     } else {
 
       try {
-        CheckBoxTreeItem<String> element = 
+        ImageView icon = new ImageView(getIconForFile(file));
-          new CheckBoxTreeItem<>(file.getName(), new ImageView(getIconForFile(file)));
+        icon.setFitHeight(iconSize);
+        icon.setFitWidth(iconSize);
+
+        CheckBoxTreeItem<String> element = new CheckBoxTreeItem<>(file.getName(), icon);
         parent.getChildren().add(element);
+
       } catch (Exception e) {
+        System.err.print(e);
         System.err.println("[ERROR]: DEFAULT FILE ICON NOT FOUND");
       }
     }
@@ -76,21 +87,20 @@ public class FiletreeOperations {
     Image icon;
 
     try {
-      String mimeType = Files.probeContentType(file.toPath()).replace('/', '-');
+      String mimeType = Files.probeContentType(file.toPath());
-
-      String iconPath = "/graphics/filetreeicons/" + mimeType + ".png";
       if (mimeType == null) throw new IOException();
 
+      String iconPath = "/graphics/filetreeicons/" + mimeType.replace('/', '-') + ".png";
       InputStream imageData = FiletreeOperations.class.getResourceAsStream(iconPath);
       if (imageData == null) throw new IOException();
 
       icon = new Image(imageData);
 
     } catch (IOException e) {
-      System.err.println("[ERROR]: ICON NOT FOUND: " + file.getPath());
+      System.err.println("[WARNING]: ICON NOT FOUND: " + file.getPath());
 
       // String iconPath = "/graphics/filetreeicons/file.png";
-      String iconPath = "/graphics/placeholder.png";
+      String iconPath = "/graphics/filetreeicons/unknown.png";
 
       InputStream imageData = FileOperations.class.getResourceAsStream(iconPath);
       if (imageData == null) throw new IOException();