/ProtoXML/ProtoFile.vb

Imports System.Xml

Imports System.Text



Public Class ProtoFile

    Implements IProtoXFile







    Public Sub New(ByVal NodeName As String)

        Me.NodeName = NodeName

    End Sub



    Public Property Doc As System.Xml.Linq.XDocument = <?xml version="1.0" encoding="UTF-8" standalone="yes"?><tvshow></tvshow> Implements IProtoXFile.Doc

    Public Property CacheDoc As System.Xml.Linq.XDocument = <?xml version="1.0" encoding="UTF-8" standalone="yes"?><tvshow></tvshow> Implements IProtoXFile.CacheDoc



    Private _node As XElement = Doc.Root

    Public Property Node As System.Xml.Linq.XElement Implements IProtoXBase.Node

        Get

            Return _node

        End Get

        Set(ByVal value As System.Xml.Linq.XElement)

            _node = value

        End Set

    End Property



    Private _cacheNode As XElement = CacheDoc.Root

    Public Property CacheNode As System.Xml.Linq.XElement Implements IProtoXBase.CacheNode

        Get

            Return _cacheNode

        End Get

        Set(value As System.Xml.Linq.XElement)

            _cacheNode = value

        End Set

    End Property



    Public Property NodeName As String Implements IProtoXBase.NodeName

        Get

            Return _node.Name.ToString

        End Get

        Set(ByVal value As String)

            _node.Name = value

            _cacheNode.Name = value

        End Set

    End Property



    Friend Sub AddChildForLoad(ByRef NewChild As IProtoXChild) Implements IProtoXBase.AddChildForLoad

        If Me.ChildrenLookup.ContainsKey(NewChild.NodeName) Then

            Throw New Exception("Already contains lookup for this node name)")

        End If



        Me.ChildrenLookup.Add(NewChild.NodeName, NewChild)

    End Sub



    Friend Property ChildrenLookup As New System.Collections.Generic.Dictionary(Of String, IProtoXChild) Implements IProtoXBase.ChildrenLookup





    Public Property IsCache As Boolean



