public static List <LogInfo> ExtractAllFiles(EngineState s, CodeCommand cmd)
{
List <LogInfo> logs = new List <LogInfo>();
Debug.Assert(cmd.Info.GetType() == typeof(CodeInfo_ExtractAllFiles));
CodeInfo_ExtractAllFiles info = cmd.Info as CodeInfo_ExtractAllFiles;
string pluginFile = StringEscaper.Preprocess(s, info.PluginFile);
string dirName = StringEscaper.Preprocess(s, info.DirName);
string destDir = StringEscaper.Preprocess(s, info.DestDir);
Plugin p = Engine.GetPluginInstance(s, cmd, s.CurrentPlugin.FullPath, pluginFile, out bool inCurrentPlugin);
if (StringEscaper.PathSecurityCheck(destDir, out string errorMsg) == false)
{
logs.Add(new LogInfo(LogState.Error, errorMsg));
return(logs);
}
List <string> dirs = p.Sections["EncodedFolders"].Lines;
bool dirNameValid = dirs.Any(d => d.Equals(dirName, StringComparison.OrdinalIgnoreCase));
if (dirNameValid == false)
{
throw new ExecuteException($"Directory [{dirName}] not exists in [{pluginFile}]");
}
if (!Directory.Exists(destDir))
{
if (File.Exists(destDir))
{
logs.Add(new LogInfo(LogState.Error, $"File [{destDir}] is not a directory"));
return(logs);
}
else
{
logs.Add(new LogInfo(LogState.Error, $"Directory [{destDir}] does not exist"));
return(logs);
}
}
List <string> lines = p.Sections[dirName].Lines;
Dictionary <string, string> fileDict = Ini.ParseIniLinesIniStyle(lines);
foreach (string file in fileDict.Keys)
{
using (MemoryStream ms = EncodedFile.ExtractFile(p, dirName, file))
using (FileStream fs = new FileStream(Path.Combine(destDir, file), FileMode.Create, FileAccess.Write))
{
ms.Position = 0;
ms.CopyTo(fs);
}
}
logs.Add(new LogInfo(LogState.Success, $"Encoded folder [{dirName}] extracted to [{destDir}]"));
return(logs);
}
public static List <LogInfo> ExtractAllFiles(EngineState s, CodeCommand cmd)
{
List <LogInfo> logs = new List <LogInfo>();
CodeInfo_ExtractAllFiles info = cmd.Info.Cast <CodeInfo_ExtractAllFiles>();
string scriptFile = StringEscaper.Preprocess(s, info.ScriptFile);
string dirName = StringEscaper.Preprocess(s, info.DirName);
string destDir = StringEscaper.Preprocess(s, info.DestDir);
Script sc = Engine.GetScriptInstance(s, s.CurrentScript.RealPath, scriptFile, out _);
// Check if encoded file exist
if (!EncodedFile.ContainsFolder(sc, dirName))
{
return(LogInfo.LogErrorMessage(logs, $"Encoded folder [{dirName}] not found in script [{sc.RealPath}]."));
}
// Filter dest path
if (!StringEscaper.PathSecurityCheck(destDir, out string errorMsg))
{
return(LogInfo.LogErrorMessage(logs, errorMsg));
}
string[] dirs = sc.Sections[ScriptSection.Names.EncodedFolders].Lines;
if (!dirs.Any(d => d.Equals(dirName, StringComparison.OrdinalIgnoreCase)))
{
return(LogInfo.LogErrorMessage(logs, $"Directory [{dirName}] not exists in [{scriptFile}]"));
}
if (!Directory.Exists(destDir))
{
if (File.Exists(destDir))
{
return(LogInfo.LogErrorMessage(logs, $"File [{destDir}] is not a directory"));
}
Directory.CreateDirectory(destDir);
}
string[] lines = sc.Sections[dirName].Lines;
Dictionary <string, string> fileDict = IniReadWriter.ParseIniLinesIniStyle(lines);
int fileCount = fileDict.Count;
s.MainViewModel.SetBuildCommandProgress("ExtractAndRun Progress", fileCount);
try
{
int i = 0;
foreach (string file in fileDict.Keys)
{
object progressLock = new object();
IProgress <double> progress = new Progress <double>(x =>
{
lock (progressLock)
{
s.MainViewModel.BuildCommandProgressText = $"Decompressing \"{file}\"\r\n({(x + i) * 100 / fileCount:0.0}%)";
s.MainViewModel.BuildCommandProgressValue = x + i;
}
});
using (FileStream fs = new FileStream(Path.Combine(destDir, file), FileMode.Create, FileAccess.Write))
{
EncodedFile.ExtractFile(sc, dirName, file, fs, progress);
}
i += 1;
}
}
finally
{
s.MainViewModel.ResetBuildCommandProgress();
}
logs.Add(new LogInfo(LogState.Success, $"Encoded folder [{dirName}] was extracted to [{destDir}]"));
return(logs);
}
