PermissionsTime/src/main/gg/frog/mc/base/utils/config/PluginConfig.java

package gg.frog.mc.base.utils.config;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.logging.Level;

import org.bukkit.command.CommandSender;
import org.bukkit.configuration.InvalidConfigurationException;
import org.bukkit.configuration.MemorySection;
import org.bukkit.configuration.file.FileConfiguration;
import org.bukkit.configuration.file.YamlConfiguration;

import com.google.common.base.Charsets;

import gg.frog.mc.base.PluginMain;
import gg.frog.mc.base.config.PluginCfg;
import gg.frog.mc.base.utils.StrUtil;

/**
 * 配置操作
 * 
 * @author QiaoPengyu
 *
 */
public abstract class PluginConfig {

	protected PluginMain pm;
	private FileConfiguration config = null;
	private File folder = null;
	private String fileName = null;
	private File configFile = null;

	protected PluginConfig(PluginMain pm) {
		this.pm = pm;
		initConfig(pm.getDataFolder(), "config.yml");
	}

	protected PluginConfig(String fileName, PluginMain pm) {
		this.pm = pm;
		initConfig(pm.getDataFolder(), fileName);
	}

	protected PluginConfig(File folder, String fileName, PluginMain pm) {
		this.pm = pm;
		initConfig(folder, fileName);
	}

	/**
	 * 初始化
	 * 
	 * @param folder
	 * @param fileName
	 */
	private void initConfig(File folder, String fileName) {
		this.folder = folder;
		this.fileName = fileName;
		configFile = new File(folder, fileName);
		if (!configFile.exists()) {
			getConfig(folder, fileName).options().copyDefaults(true);
			init();
			saveAndReloadConfig();
		} else {
			reloadConfig();
		}
	}

	/**
	 * 首次生成文件调用
	 */
	protected abstract void init();
	
	/**
	 * 加载配置后调用
	 */
	protected abstract void loadToDo(CommandSender sender);

	/**
	 * 获取配置(首次)
	 * 
	 * @return
	 */
	private FileConfiguration getConfig(File folder, String fileName) {
		if (config == null) {
			reloadConfig(folder, fileName, true);
		}
		return config;
	}

	/**
	 * 获取配置
	 * 
	 * @return
	 */
	protected FileConfiguration getConfig() {
		return config;
	}

	/**
	 * 保存配置
	 */
	public void saveConfig() {
		try {
			getConfig().save(configFile);
		} catch (IOException ex) {
			PluginMain.LOG.log(Level.SEVERE, StrUtil.messageFormat(PluginCfg.PLUGIN_PREFIX + "Could not save config to " + configFile), ex);
		}
	}

	public void saveAndReloadConfig() {
		try {
			getConfig().save(configFile);
			reloadConfig();
		} catch (IOException ex) {
			PluginMain.LOG.log(Level.SEVERE, StrUtil.messageFormat(PluginCfg.PLUGIN_PREFIX + "Could not save config to " + configFile), ex);
		}
	}

	/**
	 * 配置重载
	 */
	public void reloadConfig() {
		reloadConfig(folder, fileName, false);
	}

	/**
	 * 配置重载
	 */
	public void reloadConfig(CommandSender sender) {
		reloadConfig(folder, fileName, false, sender);
	}

	/**
	 * 配置重载
	 */
	private void reloadConfig(File folder, String fileName, boolean useRes) {
		reloadConfig(folder, fileName, useRes, null);
	}

	/**
	 * 配置重载
	 */
	private void reloadConfig(File folder, String fileName, boolean useRes, CommandSender sender) {
		YamlConfiguration tempConfig = new YamlConfiguration();
		try {
			tempConfig.load(configFile);
			config = tempConfig;
		} catch (FileNotFoundException e) {
		} catch (IOException | InvalidConfigurationException e1) {
			tempConfig = null;
			e1.printStackTrace();
		}
		if (config == null) {
			config = new YamlConfiguration();
		}
		if (useRes) {
			final InputStream defConfigStream = pm.getResource(fileName);
			if (defConfigStream == null) {
				return;
			}
			config.setDefaults(YamlConfiguration.loadConfiguration(new InputStreamReader(defConfigStream, Charsets.UTF_8)));
		}
		loadToDo(sender);
		// if (tempConfig != null) {
		// saveConfig();
		// }
	}

	protected void setObj(String path, Map<String, ? extends IConfigBean> o) {
		for (Entry<String, ? extends IConfigBean> configBean : o.entrySet()) {
			setObj(path + "." + configBean.getKey(), configBean.getValue());
		}
	}

	protected void setObj(String path, IConfigBean o) {
		getConfig().set(path, o.toConfig());
	}

	protected <T extends IConfigBean> Map<String, T> getObjMap(String path, Class<T> clazz) {
		Map<String, T> map = new HashMap<>();
		MemorySection configMap = (MemorySection) getConfig().get(path);
		if (configMap != null) {
			for (String key : configMap.getKeys(false)) {
				T bean = getObj(path + "." + key, clazz);
				if (bean != null) {
					map.put(key, bean);
				}
			}
		}
		return map;
	}

	protected <T extends IConfigBean> T getObj(String path, Class<T> clazz) {
		Object beanConfig = getConfig().get(path);
		if (beanConfig != null && beanConfig instanceof MemorySection) {
			try {
				T bean = clazz.newInstance();
				bean.toConfigBean((MemorySection) beanConfig);
				return bean;
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
		return null;
	}

	protected String setGetDefault(String path, String def) {
		if (!getConfig().contains(path)) {
			getConfig().set(path, def);
			return def;
		}
		return getConfig().getString(path);
	}

	protected int setGetDefault(String path, int def) {
		if (!getConfig().contains(path)) {
			getConfig().set(path, def);
			return def;
		}
		return getConfig().getInt(path);
	}

	protected boolean setGetDefault(String path, boolean def) {
		if (!getConfig().contains(path)) {
			getConfig().set(path, def);
			return def;
		}
		return getConfig().getBoolean(path);
	}
}