-
Notifications
You must be signed in to change notification settings - Fork 1
/
InsertUpdater.cs
106 lines (94 loc) · 3.94 KB
/
InsertUpdater.cs
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
using System.Text.RegularExpressions;
namespace KeyValue3Updater
{
/// <summary>
/// Updater that takes a block and inserts it elsewhere in a file
/// </summary>
internal abstract class InsertUpdater : Updater
{
protected virtual string InsertContainerBlockName { get; }
private Regex insertBlockRegex;
public InsertUpdater() : base()
{
insertBlockRegex = new Regex(InsertContainerBlockName + @" ?= ?\n?", RegexOptions.Compiled | RegexOptions.IgnoreCase | RegexOptions.Multiline, TimeSpan.FromSeconds(1));
}
public override string Process(ref string input)
{
string classname = GetType().Name;
Match match = findRegex.Match(input);
if (match == null || !match.Success)
{
Log($"[{classname}] Found 0 matches and did not update.");
}
while (match != null && match.Success)
{
Log($"[{classname}] Found match to move and insert elsewhere.");
//Increase match length by 1 to include a comma if found.
var length = match.Length;
char afterChar = input[match.Index + length];
if (afterChar == '\n')
{
length += 1;
}
//Remove block from old container block
input = input.Remove(match.Index, length);
//Insert block into new destination container block
input = InsertBlock(ref input, match.Value);
match = findRegex.Match(input);
}
return input;
}
protected string InsertBlock(ref string toUpdate, string foundBlock)
{
int matchIndex;
int matchLength;
//Find the start of the desired destination block
var match = insertBlockRegex.Match(toUpdate);
if(match == null || string.IsNullOrEmpty(match.Value))
{
//No destination block found, lets insert one ourselves to contain the moved block.
var oldLength = toUpdate.Length;
var newDestinationBlockContainer = GetNewInsertBlockContainer();
toUpdate = toUpdate.Insert(oldLength - 1, newDestinationBlockContainer);
//Set match index and length based on the block we just made
matchIndex = oldLength - 1;
matchLength = newDestinationBlockContainer.Length;
}
else
{
matchIndex = match.Index;
matchLength = match.Length;
//Search forward to find the blocks ending ] to get the correct end of this block
int openBrackets = 0;
for(int i = matchIndex + matchLength - 1; i < toUpdate.Length; i++)
{
char c = toUpdate[i];
if(c == '[')
{
openBrackets++;
}
else if(c == ']')
{
openBrackets--;
if (openBrackets == 0)
{
matchLength = i - matchIndex + 1;
break;
}
}
}
}
//Insert new block 1 before the end of the match to be inside the desired section
int insertIndex = matchIndex + matchLength - 1;
string newToInsert = GetBlockToInsert(foundBlock) + "\n";
var newString = toUpdate.Insert(insertIndex, newToInsert);
Log($"[{GetType().Name}] Inserted block successfully.");
return newString;
}
protected abstract string GetBlockToInsert(string foundBlock);
private string GetNewInsertBlockContainer()
{
return $"{InsertContainerBlockName} = \n[\n]";
}
}
}