public static List <LogInfo> IniReadSection(EngineState s, CodeCommand cmd) { List <LogInfo> logs = new List <LogInfo>(); CodeInfo_IniReadSection info = cmd.Info.Cast <CodeInfo_IniReadSection>(); string fileName = StringEscaper.Preprocess(s, info.FileName); string section = StringEscaper.Preprocess(s, info.Section); string delim = "|"; if (info.Delim != null) { delim = StringEscaper.Preprocess(s, info.Delim); } Debug.Assert(fileName != null, $"{nameof(fileName)} != null"); Debug.Assert(section != null, $"{nameof(section)} != null"); if (section.Length == 0) { return(LogInfo.LogErrorMessage(logs, "Section name cannot be empty")); } IniKey[] keys = IniReadWriter.ReadSection(fileName, section); if (keys != null) { List <string> kvList = new List <string>(keys.Length * 2); foreach (IniKey k in keys) { kvList.Add(k.Key); kvList.Add(k.Value); } string destStr = StringEscaper.PackListStr(kvList, delim); logs.Add(new LogInfo(LogState.Success, $"Section [{section}] read from [{fileName}]")); string escapedValue = StringEscaper.Escape(destStr, false, true); List <LogInfo> varLogs = Variables.SetVariable(s, info.DestVar, escapedValue, false, false, false); logs.AddRange(varLogs); } else { logs.Add(new LogInfo(LogState.Ignore, $"Section [{section}] does not exist in [{fileName}]")); List <LogInfo> varLogs = Variables.SetVariable(s, info.DestVar, string.Empty, false, false, false); logs.AddRange(varLogs); } return(logs); }
public static List <LogInfo> IniReadSection(EngineState s, CodeCommand cmd) { List <LogInfo> logs = new List <LogInfo>(); Debug.Assert(cmd.Info.GetType() == typeof(CodeInfo_IniReadSection)); CodeInfo_IniReadSection info = cmd.Info as CodeInfo_IniReadSection; string fileName = StringEscaper.Preprocess(s, info.FileName); string section = StringEscaper.Preprocess(s, info.Section); if (section.Equals(string.Empty, StringComparison.Ordinal)) { throw new ExecuteException("Section name cannot be empty"); } IniKey[] keys = Ini.ReadSection(fileName, section); if (keys != null) { StringBuilder b = new StringBuilder(); b.AppendLine($"[{section}]"); foreach (IniKey k in keys) { b.AppendLine($"{k.Key}={k.Value}"); } logs.Add(new LogInfo(LogState.Success, $"Section [{section}] read in [{fileName}]")); string escapedValue = StringEscaper.Escape(b.ToString(), false, true); List <LogInfo> varLogs = Variables.SetVariable(s, info.DestVar, escapedValue, false, false, false); logs.AddRange(varLogs); } else { logs.Add(new LogInfo(LogState.Ignore, $"Section [{section}] does not exist in [{fileName}]")); List <LogInfo> varLogs = Variables.SetVariable(s, info.DestVar, string.Empty, false, false, false); logs.AddRange(varLogs); } return(logs); }
public static List <LogInfo> IniReadSectionOp(EngineState s, CodeCommand cmd) { List <LogInfo> logs = new List <LogInfo>(); Debug.Assert(cmd.Info.GetType() == typeof(CodeInfo_IniReadSectionOp)); CodeInfo_IniReadSectionOp infoOp = cmd.Info as CodeInfo_IniReadSectionOp; string fileName = StringEscaper.Preprocess(s, infoOp.Infos[0].FileName); if (StringEscaper.PathSecurityCheck(fileName, out string errorMsg) == false) { logs.Add(new LogInfo(LogState.Error, errorMsg)); return(logs); } string[] sections = new string[infoOp.Cmds.Count]; string[] destVars = new string[infoOp.Cmds.Count]; for (int i = 0; i < sections.Length; i++) { CodeInfo_IniReadSection info = infoOp.Infos[i]; string section = StringEscaper.Preprocess(s, info.Section); if (section.Equals(string.Empty, StringComparison.Ordinal)) { throw new ExecuteException("Section name cannot be empty"); } sections[i] = section; destVars[i] = info.DestVar; } Dictionary <string, IniKey[]> keyDict = Ini.ReadSections(fileName, sections); int successCount = 0; for (int i = 0; i < sections.Length; i++) { string section = sections[i]; IniKey[] keys = keyDict[section]; CodeCommand subCmd = infoOp.Cmds[i]; if (keys != null) { StringBuilder b = new StringBuilder(); b.AppendLine($"[{section}]"); foreach (IniKey k in keys) { b.AppendLine($"{k.Key}={k.Value}"); } logs.Add(new LogInfo(LogState.Success, $"Section [{section}] read", subCmd)); string escapedValue = StringEscaper.Escape(b.ToString(), false, true); List <LogInfo> varLogs = Variables.SetVariable(s, destVars[i], escapedValue, false, false, false); LogInfo.AddCommand(varLogs, subCmd); logs.AddRange(varLogs); } else { logs.Add(new LogInfo(LogState.Ignore, $"Section [{section}] does not exist", subCmd)); List <LogInfo> varLogs = Variables.SetVariable(s, destVars[i], string.Empty, false, false, false); LogInfo.AddCommand(varLogs, subCmd); logs.AddRange(varLogs); } } logs.Add(new LogInfo(LogState.Success, $"Read [{successCount}] sections from [{fileName}]", cmd)); 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); }
public static List <LogInfo> IniReadSectionOp(EngineState s, CodeCommand cmd) { List <LogInfo> logs = new List <LogInfo>(); CodeInfo_IniReadSectionOp infoOp = cmd.Info.Cast <CodeInfo_IniReadSectionOp>(); string fileName = StringEscaper.Preprocess(s, infoOp.Infos[0].FileName); Debug.Assert(fileName != null, $"{nameof(fileName)} != null"); string[] sections = new string[infoOp.Cmds.Count]; string[] destVars = new string[infoOp.Cmds.Count]; string[] delims = new string[infoOp.Cmds.Count]; for (int i = 0; i < sections.Length; i++) { CodeInfo_IniReadSection info = infoOp.Infos[i]; string section = StringEscaper.Preprocess(s, info.Section); if (section.Length == 0) { return(LogInfo.LogErrorMessage(logs, "Section name cannot be empty")); } sections[i] = section; destVars[i] = info.DestVar; delims[i] = "|"; if (info.Delim != null) { delims[i] = StringEscaper.Preprocess(s, info.Delim); } } Dictionary <string, IniKey[]> keyDict = IniReadWriter.ReadSections(fileName, sections); int successCount = 0; for (int i = 0; i < sections.Length; i++) { string section = sections[i]; string delim = delims[i]; IniKey[] keys = keyDict[section]; CodeCommand subCmd = infoOp.Cmds[i]; if (keys != null) { List <string> kvList = new List <string>(keys.Length * 2); foreach (IniKey k in keys) { kvList.Add(k.Key); kvList.Add(k.Value); } string destStr = StringEscaper.PackListStr(kvList, delim); logs.Add(new LogInfo(LogState.Success, $"Section [{section}] read", subCmd)); string escapedValue = StringEscaper.Escape(destStr, false, true); List <LogInfo> varLogs = Variables.SetVariable(s, destVars[i], escapedValue, false, false, false); LogInfo.AddCommand(varLogs, subCmd); logs.AddRange(varLogs); } else { logs.Add(new LogInfo(LogState.Ignore, $"Section [{section}] does not exist", subCmd)); List <LogInfo> varLogs = Variables.SetVariable(s, destVars[i], string.Empty, false, false, false); LogInfo.AddCommand(varLogs, subCmd); logs.AddRange(varLogs); } } logs.Add(new LogInfo(LogState.Success, $"Read [{successCount}] sections from [{fileName}]", cmd)); return(logs); }