package pm.j4.petroleum.util.data;

import java.util.*;
import pm.j4.petroleum.PetroleumMod;
import pm.j4.petroleum.util.module.ModuleBase;

public class Category {
	public static Map<String, List<ModuleBase>> getCategoryMap() {
		List<ModuleBase> modules = PetroleumMod.getActiveMods();
		Map<String, List<ModuleBase>> categoryMap = new HashMap<>();
		modules.forEach(module -> {
			if(!categoryMap.containsKey(module.getCategory())) {
				List<ModuleBase> m = new ArrayList<>();
				m.add(module);
				categoryMap.put(module.getCategory(), m);
			}
			else {
				List<ModuleBase> m = categoryMap.get(module.getCategory());
				List<ModuleBase> nm = new ArrayList<>();
				nm.addAll(m);
				m.add(module);
				categoryMap.replace(module.getCategory(), nm);
			}
		});
		return categoryMap;
	}

	public static List<ModuleBase> getByCategory(String category) {
		return getCategoryMap().containsKey(category) ? getCategoryMap().get(category) : new ArrayList<>();
	}
}