Modules/GUIs.ZooFlow/Search/SearchLoader.vb

Imports System.IO
Imports System.Xml.Serialization
Imports DevExpress.Utils
Imports DevExpress.Utils.Svg
Imports DigitalData.GUIs.ZooFlow.frmSearchFlow
Imports DigitalData.GUIs.ZooFlow.Search.SearchToken
Imports DigitalData.Modules.Base
Imports DigitalData.Modules.Config
Imports DigitalData.Modules.Logging
Imports DigitalData.Modules.Language
Imports System.Reflection

Namespace Search
    Public Class SearchLoader
        Inherits BaseClass

        Private Const CUSTOM_SEARCH_DIRECTORY = "Searches"

        Private ReadOnly Images As SvgImageCollection
        Private ReadOnly ImageTable As List(Of ImageTableItem)

        Private ReadOnly Config As ConfigManager(Of SystemConfig)
        Private ReadOnly Serializer As XmlSerializer

        Public Sub New(pLogConfig As LogConfig, pConfig As ConfigManager(Of SystemConfig), pSvgImages As SvgImageCollection)
            MyBase.New(pLogConfig)
            Images = pSvgImages
            ImageTable = GetImageTable(pSvgImages)
            Config = pConfig
            Serializer = New XmlSerializer(GetType(SavedSearch.CustomSearchSerializable))
        End Sub

        Private Function GetSearchDirectoryPath() As String
            Dim oConfigPath As String = Config.UserConfigPath
            Dim oConfigDirectory As String = IO.Path.Combine(IO.Path.GetDirectoryName(oConfigPath), CUSTOM_SEARCH_DIRECTORY)

            If IO.Directory.Exists(oConfigDirectory) = False Then
                Try
                    IO.Directory.CreateDirectory(oConfigDirectory)
                Catch ex As Exception
                    Logger.Error(ex)
                End Try
            End If

            Return oConfigDirectory
        End Function

        Public Sub CreateCustomSearch(pTitle As String, pDescription As String, pTokens As List(Of Token), pImage As String)
            Dim oSearch As New SavedSearch.CustomSearch With {
                .Name = pTitle,
                .Description = pDescription,
                .Tokens = pTokens,
                .Image = Images.Item(pImage)
            }

            Try
                Dim oDirectoryPath As String = GetSearchDirectoryPath()
                Dim oFilePath As String = IO.Path.Combine(oDirectoryPath, Utils.ConvertTextToSlug(pTitle) & ".xml")
                Dim oBuffer As Byte() = SerializeSearch(oSearch)

                File.WriteAllBytes(oFilePath, oBuffer)
            Catch ex As Exception
                Logger.Warn("Custom Search could not be saved!")
                Logger.Error(ex)
            End Try
        End Sub

        Public Function LoadSearches() As List(Of SavedSearch.SavedSearch)
            Dim oSearches As New List(Of SavedSearch.SavedSearch)
            Dim oCustomSearches = LoadCustomSearches()
            Dim oPredefinedSearches = LoadPredefinedSearches()

            Return oSearches.
                Concat(oCustomSearches).
                Concat(oPredefinedSearches).
                ToList()
        End Function

        Public Function LoadCustomSearches() As List(Of SavedSearch.SavedSearch)
            Dim oSearches As New List(Of SavedSearch.SavedSearch)
            Dim oPath = GetSearchDirectoryPath()

            Dim oFiles = Directory.GetFiles(oPath, "*.xml")

            For Each oFile In oFiles
                Dim oBytes = File.ReadAllBytes(oFile)
                Dim oSearch = DeserializeSearch(oBytes)
                oSearches.Add(oSearch)
            Next

            Return oSearches
        End Function

        Public Function LoadPredefinedSearches() As List(Of SavedSearch.SavedSearch)
            Return New List(Of SavedSearch.SavedSearch) From {
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Heute",
                    .Description = "Dokumente, die heute abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.Today,
                    .Image = Images.Item("today")
                },
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Gestern",
                    .Description = "Dokumente, die gestern abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.Yesterday,
                    .Image = Images.Item("yesterday")
                },
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Letzte 7 Tage",
                    .Description = "Dokumente, die in den letzten 7 Tagen abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.Last7Days,
                    .Image = Images.Item("week")
                },
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Dieser Monat",
                    .Description = "Dokumente, die in diesem Monat abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.CurrentMonth,
                    .Image = Images.Item("month")
                },
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Letzter Monat",
                    .Description = "Dokumente, die im letzten Monat abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.LastMonth,
                    .Image = Images.Item("month")
                },
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Dieses Jahr",
                    .Description = "Dokumente, die in diesem Jahr abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.CurrentYear,
                    .Image = Images.Item("year")
                },
                New SavedSearch.PredefinedDateSearch() With {
                    .Name = "Letztes Jahr",
                    .Description = "Dokumente, die im letzten Jahr abgelegt wurden",
                    .DateConstraint = SearchRunner.DateConstraint.LastYear,
                    .Image = Images.Item("year")
                }
            }
        End Function

        Private Function ConvertSvgImageToString(pImage As SvgImage) As String
            Dim oItem = ImageTable.Where(Function(item) item.Image.Equals(pImage)).FirstOrDefault()
            If oItem Is Nothing Then
                Return Nothing
            End If

            Return oItem.Name
        End Function

        Private Function ConvertStringToImageName(pName As String) As SvgImage
            Dim oItem = ImageTable.Where(Function(item) item.Name.Equals(pName)).FirstOrDefault()
            If oItem Is Nothing Then
                Return Nothing
            End If

            Return oItem.Image
        End Function

        Private Function GetImageTable(pImageCollection As SvgImageCollection) As List(Of ImageTableItem)
            Dim oList As New List(Of ImageTableItem)

            For Each oImageName In pImageCollection.Keys
                Dim oSvgImage = pImageCollection.Item(oImageName)
                oList.Add(New ImageTableItem(oImageName, oSvgImage))
            Next

            Return oList
        End Function

        Private Function SerializeSearch(pSearch As SavedSearch.CustomSearch) As Byte()
            Try
                Dim oSearch As New SavedSearch.CustomSearchSerializable With {
                    .Name = pSearch.Name,
                    .Description = pSearch.Description,
                    .GroupText = pSearch.GroupText,
                    .Tokens = pSearch.Tokens,
                    .ImageName = ConvertSvgImageToString(pSearch.Image)
                }

                Using oStream As New MemoryStream
                    Serializer.Serialize(oStream, oSearch)
                    Return oStream.ToArray()
                End Using
            Catch ex As Exception
                Logger.Error(ex)
                Return Nothing
            End Try
        End Function

        Private Function DeserializeSearch(pBytes As Byte()) As SavedSearch.CustomSearch
            Try
                Using oStream As New MemoryStream(pBytes)
                    Dim oSearch As SavedSearch.CustomSearchSerializable = Serializer.Deserialize(oStream)

                    Return New SavedSearch.CustomSearch With {
                        .Name = oSearch.Name,
                        .Description = oSearch.Description,
                        .GroupText = oSearch.GroupText,
                        .Tokens = oSearch.Tokens,
                        .Image = ConvertStringToImageName(oSearch.ImageName)
                    }
                End Using
            Catch ex As Exception
                Logger.Error(ex)
                Return Nothing
            End Try
        End Function

        Private Class ImageTableItem
            Public ReadOnly Name As String
            Public ReadOnly Image As SvgImage

            Public Sub New(pName As String, pImage As SvgImage)
                Name = pName
                Image = pImage
            End Sub
        End Class




    End Class
End Namespace