Fix BrowserBookmark '100% CPU' issue

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

@@ -2,6 +2,7 @@
 using System.Collections.Concurrent;
 using System.Collections.Generic;
 using System.IO;
+using System.IO.Compression;
 using System.Linq;
 using System.Threading.Tasks;
 using Flow.Launcher.Plugin.BrowserBookmark.Helper;
@@ -53,7 +54,7 @@
             return bookmarks;
         }
 
         var tempDbPath = Path.Combine(_faviconCacheDir, $"tempplaces_{Guid.NewGuid()}.sqlite");
 
         try
         {
@@ -134,10 +135,6 @@
 
                 try
                 {
-                    if (string.IsNullOrEmpty(bookmark.Url))
-                        return;
-
-                    // Extract domain from URL
                     if (!Uri.TryCreate(bookmark.Url, UriKind.Absolute, out Uri uri))
                         return;
 
@@ -146,43 +143,49 @@
                     // Query for latest Firefox version favicon structure
                     using var cmd = connection.CreateCommand();
                     cmd.CommandText = @"
-                        SELECT i.data
+                        SELECT i.id, i.data
                         FROM moz_icons i
                         JOIN moz_icons_to_pages ip ON i.id = ip.icon_id
                         JOIN moz_pages_w_icons p ON ip.page_id = p.id
-                        WHERE p.page_url LIKE @url
-                        AND i.data IS NOT NULL
-                        ORDER BY i.width DESC  -- Select largest icon available
+                        WHERE p.page_url LIKE @domain
+                        ORDER BY i.width DESC
                         LIMIT 1";
 
-                    cmd.Parameters.AddWithValue("@url", $"%{domain}%");
+                    cmd.Parameters.AddWithValue("@domain", $"%{domain}%");
 
                     using var reader = cmd.ExecuteReader();
-                    if (!reader.Read() || reader.IsDBNull(0))
+                    if (!reader.Read() || reader.IsDBNull(1))
                         return;
 
+                    var iconId = reader.GetInt64(0).ToString();
                     var imageData = (byte[])reader["data"];
 
                     if (imageData is not { Length: > 0 })
                         return;
 
-                    string faviconPath;
-                    if (FaviconHelper.IsSvgData(imageData))
-                    {
-                        faviconPath = Path.Combine(_faviconCacheDir, $"firefox_{domain}.svg");
-                    }
-                    else
+                    // Check if the image data is compressed (GZip)
+                    if (imageData.Length > 2 && imageData[0] == 0x1f && imageData[1] == 0x8b)
                     {
-                        faviconPath = Path.Combine(_faviconCacheDir, $"firefox_{domain}.png");
+                        using var inputStream = new MemoryStream(imageData);
+                        using var gZipStream = new GZipStream(inputStream, CompressionMode.Decompress);
+                        using var outputStream = new MemoryStream();
+                        gZipStream.CopyTo(outputStream);
+                        imageData = outputStream.ToArray();
                     }
 
-                    // Filter out duplicate favicons
-                    if (savedPaths.TryAdd(faviconPath, true))
+                    // Convert the image data to WebP format
+                    var webpData = FaviconHelper.TryConvertToWebp(imageData);
+                    if (webpData != null)
                     {
-                        FaviconHelper.SaveBitmapData(imageData, faviconPath);
-                    }
+                        var faviconPath = Path.Combine(_faviconCacheDir, $"firefox_{domain}_{iconId}.webp");
+
+                        if (savedPaths.TryAdd(faviconPath, true))
+                        {
+                            FaviconHelper.SaveBitmapData(webpData, faviconPath);
+                        }
 
-                    bookmark.FaviconPath = faviconPath;
+                        bookmark.FaviconPath = faviconPath;
+                    }
                 }
                 catch (Exception ex)
                 {

Plugins/Flow.Launcher.Plugin.BrowserBookmark/Flow.Launcher.Plugin.BrowserBookmark.csproj

@@ -97,6 +97,8 @@
     <PackageReference Include="CommunityToolkit.Mvvm" Version="8.4.0" />
     <PackageReference Include="Flow.Launcher.Localization" Version="0.0.4" />
     <PackageReference Include="Microsoft.Data.Sqlite" Version="9.0.7" />
+    <PackageReference Include="Svg.Skia" Version="3.0.3" />
+    <PackageReference Include="SkiaSharp" Version="3.119.0" />
   </ItemGroup>
 
 </Project>

Plugins/Flow.Launcher.Plugin.BrowserBookmark/Helper/FaviconHelper.cs

@@ -1,5 +1,7 @@
 using System;
 using System.IO;
+using SkiaSharp;
+using Svg.Skia;
 
 namespace Flow.Launcher.Plugin.BrowserBookmark.Helper;
 
@@ -65,12 +67,58 @@
         }
     }
 
-    public static bool IsSvgData(byte[] data)
+    public static byte[] TryConvertToWebp(byte[] data)
     {
-        if (data.Length < 5)
-            return false;
-        string start = System.Text.Encoding.ASCII.GetString(data, 0, Math.Min(100, data.Length));
-        return start.Contains("<svg") ||
-               (start.StartsWith("<?xml") && start.Contains("<svg"));
+        if (data == null || data.Length == 0)
+            return null;
+
+        SKBitmap bitmap = null;
+
+        try
+        {
+            using (var ms = new MemoryStream(data))
+            {
+                var svg = new SKSvg();
+                if (svg.Load(ms) != null && svg.Picture != null)
+                {
+                    bitmap = new SKBitmap((int)svg.Picture.CullRect.Width, (int)svg.Picture.CullRect.Height);
+                    using (var canvas = new SKCanvas(bitmap))
+                    {
+                        canvas.Clear(SKColors.Transparent);
+                        canvas.DrawPicture(svg.Picture);
+                        canvas.Flush();
+                    }
+                }
+            }
+        }
+        catch { /* Not an SVG */ }
+
+        if (bitmap == null)
+        {
+            try
+            {
+                bitmap = SKBitmap.Decode(data);
+            }
+            catch { /* Not a decodable bitmap */ }
+        }
+
+        if (bitmap != null)
+        {
+            try
+            {
+                using (var image = SKImage.FromBitmap(bitmap))
+                using (var webp = image.Encode(SKEncodedImageFormat.Webp, 65))
+                {
+                    if (webp != null)
+                        return webp.ToArray();
+                }
+            }
+            finally
+            {
+                bitmap.Dispose();
+            }
+        }
+
+        return null;
     }
 }