using System; using System.Collections.Generic; using System.IO; using System.Linq; using static PKHeX.Core.MessageStrings; namespace PKHeX.Core { /// /// Contains extension methods for use with a . /// public static class BoxUtil { /// /// Dumps a folder of files to the . /// /// that is being dumped from. /// Folder to store files. /// Result message from the method. /// Option to save in child folders with the Box Name as the folder name. /// public static bool DumpBoxes(this SaveFile SAV, string path, out string result, bool boxFolders = false) { var boxdata = SAV.BoxData; if (boxdata == null) { result = MsgSaveBoxExportInvalid; return false; } int ctr = 0; foreach (PKM pk in boxdata) { if (pk.Species == 0 || !pk.Valid) continue; ctr++; string fileName = Util.CleanFileName(pk.FileName); string boxfolder = string.Empty; if (boxFolders) { boxfolder = SAV.GetBoxName(pk.Box - 1); Directory.CreateDirectory(Path.Combine(path, boxfolder)); } if (!File.Exists(Path.Combine(Path.Combine(path, boxfolder), fileName))) File.WriteAllBytes(Path.Combine(Path.Combine(path, boxfolder), fileName), pk.DecryptedBoxData); } result = string.Format(MsgSaveBoxExportPathCount, ctr) + Environment.NewLine + path; return true; } /// /// Dumps the to a folder with individual decrypted files. /// /// that is being dumped from. /// Folder to store files. /// Result message from the method. /// Box contents to be dumped. /// public static bool DumpBox(this SaveFile SAV, string path, out string result, int currentBox) { var boxdata = SAV.BoxData; if (boxdata == null) { result = MsgSaveBoxExportInvalid; return false; } int ctr = 0; foreach (PKM pk in boxdata) { if (pk.Species == 0 || !pk.Valid || pk.Box - 1 != currentBox) continue; ctr++; string fileName = Util.CleanFileName(pk.FileName); if (!File.Exists(Path.Combine(path, fileName))) File.WriteAllBytes(Path.Combine(path, fileName), pk.DecryptedBoxData); } result = string.Format(MsgSaveBoxExportPathCount, ctr) + Environment.NewLine + path; return true; } /// /// Loads a folder of files to the . /// /// to load folder to. /// Folder to load files from. Files are only loaded from the top directory. /// Result message from the method. /// First box to start loading to. All prior boxes are not modified. /// Instruction to clear boxes after the starting box. /// Bypass option to not modify properties when setting to Save File. /// Enumerate all files even in sub-folders. /// True if any files are imported. public static bool LoadBoxes(this SaveFile SAV, string path, out string result, int boxStart = 0, bool boxClear = false, bool? noSetb = null, bool all = false) { if (string.IsNullOrWhiteSpace(path) || !Directory.Exists(path)) { result = MsgSaveBoxExportPathInvalid; return false; } var opt = all ? SearchOption.AllDirectories : SearchOption.TopDirectoryOnly; var filepaths = Directory.EnumerateFiles(path, "*.*", opt); return SAV.LoadBoxes(filepaths, out result, boxStart, boxClear, noSetb); } /// /// Loads a folder of files to the . /// /// to load folder to. /// Files to load files from. /// Result message from the method. /// First box to start loading to. All prior boxes are not modified. /// Instruction to clear boxes after the starting box. /// Bypass option to not modify properties when setting to Save File. /// True if any files are imported. public static bool LoadBoxes(this SaveFile SAV, IEnumerable filepaths, out string result, int boxStart = 0, bool boxClear = false, bool? noSetb = null) { int generation = SAV.Generation; var pks = GetPKMsFromPaths(filepaths, generation); return SAV.LoadBoxes(pks, out result, boxStart, boxClear, noSetb); } /// /// Loads a folder of files to the . /// /// to load folder to. /// Gifts to load files from. /// Result message from the method. /// First box to start loading to. All prior boxes are not modified. /// Instruction to clear boxes after the starting box. /// Bypass option to not modify properties when setting to Save File. /// True if any files are imported. public static bool LoadBoxes(this SaveFile SAV, IEnumerable gifts, out string result, int boxStart = 0, bool boxClear = false, bool? noSetb = null) { var pks = gifts.Select(z => z.ConvertToPKM(SAV)); return SAV.LoadBoxes(pks, out result, boxStart, boxClear, noSetb); } /// /// Loads a folder of files to the . /// /// to load folder to. /// Unconverted objects to load. /// Result message from the method. /// First box to start loading to. All prior boxes are not modified. /// Instruction to clear boxes after the starting box. /// Bypass option to not modify properties when setting to Save File. /// True if any files are imported. public static bool LoadBoxes(this SaveFile SAV, IEnumerable pks, out string result, int boxStart = 0, bool boxClear = false, bool? noSetb = null) { if (!SAV.HasBox) { result = MsgSaveBoxFailNone; return false; } var compat = SAV.GetCompatible(pks); if (boxClear) SAV.ClearBoxes(boxStart); int ctr = SAV.ImportPKMs(compat, boxStart, noSetb); if (ctr <= 0) { result = MsgSaveBoxImportNoFiles; return false; } result = string.Format(MsgSaveBoxImportSuccess, ctr); return true; } private static IEnumerable GetPKMsFromPaths(IEnumerable filepaths, int generation) { return filepaths .Where(file => PKX.IsPKM(new FileInfo(file).Length)) .Select(File.ReadAllBytes) .Select(data => PKMConverter.GetPKMfromBytes(data, prefer: generation)) .Where(temp => temp != null); } } }