From f05767afe861afb681e87a2b9cabce87f35070b9 Mon Sep 17 00:00:00 2001
From: Jeremy Wu <jeremy24wu@gmail.com>
Date: Mon, 9 Dec 2019 08:27:50 +1100
Subject: [PATCH] Add exceptions for unauth and arg not found when searching

---
 Plugins/Wox.Plugin.Folder/Main.cs | 36 +++++++++++++++++++++----------
 1 file changed, 25 insertions(+), 11 deletions(-)

diff --git a/Plugins/Wox.Plugin.Folder/Main.cs b/Plugins/Wox.Plugin.Folder/Main.cs
index 0e81623785..487974ceb4 100644
--- a/Plugins/Wox.Plugin.Folder/Main.cs
+++ b/Plugins/Wox.Plugin.Folder/Main.cs
@@ -1,4 +1,4 @@
-using System;
+using System;
 using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO;
@@ -157,18 +157,32 @@ namespace Wox.Plugin.Folder
                 incompleteName = incompleteName.Substring(1);
             }
 
-            // search folder and add results
-            var fileSystemInfos = directoryInfo.GetFileSystemInfos(incompleteName, searchOption);
-
-            foreach (var fileSystemInfo in fileSystemInfos)
+            try
             {
-                if ((fileSystemInfo.Attributes & FileAttributes.Hidden) == FileAttributes.Hidden) continue;
+                // search folder and add results
+                var fileSystemInfos = directoryInfo.GetFileSystemInfos(incompleteName, searchOption);
 
-                var result =
-                    fileSystemInfo is DirectoryInfo
-                        ? CreateFolderResult(fileSystemInfo.Name, fileSystemInfo.FullName, query.ActionKeyword)
-                        : CreateFileResult(fileSystemInfo.FullName);
-                results.Add(result);
+                foreach (var fileSystemInfo in fileSystemInfos)
+                {
+                    if ((fileSystemInfo.Attributes & FileAttributes.Hidden) == FileAttributes.Hidden) continue;
+
+                    var result =
+                        fileSystemInfo is DirectoryInfo
+                            ? CreateFolderResult(fileSystemInfo.Name, fileSystemInfo.FullName, query.ActionKeyword)
+                            : CreateFileResult(fileSystemInfo.FullName);
+                    results.Add(result);
+                }
+            }
+            catch(Exception e)
+            {
+                if (e is UnauthorizedAccessException || e is ArgumentException)
+                {
+                    results.Add(new Result { Title = e.Message, Score = 501 });
+
+                    return results;
+                }
+
+                throw;
             }
 
             return results;