package org.dbsyncer.parser;

import org.apache.commons.io.FileUtils;
import org.dbsyncer.common.util.JsonUtil;
import org.dbsyncer.connector.config.ConnectorConfig;
import org.dbsyncer.connector.enums.ConnectorEnum;
import org.dbsyncer.parser.model.Connector;
import org.dbsyncer.parser.model.Mapping;
import org.dbsyncer.parser.model.TableGroup;
import org.json.JSONException;
import org.json.JSONObject;
import org.junit.Test;

import java.io.File;
import java.io.IOException;
import java.net.URL;

/**
 * @author AE86
 * @version 1.0.0
 * @date 2019/10/9 23:46
 */
public class ConnectorParserTest {

    @Test
    public void testConnector() throws IOException, JSONException {
        String json = readJson("Connector.json");
        System.out.println(json);

        // 解析基本信息
        JSONObject conn = new JSONObject(json);
        JSONObject config = (JSONObject) conn.remove("config");
        Connector connector = JsonUtil.jsonToObj(conn.toString(), Connector.class);

        // 解析配置
        String connectorType = config.getString("connectorType");
        Class<?> configClass = ConnectorEnum.getConfigClass(connectorType);
        Object obj = JsonUtil.jsonToObj(config.toString(), configClass);
        connector.setConfig((ConnectorConfig) obj);
        System.out.println(connector);
    }

    @Test
    public void testMapping() throws IOException, JSONException {
        String json = readJson("Mapping.json");
        System.out.println(json);

        // 解析基本信息
        JSONObject map = new JSONObject(json);
        Mapping mapping = JsonUtil.jsonToObj(map.toString(), Mapping.class);
        System.out.println(mapping);
    }

    @Test
    public void testTableGroup() throws IOException, JSONException {
        String json = readJson("TableGroup.json");
        System.out.println(json);
        // 解析基本信息
        JSONObject group = new JSONObject(json);
        TableGroup tableGroup = JsonUtil.jsonToObj(group.toString(), TableGroup.class);
        System.out.println(tableGroup);
    }

    /**
     * 读取JSON文件
     *
     * @param fileName
     * @return
     * @throws IOException
     */
    private String readJson(String fileName) throws IOException {
        ClassLoader loader = this.getClass().getClassLoader();
        URL fileURL = loader.getResource(fileName);
        if (null != fileURL) {
            String filePath = fileURL.getFile();
            File file = new File(filePath);
            return FileUtils.readFileToString(file, "UTF-8");
        }
        return "";
    }

}