456 lines
19 KiB
VB.net
456 lines
19 KiB
VB.net
Imports System.Text.RegularExpressions
|
|
Imports System.Windows.Forms
|
|
Imports DigitalData.Modules.Language.Utils
|
|
Imports DigitalData.Modules.Logging
|
|
Imports DigitalData.Modules.Windows
|
|
Imports DigitalData.Modules.ZooFlow.Params
|
|
|
|
Public Class ProfileFilter
|
|
Private _ProfileMatch As ProfileMatch
|
|
Private _ProfileTable As DataTable
|
|
Private _ProcessTable As DataTable
|
|
Private _WindowTable As DataTable
|
|
Private _ControlTable As DataTable
|
|
Private _Profiles As List(Of ProfileData)
|
|
Private _TreeView As TreeView
|
|
Private _LogConfig As LogConfig
|
|
Private _Logger As Logger
|
|
|
|
Public ReadOnly Property Profiles As List(Of ProfileData)
|
|
Get
|
|
Return _Profiles
|
|
End Get
|
|
End Property
|
|
|
|
Public Sub New(LogConfig As LogConfig, ProfileDatatable As DataTable, ProcessTable As DataTable, WindowDatatable As DataTable, ControlDatatable As DataTable, TreeView As TreeView)
|
|
Try
|
|
_ProfileMatch = New ProfileMatch(LogConfig)
|
|
_ProfileTable = ProfileDatatable
|
|
_ProcessTable = ProcessTable
|
|
_WindowTable = WindowDatatable
|
|
_ControlTable = ControlDatatable
|
|
_Profiles = TransformProfiles()
|
|
_TreeView = TreeView
|
|
_LogConfig = LogConfig
|
|
_Logger = LogConfig.GetLogger()
|
|
Catch ex As Exception
|
|
_Logger.Error(ex)
|
|
Throw ex
|
|
End Try
|
|
End Sub
|
|
|
|
Public Function ToList() As List(Of ProfileData)
|
|
Return _Profiles
|
|
End Function
|
|
|
|
Public Function FilterProfilesByClipboardRegex(Profiles As List(Of ProfileData), ClipboardContents As String) As List(Of ProfileData)
|
|
Dim oFilteredProfiles As New List(Of ProfileData)
|
|
|
|
Dim oRootNode As TreeNode = New TreeNode() With {
|
|
.Text = $"Suche nach '{ClipboardContents}'",
|
|
.Tag = "ROOT"
|
|
}
|
|
_TreeView.Nodes.Add(oRootNode)
|
|
|
|
For Each oProfile In Profiles
|
|
_Logger.Debug("Current Profile: {0}", oProfile.Name)
|
|
|
|
Dim oNode = _ProfileMatch.NewProfileNode(oProfile)
|
|
oRootNode.Nodes.Add(oNode)
|
|
|
|
Try
|
|
Dim oRegex As New Regex(oProfile.Regex)
|
|
Dim oMatch = oRegex.Match(ClipboardContents)
|
|
If oMatch.Success Then
|
|
_Logger.Debug("FilterProfilesByClipboardRegex: Clipboard Regex Matched: {0}", ClipboardContents)
|
|
oFilteredProfiles.Add(oProfile)
|
|
oProfile.IsMatched = True
|
|
|
|
Dim oSubnode = _ProfileMatch.NewClipboardRegexNode(oProfile, oMatch.Success)
|
|
oNode.Nodes.Add(oSubnode)
|
|
End If
|
|
Catch ex As Exception
|
|
_Logger.Warn("Regex '{0}' could not be processed for input '{1}'", oProfile.Regex, ClipboardContents)
|
|
_Logger.Error(ex)
|
|
End Try
|
|
Next
|
|
|
|
Return oFilteredProfiles
|
|
End Function
|
|
Public Function FilterProfilesByProcess(Profiles As List(Of ProfileData), CurrentProcessName As String) As List(Of ProfileData)
|
|
Dim oFilteredProfiles As New List(Of ProfileData)
|
|
Try
|
|
For Each oProfile In Profiles
|
|
Dim oGuid = oProfile.Guid
|
|
|
|
If oProfile.IsMatched = False Then
|
|
Continue For
|
|
End If
|
|
|
|
Dim oProcesses As New List(Of ProcessData)
|
|
For Each oProcessDef As ProcessData In oProfile.Processes
|
|
If oProcessDef.PROFILE_ID <> oGuid Then
|
|
Continue For
|
|
End If
|
|
|
|
Dim oIsMatch = oProcessDef.ProcessName.ToLower = CurrentProcessName.ToLower
|
|
Dim oParent = _ProfileMatch.FindNodeByTag(_TreeView.Nodes, oProfile.Name & "-REGEX")
|
|
If oParent IsNot Nothing Then
|
|
Dim oNode = _ProfileMatch.NewProcessNode(oProfile, oProcessDef, oIsMatch)
|
|
oParent.Nodes.Add(oNode)
|
|
End If
|
|
|
|
_Logger.Debug($"FilterProfilesByProcess: Checking Profile: {oProfile.Name} ...")
|
|
If oIsMatch Then
|
|
_Logger.Debug($"Yes...Processname Matched: {oProcessDef.ProcessName}")
|
|
'TODO: Add Debug Data
|
|
oFilteredProfiles.Add(oProfile)
|
|
oProfile.MatchedProcessID = oProcessDef.Guid
|
|
oProcessDef.IsMatched = True
|
|
oProcesses.Add(oProcessDef)
|
|
oProfile.IsMatched = True
|
|
oProfile.MatchedProcessID = oProcessDef.Guid
|
|
|
|
End If
|
|
Next
|
|
If oFilteredProfiles.Count > 0 Then
|
|
oProfile.Processes = oProcesses
|
|
End If
|
|
Next
|
|
|
|
Return oFilteredProfiles
|
|
Catch ex As Exception
|
|
_Logger.Warn("Unexpected error in FilterProfilesByProcess...")
|
|
_Logger.Error(ex)
|
|
End Try
|
|
|
|
End Function
|
|
Public Function FilterWindowsByWindowTitleRegex(Profiles As List(Of ProfileData), WindowTitle As String) As List(Of ProfileData)
|
|
Dim oProfiles As New List(Of ProfileData)
|
|
|
|
For Each oProfile As ProfileData In Profiles
|
|
_Logger.Debug("Checking WindowDefinition for profile: {0}...", oProfile.Name)
|
|
If oProfile.IsMatched = False Then Continue For
|
|
Dim oWindows As New List(Of WindowData)
|
|
|
|
For Each oWindowDef As WindowData In oProfile.Windows
|
|
If oWindowDef.WindowProcessID <> oProfile.MatchedProcessID Then Continue For
|
|
Try
|
|
If oWindowDef.Regex = String.Empty Then
|
|
oProfile.MatchedWindowID = oWindowDef.Guid
|
|
oWindowDef.IsMatched = True
|
|
oWindows.Add(oWindowDef)
|
|
Exit For
|
|
End If
|
|
|
|
Dim oRegex As New Regex(oWindowDef.Regex)
|
|
Dim oMatch = oRegex.Match(WindowTitle)
|
|
|
|
If oMatch.Success Then
|
|
_Logger.Debug("MATCH on WindowTitle: {0}", WindowTitle)
|
|
oProfile.MatchedWindowID = oWindowDef.Guid
|
|
oWindowDef.IsMatched = True
|
|
oWindows.Add(oWindowDef)
|
|
End If
|
|
|
|
Dim oParent = _ProfileMatch.FindNodeByTag(_TreeView.Nodes, oWindowDef.WindowProcessID & "-PROCESS")
|
|
If oParent IsNot Nothing Then
|
|
Dim oNode = _ProfileMatch.NewWindowNode(oProfile, oWindowDef, oMatch.Success)
|
|
oParent.Nodes.Add(oNode)
|
|
End If
|
|
|
|
Catch ex As Exception
|
|
_Logger.Warn("Regex '{0}' could not be processed for input '{1}'", oWindowDef.Regex, WindowTitle)
|
|
_Logger.Error(ex)
|
|
End Try
|
|
Next
|
|
|
|
If oWindows.Count > 0 Then
|
|
oProfile.Windows = oWindows
|
|
oProfile.IsMatched = True
|
|
oProfiles.Add(oProfile)
|
|
End If
|
|
Next
|
|
|
|
Return oProfiles
|
|
End Function
|
|
|
|
Public Function FilterProfilesByFocusedControlLocation(Profiles As List(Of ProfileData), ClipboardContents As String, WindowHandle As IntPtr) As List(Of ProfileData)
|
|
Dim oFilteredProfiles As New List(Of ProfileData)
|
|
Dim oWindow As New Window(_LogConfig)
|
|
|
|
For Each oProfile In Profiles
|
|
If oProfile.IsMatched = False Then Continue For
|
|
|
|
If oProfile.Controls.Count = 0 Then
|
|
oFilteredProfiles.Add(oProfile)
|
|
Continue For
|
|
End If
|
|
|
|
Dim oControls As New List(Of ControlData)
|
|
|
|
For Each oControl In oProfile.Controls
|
|
Dim oControlBounds = oWindow.GetFocusedControlLocation(WindowHandle)
|
|
Dim oFound As Boolean = False
|
|
|
|
For Each oItem As KeyValuePair(Of String, Window.RectangleInfo) In oControlBounds
|
|
Select Case oItem.Key
|
|
Case "TOPLEFT"
|
|
If oControl.TopLeft.Top = oItem.Value.Top And oControl.TopLeft.Left = oItem.Value.Left Then
|
|
oFound = True
|
|
End If
|
|
Case "TOPRIGHT"
|
|
If oControl.TopRight.Top = oItem.Value.Top And oControl.TopLeft.Right = oItem.Value.Right Then
|
|
oFound = True
|
|
End If
|
|
Case "BOTTOMLEFT"
|
|
If oControl.BottomLeft.Bottom = oItem.Value.Bottom And oControl.TopLeft.Left = oItem.Value.Left Then
|
|
oFound = True
|
|
End If
|
|
Case "BOTTOMRIGHT"
|
|
If oControl.BottomRight.Bottom = oItem.Value.Bottom And oControl.TopLeft.Right = oItem.Value.Right Then
|
|
oFound = True
|
|
End If
|
|
End Select
|
|
Next
|
|
|
|
If oFound Then
|
|
oFilteredProfiles.Add(oProfile)
|
|
End If
|
|
|
|
Dim oParent = _ProfileMatch.FindNodeByTag(_TreeView.Nodes, oControl.WindowId & "-WINDOW")
|
|
If oParent IsNot Nothing Then
|
|
Dim oNode = _ProfileMatch.NewControlNode(oProfile, oControl, oFound)
|
|
oParent.Nodes.Add(oNode)
|
|
End If
|
|
Next
|
|
|
|
If oControls.Count > 0 Then
|
|
oProfile.Controls = oControls
|
|
oFilteredProfiles.Add(oProfile)
|
|
End If
|
|
Next
|
|
|
|
|
|
Return oFilteredProfiles
|
|
End Function
|
|
|
|
'Public Function FilterProfilesBySearchResults(Profiles As List(Of ProfileData), DataSearchTable As DataTable, DocSearchTable As DataTable) As List(Of ProfileData)
|
|
' Dim oProfiles As New List(Of ProfileData)
|
|
|
|
' For Each oProfile In Profiles
|
|
' Dim oResultDocs As Integer = 0
|
|
' Dim oResultData As Integer = 0
|
|
|
|
' Dim oPatterns As New ClassPatterns(_LogConfig)
|
|
' Dim oDataSearches As DataTable = DataSearchTable
|
|
' Dim oDocSearches As DataTable = DocSearchTable
|
|
|
|
' For Each oRow As DataRow In oDataSearches.Rows
|
|
' Dim oCountCommand = String.Empty
|
|
' Try
|
|
' oCountCommand = NotNull(oRow.Item("COUNT_COMMAND"), String.Empty)
|
|
|
|
' If oCountCommand = String.Empty Then
|
|
' Continue For
|
|
' End If
|
|
|
|
' oCountCommand = oPatterns.ReplaceAllValues(oCountCommand, My.Application.User)
|
|
' oResultData += NotNull(Of Integer)(My.Database.GetScalarValue(oCountCommand), 0)
|
|
' Catch ex As Exception
|
|
' _Logger.Warn("Invalid SQL Query for Counting Data in Profile {0}: {1}", oProfile.Guid, oCountCommand)
|
|
' End Try
|
|
' Next
|
|
|
|
' For Each oRow As DataRow In oDocSearches.Rows
|
|
' Dim oCountCommand = String.Empty
|
|
' Try
|
|
' oCountCommand = NotNull(oRow.Item("COUNT_COMMAND"), String.Empty)
|
|
|
|
' If oCountCommand = String.Empty Then
|
|
' Continue For
|
|
' End If
|
|
|
|
' oCountCommand = oPatterns.ReplaceAllValues(oCountCommand, My.Application.User)
|
|
' oResultDocs += NotNull(Of Integer)(My.Database.GetScalarValue(oCountCommand), 0)
|
|
' Catch ex As Exception
|
|
' _Logger.Warn("Invalid SQL Query for Counting Data in Profile {0}: {1}", oProfile.Guid, oCountCommand)
|
|
' End Try
|
|
' Next
|
|
|
|
' If oResultData > 0 Or oResultDocs > 0 Then
|
|
' oProfile.CountData = oResultData
|
|
' oProfile.CountDocs = oResultDocs
|
|
' oProfiles.Add(oProfile)
|
|
' End If
|
|
' Next
|
|
|
|
' Return oProfiles
|
|
'End Function
|
|
|
|
Public Function FilterWindowsByWindowClipboardRegex(Profiles As List(Of ProfileData), ClipboardContents As String) As List(Of ProfileData)
|
|
Dim oProfiles As New List(Of ProfileData)
|
|
|
|
For Each oProfile As ProfileData In Profiles
|
|
_Logger.Debug("Current Profile: {0}", oProfile.Name)
|
|
|
|
Dim oWindows As New List(Of WindowData)
|
|
|
|
For Each w As WindowData In oProfile.Windows
|
|
Try
|
|
If w.Regex = String.Empty Then
|
|
oWindows.Add(w)
|
|
End If
|
|
|
|
Dim oRegex As New Regex(w.Regex)
|
|
Dim oMatch = oRegex.Match(ClipboardContents)
|
|
|
|
If oMatch.Success Then
|
|
_Logger.Debug("Window Clipboard Regex Matched: {0}", ClipboardContents)
|
|
oWindows.Add(w)
|
|
End If
|
|
|
|
Dim oResult As TreeNode = _ProfileMatch.FindNodeByTag(_TreeView.Nodes, oProfile.Name & "-REGEX")
|
|
If Not IsNothing(oResult) Then
|
|
Dim oNode As New TreeNode($"MATCH on WINDOW Clipboard Regex [{w.Regex}]: {oMatch.Success}")
|
|
oNode.ImageIndex = 2
|
|
oNode.Tag = oProfile.Name & "-WINDOW_REGEX"
|
|
oResult.Nodes.Add(oNode)
|
|
End If
|
|
Catch ex As Exception
|
|
_Logger.Warn("Regex '{0}' could not be processed for input '{1}'", w.Regex, ClipboardContents)
|
|
_Logger.Error(ex)
|
|
End Try
|
|
Next
|
|
|
|
If oWindows.Count > 0 Then
|
|
oProfile.Windows = oWindows
|
|
oProfiles.Add(oProfile)
|
|
End If
|
|
Next
|
|
|
|
Return oProfiles
|
|
End Function
|
|
|
|
Public Function ClearNotMatchedProfiles(Profiles As List(Of ProfileData)) As List(Of ProfileData)
|
|
Dim oFilteredProfiles As New List(Of ProfileData)
|
|
For Each oProfile In Profiles
|
|
If oProfile.IsMatched Then
|
|
oFilteredProfiles.Add(oProfile)
|
|
End If
|
|
Next
|
|
Return oFilteredProfiles
|
|
End Function
|
|
|
|
Public Function ClearDuplicateProfiles(Profiles As List(Of ProfileData)) As List(Of ProfileData)
|
|
Return Profiles.
|
|
GroupBy(Function(Profile) Profile.Guid).
|
|
Select(Function(GroupedProfiles) GroupedProfiles.First).
|
|
ToList()
|
|
End Function
|
|
|
|
Private Function TransformProfiles() As List(Of ProfileData)
|
|
Dim oList As New List(Of ProfileData)
|
|
|
|
For Each oRow As DataRow In _ProfileTable.Rows
|
|
Dim oProfileId = oRow.Item("GUID")
|
|
Dim oProcessList As List(Of ProcessData) = TransformProcesses(oProfileId, _ProcessTable)
|
|
Dim oWindowList As List(Of WindowData) = TransformWindows(oProfileId, _WindowTable)
|
|
Dim oControlList As List(Of ControlData) = TransformControls(oProfileId, _ControlTable)
|
|
|
|
oList.Add(New ProfileData() With {
|
|
.Guid = oRow.Item("GUID"),
|
|
.Regex = oRow.Item("REGEX_EXPRESSION"),
|
|
.Name = NotNull(oRow.Item("NAME"), String.Empty),
|
|
.Comment = NotNull(oRow.Item("COMMENT"), String.Empty),
|
|
.ProfileType = NotNull(oRow.Item("PROFILE_TYPE"), String.Empty),
|
|
.Processes = oProcessList,
|
|
.Windows = oWindowList,
|
|
.Controls = oControlList
|
|
})
|
|
Next
|
|
|
|
oList = oList.
|
|
Distinct().
|
|
ToList()
|
|
|
|
Return oList
|
|
End Function
|
|
|
|
Private Function TransformControls(ProfileId As Integer, ControlDatatable As DataTable) As List(Of ControlData)
|
|
Dim oControlList As New List(Of ControlData)
|
|
|
|
For Each oRow As DataRow In ControlDatatable.Rows
|
|
If oRow.Item("PROFILE_ID") = ProfileId Then
|
|
oControlList.Add(New ControlData() With {
|
|
.Guid = oRow.Item("GUID"),
|
|
.Description = NotNull(oRow.Item("DESCRIPTION"), String.Empty),
|
|
.WindowId = oRow.Item("WINDOW_ID"),
|
|
.TopLeft = New ControlBounds() With {
|
|
.Left = oRow.Item("TOPLEFT_LEFT"),
|
|
.Right = oRow.Item("TOPLEFT_RIGHT"),
|
|
.Top = oRow.Item("TOPLEFT_TOP"),
|
|
.Bottom = oRow.Item("TOPLEFT_BOTTOM")
|
|
},
|
|
.TopRight = New ControlBounds() With {
|
|
.Left = oRow.Item("TOPRIGHT_LEFT"),
|
|
.Right = oRow.Item("TOPRIGHT_RIGHT"),
|
|
.Top = oRow.Item("TOPRIGHT_TOP"),
|
|
.Bottom = oRow.Item("TOPRIGHT_BOTTOM")
|
|
},
|
|
.BottomLeft = New ControlBounds() With {
|
|
.Left = oRow.Item("BOTTOMLEFT_LEFT"),
|
|
.Right = oRow.Item("BOTTOMLEFT_RIGHT"),
|
|
.Top = oRow.Item("BOTTOMLEFT_TOP"),
|
|
.Bottom = oRow.Item("BOTTOMLEFT_BOTTOM")
|
|
},
|
|
.BottomRight = New ControlBounds() With {
|
|
.Left = oRow.Item("BOTTOMRIGHT_LEFT"),
|
|
.Right = oRow.Item("BOTTOMRIGHT_RIGHT"),
|
|
.Top = oRow.Item("BOTTOMRIGHT_TOP"),
|
|
.Bottom = oRow.Item("BOTTOMRIGHT_BOTTOM")
|
|
}
|
|
})
|
|
End If
|
|
Next
|
|
|
|
Return oControlList
|
|
End Function
|
|
Private Function TransformProcesses(ProfileId As Integer, ProcessDatatable As DataTable) As List(Of ProcessData)
|
|
Dim oProcessList As New List(Of ProcessData)
|
|
|
|
For Each oRow As DataRow In ProcessDatatable.Rows
|
|
If oRow.Item("PROFILE_ID") = ProfileId Then
|
|
oProcessList.Add(New ProcessData() With {
|
|
.Guid = oRow.Item("GUID"),
|
|
.PROFILE_ID = oRow.Item("PROFILE_ID"),
|
|
.ProcessName = NotNull(oRow.Item("PROC_NAME"), String.Empty)
|
|
})
|
|
End If
|
|
Next
|
|
|
|
oProcessList = oProcessList.
|
|
Distinct().
|
|
ToList()
|
|
|
|
Return oProcessList
|
|
End Function
|
|
Private Function TransformWindows(ProfileId As Integer, WindowDatatable As DataTable) As List(Of WindowData)
|
|
Dim oWindowList As New List(Of WindowData)
|
|
|
|
For Each oRow As DataRow In WindowDatatable.Rows
|
|
If oRow.Item("PROFILE_ID") = ProfileId Then
|
|
oWindowList.Add(New WindowData() With {
|
|
.Guid = oRow.Item("GUID"),
|
|
.WindowProcessID = oRow.Item("PROCESS_ID"),
|
|
.Title = NotNull(oRow.Item("DESCRIPTION"), String.Empty),
|
|
.Regex = NotNull(oRow.Item("REGEX"), String.Empty),
|
|
.Sequence = NotNull(oRow.Item("SEQUENCE"), 0)
|
|
})
|
|
End If
|
|
Next
|
|
|
|
Return oWindowList
|
|
End Function
|
|
End Class
|