CompleteConfig/src/main/java/me/lortseam/completeconfig/ConfigSource.java

package me.lortseam.completeconfig;

import lombok.AccessLevel;
import lombok.Getter;
import me.lortseam.completeconfig.data.Config;
import me.lortseam.completeconfig.util.TypeUtils;
import net.fabricmc.loader.api.FabricLoader;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.spongepowered.configurate.CommentedConfigurationNode;
import org.spongepowered.configurate.ConfigurateException;
import org.spongepowered.configurate.hocon.HoconConfigurationLoader;
import org.spongepowered.configurate.serialize.TypeSerializerCollection;

import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Objects;
import java.util.Set;

final class ConfigSource {

    private static final Logger LOGGER = LogManager.getLogger();
    private static final Set<ConfigSource> sources = new HashSet<>();
    private static TypeSerializerCollection globalTypeSerializers;

    static void registerGlobalTypeSerializers(TypeSerializerCollection typeSerializers) {
        globalTypeSerializers = TypeUtils.mergeSerializers(globalTypeSerializers, typeSerializers);
    }

    @Getter(AccessLevel.PACKAGE)
    private final String modID;
    private final String[] branch;
    private final HoconConfigurationLoader loader;

    ConfigSource(String modID, String[] branch, TypeSerializerCollection typeSerializers) {
        this.modID = modID;
        this.branch = branch;
        if (!sources.add(this)) {
            throw new IllegalArgumentException("A config of the mod " + modID + " with the specified branch " + Arrays.toString(branch) + " already exists!");
        }
        String[] subPath = ArrayUtils.add(branch, 0, modID);
        subPath[subPath.length - 1] = subPath[subPath.length - 1] + ".conf";
        Path filePath = Paths.get(FabricLoader.getInstance().getConfigDir().toString(), subPath);
        loader = HoconConfigurationLoader.builder()
                .path(filePath)
                .defaultOptions(options -> {
                    if (globalTypeSerializers != null) {
                        options.serializers(builder -> builder.registerAll(globalTypeSerializers));
                    }
                    if (typeSerializers != null) {
                        options.serializers(builder -> builder.registerAll(typeSerializers));
                    }
                    return options;
                })
                .build();
    }

    void load(Config config) {
        try {
            CommentedConfigurationNode root = loader.load();
            if (!root.virtual()) {
                config.apply(root);
            }
        } catch (ConfigurateException e) {
            LOGGER.error("[CompleteConfig] Failed to load config from file!", e);
        }
    }

    void save(Config config) {
        CommentedConfigurationNode root = loader.createNode();
        config.fetch(root);
        try {
            loader.save(root);
        } catch (ConfigurateException e) {
            LOGGER.error("[CompleteConfig] Failed to save config to file!", e);
        }
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        ConfigSource that = (ConfigSource) o;
        return modID.equals(that.modID) && Arrays.equals(branch, that.branch);
    }

    @Override
    public int hashCode() {
        int result = Objects.hash(modID);
        result = 31 * result + Arrays.hashCode(branch);
        return result;
    }

}