public static List <LogInfo> TXTDelLine(EngineState s, CodeCommand cmd)
{
List <LogInfo> logs = new List <LogInfo>();
CodeInfo_TXTDelLine info = cmd.Info.Cast <CodeInfo_TXTDelLine>();
string fileName = StringEscaper.Preprocess(s, info.FileName);
string deleteLine = StringEscaper.Preprocess(s, info.DeleteLine);
if (!StringEscaper.PathSecurityCheck(fileName, out string errorMsg))
{
return(LogInfo.LogErrorMessage(logs, errorMsg));
}
if (!File.Exists(fileName))
{
return(LogInfo.LogErrorMessage(logs, $"File [{fileName}] does not exist"));
}
// Detect encoding of text.
Encoding encoding = EncodingHelper.SmartDetectEncoding(fileName, deleteLine);
int count = 0;
string tempPath = FileHelper.GetTempFile();
using (StreamReader r = new StreamReader(fileName, encoding, false))
using (StreamWriter w = new StreamWriter(tempPath, false, encoding))
{
string srcLine;
while ((srcLine = r.ReadLine()) != null)
{
// Strange enough, WB082 treat [deleteLine] as case sensitive string.
if (srcLine.StartsWith(deleteLine, StringComparison.Ordinal))
{
count++;
}
else
{
w.WriteLine(srcLine);
}
}
}
FileHelper.FileReplaceEx(tempPath, fileName);
logs.Add(new LogInfo(LogState.Success, $"Line [{deleteLine}] deleted from [{fileName}]"));
logs.Add(new LogInfo(LogState.Success, $"Deleted [{count}] lines"));
return(logs);
}
public static List <LogInfo> TXTDelLine(EngineState s, CodeCommand cmd)
{
List <LogInfo> logs = new List <LogInfo>();
Debug.Assert(cmd.Info.GetType() == typeof(CodeInfo_TXTDelLine));
CodeInfo_TXTDelLine info = cmd.Info as CodeInfo_TXTDelLine;
string fileName = StringEscaper.Preprocess(s, info.FileName);
string deleteLine = StringEscaper.Preprocess(s, info.DeleteLine);
if (StringEscaper.PathSecurityCheck(fileName, out string errorMsg) == false)
{
logs.Add(new LogInfo(LogState.Error, errorMsg));
return(logs);
}
if (File.Exists(fileName) == false)
{
logs.Add(new LogInfo(LogState.Error, $"File [{fileName}] not exists"));
return(logs);
}
Encoding encoding = FileHelper.DetectTextEncoding(fileName);
int i = 0;
string tempPath = Path.GetTempFileName();
using (StreamReader reader = new StreamReader(fileName, encoding))
using (StreamWriter writer = new StreamWriter(tempPath, false, encoding))
{
string srcLine;
while ((srcLine = reader.ReadLine()) != null)
{
// Strange enough, WB082 treat [deleteLine] as case sensitive string.
if (srcLine.StartsWith(deleteLine, StringComparison.Ordinal))
{
i++;
continue;
}
writer.WriteLine(srcLine);
}
}
FileHelper.FileReplaceEx(tempPath, fileName);
logs.Add(new LogInfo(LogState.Success, $"Deleted [{i}] lines from [{fileName}]"));
return(logs);
}
public static List <LogInfo> TXTDelLineOp(EngineState s, CodeCommand cmd)
{
List <LogInfo> logs = new List <LogInfo>();
CodeInfo_TXTDelLineOp infoOp = cmd.Info.Cast <CodeInfo_TXTDelLineOp>();
CodeInfo_TXTDelLine firstInfo = infoOp.Infos[0];
string fileName = StringEscaper.Preprocess(s, firstInfo.FileName);
if (!StringEscaper.PathSecurityCheck(fileName, out string errorMsg))
{
return(LogInfo.LogErrorMessage(logs, errorMsg));
}
if (!File.Exists(fileName))
{
return(LogInfo.LogErrorMessage(logs, $"File [{fileName}] does not exist"));
}
List <(CodeCommand, string)> prepDeleteLine = new List <(CodeCommand, string)>(infoOp.Cmds.Count);
foreach (CodeCommand subCmd in infoOp.Cmds)
{
CodeInfo_TXTDelLine info = subCmd.Info.Cast <CodeInfo_TXTDelLine>();
string deleteLine = StringEscaper.Preprocess(s, info.DeleteLine);
prepDeleteLine.Add((subCmd, deleteLine));
}
// Detect encoding of text.
Encoding encoding = EncodingHelper.SmartDetectEncoding(fileName, prepDeleteLine.Select(t => t.Item2));
int count = 0;
string tempPath = FileHelper.GetTempFile();
using (StreamReader r = new StreamReader(fileName, encoding, false))
using (StreamWriter w = new StreamWriter(tempPath, false, encoding))
{
string srcLine;
while ((srcLine = r.ReadLine()) != null)
{
bool writeLine = true;
foreach ((CodeCommand _, string deleteLine) in prepDeleteLine)
{
// Strange enough, WB082 treat [deleteLine] as case sensitive string.
if (srcLine.StartsWith(deleteLine, StringComparison.Ordinal))
{
writeLine = false;
count++;
break;
}
}
if (writeLine)
{
w.WriteLine(srcLine);
}
}
}
FileHelper.FileReplaceEx(tempPath, fileName);
foreach ((CodeCommand subCmd, string deleteLine) in prepDeleteLine)
{
logs.Add(new LogInfo(LogState.Success, $"Line [{deleteLine}] deleted from [{fileName}]", subCmd));
}
logs.Add(new LogInfo(LogState.Success, $"Deleted [{count}] lines from [{fileName}]"));
return(logs);
}
private static List <CodeCommand> InternalOptimize(List <CodeCommand> cmds)
{
List <CodeCommand> optimized = new List <CodeCommand>(cmds.Count);
Dictionary <CodeType, List <CodeCommand> > opDict = OptimizedCodeTypes.ToDictionary(x => x, x => new List <CodeCommand>(cmds.Count / 2));
CodeType s = CodeType.None;
foreach (CodeCommand cmd in cmds)
{
bool loopAgain;
do
{
loopAgain = false;
switch (s)
{
#region Default
case CodeType.None:
switch (cmd.Type)
{
case CodeType.TXTAddLine:
case CodeType.TXTReplace:
case CodeType.TXTDelLine:
case CodeType.IniRead:
case CodeType.IniWrite:
case CodeType.IniDelete:
case CodeType.IniReadSection:
case CodeType.IniAddSection:
case CodeType.IniDeleteSection:
case CodeType.IniWriteTextLine:
case CodeType.Visible:
case CodeType.ReadInterface:
case CodeType.WriteInterface:
case CodeType.WimExtract:
s = cmd.Type;
opDict[cmd.Type].Add(cmd);
break;
case CodeType.WimPathAdd:
case CodeType.WimPathDelete:
case CodeType.WimPathRename:
s = CodeType.WimPathAdd; // Use WimPathAdd as representative
opDict[CodeType.WimPathAdd].Add(cmd);
break;
default:
optimized.Add(cmd);
break;
}
break;
#endregion
#region TXTAddLine
case CodeType.TXTAddLine:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_TXTAddLine), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.TXTAddLine:
{
CodeInfo_TXTAddLine firstInfo = opDict[s][0].Info.Cast <CodeInfo_TXTAddLine>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region TXTReplace
case CodeType.TXTReplace:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_TXTReplace));
switch (cmd.Type)
{
case CodeType.TXTReplace:
{
CodeInfo_TXTReplace firstInfo = opDict[s][0].Info.Cast <CodeInfo_TXTReplace>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region TXTDelLine
case CodeType.TXTDelLine:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_TXTDelLine), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.TXTDelLine:
{
CodeInfo_TXTDelLine firstInfo = opDict[s][0].Info.Cast <CodeInfo_TXTDelLine>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IniRead
case CodeType.IniRead:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniRead), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.IniRead:
{
CodeInfo_IniRead firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniRead>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IniWrite
case CodeType.IniWrite:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniWrite), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.IniWrite:
{
CodeInfo_IniWrite firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniWrite>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IinDelete
case CodeType.IniDelete:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniDelete), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.IniDelete:
{
CodeInfo_IniDelete firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniDelete>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IniReadSection
case CodeType.IniReadSection:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniReadSection));
switch (cmd.Type)
{
case CodeType.IniReadSection:
{
CodeInfo_IniReadSection firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniReadSection>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IniAddSection
case CodeType.IniAddSection:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniAddSection), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.IniAddSection:
{
CodeInfo_IniAddSection firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniAddSection>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IniDeleteSection
case CodeType.IniDeleteSection:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniDeleteSection), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.IniDeleteSection:
{
CodeInfo_IniDeleteSection firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniDeleteSection>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region IniWriteTextLine
case CodeType.IniWriteTextLine:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_IniWriteTextLine), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.IniWriteTextLine:
{
CodeInfo_IniWriteTextLine firstInfo = opDict[s][0].Info.Cast <CodeInfo_IniWriteTextLine>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region Visible
case CodeType.Visible:
switch (cmd.Type)
{
case CodeType.Visible:
opDict[s].Add(cmd);
break;
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region ReadInterface
case CodeType.ReadInterface:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_ReadInterface), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.ReadInterface:
{
CodeInfo_ReadInterface firstInfo = opDict[s][0].Info.Cast <CodeInfo_ReadInterface>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region WriteInterface
case CodeType.WriteInterface:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_WriteInterface), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.WriteInterface:
{
CodeInfo_WriteInterface firstInfo = opDict[s][0].Info.Cast <CodeInfo_WriteInterface>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region WimExtract
case CodeType.WimExtract:
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_WimExtract), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.WimExtract:
{
CodeInfo_WimExtract firstInfo = opDict[s][0].Info.Cast <CodeInfo_WimExtract>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region WimPath Series
case CodeType.WimPathAdd: // Use WimPathAdd as a representative of WimPath{Add, Delete, Rename}
Debug.Assert(opDict[s][0].Info.GetType() == typeof(CodeInfo_WimPathAdd) ||
opDict[s][0].Info.GetType() == typeof(CodeInfo_WimPathDelete) ||
opDict[s][0].Info.GetType() == typeof(CodeInfo_WimPathRename), "Invalid CodeInfo");
switch (cmd.Type)
{
case CodeType.WimPathAdd:
case CodeType.WimPathDelete:
case CodeType.WimPathRename:
{
CodeCommand firstCmd = opDict[s][0];
if (firstCmd.Type == CodeType.WimPathAdd)
{
CodeInfo_WimPathAdd firstInfo = opDict[s][0].Info.Cast <CodeInfo_WimPathAdd>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
}
else if (firstCmd.Type == CodeType.WimPathDelete)
{
CodeInfo_WimPathDelete firstInfo = opDict[s][0].Info.Cast <CodeInfo_WimPathDelete>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
}
else if (firstCmd.Type == CodeType.WimPathRename)
{
CodeInfo_WimPathRename firstInfo = opDict[s][0].Info.Cast <CodeInfo_WimPathRename>();
if (firstInfo.OptimizeCompare(cmd.Info))
{
opDict[s].Add(cmd);
}
else
{
goto default;
}
}
break;
}
case CodeType.Comment: // Remove comments
break;
default: // Optimize them
FinalizeSequence(s, opDict[s]);
s = CodeType.None;
loopAgain = true;
break;
}
break;
#endregion
#region Error
default:
throw new InternalException("Internal Logic Error at CodeOptimizer.InternalOptimize()");
#endregion
}
}while (loopAgain);
}
#region Finish
foreach (var kv in opDict)
{
FinalizeSequence(kv.Key, kv.Value);
}
#endregion
#region FinalizeSequence
void FinalizeSequence(CodeType state, List <CodeCommand> cmdSeq)
{
CodeCommand opCmd;
if (cmdSeq.Count == 1)
{
opCmd = cmdSeq[0];
}
else if (1 < cmdSeq.Count)
{
opCmd = PackCommand(state, new List <CodeCommand>(cmdSeq));
}
else // if (cmds.Count <= 0)
{
return;
}
Debug.Assert(opCmd != null, "Internal Logic Error in CodeOptimizer.Optimize");
optimized.Add(opCmd);
cmdSeq.Clear();
}
#endregion
return(optimized);
}
