github/tai

公开

Tai · Windows 软件与网站使用时长统计。上游 Planshit/Tai 的源码镜像，非灵栈原创；MIT 许可。来源：https://github.com/Planshit/Tai 。本次为手动同步，暂不提供安装包。

修复浏览器地址栏隐藏协议头时获取favicon失败的问题

作者: noberumotto
时间: 2023/03/02 17:11
提交: ca5beef
上一个: 6417efc

修改Core/Librarys/Browser/Favicon/ChromeFaviconManager.cs+23 −13看文件

@@ -3,10 +3,12 @@ using System;
3	3	using System.Collections.Generic;
4	4	using System.Data.SQLite;
5	5	using System.Diagnostics;
	6	新增：using System.Drawing;
6	7	using System.IO;
7	8	using System.Linq;
8	9	using System.Net;
9	10	using System.Security.Cryptography;
	11	新增：using System.Security.Policy;
10	12	using System.Text;
11	13	using System.Threading;
12	14	using System.Threading.Tasks;
@@ -35,13 +37,13 @@ namespace Core.Librarys.Browser.Favicon
35	37	{
36	38	Directory.CreateDirectory(_taiFaviconsPath);
37	39	}
38		删除： Debug.WriteLine(_chromeFaviconsPath);
39	40	}
40	41	public async Task<object> GetIconDataAsync(string url)
41	42	{
42	43	if (string.IsNullOrEmpty(url)) { throw new ArgumentNullException("url"); }
43	44
44	45	url = UrlHelper.ChromeURLEncode(url);
	46	新增：
45	47	var result = await Task.Run(() =>
46	48	{
47	49	byte[] icon = { };
@@ -52,18 +54,9 @@ namespace Core.Librarys.Browser.Favicon
52	54	using (var con = new SQLiteConnection($"Data Source={_faviconsTempPath}"))
53	55	{
54	56	con.Open();
55		删除： string iconId = string.Empty;
56		删除： using (var cmd = new SQLiteCommand($"select * from icon_mapping where page_url like'{url}%' LIMIT 1", con))
57		删除： {
58		删除： using (SQLiteDataReader dr = cmd.ExecuteReader())
59		删除： {
60		删除： if (dr.Read())
61		删除： {
62		删除： iconId = dr["icon_id"].ToString();
63		删除： }
64		删除： }
65		删除： cmd.CommandText = "select * from ";
66		删除： }
	57	新增：
	58	新增： string iconId = GetIconId(url, con);
	59	新增：
67	60	if (iconId != string.Empty)
68	61	{
69	62	using (var cmd = new SQLiteCommand($"select * from favicon_bitmaps where icon_id={iconId} order by width desc LIMIT 1", con))
@@ -91,6 +84,23 @@ namespace Core.Librarys.Browser.Favicon
91	84	return result.Length == 0 ? null : result;
92	85	}
93	86
	87	新增： private string GetIconId(string url_, SQLiteConnection con_)
	88	新增： {
	89	新增： string iconId = string.Empty;
	90	新增： string sqlStr = url_.IndexOf("://") == -1 ? $"%{url_}%" : $"{url_}%";
	91	新增： using (var cmd = new SQLiteCommand($"select * from icon_mapping where page_url like'{sqlStr}' LIMIT 1", con_))
	92	新增： {
	93	新增： using (SQLiteDataReader dr = cmd.ExecuteReader())
	94	新增： {
	95	新增： if (dr.Read())
	96	新增： {
	97	新增： iconId = dr["icon_id"].ToString();
	98	新增： }
	99	新增： }
	100	新增： }
	101	新增： return iconId;
	102	新增： }
	103	新增：
94	104	public async Task<string> SaveToLocalIconAsync(object data)
95	105	{
96	106	if (data == null) { throw new ArgumentNullException("data"); }