keeweb
/
keeweb
mirror of https://github.com/keeweb/keeweb.git
1
0
Fork 0
Code Issues Releases Wiki Activity

better browser url detection

This commit is contained in:
antelle 2016-04-27 23:54:13 +03:00
parent 46f3d76380
commit ad6fb55e9d
1 changed files with 32 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
helper/win32/KeeWebHelper/WindowHelper.cs
View File

@ -8,25 +8,31 @@ namespace KeeWebHelper
{ {
class WindowHelper class WindowHelper
{ {
static readonly string[] BrowserProcessNames = new[] { "chrome", "firefox", "iexplore", "edge", "opera", "browser" };
static readonly string[] BrowserWindowClasses = new[] { "Chrome_WidgetWin_1" };
[DllImport("user32.dll")] [DllImport("user32.dll")]
static extern IntPtr GetForegroundWindow(); static extern IntPtr GetForegroundWindow();
[DllImport("user32.dll")] [DllImport("user32.dll")]
static extern int GetWindowText(IntPtr hWnd, StringBuilder text, int count); static extern int GetWindowText(IntPtr hWnd, StringBuilder text, int count);
[DllImport("user32.dll", SetLastError = true, CharSet = CharSet.Auto)]
static extern int GetClassName(IntPtr hWnd, StringBuilder lpClassName, int nMaxCount);
[DllImport("user32.dll", SetLastError = true)] [DllImport("user32.dll", SetLastError = true)]
static extern uint GetWindowThreadProcessId(IntPtr hWnd, out uint processId); static extern uint GetWindowThreadProcessId(IntPtr hWnd, out uint processId);
public static WindowInfo GetActiveWindowInfo() public static WindowInfo GetActiveWindowInfo()
{ {
const int nChars = 2048; const int nChars = 2048;
StringBuilder Buff = new StringBuilder(nChars); var buff = new StringBuilder(nChars);
IntPtr hwnd = GetForegroundWindow(); IntPtr hwnd = GetForegroundWindow();
var result = new WindowInfo(); var result = new WindowInfo();
if (GetWindowText(hwnd, Buff, nChars) > 0) if (GetWindowText(hwnd, buff, nChars) > 0)
{ {
result.Title = Buff.ToString(); result.Title = buff.ToString();
} }
try try
@ -47,11 +53,17 @@ namespace KeeWebHelper
return null; return null;
} }
var process = Process.GetProcessById((int)pid); var process = Process.GetProcessById((int)pid);
return GetUrlAcc(process);
}
static string GetUrlAcc(Process process) if (!IsBrowser(process.ProcessName))
{ {
const int maxChars = 64;
var cls = new StringBuilder(maxChars);
var clsId = GetClassName(hwnd, cls, maxChars);
if (Array.IndexOf(BrowserWindowClasses, cls.ToString()) < 0) {
return null;
}
}
var el = AutomationElement.FromHandle(process.MainWindowHandle); var el = AutomationElement.FromHandle(process.MainWindowHandle);
if (el == null) if (el == null)
{ {
@ -79,5 +91,18 @@ namespace KeeWebHelper
var val = (ValuePattern)urlEl.GetCurrentPattern(ValuePattern.Pattern); var val = (ValuePattern)urlEl.GetCurrentPattern(ValuePattern.Pattern);
return val.Current.Value; return val.Current.Value;
} }
static bool IsBrowser(string processName)
{
processName = processName.ToLower();
foreach (var br in BrowserProcessNames)
{
if (processName.IndexOf(br) >= 0)
{
return true;
}
}
return false;
}
} }
} }