1 files changed, 12 insertions, 0 deletions

diff --git a/viewer/src/tools.ts b/viewer/src/tools.ts
index 57a889d..94fdf33 100644
--- a/viewer/src/tools.ts
+++ b/viewer/src/tools.ts
@@ -43,5 +43,17 @@ export default class Tools {
     ];
   }
 
+  public static getIcon(item: Gallery.Item): string {
+    if (item.path.length <= 1) return "home";
+    switch (item.properties.type) {
+      case "picture":
+        return "image";
+      case "directory":
+        return "folder";
+      case "other":
+      default:
+        return "question-circle";
+    }
+  }
 
 }
\ No newline at end of file