2014-01-08 23:21:37 +08:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Diagnostics;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Runtime.InteropServices;
|
|
|
|
|
using System.Text;
|
|
|
|
|
using System.Text.RegularExpressions;
|
|
|
|
|
using System.Windows.Forms;
|
|
|
|
|
using Newtonsoft.Json;
|
2014-02-06 22:22:02 +08:00
|
|
|
|
using Wox.Infrastructure;
|
2014-01-08 23:21:37 +08:00
|
|
|
|
|
2014-01-29 18:33:24 +08:00
|
|
|
|
namespace Wox.Plugin.System
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-01-15 22:45:02 +08:00
|
|
|
|
public class BrowserBookmarks : BaseSystemPlugin
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-21 03:53:18 +08:00
|
|
|
|
private PluginInitContext context;
|
2014-01-08 23:21:37 +08:00
|
|
|
|
private List<Bookmark> bookmarks = new List<Bookmark>();
|
|
|
|
|
|
2014-01-15 22:45:02 +08:00
|
|
|
|
protected override List<Result> QueryInternal(Query query)
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
|
|
|
|
if (string.IsNullOrEmpty(query.RawQuery) || query.RawQuery.EndsWith(" ") || query.RawQuery.Length <= 1) return new List<Result>();
|
|
|
|
|
|
2014-03-17 03:37:52 +08:00
|
|
|
|
var fuzzyMather = FuzzyMatcher.Create(query.RawQuery);
|
2014-03-25 13:36:33 +08:00
|
|
|
|
List<Bookmark> returnList = bookmarks.Where(o => MatchProgram(o, fuzzyMather)).ToList();
|
|
|
|
|
returnList = returnList.OrderByDescending(o => o.Score).ToList();
|
2014-01-08 23:21:37 +08:00
|
|
|
|
return returnList.Select(c => new Result()
|
|
|
|
|
{
|
|
|
|
|
Title = c.Name,
|
|
|
|
|
SubTitle = "Bookmark: " + c.Url,
|
2014-03-22 16:50:47 +08:00
|
|
|
|
IcoPath = Path.GetDirectoryName(Application.ExecutablePath) + @"\Images\bookmark.png",
|
2014-01-08 23:21:37 +08:00
|
|
|
|
Score = 5,
|
2014-03-21 03:53:18 +08:00
|
|
|
|
Action = (e) =>
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-21 03:53:18 +08:00
|
|
|
|
context.HideApp();
|
|
|
|
|
context.ShellRun(c.Url);
|
2014-02-28 23:21:01 +08:00
|
|
|
|
return true;
|
2014-01-08 23:21:37 +08:00
|
|
|
|
}
|
|
|
|
|
}).ToList();
|
|
|
|
|
}
|
2014-03-25 13:36:33 +08:00
|
|
|
|
private bool MatchProgram(Bookmark bookmark, FuzzyMatcher matcher)
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-25 13:36:33 +08:00
|
|
|
|
if ((bookmark.Score = matcher.Evaluate(bookmark.Name).Score) > 0) return true;
|
|
|
|
|
if ((bookmark.Score = matcher.Evaluate(bookmark.PinyinName).Score) > 0) return true;
|
|
|
|
|
if ((bookmark.Score = matcher.Evaluate(bookmark.Url).Score / 10) > 0) return true;
|
|
|
|
|
|
2014-01-08 23:21:37 +08:00
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
2014-01-15 22:45:02 +08:00
|
|
|
|
protected override void InitInternal(PluginInitContext context)
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-17 03:37:52 +08:00
|
|
|
|
bookmarks.Clear();
|
2014-01-08 23:21:37 +08:00
|
|
|
|
LoadChromeBookmarks();
|
2014-03-17 12:32:19 +08:00
|
|
|
|
|
2014-03-17 03:37:52 +08:00
|
|
|
|
bookmarks = bookmarks.Distinct().ToList();
|
2014-03-21 03:53:18 +08:00
|
|
|
|
this.context = context;
|
2014-01-08 23:21:37 +08:00
|
|
|
|
}
|
|
|
|
|
|
2014-03-17 12:32:19 +08:00
|
|
|
|
private void ParseChromeBookmarks(String path, string source)
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-17 12:32:19 +08:00
|
|
|
|
if (!File.Exists(path)) return;
|
|
|
|
|
|
|
|
|
|
string all = File.ReadAllText(path);
|
|
|
|
|
Regex nameRegex = new Regex("\"name\": \"(?<name>.*?)\"");
|
|
|
|
|
MatchCollection nameCollection = nameRegex.Matches(all);
|
|
|
|
|
Regex typeRegex = new Regex("\"type\": \"(?<type>.*?)\"");
|
|
|
|
|
MatchCollection typeCollection = typeRegex.Matches(all);
|
|
|
|
|
Regex urlRegex = new Regex("\"url\": \"(?<url>.*?)\"");
|
|
|
|
|
MatchCollection urlCollection = urlRegex.Matches(all);
|
2014-01-08 23:21:37 +08:00
|
|
|
|
|
2014-03-17 12:32:19 +08:00
|
|
|
|
List<string> names = (from Match match in nameCollection select match.Groups["name"].Value).ToList();
|
|
|
|
|
List<string> types = (from Match match in typeCollection select match.Groups["type"].Value).ToList();
|
|
|
|
|
List<string> urls = (from Match match in urlCollection select match.Groups["url"].Value).ToList();
|
|
|
|
|
|
|
|
|
|
int urlIndex = 0;
|
|
|
|
|
for (int i = 0; i < names.Count; i++)
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-17 12:32:19 +08:00
|
|
|
|
string name = DecodeUnicode(names[i]);
|
|
|
|
|
string type = types[i];
|
|
|
|
|
if (type == "url")
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-17 12:32:19 +08:00
|
|
|
|
string url = urls[urlIndex];
|
|
|
|
|
urlIndex++;
|
|
|
|
|
|
|
|
|
|
if (url == null) continue;
|
|
|
|
|
if (url.StartsWith("javascript:", StringComparison.OrdinalIgnoreCase)) continue;
|
|
|
|
|
if (url.StartsWith("vbscript:", StringComparison.OrdinalIgnoreCase)) continue;
|
|
|
|
|
|
|
|
|
|
bookmarks.Add(new Bookmark()
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-17 12:32:19 +08:00
|
|
|
|
Name = name,
|
|
|
|
|
Url = url,
|
|
|
|
|
Source = source
|
|
|
|
|
});
|
2014-01-08 23:21:37 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-03-17 12:32:19 +08:00
|
|
|
|
private void LoadChromeBookmarks(string path, string name)
|
|
|
|
|
{
|
|
|
|
|
if (!Directory.Exists(path)) return;
|
|
|
|
|
var paths = Directory.GetDirectories(path);
|
|
|
|
|
|
|
|
|
|
foreach (var profile in paths)
|
|
|
|
|
{
|
|
|
|
|
if (File.Exists(Path.Combine(profile, "Bookmarks")))
|
|
|
|
|
ParseChromeBookmarks(Path.Combine(profile, "Bookmarks"), name + (Path.GetFileName(profile) == "Default" ? "" : (" (" + Path.GetFileName(profile) + ")")));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void LoadChromeBookmarks()
|
|
|
|
|
{
|
|
|
|
|
String platformPath = Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData);
|
|
|
|
|
LoadChromeBookmarks(Path.Combine(platformPath, @"Google\Chrome\User Data"), "Google Chrome");
|
|
|
|
|
LoadChromeBookmarks(Path.Combine(platformPath, @"Google\Chrome SxS\User Data"), "Google Chrome Canary");
|
|
|
|
|
LoadChromeBookmarks(Path.Combine(platformPath, @"Chromium\User Data"), "Chromium");
|
|
|
|
|
}
|
|
|
|
|
|
2014-01-08 23:21:37 +08:00
|
|
|
|
private String DecodeUnicode(String dataStr)
|
|
|
|
|
{
|
|
|
|
|
Regex reg = new Regex(@"(?i)\\[uU]([0-9a-f]{4})");
|
|
|
|
|
return reg.Replace(dataStr, m => ((char)Convert.ToInt32(m.Groups[1].Value, 16)).ToString());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-03-17 03:37:52 +08:00
|
|
|
|
public class Bookmark : IEquatable<Bookmark>, IEqualityComparer<Bookmark>
|
2014-01-08 23:21:37 +08:00
|
|
|
|
{
|
2014-03-17 03:37:52 +08:00
|
|
|
|
private string m_Name;
|
|
|
|
|
public string Name {
|
|
|
|
|
get{
|
|
|
|
|
return m_Name;
|
|
|
|
|
}
|
|
|
|
|
set
|
|
|
|
|
{
|
|
|
|
|
m_Name = value;
|
|
|
|
|
PinyinName = ChineseToPinYin.ToPinYin(m_Name).Replace(" ", "").ToLower();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
public string PinyinName { get; private set; }
|
2014-01-08 23:21:37 +08:00
|
|
|
|
public string Url { get; set; }
|
|
|
|
|
public string Source { get; set; }
|
2014-03-25 13:36:33 +08:00
|
|
|
|
public int Score { get; set; }
|
2014-03-17 03:37:52 +08:00
|
|
|
|
|
|
|
|
|
/* TODO: since Source maybe unimportant, we just need to compare Name and Url */
|
|
|
|
|
public bool Equals(Bookmark other)
|
|
|
|
|
{
|
|
|
|
|
return Equals(this, other);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public bool Equals(Bookmark x, Bookmark y)
|
|
|
|
|
{
|
|
|
|
|
if (Object.ReferenceEquals(x, y)) return true;
|
|
|
|
|
if (Object.ReferenceEquals(x, null) || Object.ReferenceEquals(y, null))
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
|
|
return x.Name == y.Name && x.Url == y.Url;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public int GetHashCode(Bookmark bookmark)
|
|
|
|
|
{
|
|
|
|
|
if (Object.ReferenceEquals(bookmark, null)) return 0;
|
|
|
|
|
int hashName = bookmark.Name == null ? 0 : bookmark.Name.GetHashCode();
|
|
|
|
|
int hashUrl = bookmark.Url == null ? 0 : bookmark.Url.GetHashCode();
|
|
|
|
|
return hashName ^ hashUrl;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override int GetHashCode()
|
|
|
|
|
{
|
|
|
|
|
return GetHashCode(this);
|
|
|
|
|
}
|
2014-01-08 23:21:37 +08:00
|
|
|
|
}
|
|
|
|
|
}
|