ConfigurableFeatureManager.java

package openmods.config.game;

import com.google.common.base.Preconditions;
import com.google.common.collect.HashBasedTable;
import com.google.common.collect.ImmutableTable;
import com.google.common.collect.Table;
import java.lang.reflect.Field;
import java.util.Set;
import net.minecraftforge.common.config.Configuration;
import net.minecraftforge.common.config.Property;
import openmods.config.BlockInstances;
import openmods.config.ItemInstances;

public class ConfigurableFeatureManager extends AbstractFeatureManager {

	private static class FeatureEntry {
		public boolean isEnabled;
		public final boolean isConfigurable;

		private FeatureEntry(boolean isEnabled, boolean isConfigurable) {
			this.isEnabled = isEnabled;
			this.isConfigurable = isConfigurable;
		}
	}

	public interface CustomFeatureRule {
		public boolean isEnabled(boolean flag);
	}

	private final Table<String, String, CustomFeatureRule> customRules = HashBasedTable.create();

	private final Table<String, String, FeatureEntry> features = HashBasedTable.create();

	public void collectFromItems(Class<? extends ItemInstances> itemContainer) {
		for (Field f : itemContainer.getFields()) {
			RegisterItem item = f.getAnnotation(RegisterItem.class);
			if (item == null) continue;
			features.put(CATEGORY_ITEMS, item.name(), new FeatureEntry(item.isEnabled(), item.isConfigurable()));
		}
	}

	public void collectFromBlocks(Class<? extends BlockInstances> blockContainer) {
		for (Field f : blockContainer.getFields()) {
			RegisterBlock item = f.getAnnotation(RegisterBlock.class);
			if (item == null) continue;
			features.put(CATEGORY_BLOCKS, item.name(), new FeatureEntry(item.isEnabled(), item.isConfigurable()));
		}
	}

	public Table<String, String, Property> loadFromConfiguration(Configuration config) {
		final Table<String, String, Property> properties = HashBasedTable.create();
		for (Table.Cell<String, String, FeatureEntry> cell : features.cellSet()) {
			final FeatureEntry entry = cell.getValue();
			if (!entry.isConfigurable) continue;

			final String categoryName = cell.getRowKey();
			final String featureName = cell.getColumnKey();
			final Property prop = config.get(categoryName, featureName, entry.isEnabled);
			properties.put(categoryName, featureName, prop);
			if (!prop.wasRead()) continue;

			if (!prop.isBooleanValue()) prop.set(entry.isEnabled);
			else entry.isEnabled = prop.getBoolean(entry.isEnabled);
		}

		return ImmutableTable.copyOf(properties);
	}

	@Override
	public boolean isEnabled(String category, String name) {
		FeatureEntry result = features.get(category, name);
		if (result == null) return false;

		CustomFeatureRule rule = customRules.get(category, name);
		return rule != null? rule.isEnabled(result.isEnabled) : result.isEnabled;
	}

	public void addCustomRule(String category, String name, CustomFeatureRule rule) {
		CustomFeatureRule prev = customRules.put(category, name, rule);
		Preconditions.checkState(prev == null, "Duplicate rule on %s:%s", category, name);
	}

	@Override
	public Set<String> getCategories() {
		return features.rowKeySet();
	}

	@Override
	public Set<String> getFeaturesInCategory(String category) {
		return features.row(category).keySet();
	}
}