#Region "File Access"

    Private _NfoFilePath As String

    Public Property NfoFilePath As String Implements IProtoXFile.NfoFilePath

        Get

            Return _NfoFilePath

        End Get

        Set(ByVal value As String)

            If _NfoFilePath <> value And _NfoFilePath IsNot Nothing Then Me.IsAltered = True

            Dim Parts() As String

            Parts = value.Split("\")

            _FolderPath = Parts(0)

            For I = 1 To Parts.GetUpperBound(0) - 1

                _FolderPath &= "\" & Parts(I)

            Next

            _FolderPath &= "\"

            _NfoFilePath = value

            EditAttribute("NfoPath", _NfoFilePath)

        End Set

    End Property



    Protected Friend Sub EditAttribute(ByVal Name As String, ByVal Value As String)



        If Me.Node.Attribute(Name) Is Nothing Then

            If Value IsNot Nothing Then

                Me.Node.Add(New XAttribute(Name, Value))

            Else

                Exit Sub

            End If

        Else

            If Value Is Nothing Then

                Me.Node.Attribute(Name).Remove()



                Exit Sub

            End If

        End If



        Me.Node.Attribute(Name).SetValue(Value)



        If Me.CacheNode.Attribute(Name) Is Nothing Then

            If Value IsNot Nothing Then

                Me.CacheNode.Add(New XAttribute(Name, Value))

            Else

                Exit Sub

            End If

        Else

            If Value Is Nothing Then

                Me.CacheNode.Attribute(Name).Remove()



                Exit Sub

            End If

        End If



        Me.CacheNode.Attribute(Name).SetValue(Value)

    End Sub





    Protected Friend Function GetAttribute(ByVal Name As String) As String



        If Me.Node.Attribute(Name) IsNot Nothing Then

            Return Me.Node.Attribute(Name).Value

        ElseIf Me.CacheNode.Attribute(Name) IsNot Nothing Then

            Return Me.CacheNode.Attribute(Name).Value

        Else

            Return Nothing

        End If

    End Function



    Private _FolderPath As String

    Public Property FolderPath As String Implements IProtoXBase.FolderPath

        Get

            Return _FolderPath

        End Get

        Set(value As String)

            _FolderPath = value

            If _NfoFilePath IsNot Nothing Then

                _NfoFilePath = IO.Path.Combine(value, IO.Path.GetFileName(_NfoFilePath))

            End If



        End Set

    End Property



    Public Overridable Sub Load() Implements IProtoXFile.Load

        Me.Load(Me.NfoFilePath)

        Me.IsAltered = False

        Me.IsCache = False

    End Sub



    Public Property FailedLoad As Boolean



    Public Overridable Sub Load(ByVal Path As String) Implements IProtoXFile.Load

        Me.IsAltered = True

        Me.CleanDoc()

        If IO.File.Exists(Path) Then

            Try

                Me.Doc = XDocument.Load(Path)

                Me.Doc.DescendantNodes.OfType(Of XComment)().Remove()

            Catch

                FailedLoad = True

                Exit Sub

            End Try

        Else

            Exit Sub

        End If

        LoadDoc()

    End Sub





    Public Sub LoadXml(ByVal Input As XNode)

        Me.Doc = New XDocument(Input)

        LoadDoc()

        Me.IsAltered = False    'These two properties have been moved here from TvCache.vb:Load() as this function is the

        Me.IsCache = True       'only one that uses LoadXml(XNode) so far. Please note if using this. - HueyHQ 27Feb2013

    End Sub





    Public Sub LoadXml(ByVal Input As String)

        Me.IsAltered = True

        Me.Doc = XDocument.Parse(Input)



        LoadDoc()

    End Sub



    Protected Overridable Sub LoadDoc()

        If Me.Doc.Root Is Nothing Then Throw New Exception("Invalid NFO file")

        Me._node = Me.Doc.Root

        Dim Root As XElement = Me.Doc.Root



        Dim ChildProperty As IProtoXChild

        Dim XElementList As New List(Of XElement)

        If Root.Name = "multiepisodenfo" Then

            For Each episode As XElement In Root.Nodes

                For Each Child As XNode In episode.Nodes

                    If TypeOf Child Is XElement Then

                        XElementList.Add(Child)

                    Else

                        Dim Test As Boolean = False

                    End If

                Next



            Next

        Else

            For Each Child As XNode In Root.Nodes

                If TypeOf Child Is XElement Then

                    XElementList.Add(Child)

                Else

                    Dim Test As Boolean = False

                End If

            Next

        End If



        For Each Child As XElement In XElementList

            If Me.ChildrenLookup.ContainsKey(Child.Name.ToString.ToLower) Then

                ChildProperty = Me.ChildrenLookup.Item(Child.Name.ToString.ToLower)



                ChildProperty.ProcessNode(Child)

                Dim ChanceToSeeChild As Boolean = True

            End If

        Next



        Me.CleanDoc()

        Me.IsCache = False

    End Sub



    Public Sub DeleteElement(ByVal elementName As String)

        CleanNode(Doc.Root, elementName)

        'Me.SurpressAlter = True

    End Sub



    Public Sub Save() Implements IProtoXFile.Save

        Me.IsAltered = False

        Me.Save(Me.NfoFilePath)

    End Sub



    Public Sub Save(ByVal Path As String) Implements IProtoXFile.Save

        'IsAltred shouldn't be set here, Save() isn't called the actual file referenced in NfoFilePath may not match what is in the current file

        Me.CleanDoc()

        Dim myloop As Boolean = True

        Do

            If Not DirectCast(Doc.FirstNode, System.Xml.Linq.XElement).FirstAttribute Is Nothing Then

                DirectCast(Doc.FirstNode, System.Xml.Linq.XElement).FirstAttribute.Remove() '   Value = "Removed - Now Not Used!"

            Else

                myloop = False

            End If

        Loop While myloop

        Doc.Save(Path)

        'Dim settings As New XmlWriterSettings()

        'settings.Encoding = New UTF8Encoding(False)

        'settings.Indent = True

        'settings.IndentChars = (ControlChars.Tab)

        'settings.NewLineHandling = NewLineHandling.None

        'Dim writer As XmlWriter = XmlWriter.Create(Path, settings)

        'Doc.Save(writer)

        'writer.Close()

        Me.IsCache = False

    End Sub



    Private Sub CleanDoc()

        CleanNode(Doc.Root)

    End Sub



    Private Sub CleanNode(ByVal Element As XElement, ByVal Optional elementName As String = "")

        Dim Cursor As XElement

        Dim NextOne As XNode

        If Element Is Nothing Then Exit Sub

        If Element.FirstNode Is Nothing Then

            Element.Remove()

            Exit Sub

        End If



        If Element.FirstNode.NodeType = Xml.XmlNodeType.Text Then

            Exit Sub

        End If



        Cursor = Element.FirstNode



        Do

            NextOne = Cursor.NextNode

            If Cursor.Nodes.Count = 0 AndAlso Cursor.Attributes.Count = 0 OrElse Cursor.Name.LocalName = elementName Then

                Cursor.Remove()

            Else

                CleanNode(Cursor)

            End If

            Cursor = NextOne

        Loop Until Cursor Is Nothing



        If Element.Nodes.Count = 0 AndAlso Element.Attributes.Count = 0 Then

            Element.Remove()

        End If

    End Sub



#End Region



    Public ReadOnly Property FileContainsReadableXml As Boolean Implements IProtoXFile.FileContainsReadableXml

        Get

            Try

                If Me.FileExists Then

                    Dim Test As XDocument = XDocument.Load(Me.NfoFilePath)

                Else

                    Return False

                End If

            Catch ex As Exception

                Return False

            End Try

            Return True

        End Get

    End Property



    Public ReadOnly Property FileExists As Boolean Implements IProtoXFile.FileExists

        Get

            Return IO.File.Exists(Me.NfoFilePath)

        End Get

    End Property



    Public Sub HandleChildValueChanged(ByRef ProtoChild As ProtoXChildBase) Implements IProtoXBase.HandleChildValueChanged

        Me.IsAltered = True

        RaiseEvent ValueChanged(ProtoChild)

    End Sub



    Public Sub RaiseValueChanged(ByRef ProtoChild As ProtoXChildBase) Implements IProtoXBase.RaiseValueChanged

        Me.IsAltered = True

        RaiseEvent ValueChanged(ProtoChild)

    End Sub



    Public Event ValueChanged(ByRef ProtoChild As ProtoXChildBase) Implements IProtoXBase.ValueChanged



    Public Property IsAltered As Boolean Implements IProtoXBase.IsAltered



    Private Property SurpressAlter As Boolean Implements IProtoXBase.SurpressAlter



    Public Sub SortNodes()

        For Each Item In ChildrenLookup.Values

            Item.Node.Remove()

        Next



        'Sort decending so that we can add nodes with an order first, and ones without at the bottom

        For Each Item In (From Items As IProtoXChild In ChildrenLookup.Values Order By Items.SortIndex Descending)

            If Item.SortIndex = -1 Then

                Me.Node.Add(Item.Node)

            Else

                Me.Node.AddFirst(Item.Node)

            End If



        Next

    End Sub



End Class