public void AttachFile_Template(string fileName, EncodedFile.EncodeMode encodeMode) { EngineState s = EngineTests.CreateEngineState(); string dirPath = StringEscaper.Preprocess(s, Path.Combine("%TestBench%", "EncodedFile")); string blankPath = Path.Combine(dirPath, "EncodeFileTests_Blank.script"); string pPath = Path.Combine(dirPath, "EncodeFileTests.script"); File.Copy(blankPath, pPath, true); Script p = s.Project.LoadScriptMonkeyPatch(pPath); string originFile = Path.Combine(dirPath, fileName); p = EncodedFile.AttachFile(p, "FolderExample", fileName, originFile, encodeMode); try { // Check whether file was successfully encoded Assert.IsTrue(p.Sections.ContainsKey("EncodedFolders")); List <string> folders = p.Sections["EncodedFolders"].GetLines(); folders = folders.Where(x => x.Equals(string.Empty, StringComparison.Ordinal) == false).ToList(); Assert.IsTrue(folders.Count == 1); Assert.IsTrue(folders[0].Equals("FolderExample", StringComparison.Ordinal)); Assert.IsTrue(p.Sections.ContainsKey("FolderExample")); List <string> fileInfos = p.Sections["FolderExample"].GetLinesOnce(); fileInfos = fileInfos.Where(x => x.Equals(string.Empty, StringComparison.Ordinal) == false).ToList(); Assert.IsTrue(fileInfos[0].StartsWith($"{fileName}=", StringComparison.Ordinal)); Assert.IsTrue(p.Sections.ContainsKey($"EncodedFile-FolderExample-{fileName}")); List <string> encodedFile = p.Sections[$"EncodedFile-FolderExample-{fileName}"].GetLinesOnce(); encodedFile = encodedFile.Where(x => x.Equals(string.Empty, StringComparison.Ordinal) == false).ToList(); Assert.IsTrue(1 < encodedFile.Count); Assert.IsTrue(encodedFile[0].StartsWith("lines=", StringComparison.Ordinal)); // Check whether file can be successfully extracted byte[] extractDigest; using (MemoryStream ms = EncodedFile.ExtractFile(p, "FolderExample", fileName)) { extractDigest = HashHelper.CalcHash(HashType.SHA256, ms); } byte[] originDigest; using (FileStream fs = new FileStream(originFile, FileMode.Open)) { originDigest = HashHelper.CalcHash(HashType.SHA256, fs); } Assert.IsTrue(originDigest.SequenceEqual(extractDigest)); } finally { File.Delete(pPath); } }
public static List <LogInfo> Encode(EngineState s, CodeCommand cmd) { List <LogInfo> logs = new List <LogInfo>(); CodeInfo_Encode info = cmd.Info.Cast <CodeInfo_Encode>(); string scriptFile = StringEscaper.Preprocess(s, info.ScriptFile); string dirName = StringEscaper.Preprocess(s, info.DirName); string filePath = StringEscaper.Preprocess(s, info.FilePath); EncodedFile.EncodeMode mode = EncodedFile.EncodeMode.ZLib; if (info.Compression != null) { string encodeModeStr = StringEscaper.Preprocess(s, info.Compression); if (encodeModeStr.Equals("None", StringComparison.OrdinalIgnoreCase)) { mode = EncodedFile.EncodeMode.Raw; } else if (encodeModeStr.Equals("Deflate", StringComparison.OrdinalIgnoreCase)) { mode = EncodedFile.EncodeMode.ZLib; } else if (encodeModeStr.Equals("LZMA2", StringComparison.OrdinalIgnoreCase)) { mode = EncodedFile.EncodeMode.XZ; } else { return(LogInfo.LogErrorMessage(logs, $"[{encodeModeStr}] is invalid compression")); } } Script sc = Engine.GetScriptInstance(s, s.CurrentScript.RealPath, scriptFile, out _); // Check srcFileName contains wildcard if (filePath.IndexOfAny(new char[] { '*', '?' }) == -1) { // No Wildcard s.MainViewModel.SetBuildCommandProgress("Encode Progress", 1); try { object progressLock = new object(); IProgress <double> progress = new Progress <double>(x => { lock (progressLock) { s.MainViewModel.BuildCommandProgressValue = x; if (x < EncodedFile.CompReportFactor) // [Stage 1] Compress { s.MainViewModel.BuildCommandProgressText = $"Compressing \"{filePath}\"\r\n({x * 100:0.0}%)"; } else // [Stage 2] Base64 { s.MainViewModel.BuildCommandProgressText = $"Writing \"{filePath}\" to script\r\n({x * 100:0.0}%)"; } } }); EncodedFile.AttachFile(sc, dirName, Path.GetFileName(filePath), filePath, mode, progress); logs.Add(new LogInfo(LogState.Success, $"[{filePath}] was encoded into [{sc.RealPath}]", cmd)); } finally { s.MainViewModel.ResetBuildCommandProgress(); } } else { // With Wildcard // Use FileHelper.GetDirNameEx to prevent ArgumentException of Directory.GetFiles string srcDirToFind = FileHelper.GetDirNameEx(filePath); string[] files = Directory.GetFiles(srcDirToFind, Path.GetFileName(filePath)); // No file will be compressed if (files.Length == 0) { logs.Add(new LogInfo(LogState.Warning, $"Files matching wildcard [{filePath}] were not found", cmd)); return(logs); } s.MainViewModel.SetBuildCommandProgress("Encode Progress", files.Length); try { int i = 0; object progressLock = new object(); IProgress <double> progress = new Progress <double>(x => { lock (progressLock) { s.MainViewModel.BuildCommandProgressText = $"Attaching {filePath}...\r\n({(x + i) * 100:0.0}%)"; s.MainViewModel.BuildCommandProgressValue = x + i; } }); // One or more file will be copied logs.Add(new LogInfo(LogState.Success, $"[{filePath}] will be encoded into [{sc.RealPath}]", cmd)); foreach (string f in files) { EncodedFile.AttachFile(sc, dirName, Path.GetFileName(f), f, mode, progress); logs.Add(new LogInfo(LogState.Success, $"[{f}] encoded ({i + 1}/{files.Length})", cmd)); i += 1; } logs.Add(new LogInfo(LogState.Success, $"[{files.Length}] files copied", cmd)); } finally { s.MainViewModel.ResetBuildCommandProgress(); } } return(logs); }