CarryOn/Fabric/src/main/java/tschipp/carryon/config/fabric/ConfigLoaderImpl.java

/*
 * GNU Lesser General Public License v3
 * Copyright (C) 2024 Tschipp
 * mrtschipp@gmail.com
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

package tschipp.carryon.config.fabric;

import com.google.gson.*;
import net.fabricmc.loader.api.FabricLoader;
import org.apache.commons.io.FileUtils;
import tschipp.carryon.Constants;
import tschipp.carryon.common.config.CarryConfig;
import tschipp.carryon.common.config.ListHandler;
import tschipp.carryon.config.*;

import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.util.*;

public class ConfigLoaderImpl {

    private static final Gson GSON = new GsonBuilder().disableHtmlEscaping().setPrettyPrinting().create();

    //Default JSON and config data.
    public static final Map<JsonObject, BuiltConfig> CONFIGS = new LinkedHashMap<>();

    public static void initialize() throws IOException {
        Path cfgPath = FabricLoader.getInstance().getConfigDir();

        for (Map.Entry<JsonObject, BuiltConfig> entry : CONFIGS.entrySet()) {
            File cfgFile = new File(cfgPath.toFile(), entry.getValue().fileName+".json");
            if (!cfgFile.exists()) {
                cfgPath.toFile().mkdirs();
                FileUtils.write(cfgFile, GSON.toJson(entry.getKey()), StandardCharsets.UTF_8);
            } else {
                JsonObject cfgJson = GSON.fromJson(FileUtils.readFileToString(cfgFile, StandardCharsets.UTF_8), JsonObject.class);
                if(cfgJson == null)
                {
                    cfgPath.toFile().mkdirs();
                    FileUtils.write(cfgFile, GSON.toJson(entry.getKey()), StandardCharsets.UTF_8);
                }
                FileUtils.write(cfgFile, GSON.toJson(loadConfig(entry.getValue(), cfgJson)), StandardCharsets.UTF_8);
            }
        }
    }

    private static JsonObject loadConfig(BuiltCategory category, JsonObject config) {
        config.entrySet().forEach((entry) -> {
            String id = entry.getKey();
            if (!id.startsWith("//")) {
                JsonElement value = entry.getValue();
                if (value instanceof JsonPrimitive configValue) {
                    category.getProperty(id).ifPresent(data -> {
                        if (configValue.isBoolean() && data.getData().type().equals(PropertyType.BOOLEAN))
                            data.setBoolean(configValue.getAsBoolean());
                        if (configValue.isNumber() && data.getData().type().equals(PropertyType.INT)) {
                            int configInt = configValue.getAsInt();
                            if (configInt > data.getData().max() || configInt < data.getData().min()) {
                                try {
                                    config.addProperty(id, data.getInt());
                                } catch (IllegalAccessException ignored) {
                                }
                            } else {
                                data.setInt(configInt);
                            }
                        }
                        if (configValue.isNumber() && data.getData().type().equals(PropertyType.DOUBLE)) {
                            double configDouble = configValue.getAsDouble();
                            if (configDouble > data.getData().maxD() || configDouble < data.getData().minD()) {
                                try {
                                    config.addProperty(id, data.getDouble());
                                } catch (IllegalAccessException ignored) {
                                }
                            } else {
                                data.setDouble(configDouble);
                            }
                        }
                    });
                } else if (value instanceof JsonObject subConfig) {
                    category.getCategory(id).ifPresent(cat -> loadConfig(cat, subConfig));
                } else if (value instanceof JsonArray list) {
                    category.getProperty(id).ifPresent(data -> {
                        if(data.getData().type() == PropertyType.STRING_ARRAY)
                        {
                            List<String> ls = new ArrayList<>();
                            for(JsonElement arrEle : list)
                            {
                                if(arrEle instanceof JsonPrimitive p && p.isString())
                                {
                                    ls.add(p.getAsString());
                                }
                            }
                            data.setStringArray(ls.toArray(new String[ls.size()]));
                        }
                    });
                }
            }
        });
        return config;
    }

    public static void saveConfigs() {
        try {
            Path cfgPath = FabricLoader.getInstance().getConfigDir();
            for (Map.Entry<JsonObject, BuiltConfig> entry : CONFIGS.entrySet()) {
                File cfgFile = new File(cfgPath.toFile(), entry.getValue().fileName+".json");
                cfgPath.toFile().mkdirs();
                FileUtils.write(cfgFile, GSON.toJson(entry.getKey()), StandardCharsets.UTF_8);
            }
            ListHandler.initConfigLists();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void registerConfig(BuiltConfig config) {
        try {
            JsonObject configJson = new JsonObject();
            for (PropertyData property : config.properties) buildProperty(configJson, property);
            for (BuiltCategory category : config.categories) buildCategory(configJson, category);
            CONFIGS.put(configJson, config);
        }catch (Exception e) {
            e.printStackTrace();
        }
    }

    private static void buildCategory(JsonObject builder, BuiltCategory category) throws IllegalAccessException {
        JsonObject categoryJson = new JsonObject();
        if (category.categoryDesc != null) categoryJson.addProperty("//"+category.category, category.categoryDesc);
        for (PropertyData property : category.properties) buildProperty(categoryJson, property);
        for (BuiltCategory builtCategory : category.categories) buildCategory(categoryJson, builtCategory);
        builder.add(category.category, categoryJson);
    }

    private static void buildProperty(JsonObject builder, PropertyData data) throws IllegalAccessException {
        AnnotationData annotationData = data.getData();
        builder.addProperty("//"+data.getId(), annotationData.description());

        switch (annotationData.type()) {
            case BOOLEAN:
                builder.addProperty(data.getId(), data.getBoolean());
                data.setSetter((b) -> {builder.addProperty(data.getId(), (boolean)b); data.setBoolean((boolean)b);});
                break;
            case INT:
                builder.addProperty(data.getId(), data.getInt());
                data.setSetter((b) -> {builder.addProperty(data.getId(), (int)b); data.setInt((int)b);});
                break;
            case DOUBLE:
                builder.addProperty(data.getId(), data.getDouble());
                data.setSetter((b) -> {builder.addProperty(data.getId(), (double)b); data.setDouble((double)b);});
                break;
            case STRING_ARRAY:
                JsonArray arr = new JsonArray();
                for(String s : data.getStringArray())
                    arr.add(s);
                builder.add(data.getId(), arr);
                data.setSetter(list -> {
                    JsonArray overwrite = new JsonArray();
                    for(String s : (List<String>)list)
                        overwrite.add(s);
                    builder.add(data.getId(), overwrite);
                    data.setStringArray(((List<?>) list).toArray(new String[0]));
                });
                break;
            default:
                throw new IllegalAccessException("Unknown property type.");
        }
    }
}