Working on storage.

1 files changed, 17 insertions, 10 deletions

diff --git a/Software/Visual_Studio/Tango.FileSystem/FileSystemManager.cs b/Software/Visual_Studio/Tango.FileSystem/FileSystemManager.cs
index c18df97c9..f72785f81 100644
--- a/Software/Visual_Studio/Tango.FileSystem/FileSystemManager.cs
+++ b/Software/Visual_Studio/Tango.FileSystem/FileSystemManager.cs
@@ -73,15 +73,18 @@ namespace Tango.FileSystem
                 });
             }
 
-            foreach (var file in Directory.GetFiles(request.Path))
+            if (!request.FoldersOnly)
             {
-                items.Add(new FileSystemItemDTO()
+                foreach (var file in Directory.GetFiles(request.Path, request.Filter != null ? request.Filter : "*.*"))
                 {
-                    Path = file,
-                    Type = FileSystemItemType.File,
-                    DateModified = File.GetLastWriteTimeUtc(file),
-                    Size = new FileInfo(file).Length
-                });
+                    items.Add(new FileSystemItemDTO()
+                    {
+                        Path = file,
+                        Type = FileSystemItemType.File,
+                        DateModified = File.GetLastWriteTimeUtc(file),
+                        Size = new FileInfo(file).Length
+                    });
+                }
             }
 
             return new FileSystemItemDTO()
@@ -92,24 +95,28 @@ namespace Tango.FileSystem
             };
         }
 
-        public Task<FileSystemItem> GetFolder(String path)
+        public Task<FileSystemItem> GetFolder(String path, bool foldersOnly = false, String filter = "*.*")
         {
             return Task.Factory.StartNew<FileSystemItem>(() =>
             {
                 return FileSystemItem.FromDTO(GetFolder(new GetFileSystemItemRequest()
                 {
                     Path = path,
+                    FoldersOnly = foldersOnly,
+                    Filter = filter,
                 }));
             });
         }
 
-        public Task<FileSystemItem> GetFolder(Environment.SpecialFolder specialFolder)
+        public Task<FileSystemItem> GetFolder(Environment.SpecialFolder specialFolder, bool foldersOnly = false, String filter = "*.*")
         {
             return Task.Factory.StartNew<FileSystemItem>(() =>
             {
                 return FileSystemItem.FromDTO(GetFolder(new GetFileSystemItemRequest()
                 {
-                    SpecialFolder = specialFolder
+                    SpecialFolder = specialFolder,
+                    FoldersOnly = foldersOnly,
+                    Filter = filter,
                 }));
             });
         }