detect browserbookmark file change

Author: taooceros

Plugins/Flow.Launcher.Plugin.BrowserBookmark/ChromiumBookmarkLoader.cs

@@ -20,6 +20,8 @@ protected List<Bookmark> LoadBookmarks(string browserDataPath, string name)
                 var bookmarkPath = Path.Combine(profile, "Bookmarks");
                 if (!File.Exists(bookmarkPath))
                     continue;
+                
+                Main.RegisterBookmarkFile(bookmarkPath);
 
                 var source = name + (Path.GetFileName(profile) == "Default" ? "" : $" ({Path.GetFileName(profile)})");
                 bookmarks.AddRange(LoadBookmarksFromFile(bookmarkPath, source));
@@ -31,6 +33,9 @@ protected List<Bookmark> LoadBookmarksFromFile(string path, string source)
         {
             if (!File.Exists(path))
                 return new();
+
+            Main.RegisterBookmarkFile(path);
+            
             var bookmarks = new List<Bookmark>();
             using var jsonDocument = JsonDocument.Parse(File.ReadAllText(path));
             if (!jsonDocument.RootElement.TryGetProperty("roots", out var rootElement))

Plugins/Flow.Launcher.Plugin.BrowserBookmark/Commands/BookmarkLoader.cs

@@ -3,6 +3,10 @@
 using Flow.Launcher.Infrastructure;
 using Flow.Launcher.Plugin.BrowserBookmark.Models;
 using Flow.Launcher.Plugin.SharedModels;
+using Microsoft.AspNetCore.Authentication;
+using System.IO;
+using System.Threading.Channels;
+using System.Threading.Tasks;
 
 namespace Flow.Launcher.Plugin.BrowserBookmark.Commands
 {
@@ -17,6 +21,7 @@ internal static MatchResult MatchProgram(Bookmark bookmark, string queryString)
             return StringMatcher.FuzzySearch(queryString, bookmark.Url);
         }
 
+
         internal static List<Bookmark> LoadAllBookmarks(Settings setting)
         {
 

Plugins/Flow.Launcher.Plugin.BrowserBookmark/FirefoxBookmarkLoader.cs

@@ -29,6 +29,8 @@ public List<Bookmark> GetBookmarks()
                 return new List<Bookmark>();
 
             var bookmarkList = new List<Bookmark>();
+            
+            Main.RegisterBookmarkFile(PlacesPath);
 
             // create the connection string and init the connection
             string dbPath = string.Format(dbPathFormat, PlacesPath);

Plugins/Flow.Launcher.Plugin.BrowserBookmark/Main.cs

@@ -9,24 +9,29 @@
 using Flow.Launcher.Plugin.BrowserBookmark.Models;
 using Flow.Launcher.Plugin.BrowserBookmark.Views;
 using Flow.Launcher.Plugin.SharedCommands;
+using System.IO;
+using System.Threading.Channels;
+using System.Threading.Tasks;
 
 namespace Flow.Launcher.Plugin.BrowserBookmark
 {
-    public class Main : ISettingProvider, IPlugin, IReloadable, IPluginI18n, IContextMenu
+    public class Main : ISettingProvider, IPlugin, IReloadable, IPluginI18n, IContextMenu, IDisposable
     {
         private PluginInitContext context;
 
         private List<Bookmark> cachedBookmarks = new List<Bookmark>();
 
-        private Settings _settings { get; set;}
+        private Settings _settings { get; set; }
 
         public void Init(PluginInitContext context)
         {
             this.context = context;
-            
+
             _settings = context.API.LoadSettingJsonStorage<Settings>();
 
             cachedBookmarks = BookmarkLoader.LoadAllBookmarks(_settings);
+
+            _ = MonitorRefreshQueue();
         }
 
         public List<Result> Query(Query query)
@@ -52,7 +57,10 @@ public List<Result> Query(Query query)
 
                         return true;
                     },
-                    ContextData = new BookmarkAttributes { Url = c.Url }
+                    ContextData = new BookmarkAttributes
+                    {
+                        Url = c.Url
+                    }
                 }).Where(r => r.Score > 0);
                 return returnList.ToList();
             }
@@ -69,11 +77,64 @@ public List<Result> Query(Query query)
                         context.API.OpenUrl(c.Url);
                         return true;
                     },
-                    ContextData = new BookmarkAttributes { Url = c.Url }
+                    ContextData = new BookmarkAttributes
+                    {
+                        Url = c.Url
+                    }
                 }).ToList();
             }
         }
 
+
+        private static Channel<byte> refreshQueue = Channel.CreateBounded<byte>(1);
+
+        private async Task MonitorRefreshQueue()
+        {
+            var reader = refreshQueue.Reader;
+            while (await reader.WaitToReadAsync())
+            {
+                await Task.Delay(2000);
+                if (reader.TryRead(out _))
+                {
+                    ReloadData();
+                }
+            }
+        }
+
+        private static readonly List<FileSystemWatcher> Watchers = new();
+
+        internal static void RegisterBookmarkFile(string path)
+        {
+            var directory = Path.GetDirectoryName(path);
+            if (!Directory.Exists(directory))
+                return;
+            var watcher = new FileSystemWatcher(directory!);
+            if (File.Exists(path))
+            {
+                var fileName = Path.GetFileName(path);
+                watcher.Filter = fileName;
+            }
+
+            watcher.NotifyFilter = NotifyFilters.FileName |
+                                   NotifyFilters.LastAccess |
+                                   NotifyFilters.LastWrite |
+                                   NotifyFilters.Size;
+
+            watcher.Changed += static (_, _) =>
+            {
+                refreshQueue.Writer.TryWrite(default);
+            };
+
+            watcher.Renamed += static (_, _) =>
+            {
+                refreshQueue.Writer.TryWrite(default);
+            };
+
+            watcher.EnableRaisingEvents = true;
+
+            Watchers.Add(watcher);
+        }
+
         public void ReloadData()
         {
             cachedBookmarks.Clear();
@@ -98,7 +159,8 @@ public Control CreateSettingPanel()
 
         public List<Result> LoadContextMenus(Result selectedResult)
         {
-            return new List<Result>() {
+            return new List<Result>()
+            {
                 new Result
                 {
                     Title = context.API.GetTranslation("flowlauncher_plugin_browserbookmark_copyurl_title"),
@@ -114,20 +176,28 @@ public List<Result> LoadContextMenus(Result selectedResult)
                         catch (Exception e)
                         {
                             var message = "Failed to set url in clipboard";
-                            Log.Exception("Main",message, e, "LoadContextMenus");
+                            Log.Exception("Main", message, e, "LoadContextMenus");
 
                             context.API.ShowMsg(message);
 
                             return false;
                         }
                     },
                     IcoPath = "Images\\copylink.png"
-                }};
+                }
+            };
         }
 
         internal class BookmarkAttributes
         {
             internal string Url { get; set; }
         }
+        public void Dispose()
+        {
+            foreach (var watcher in Watchers)
+            {
+                watcher.Dispose();
+            }
+        }
     }
-}
+}