add search for Firefox bookmarks

2026-04-09 04:37:30 +02:00 · 2014-07-02 18:01:31 +03:00
parent e275ce6063
commit 6807dcbd5d
8 changed files with 250 additions and 111 deletions
--- a/Plugins/Wox.Plugin.BrowserBookmark/FirefoxBookmarks.cs
+++ b/Plugins/Wox.Plugin.BrowserBookmark/FirefoxBookmarks.cs
@@ -0,0 +1,89 @@
+using System;
+using System.Collections.Generic;
+using System.Data.SQLite;
+using System.IO;
+using System.Linq;
+
+namespace Wox.Plugin.BrowserBookmark
+{
+    public class FirefoxBookmarks
+    {
+        private const string queryBookmarks = @"SELECT url, title
+              FROM moz_places
+              WHERE id in (
+                SELECT bm.fk FROM moz_bookmarks bm WHERE bm.fk NOT NULL
+              )
+              AND ( url LIKE '%{0}%' OR title LIKE '%{0}%' )
+              ORDER BY visit_count DESC
+              LIMIT 20
+            ";
+
+        private const string queryTopBookmarks = @"SELECT url, title
+              FROM moz_places
+              WHERE id in (
+                SELECT bm.fk FROM moz_bookmarks bm WHERE bm.fk NOT NULL
+              )
+              ORDER BY visit_count DESC
+              LIMIT 20
+            ";
+
+        private const string dbPathFormat = "Data Source ={0};Version=3;New=False;Compress=True;";
+
+        public List<Bookmark> GetBookmarks(string search = null, bool top = false)
+        {
+            // Create the query command for the given case
+            string query = top ? queryTopBookmarks : string.Format(queryBookmarks, search);
+
+            return GetResults(query);
+        }
+
+        /// <summary>
+        /// Searches the places.sqlite db based on the given query and returns the results
+        /// </summary>
+        private List<Bookmark> GetResults(string query)
+        {
+            // create the connection string and init the connection
+            string dbPath = string.Format(dbPathFormat, PlacesPath);
+            var dbConnection = new SQLiteConnection(dbPath);
+
+            // Open connection to the database file and execute the query
+            dbConnection.Open();
+            var reader = new SQLiteCommand(query, dbConnection).ExecuteReader();
+
+            // return results in List<Bookmark> format
+            return reader.Select(x => new Bookmark()
+            {
+                Name = (x["title"] is DBNull) ? string.Empty : x["title"].ToString(),
+                Url = x["url"].ToString()
+            }).ToList();
+        }
+
+        /// <summary>
+        /// Path to places.sqlite
+        /// </summary>
+        private string PlacesPath
+        {
+            get
+            {
+                var profilesPath = Environment.ExpandEnvironmentVariables(@"%appdata%\Mozilla\Firefox\Profiles\");
+                var folders = new DirectoryInfo(profilesPath).GetDirectories().Select(x => x.FullName).ToList();
+
+                // Look for the default profile folder
+                return string.Format(@"{0}\places.sqlite",
+                                     folders.FirstOrDefault(d => File.Exists(d + @"\places.sqlite") && d.EndsWith(".default"))
+                                     ?? folders.First(d => File.Exists(d + @"\places.sqlite")));
+            }
+        }
+    }
+
+    public static class Extensions
+    {
+        public static IEnumerable<T> Select<T>(this SQLiteDataReader reader, Func<SQLiteDataReader, T> projection)
+        {
+            while (reader.Read())
+            {
+                yield return projection(reader);
+            }
+        }
+    }
+}