github/tai

公开

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

完善观察服务

作者: noberumotto
时间: 2022/10/24 15:13
提交: 1d1e358
上一个: 398eae6

修改Core/Servicers/Instances/Observer.cs+92 −71看文件

@@ -18,11 +18,8 @@ namespace Core.Servicers.Instances
18	18	private IntPtr hook;
19	19	Win32API.WinEventDelegate winEventDelegate;
20	20
21		删除：
22	21	public event ObserverEventHandler OnAppActive;
23	22
24		删除：
25		删除： //private string activeProcessName, activeProcessFile = null;
26	23	public Observer()
27	24	{
28	25	winEventDelegate = new Win32API.WinEventDelegate(WinEventProc);
@@ -31,70 +28,107 @@ namespace Core.Servicers.Instances
31	28	{
32	29	hook = Win32API.SetWinEventHook(0x0003, 0x0003, IntPtr.Zero,
33	30	winEventDelegate, 0, 0, 0);
	31	新增：
	32	新增： var window = Win32API.GetForegroundWindow();
	33	新增： Handle(window);
	34	新增： }
	35	新增：
	36	新增： private void WinEventProc(IntPtr hWinEventHook, uint eventType, IntPtr hwnd, int idObject, int idChild, uint dwEventThread, uint dwmsEventTime)
	37	新增： {
	38	新增： Handle(hwnd);
34	39	}
35	40
36		删除： private async void WinEventProc(IntPtr hWinEventHook, uint eventType, IntPtr hwnd, int idObject, int idChild, uint dwEventThread, uint dwmsEventTime)
	41	新增： private async void Handle(IntPtr hwnd)
37	42	{
38	43	string processName = String.Empty, processFileName = String.Empty, processDescription = String.Empty;
	44	新增：
39	45	var processResult = await Task.Run(() =>
40		删除： {
41		删除： try
42		删除： {
43		删除： string[] processInfo = GetProcessInfoByWindowHandle(hwnd);
44		删除： int processID = int.Parse(processInfo[2]);
45		删除： processName = processInfo[0];
46		删除： processFileName = processInfo[1];
47		删除：
48		删除：
49		删除： // handle uwp app
50		删除： if (processFileName.IndexOf("ApplicationFrameHost.exe") != -1)
51		删除： {
52		删除： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
53		删除： if (processFileName == null)
54		删除： {
55		删除： Thread.Sleep(1500);
56		删除：
57		删除： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
58		删除： processName = GetUWPAPPName(processFileName);
59		删除：
60		删除： IntPtr foregeroundHwnd = Win32API.GetForegroundWindow();
61		删除：
62		删除： if (foregeroundHwnd != hwnd)
63		删除： {
64		删除： // 获取后焦点变了
65		删除： processInfo = GetProcessInfoByWindowHandle(foregeroundHwnd);
66		删除： processID = int.Parse(processInfo[2]);
67		删除： processName = processInfo[0];
68		删除： processFileName = processInfo[1];
69		删除： }
70		删除：
71		删除： }
72		删除： else
73		删除： {
74		删除： processName = GetUWPAPPName(processFileName);
75		删除： }
76		删除：
77		删除： }
78		删除： Debug.WriteLine("file name: ---------> " + " \| " + hwnd + " \| " + processID + processFileName);
79		删除： Debug.WriteLine("app name: ---------> " + processName);
80		删除： FileVersionInfo info = FileVersionInfo.GetVersionInfo(processFileName);
81		删除： processDescription = info.FileDescription;
82		删除：
83		删除：
84		删除： //Debug.WriteLine("file name: ---------> " + processFileName);
85		删除： //Debug.WriteLine("app name: ---------> " + processName);
86		删除： //Debug.WriteLine("processDescription: ---------> " + processDescription);
87		删除： }
88		删除： catch (Exception e)
89		删除： {
90		删除： Logger.Warn(e.Message + " \| Process Name:" + processName + " \| Process File:" + processFileName + " \| Process Description:" + processDescription);
91		删除： }
92		删除： return processName;
93		删除： });
	46	新增： {
	47	新增： try
	48	新增： {
	49	新增： string[] processInfo = GetProcessInfoByWindowHandle(hwnd);
	50	新增： int processID = int.Parse(processInfo[2]);
	51	新增： processName = processInfo[0];
	52	新增： processFileName = processInfo[1];
	53	新增：
	54	新增：
	55	新增： // handle uwp app
	56	新增： if (processFileName.IndexOf("ApplicationFrameHost.exe") != -1)
	57	新增： {
	58	新增： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
	59	新增： if (processFileName == null)
	60	新增： {
	61	新增： Thread.Sleep(1500);
	62	新增：
	63	新增： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
	64	新增： processName = GetUWPAPPName(processFileName);
	65	新增：
	66	新增： IntPtr foregeroundHwnd = Win32API.GetForegroundWindow();
	67	新增：
	68	新增： if (foregeroundHwnd != hwnd)
	69	新增： {
	70	新增： // 获取后焦点变了
	71	新增： processInfo = GetProcessInfoByWindowHandle(foregeroundHwnd);
	72	新增： processID = int.Parse(processInfo[2]);
	73	新增： processName = processInfo[0];
	74	新增： processFileName = processInfo[1];
	75	新增： }
	76	新增：
	77	新增： }
	78	新增： else
	79	新增： {
	80	新增： processName = GetUWPAPPName(processFileName);
	81	新增： }
	82	新增：
	83	新增： }
	84	新增： FileVersionInfo info = FileVersionInfo.GetVersionInfo(processFileName);
	85	新增： processDescription = info.FileDescription;
	86	新增：
	87	新增：
	88	新增： int titleCapacity = Win32API.GetWindowTextLength(hwnd) * 2;
	89	新增： StringBuilder stringBuilder = new StringBuilder(titleCapacity);
	90	新增： Win32API.GetWindowText(hwnd, stringBuilder, stringBuilder.Capacity);
	91	新增： if (IsSystemProcess(processName, stringBuilder.ToString()))
	92	新增： {
	93	新增： // 过滤掉一些系统进程
	94	新增： processName = string.Empty;
	95	新增： }
	96	新增： }
	97	新增： catch (Exception e)
	98	新增： {
	99	新增： Logger.Warn(e.Message + " \| Process Name:" + processName + " \| Process File:" + processFileName + " \| Process Description:" + processDescription);
	100	新增： }
	101	新增： return processName;
	102	新增： });
94	103
95	104	EventInvoke(processName, processDescription, processFileName);
96	105	}
97	106
	107	新增： /// <summary>
	108	新增： /// 指示指定进程是否属于系统进程
	109	新增： /// </summary>
	110	新增： /// <param name="processName">进程名称</param>
	111	新增： /// <param name="title">窗口标题</param>
	112	新增： /// <returns></returns>
	113	新增： private bool IsSystemProcess(string processName, string title)
	114	新增： {
	115	新增： processName = processName.ToLower();
	116	新增： int length = title.Length;
	117	新增： string[] titleArr = { "UnlockingWindow", "Program Manager" };
	118	新增：
	119	新增： if (processName == "explorer")
	120	新增： {
	121	新增： if (length == 0)
	122	新增： {
	123	新增： // 可能是任务栏
	124	新增： return true;
	125	新增： }
	126	新增： return titleArr.Contains(title);
	127	新增：
	128	新增： }
	129	新增： return false;
	130	新增： }
	131	新增：
98	132	private string GetUWPAPPName(string processFileName)
99	133	{
100	134	string processName = string.Empty;
@@ -137,19 +171,6 @@ namespace Core.Servicers.Instances
137	171	}
138	172	private void EventInvoke(string processName, string description, string filename)
139	173	{
140		删除： // 防止重复和错误响应
141		删除： //if (string.IsNullOrEmpty(processName) \|\| string.IsNullOrEmpty(filename) \|\| !File.Exists(filename))
142		删除： //{
143		删除： // return;
144		删除： //}
145		删除： //if (filename == activeProcessFile && processName == activeProcessName)
146		删除： //{
147		删除： // return;
148		删除： //}
149		删除：
150		删除： //activeProcessName = processName;
151		删除： //activeProcessFile = filename;
152		删除：
153	174	OnAppActive?.Invoke(processName, description, filename);
154	175	}
155	176