package emu.grasscutter.data; import emu.grasscutter.Grasscutter; import emu.grasscutter.server.http.handlers.GachaHandler; import emu.grasscutter.tools.Tools; import emu.grasscutter.utils.FileUtils; import emu.grasscutter.utils.Utils; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.InputStream; import java.nio.file.Path; import java.util.List; import static emu.grasscutter.Configuration.DATA; public class DataLoader { /** * Load a data file by its name. If the file isn't found within the /data directory then it will fallback to the default within the jar resources * * @param resourcePath The path to the data file to be loaded. * @return InputStream of the data file. * @throws FileNotFoundException * @see #load(String, boolean) */ public static InputStream load(String resourcePath) throws FileNotFoundException { return load(resourcePath, true); } /** * Load a data file by its name. * * @param resourcePath The path to the data file to be loaded. * @param useFallback If the file does not exist in the /data directory, should it use the default file in the jar? * @return InputStream of the data file. * @throws FileNotFoundException */ public static InputStream load(String resourcePath, boolean useFallback) throws FileNotFoundException { if (Utils.fileExists(DATA(resourcePath))) { // Data is in the resource directory return new FileInputStream(DATA(resourcePath)); } else { if (useFallback) { return FileUtils.readResourceAsStream("/defaults/data/" + resourcePath); } } return null; } public static void CheckAllFiles() { try { List filenames = FileUtils.getPathsFromResource("/defaults/data/"); if (filenames == null) { Grasscutter.getLogger().error("We were unable to locate your default data files."); } else for (Path file : filenames) { String relativePath = String.valueOf(file).split("defaults[\\\\\\/]data[\\\\\\/]")[1]; CheckAndCopyData(relativePath); } } catch (Exception e) { Grasscutter.getLogger().error("An error occurred while trying to check the data folder.", e); } GenerateGachaMappings(); } private static void CheckAndCopyData(String name) { String filePath = Utils.toFilePath(DATA(name)); if (!Utils.fileExists(filePath)) { // Check if file is in subdirectory if (name.indexOf("/") != -1) { String[] path = name.split("/"); String folder = ""; for (int i = 0; i < (path.length - 1); i++) { folder += path[i] + "/"; // Make sure the current folder exists String folderToCreate = Utils.toFilePath(DATA(folder)); if (!Utils.fileExists(folderToCreate)) { Grasscutter.getLogger().info("Creating data folder '" + folder + "'"); Utils.createFolder(folderToCreate); } } } Grasscutter.getLogger().info("Creating default '" + name + "' data"); FileUtils.copyResource("/defaults/data/" + name, filePath); } } private static void GenerateGachaMappings() { if (!Utils.fileExists(GachaHandler.gachaMappings)) { try { Grasscutter.getLogger().info("Creating default '" + GachaHandler.gachaMappings + "' data"); Tools.createGachaMapping(GachaHandler.gachaMappings); } catch (Exception exception) { Grasscutter.getLogger().warn("Failed to create gacha mappings. \n" + exception); } } } }