Add local favicon load

Author: onesounds

Flow.Launcher.Core/Flow.Launcher.Core.csproj

@@ -57,6 +57,7 @@
     <PackageReference Include="FSharp.Core" Version="9.0.201" />
     <PackageReference Include="Meziantou.Framework.Win32.Jobs" Version="3.4.0" />
     <PackageReference Include="Microsoft.IO.RecyclableMemoryStream" Version="3.0.1" />
+    <PackageReference Include="SkiaSharp" Version="3.116.1" />
     <PackageReference Include="squirrel.windows" Version="1.5.2" NoWarn="NU1701" />
     <PackageReference Include="StreamJsonRpc" Version="2.21.10" />
   </ItemGroup>

Flow.Launcher.Infrastructure/Flow.Launcher.Infrastructure.csproj

@@ -67,6 +67,7 @@
     </PackageReference>
     <PackageReference Include="NLog" Version="4.7.10" />
     <PackageReference Include="PropertyChanged.Fody" Version="3.4.0" />
+    <PackageReference Include="SkiaSharp" Version="3.116.1" />
     <PackageReference Include="System.Drawing.Common" Version="7.0.0" />
     <!--ToolGood.Words.Pinyin v3.0.2.6 results in high memory usage when search with pinyin is enabled-->
     <!--Bumping to it or higher needs to test and ensure this is no longer a problem-->

Flow.Launcher.Plugin/Flow.Launcher.Plugin.csproj

@@ -77,6 +77,7 @@
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
     <PackageReference Include="PropertyChanged.Fody" Version="3.4.0" />
+    <PackageReference Include="SkiaSharp" Version="3.116.1" />
   </ItemGroup>
 
 </Project>

Flow.Launcher.Test/Flow.Launcher.Test.csproj

@@ -55,6 +55,7 @@
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.12.0" />
+    <PackageReference Include="SkiaSharp" Version="3.116.1" />
   </ItemGroup>
 
 </Project>

Flow.Launcher/Flow.Launcher.csproj

@@ -104,6 +104,7 @@
     <PackageReference Include="NHotkey.Wpf" Version="3.0.0" />
     <PackageReference Include="PropertyChanged.Fody" Version="3.4.0" />
     <PackageReference Include="SemanticVersioning" Version="3.0.0" />
+    <PackageReference Include="SkiaSharp" Version="3.116.1" />
     <PackageReference Include="TaskScheduler" Version="2.12.1" />
     <PackageReference Include="VirtualizingWrapPanel" Version="2.1.1" />
   </ItemGroup>

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

@@ -3,11 +3,24 @@
 using System.IO;
 using System.Text.Json;
 using Flow.Launcher.Infrastructure.Logger;
+using System;
+using System.Data.SQLite;
+using SkiaSharp;
 
 namespace Flow.Launcher.Plugin.BrowserBookmark;
 
 public abstract class ChromiumBookmarkLoader : IBookmarkLoader
 {
+    private readonly string _faviconCacheDir;
+
+    protected ChromiumBookmarkLoader()
+    {
+        _faviconCacheDir = Path.Combine(
+            Path.GetDirectoryName(typeof(ChromiumBookmarkLoader).Assembly.Location),
+            "FaviconCache");
+        Directory.CreateDirectory(_faviconCacheDir);
+    }
+
     public abstract List<Bookmark> GetBookmarks();
 
     protected List<Bookmark> LoadBookmarks(string browserDataPath, string name)
@@ -22,10 +35,30 @@ protected List<Bookmark> LoadBookmarks(string browserDataPath, string name)
             if (!File.Exists(bookmarkPath))
                 continue;
 
-            Main.RegisterBookmarkFile(bookmarkPath);
+            // Register bookmark file monitoring (direct call to Main.RegisterBookmarkFile)
+            try
+            {
+                if (File.Exists(bookmarkPath))
+                {
+                    //Main.RegisterBookmarkFile(bookmarkPath);
+                }
+            }
+            catch (Exception ex)
+            {
+                Log.Exception($"Failed to register bookmark file monitoring: {bookmarkPath}", ex);
+            }
 
             var source = name + (Path.GetFileName(profile) == "Default" ? "" : $" ({Path.GetFileName(profile)})");
-            bookmarks.AddRange(LoadBookmarksFromFile(bookmarkPath, source));
+            var profileBookmarks = LoadBookmarksFromFile(bookmarkPath, source);
+
+            // Load favicons after loading bookmarks
+            var faviconDbPath = Path.Combine(profile, "Favicons");
+            if (File.Exists(faviconDbPath))
+            {
+                LoadFaviconsFromDb(faviconDbPath, profileBookmarks);
+            }
+
+            bookmarks.AddRange(profileBookmarks);
         }
 
         return bookmarks;
