From 80bf87d71df7701e6b2e12b5d468d08e93662da5 Mon Sep 17 00:00:00 2001
From: Thorsten Sommer <mail@tsommer.org>
Date: Tue, 28 Oct 2025 10:08:54 +0100
Subject: [PATCH] Add video file filtering and validation methods

---
 .../Components/AttachDocuments.razor.cs       | 45 +++++++++++++------
 .../Components/ReadFileContent.razor.cs       |  6 +++
 .../Tools/Rust/FileTypeFilter.cs              |  4 +-
 3 files changed, 41 insertions(+), 14 deletions(-)

diff --git a/app/MindWork AI Studio/Components/AttachDocuments.razor.cs b/app/MindWork AI Studio/Components/AttachDocuments.razor.cs
index 576c9b1b..8fc7370c 100644
--- a/app/MindWork AI Studio/Components/AttachDocuments.razor.cs	
+++ b/app/MindWork AI Studio/Components/AttachDocuments.razor.cs	
@@ -55,9 +55,14 @@ public partial class AttachDocuments : MSGComponentBase
                     return;
                 }
                 
-                #warning Filter unsupported files
                 foreach (var path in paths)
+                {
+                    if(!await this.IsFileExtensionValid(path))
+                        continue;
+                    
                     this.DocumentPaths.Add(path);
+                }
+
                 await this.DocumentPathsChanged.InvokeAsync(this.DocumentPaths);
                 await this.OnChange(this.DocumentPaths);
                 this.StateHasChanged();
@@ -82,24 +87,38 @@ public partial class AttachDocuments : MSGComponentBase
         if (!File.Exists(selectedFile.SelectedFilePath))
             return;
 
-        var ext = Path.GetExtension(selectedFile.SelectedFilePath).TrimStart('.');
-        if (Array.Exists(FileTypeFilter.Executables.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
-        {
-            await MessageBus.INSTANCE.SendError(new(Icons.Material.Filled.AppBlocking, T("Executables are not allowed")));
+        if (!await this.IsFileExtensionValid(selectedFile.SelectedFilePath))
             return;
-        }
-
-        if (Array.Exists(FileTypeFilter.AllImages.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
-        {
-            await MessageBus.INSTANCE.SendWarning(new(Icons.Material.Filled.ImageNotSupported, T("Images are not supported yet")));
-            return;
-        }
 
         this.DocumentPaths.Add(selectedFile.SelectedFilePath);
         await this.DocumentPathsChanged.InvokeAsync(this.DocumentPaths);
         await this.OnChange(this.DocumentPaths);
     }
-    
+
+    private async Task<bool> IsFileExtensionValid(string selectedFile)
+    {
+        var ext = Path.GetExtension(selectedFile).TrimStart('.');
+        if (Array.Exists(FileTypeFilter.Executables.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        {
+            await MessageBus.INSTANCE.SendError(new(Icons.Material.Filled.AppBlocking, this.T("Executables are not allowed")));
+            return false;
+        }
+
+        if (Array.Exists(FileTypeFilter.AllImages.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        {
+            await MessageBus.INSTANCE.SendWarning(new(Icons.Material.Filled.ImageNotSupported, this.T("Images are not supported yet")));
+            return false;
+        }
+        
+        if (Array.Exists(FileTypeFilter.AllVideos.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        {
+            await MessageBus.INSTANCE.SendWarning(new(Icons.Material.Filled.FeaturedVideo, this.T("Videos are not supported yet")));
+            return false;
+        }
+
+        return true;
+    }
+
     private async Task ClearAllFiles()
     {
         this.DocumentPaths.Clear();
diff --git a/app/MindWork AI Studio/Components/ReadFileContent.razor.cs b/app/MindWork AI Studio/Components/ReadFileContent.razor.cs
index 86bafebe..17814e16 100644
--- a/app/MindWork AI Studio/Components/ReadFileContent.razor.cs	
+++ b/app/MindWork AI Studio/Components/ReadFileContent.razor.cs	
@@ -38,6 +38,12 @@ public partial class ReadFileContent : MSGComponentBase
             return;
         }
         
+        if (Array.Exists(FileTypeFilter.AllVideos.FilterExtensions, x => x.Equals(ext, StringComparison.OrdinalIgnoreCase)))
+        {
+            await MessageBus.INSTANCE.SendWarning(new(Icons.Material.Filled.FeaturedVideo, this.T("Videos are not supported yet")));
+            return;
+        }
+        
         var fileContent = await this.RustService.ReadArbitraryFileData(selectedFile.SelectedFilePath, int.MaxValue);
         await this.FileContentChanged.InvokeAsync(fileContent);
     }
diff --git a/app/MindWork AI Studio/Tools/Rust/FileTypeFilter.cs b/app/MindWork AI Studio/Tools/Rust/FileTypeFilter.cs
index bf0ac5b2..d755184e 100644
--- a/app/MindWork AI Studio/Tools/Rust/FileTypeFilter.cs	
+++ b/app/MindWork AI Studio/Tools/Rust/FileTypeFilter.cs	
@@ -19,7 +19,9 @@ public readonly record struct FileTypeFilter(string FilterName, string[] FilterE
     
     public static FileTypeFilter AllOffice => new(TB("All Office Files"), ["docx", "xlsx", "pptx", "doc", "xls", "ppt", "pdf"]);
     
-    public static FileTypeFilter AllImages => new(TB("All Image Files"), ["jpg", "jpeg", "png", "gif", "bmp", "tiff"]);
+    public static FileTypeFilter AllImages => new(TB("All Image Files"), ["jpg", "jpeg", "png", "gif", "bmp", "tiff", "svg", "webp", "heic"]);
+    
+    public static FileTypeFilter AllVideos => new(TB("All Video Files"), ["mp4", "avi", "mkv", "mov", "wmv", "flv", "webm"]);
     
     public static FileTypeFilter Executables => new(TB("Executable Files"), ["exe", "app", "bin", "appimage"]);
 }
\ No newline at end of file