EasyCode/src/test/java/com/sjhy/plugin/GenerateDefaultConfigTest.java

package com.sjhy.plugin;

import com.fasterxml.jackson.core.type.TypeReference;
import com.intellij.openapi.util.io.FileUtilRt;
import com.sjhy.plugin.dict.GlobalDict;
import com.sjhy.plugin.dto.SettingsStorageDTO;
import com.sjhy.plugin.entity.*;
import com.sjhy.plugin.tool.JSON;
import org.junit.Test;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * 生成默认配置测试
 * 由于resources目录下的配置文件不便与遍历，利用测试用例提前遍历并生成配置文件。运行时再直接加载配置
 *
 * @author makejava
 * @version 1.0.0
 * @date 2021/08/14 11:11
 */
public class GenerateDefaultConfigTest {


    @Test
    public void generate() throws IOException {
        SettingsStorageDTO settingsStorage = new SettingsStorageDTO();
        settingsStorage.setAuthor(GlobalDict.AUTHOR);
        settingsStorage.setVersion(GlobalDict.VERSION);
        settingsStorage.setUserSecure("");
        File templateDir = new File(GenerateDefaultConfigTest.class.getResource("/template").getFile());
        loadTemplate(settingsStorage, templateDir);

        File globalConfigDir = new File(GenerateDefaultConfigTest.class.getResource("/globalConfig").getFile());
        loadGlobalConfig(settingsStorage, globalConfigDir);

        File typeMapperDir = new File(GenerateDefaultConfigTest.class.getResource("/typeMapper").getFile());
        loadTypeMapper(settingsStorage, typeMapperDir);

        File columnConfigDir = new File(GenerateDefaultConfigTest.class.getResource("/columnConfig").getFile());
        loadColumnConfig(settingsStorage, columnConfigDir);

        String json = JSON.toJson(settingsStorage);
        // 所有的换行符号均改为\n
        // 1.windows处理
        json = json.replace("\\r\\n", "\\n");
        // 2.mac处理
        json = json.replace("\\r", "\\n");
        File buildPath = new File(GenerateDefaultConfigTest.class.getResource("/").getPath());
        // 找到根目录
        while (!buildPath.getName().equals("EasyCode")) {
            buildPath = buildPath.getParentFile();
        }
        Files.write(new File(buildPath, "src/main/resources/defaultConfig.json").toPath(), json.getBytes());
    }

    private void loadTemplate(SettingsStorageDTO settingsStorage, File root) throws IOException {
        settingsStorage.setCurrTemplateGroupName(GlobalDict.DEFAULT_GROUP_NAME);
        settingsStorage.setTemplateGroupMap(new HashMap<>(root.listFiles().length));
        for (File dir : root.listFiles()) {
            TemplateGroup templateGroup = new TemplateGroup();
            templateGroup.setName(dir.getName());
            templateGroup.setElementList(new ArrayList<>());
            if (dir.isDirectory()) {
                for (File file : dir.listFiles()) {
                    Template template = new Template();
                    template.setName(file.getName());
                    template.setCode(FileUtilRt.loadFile(file));
                    templateGroup.getElementList().add(template);
                }
            }
            settingsStorage.getTemplateGroupMap().put(templateGroup.getName(), templateGroup);
        }
    }

    private void loadGlobalConfig(SettingsStorageDTO settingsStorage, File root) throws IOException {
        settingsStorage.setCurrGlobalConfigGroupName(GlobalDict.DEFAULT_GROUP_NAME);
        settingsStorage.setGlobalConfigGroupMap(new HashMap<>(root.listFiles().length));
        for (File dir : root.listFiles()) {
            GlobalConfigGroup globalConfigGroup = new GlobalConfigGroup();
            globalConfigGroup.setName(dir.getName());
            globalConfigGroup.setElementList(new ArrayList<>());
            if (dir.isDirectory()) {
                for (File file : dir.listFiles()) {
                    GlobalConfig globalConfig = new GlobalConfig();
                    globalConfig.setName(file.getName());
                    globalConfig.setValue(FileUtilRt.loadFile(file));
                    globalConfigGroup.getElementList().add(globalConfig);
                }
            }
            settingsStorage.getGlobalConfigGroupMap().put(globalConfigGroup.getName(), globalConfigGroup);
        }
    }

    private void loadTypeMapper(SettingsStorageDTO settingsStorage, File root) throws IOException {
        settingsStorage.setCurrTypeMapperGroupName(GlobalDict.DEFAULT_GROUP_NAME);
        settingsStorage.setTypeMapperGroupMap(new HashMap<>(root.listFiles().length));
        for (File file : root.listFiles()) {
            TypeMapperGroup typeMapperGroup = new TypeMapperGroup();
            typeMapperGroup.setName(file.getName().replace(".json", ""));
            if (file.isFile()) {
                String json = FileUtilRt.loadFile(file);
                typeMapperGroup.setElementList(JSON.parse(json, new TypeReference<List<TypeMapper>>() {
                }));
            }
            settingsStorage.getTypeMapperGroupMap().put(typeMapperGroup.getName(), typeMapperGroup);
        }
    }

    private void loadColumnConfig(SettingsStorageDTO settingsStorage, File root) throws IOException {
        settingsStorage.setCurrColumnConfigGroupName(GlobalDict.DEFAULT_GROUP_NAME);
        settingsStorage.setColumnConfigGroupMap(new HashMap<>(root.listFiles().length));
        for (File file : root.listFiles()) {
            ColumnConfigGroup columnConfigGroup = new ColumnConfigGroup();
            columnConfigGroup.setName(file.getName().replace(".json", ""));
            if (file.isFile()) {
                String json = FileUtilRt.loadFile(file);
                columnConfigGroup.setElementList(JSON.parse(json, new TypeReference<List<ColumnConfig>>() {
                }));
            }
            settingsStorage.getColumnConfigGroupMap().put(columnConfigGroup.getName(), columnConfigGroup);
        }
    }

}