aboutsummaryrefslogtreecommitdiff
path: root/src/wix/WixToolset.Converters/ConversionLab.cs
blob: 727ff1b1b4766080116da2337c3b3ae7ccee9460 (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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// 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 InsertElementBeforeTargetElement(XElement newElement)
        {
            var index = this.index - 1;

            if (0 <= index
         && this.siblingNodes[index] is XText leadingText
         && String.IsNullOrWhiteSpace(leadingText.Value))
            {
                this.siblingNodes.Insert(index, newElement);
                this.siblingNodes.Insert(index, new XText(leadingText.Value));
                this.index += 2;
            }
        }

        private bool IsUniqueElement(XElement newElement)
        {
            foreach (XNode node in this.siblingNodes)
            {
                if (node is XElement element)
                {
                    if (element.Name == newElement.Name)
                    {
                        return false;
                    }
                }
            }

            return true;
        }

        public void InsertUniqueElementBeforeTargetElement(XElement newElement)
        {
            if (this.IsUniqueElement(newElement))
            {
                this.InsertElementBeforeTargetElement(newElement);
            }
        }

        public void RemoveTargetElement()
        {
            this.siblingNodes.RemoveAt(this.index);

            var index = this.index - 1;

            if (0 <= index
             && this.siblingNodes[index] is XText leadingText
             && String.IsNullOrWhiteSpace(leadingText.Value))
            {
                this.siblingNodes.RemoveAt(index);
            }

            this.RemoveOrphanTextNodes();
        }

        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);
        }
    }
}