/*
 * Hello Minecraft! Launcher
 * Copyright (C) 2020  huangyuhui <huanghongxun2008@126.com> and contributors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
package org.jackhuang.hmcl.setting;

import com.google.gson.Gson;
import org.jackhuang.hmcl.util.StringUtils;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import static org.jackhuang.hmcl.util.Lang.tryCast;
import static org.jackhuang.hmcl.util.logging.Logger.LOG;

final class ConfigUpgrader {
    private ConfigUpgrader() {
    }

    private static final int CURRENT_VERSION = 2;

    /**
     * This method is for the compatibility with old HMCL versions.
     *
     * @param deserialized deserialized config settings
     * @param rawContent   raw json content of the config settings without modification
     */
    static void upgradeConfig(Config deserialized, String rawContent) {
        int configVersion = deserialized.getConfigVersion();

        if (configVersion == CURRENT_VERSION) {
            return;
        }

        if (configVersion > CURRENT_VERSION) {
            LOG.warning(String.format("Current HMCL only support the configuration version up to %d. However, the version now is %d.", CURRENT_VERSION, configVersion));
            deserialized.setConfigVersion(CURRENT_VERSION);
            return;
        }

        LOG.info(String.format("Updating configuration from %d to %d.", configVersion, CURRENT_VERSION));
        Map<?, ?> rawJson = Collections.unmodifiableMap(new Gson().<Map<?, ?>>fromJson(rawContent, Map.class));

        if (configVersion < 1) {
            // Upgrade configuration of HMCL 2.x: Convert OfflineAccounts whose stored uuid is important.
            tryCast(rawJson.get("auth"), Map.class).ifPresent(auth -> {
                tryCast(auth.get("offline"), Map.class).ifPresent(offline -> {
                    String selected = rawJson.containsKey("selectedAccount") ? null
                            : tryCast(offline.get("IAuthenticator_UserName"), String.class).orElse(null);

                    tryCast(offline.get("uuidMap"), Map.class).ifPresent(uuidMap -> {
                        ((Map<?, ?>) uuidMap).forEach((key, value) -> {
                            Map<Object, Object> storage = new HashMap<>();
                            storage.put("type", "offline");
                            storage.put("username", key);
                            storage.put("uuid", value);
                            if (key.equals(selected)) {
                                storage.put("selected", true);
                            }
                            deserialized.getAccountStorages().add(storage);
                        });
                    });
                });
            });

            // Upgrade configuration of HMCL earlier than 3.1.70
            if (!rawJson.containsKey("commonDirType"))
                deserialized.setCommonDirType(deserialized.getCommonDirectory().equals(Settings.getDefaultCommonDirectory()) ? EnumCommonDirectory.DEFAULT : EnumCommonDirectory.CUSTOM);
            if (!rawJson.containsKey("backgroundType"))
                deserialized.setBackgroundImageType(StringUtils.isNotBlank(deserialized.getBackgroundImage()) ? EnumBackgroundImage.CUSTOM : EnumBackgroundImage.DEFAULT);
            if (!rawJson.containsKey("hasProxy"))
                deserialized.setHasProxy(StringUtils.isNotBlank(deserialized.getProxyHost()));
            if (!rawJson.containsKey("hasProxyAuth"))
                deserialized.setHasProxyAuth(StringUtils.isNotBlank(deserialized.getProxyUser()));

            if (!rawJson.containsKey("downloadType")) {
                tryCast(rawJson.get("downloadtype"), Number.class)
                        .map(Number::intValue)
                        .ifPresent(id -> {
                            if (id == 0) {
                                deserialized.setDownloadType("mojang");
                            } else if (id == 1) {
                                deserialized.setDownloadType("bmclapi");
                            }
                        });
            }
        }

        deserialized.setConfigVersion(CURRENT_VERSION);
    }
}