@@ -52,8 +85,7 @@ private void EnumerateRoot(JsonElement rootElement, ICollection<Bookmark> bookma
             if (folder.Value.ValueKind != JsonValueKind.Object)
                 continue;
 
-            // Fix for Opera. It stores bookmarks slightly different than chrome. See PR and bug report for this change for details.
-            // If various exceptions start to build up here consider splitting this Loader into multiple separate ones.
+            // Fix for Opera. It stores bookmarks slightly different than chrome.
             if (folder.Name == "custom_root")
                 EnumerateRoot(folder.Value, bookmarks, source);
             else
@@ -91,4 +123,108 @@ private void EnumerateFolderBookmark(JsonElement folderElement, ICollection<Book
             }
         }
     }
+
+    private void LoadFaviconsFromDb(string dbPath, List<Bookmark> bookmarks)
+    {
+        try
+        {
+            // Use a copy to avoid lock issues with the original file
+            var tempDbPath = Path.Combine(_faviconCacheDir, $"tempfavicons_{Guid.NewGuid()}.db");
+
+            try
+            {
+                File.Copy(dbPath, tempDbPath, true);
+            }
+            catch (Exception ex)
+            {
+                Log.Exception($"Failed to copy favicon DB: {dbPath}", ex);
+                return;
+            }
+
+            try
+            {
+                using var connection = new SQLiteConnection($"Data Source={tempDbPath};Version=3;Read Only=True;");
+                connection.Open();
+
+                foreach (var bookmark in bookmarks)
+                {
+                    try
+                    {
+                        var url = bookmark.Url;
+                        if (string.IsNullOrEmpty(url)) continue;
+
+                        // Extract domain from URL
+                        if (!Uri.TryCreate(url, UriKind.Absolute, out Uri uri))
+                            continue;
+
+                        var domain = uri.Host;
+
+                        using var cmd = connection.CreateCommand();
+                        cmd.CommandText = @"
+                            SELECT f.id, b.image_data
+                            FROM favicons f
+                            JOIN favicon_bitmaps b ON f.id = b.icon_id
+                            JOIN icon_mapping m ON f.id = m.icon_id
+                            WHERE m.page_url LIKE @url
+                            ORDER BY b.width DESC
+                            LIMIT 1";
+
+                        cmd.Parameters.AddWithValue("@url", $"%{domain}%");
+
+                        using var reader = cmd.ExecuteReader();
+                        if (reader.Read() && !reader.IsDBNull(1))
+                        {
+                            var iconId = reader.GetInt64(0).ToString();
+                            var imageData = (byte[])reader["image_data"];
+
+                            if (imageData != null && imageData.Length > 0)
+                            {
+                                var faviconPath = Path.Combine(_faviconCacheDir, $"{domain}_{iconId}.png");
+                                if (!File.Exists(faviconPath))
+                                {
+                                    SaveBitmapData(imageData, faviconPath);
+                                }
+                                bookmark.FaviconPath = faviconPath;
+                            }
+                        }
+                    }
+                    catch (Exception ex)
+                    {
+                        Log.Exception($"Failed to extract bookmark favicon: {bookmark.Url}", ex);
+                    }
+                }
+            }
+            catch (Exception ex)
+            {
+                Log.Exception($"Failed to connect to SQLite: {tempDbPath}", ex);
+            }
+
+            // Delete temporary file
+            try { File.Delete(tempDbPath); } catch { /* Ignore */ }
+        }
+        catch (Exception ex)
+        {
+            Log.Exception($"Failed to load favicon DB: {dbPath}", ex);
+        }
+    }
+
+    private void SaveBitmapData(byte[] imageData, string outputPath)
+    {
+        try
+        {
+            using var ms = new MemoryStream(imageData);
+            using var bitmap = SKBitmap.Decode(ms);
+            if (bitmap != null)
+            {
+                using var image = SKImage.FromBitmap(bitmap);
+                using var data = image.Encode(SKEncodedImageFormat.Png, 100);
+                using var fs = File.OpenWrite(outputPath);
+                data.SaveTo(fs);
+            }
+        }
+        catch (Exception ex)
+        {
+            Log.Exception($"Failed to save image: {outputPath}", ex);
+        }
+    }
 }