1 files changed, 11 insertions, 0 deletions
diff --git a/Hotline/Models/FileInfo.swift b/Hotline/Models/FileInfo.swift
index 9caa976..0cd9b2b 100644
--- a/Hotline/Models/FileInfo.swift
+++ b/Hotline/Models/FileInfo.swift
@@ -1,4 +1,5 @@
 import SwiftUI
+import UniformTypeIdentifiers
 
 @Observable class FileInfo: Identifiable, Hashable {
   let id: UUID
@@ -15,6 +16,16 @@ import SwiftUI
   var expanded: Bool = false
   var children: [FileInfo]? = nil
   
+  var isImage: Bool {
+    let fileExtension = (self.name as NSString).pathExtension
+    if let fileType = UTType(filenameExtension: fileExtension) {
+      if fileType.isSubtype(of: .image) {
+        return true
+      }
+    }
+    return false
+  }
+  
   init(hotlineFile: HotlineFile) {
     self.id = UUID()
     self.path = hotlineFile.path