aboutsummaryrefslogtreecommitdiff
path: root/src/wix/WixToolset.Converters/ConversionLab.cs
blob: 903071f86f10caf5b68ef447e38913b17997413b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.

namespace WixToolset.Converters
{
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Xml.Linq;

    internal class ConversionLab : IDisposable
    {
        private readonly XElement targetElement;
        private readonly XElement parentElement;
        private readonly List<XNode> siblingNodes;
        private int index;

        public ConversionLab(XElement targetElement)
        {
            this.targetElement = targetElement;
            this.parentElement = this.targetElement.Parent;
            this.siblingNodes = this.parentElement.Nodes().ToList();

            foreach (var siblingNode in this.siblingNodes)
            {
                siblingNode.Remove();
            }

            this.index = this.siblingNodes.IndexOf(this.targetElement);
        }

        public void RemoveOrphanTextNodes()
        {
            if (!this.targetElement.HasElements)
            {
                var childNodes = this.targetElement.Nodes().ToList();
                foreach (var childNode in childNodes)
                {
                    childNode.Remove();
                }
            }
        }

        public void RemoveTargetElement()
        {
            if (this.index + 1 < this.siblingNodes.Count
             && this.siblingNodes[this.index + 1] is XText trailingText
             && String.IsNullOrWhiteSpace(trailingText.Value))
            {
                this.siblingNodes.RemoveAt(this.index + 1);
            }
            this.siblingNodes.RemoveAt(this.index);
            if (0 < this.index
             && this.siblingNodes[this.index - 1] is XText leadingText
             && String.IsNullOrWhiteSpace(leadingText.Value))
            {
                this.siblingNodes.RemoveAt(this.index - 1);
            }
        }

        public void ReplaceTargetElement(XElement replacement)
        {
            this.siblingNodes[this.index] = replacement;
        }

        public void AddCommentsAsSiblings(List<XNode> comments)
        {
            if (0 < this.index
             && this.siblingNodes[this.index - 1] is XText leadingText
             && String.IsNullOrWhiteSpace(leadingText.Value))
            {
                var leadingWhitespace = leadingText.Value;
                --this.index;
                var newComments = new List<XNode>();
                foreach(var comment in comments)
                {
                    newComments.Add(new XText(leadingWhitespace));
                    newComments.Add(comment);
                }
                comments = newComments;
            }

            this.siblingNodes.InsertRange(this.index, comments);
            this.index = this.siblingNodes.IndexOf(this.targetElement);
        }

        public void Dispose()
        {
            this.parentElement.Add(this.siblingNodes);
        }
    }
}