diff --git a/src/main/scala/io/izzel/taboolib/module/db/local/SecuredFile.java b/src/main/scala/io/izzel/taboolib/module/db/local/SecuredFile.java new file mode 100644 index 0000000..70ec370 --- /dev/null +++ b/src/main/scala/io/izzel/taboolib/module/db/local/SecuredFile.java @@ -0,0 +1,82 @@ +1package io.izzel.taboolib.module.db.local; + +import io.izzel.taboolib.util.Files; +import org.bukkit.Bukkit; +import org.bukkit.configuration.InvalidConfigurationException; +import org.bukkit.configuration.file.YamlConfiguration; + +import java.io.File; +import java.io.IOException; +import java.text.SimpleDateFormat; +import java.util.logging.Level; + +/** + * @Author sky + * @Since 2020-02-28 11:14 + */ +public class SecuredFile extends YamlConfiguration { + + @Override + public synchronized void set(String path, Object value) { + super.set(path, value); + } + + @Override + public synchronized void save(File file) throws IOException { + super.save(file); + } + + @Override + public synchronized void save(String file) throws IOException { + super.save(file); + } + + /** + * 如果文件读取失败则创建备份 + * 以防出现不可逆的损伤 + */ + @Override + public void load(File file) throws InvalidConfigurationException { + String content = Files.readFromFile(file); + try { + super.loadFromString(Files.readFromFile(file)); + } catch (InvalidConfigurationException t) { + Files.copy(file, new File(file.getParent(), file.getName() + "_" + new SimpleDateFormat("yyyyMMddHHmmss").format(System.currentTimeMillis()) + ".bak")); + throw t; + } + } + + /** + * 如果文本读取失败则打印到日志 + * 以防出现不可逆的损伤 + */ + @Override + public void loadFromString(String contents) throws InvalidConfigurationException { + try { + super.loadFromString(contents); + } catch (InvalidConfigurationException t) { + System.out.println("Source: \n" + contents); + throw t; + } + } + + public static SecuredFile loadConfiguration(String contents) { + SecuredFile config = new SecuredFile(); + try { + config.loadFromString(contents); + } catch (Throwable t) { + Bukkit.getLogger().log(Level.SEVERE, "Cannot load source", t); + } + return config; + } + + public static SecuredFile loadConfiguration(File file) { + SecuredFile config = new SecuredFile(); + try { + config.load(file); + } catch (Throwable t) { + Bukkit.getLogger().log(Level.SEVERE, "Cannot load " + file, t); + } + return config; + } +} diff --git a/src/main/scala/io/izzel/taboolib/util/Files.java b/src/main/scala/io/izzel/taboolib/util/Files.java index 035f5b7..179338d 100644 --- a/src/main/scala/io/izzel/taboolib/util/Files.java +++ b/src/main/scala/io/izzel/taboolib/util/Files.java @@ -2,6 +2,7 @@ package io.izzel.taboolib.util; import io.izzel.taboolib.TabooLib; import io.izzel.taboolib.common.plugin.InternalPlugin; +import io.izzel.taboolib.module.db.local.SecuredFile; import io.izzel.taboolib.module.inject.TSchedule; import io.izzel.taboolib.util.plugin.PluginUtils; import org.bukkit.configuration.file.FileConfiguration; @@ -342,7 +343,7 @@ public class Files { } public static FileConfiguration decodeYAML(String args) { - return YamlConfiguration.loadConfiguration(new StringReader(Base64Coder.decodeString(args))); + return SecuredFile.loadConfiguration(Base64Coder.decodeString(args)); } public static FileConfiguration load(File file) { @@ -350,12 +351,7 @@ public class Files { } public static YamlConfiguration loadYaml(File file) { - try { - return YamlConfiguration.loadConfiguration(file); - } catch (Exception e) { - e.printStackTrace(); - } - return new YamlConfiguration(); + return SecuredFile.loadConfiguration(file); } public static String getFileHash(File file, String algorithm) {