Files
KittyBot/src/core/CommandEnabler.java
T

109 lines
2.8 KiB
Java

package core;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Vector;
import dataStructures.Pair;
import utils.GlobalLog;
import utils.LogFilter;
// While some of the patterns in this file are similar to the localization files,
// commands that are being looked up will behave slightly differently so trimming
// rules for this file are different than the localization ones - this is more
// aggresive with whitespace removal.
public class CommandEnabler extends BaseKeyValueFile
{
// Config/const variables
public static final String enabled = "1";
public static final String disabled = "0";
public static final boolean defaultEnabledState = true;
// Local variables
private HashMap<String, Boolean> enabledMap; // Quick lookup
private ArrayList<String> keyList; // Tracking ordering for later
private final static String name = "commands.config";
public CommandEnabler()
{
super(name);
// Create/Init variables
GlobalLog.log(LogFilter.Core, "Initializing " + this.getClass().getSimpleName());
enabledMap = new HashMap<>();
keyList = new ArrayList<>();
// Startup
readIn();
getTrackedCommands();
writeOut();
}
// Reads in the config file and parses it, keeping tabs on the order it read things
private void readIn()
{
parse((pair) ->{
String key = pair.First;
String value = pair.Second;
keyList.add(key);
if(value.equalsIgnoreCase(enabled))
enabledMap.putIfAbsent(key, true);
else
enabledMap.putIfAbsent(key, false);
});
}
// Look up the already scraped values from the localizer and store them if they
// don't already exist in the lookup. Defaults to defaultEnabledState.
private void getTrackedCommands()
{
ArrayList<String> unloc = LocCommands.getUnlocalizedCommands();
for(int i = 0; i < unloc.size(); ++i)
{
String command = unloc.get(i).toLowerCase();
if(enabledMap.putIfAbsent(command, defaultEnabledState) == null)
{
GlobalLog.log(LogFilter.Strings, "Identified new toggleable raw command: " + command);
keyList.add(command);
}
}
}
// Write out enabled/disabled file info.
private void writeOut()
{
List<Pair<String, String>> list = new Vector<Pair<String, String>>();
for(int i = 0; i < keyList.size(); ++i)
{
String key = keyList.get(i).toLowerCase();
String value = enabled.toLowerCase();
if(enabledMap.get(key) == false)
value = disabled.toLowerCase();
list.add(new Pair<String, String>(key, value));
}
Collections.sort(list, (c1, c2) -> { return c1.First.compareTo(c2.First); });
write(list);
}
// Looks up a key to see if it's enabled or not
public boolean isEnabled(String key)
{
String toCheck = key.toLowerCase();
if(enabledMap.containsKey(toCheck))
return enabledMap.get(toCheck);
return true;
}
}