package org.anjocaido.groupmanager; import java.io.File; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.io.OutputStreamWriter; import java.io.UnsupportedEncodingException; import java.util.Collection; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Set; import java.util.logging.Level; import org.anjocaido.groupmanager.data.Group; import org.anjocaido.groupmanager.utils.PermissionCheckResult; import org.anjocaido.groupmanager.utils.Tasks; import org.bukkit.configuration.MemorySection; import org.bukkit.configuration.file.YamlConfiguration; import org.yaml.snakeyaml.DumperOptions; import org.yaml.snakeyaml.Yaml; /** * @author ElgarL * */ public class GlobalGroups { private GroupManager plugin; private YamlConfiguration GGroups; private Map groups; protected long timeStampGroups = 0; protected boolean haveGroupsChanged = false; protected File GlobalGroupsFile = null; public GlobalGroups(GroupManager plugin) { this.plugin = plugin; load(); } /** * @return the haveGroupsChanged */ public boolean haveGroupsChanged() { if (this.haveGroupsChanged) { return true; } for (Group g : groups.values()) { if (g.isChanged()) { return true; } } return false; } /** * @return the timeStampGroups */ public long getTimeStampGroups() { return timeStampGroups; } /** * @param timeStampGroups the timeStampGroups to set */ protected void setTimeStampGroups(long timeStampGroups) { this.timeStampGroups = timeStampGroups; } /** * @param haveGroupsChanged * the haveGroupsChanged to set */ public void setGroupsChanged(boolean haveGroupsChanged) { this.haveGroupsChanged = haveGroupsChanged; } @SuppressWarnings("unchecked") public void load() { GGroups = new YamlConfiguration(); GroupManager.setLoaded(false); // READ globalGroups FILE if (GlobalGroupsFile == null) GlobalGroupsFile = new File(plugin.getDataFolder(), "globalgroups.yml"); if (!GlobalGroupsFile.exists()) { try { // Create a new file if it doesn't exist. Tasks.copy(plugin.getResourceAsStream("globalgroups.yml"), GlobalGroupsFile); } catch (IOException ex) { GroupManager.logger.log(Level.SEVERE, null, ex); } } try { GGroups.load(GlobalGroupsFile); } catch (Exception ex) { throw new IllegalArgumentException("The following file couldn't pass on Parser.\n" + GlobalGroupsFile.getPath(), ex); } // Read all global groups Map allGroups = (Map) GGroups.getConfigurationSection("groups").getValues(false); // Load each groups permissions list. if (allGroups != null) { // Clear out old groups resetGlobalGroups(); for (String groupName : allGroups.keySet()) { Group newGroup = new Group(groupName.toLowerCase()); Object element; // Permission nodes element = GGroups.get("groups." + groupName + ".permissions"); if (element != null) if (element instanceof List) { for (String node : (List) element) { newGroup.addPermission(node); } } else if (element instanceof String) { newGroup.addPermission((String) element); } else throw new IllegalArgumentException("Unknown type of permission node for global group: " + groupName); // Info nodes element = GGroups.get("groups." + groupName + ".info"); if (element != null) if (element instanceof MemorySection) { Map vars = new HashMap(); for (String key : ((MemorySection) element).getKeys(false)) { vars.put(key, ((MemorySection) element).get(key)); } newGroup.setVariables(vars); } else throw new IllegalArgumentException("Unknown type of info node for global group: " + groupName); // Push a new group addGroup(newGroup); } } removeGroupsChangedFlag(); setTimeStampGroups(GlobalGroupsFile.lastModified()); GroupManager.setLoaded(true); //GlobalGroupsFile = null; } /** * Write the globalgroups.yml file */ public void writeGroups(boolean overwrite) { //File GlobalGroupsFile = new File(plugin.getDataFolder(), "globalgroups.yml"); if (haveGroupsChanged()) { if (overwrite || (!overwrite && (getTimeStampGroups() >= GlobalGroupsFile.lastModified()))) { Map root = new HashMap(); Map groupsMap = new HashMap(); root.put("groups", groupsMap); for (String groupKey : groups.keySet()) { Group group = groups.get(groupKey); // Group header Map aGroupMap = new HashMap(); groupsMap.put(group.getName(), aGroupMap); // Info nodes Map infoMap = new HashMap(); aGroupMap.put("info", infoMap); for (String infoKey : group.getVariables().getVarKeyList()) { infoMap.put(infoKey, group.getVariables().getVarObject(infoKey)); } // Permission nodes aGroupMap.put("permissions", group.getPermissionList()); } if (!root.isEmpty()) { DumperOptions opt = new DumperOptions(); opt.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK); final Yaml yaml = new Yaml(opt); try { yaml.dump(root, new OutputStreamWriter(new FileOutputStream(GlobalGroupsFile), "UTF-8")); } catch (UnsupportedEncodingException ex) { } catch (FileNotFoundException ex) { } } setTimeStampGroups(GlobalGroupsFile.lastModified()); } else { // Newer file found. GroupManager.logger.log(Level.WARNING, "Newer GlobalGroups file found, but we have local changes!"); throw new IllegalStateException("Unable to save unless you issue a '/mansave force'"); } removeGroupsChangedFlag(); } else { //Check for newer file as no local changes. if (getTimeStampGroups() < GlobalGroupsFile.lastModified()) { System.out.print("Newer GlobalGroups file found (Loading changes)!"); // Backup GlobalGroups file backupFile(); load(); } } } /** * Backup the BlobalGroups file * @param w */ private void backupFile() { File backupFile = new File(plugin.getBackupFolder(), "bkp_ggroups_" + Tasks.getDateString() + ".yml"); try { Tasks.copy(GlobalGroupsFile, backupFile); } catch (IOException ex) { GroupManager.logger.log(Level.SEVERE, null, ex); } } /** * Adds a group, or replaces an existing one. * * @param groupToAdd */ public void addGroup(Group groupToAdd) { // Create a new group if it already exists if (hasGroup(groupToAdd.getName())) { groupToAdd = groupToAdd.clone(); removeGroup(groupToAdd.getName()); } newGroup(groupToAdd); haveGroupsChanged = true; } /** * Creates a new group if it doesn't already exist. * * @param newGroup */ public Group newGroup(Group newGroup) { // Push a new group if (!groups.containsKey(newGroup.getName().toLowerCase())) { groups.put(newGroup.getName().toLowerCase(), newGroup); this.setGroupsChanged(true); return newGroup; } return null; } /** * Delete a group if it exist. * * @param groupName */ public boolean removeGroup(String groupName) { // Push a new group if (groups.containsKey(groupName.toLowerCase())) { groups.remove(groupName.toLowerCase()); this.setGroupsChanged(true); return true; } return false; } /** * Returns true if the Global Group exists in the globalgroups.yml * * @param groupName * @return true if the group exists */ public boolean hasGroup(String groupName) { return groups.containsKey(groupName.toLowerCase()); } /** * Returns true if the group has the correct permission node. * * @param groupName * @param permissionNode * @return true if node exists */ public boolean hasPermission(String groupName, String permissionNode) { if (!hasGroup(groupName.toLowerCase())) return false; return groups.get(groupName.toLowerCase()).hasSamePermissionNode(permissionNode); } /** * Returns a PermissionCheckResult of the permission node for the group to * be tested against. * * @param groupName * @param permissionNode * @return PermissionCheckResult object */ public PermissionCheckResult checkPermission(String groupName, String permissionNode) { PermissionCheckResult result = new PermissionCheckResult(); result.askedPermission = permissionNode; result.resultType = PermissionCheckResult.Type.NOTFOUND; if (!hasGroup(groupName.toLowerCase())) return result; Group tempGroup = groups.get(groupName.toLowerCase()); if (tempGroup.hasSamePermissionNode(permissionNode)) result.resultType = PermissionCheckResult.Type.FOUND; if (tempGroup.hasSamePermissionNode("-" + permissionNode)) result.resultType = PermissionCheckResult.Type.NEGATION; if (tempGroup.hasSamePermissionNode("+" + permissionNode)) result.resultType = PermissionCheckResult.Type.EXCEPTION; return result; } /** * Returns a List of all permission nodes for this group null if none * * @param groupName * @return List of all group names */ public List getGroupsPermissions(String groupName) { if (!hasGroup(groupName.toLowerCase())) return null; return groups.get(groupName.toLowerCase()).getPermissionList(); } /** * Returns a Set of all global group names. * * @return Set containing all group names. */ public Set getGlobalGroups() { return groups.keySet(); } /** * Resets GlobalGroups. */ public void resetGlobalGroups() { this.groups = new HashMap(); } /** * * @return a collection of the groups */ public Collection getGroupList() { return groups.values(); } /** * Returns the Global Group or null if it doesn't exist. * * @param groupName * @return Group object */ public Group getGroup(String groupName) { if (!hasGroup(groupName.toLowerCase())) return null; return groups.get(groupName.toLowerCase()); } /** * @return the globalGroupsFile */ public File getGlobalGroupsFile() { return GlobalGroupsFile; } /** * */ public void removeGroupsChangedFlag() { setGroupsChanged(false); for (Group g : groups.values()) { g.flagAsSaved(); } } }