package io.dataease.service.sys;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.ZipUtil;
import com.google.gson.Gson;
import io.dataease.dto.MyPluginDTO;
import io.dataease.ext.ExtSysPluginMapper;
import io.dataease.ext.query.GridExample;
import io.dataease.commons.constants.AuthConstants;
import io.dataease.commons.exception.DEException;
import io.dataease.commons.utils.CodingUtil;
import io.dataease.commons.utils.DeFileUtils;
import io.dataease.commons.utils.LogUtil;
import io.dataease.commons.utils.ZipUtils;
import io.dataease.controller.sys.base.BaseGridRequest;
import io.dataease.i18n.Translator;
import io.dataease.listener.util.CacheUtils;
import io.dataease.plugins.common.base.domain.MyPlugin;
import io.dataease.plugins.common.base.mapper.MyPluginMapper;
import io.dataease.plugins.config.LoadjarUtil;
import io.dataease.plugins.config.SpringContextUtil;
import io.dataease.service.datasource.DatasourceService;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;


@Service
public class PluginService {

    @Value("${dataease.plugin.dir:/opt/dataease/plugins/}")
    private String pluginDir;

    private final static String pluginJsonName = "plugin.json";

    @Resource
    private ExtSysPluginMapper extSysPluginMapper;

    @Resource
    private MyPluginMapper myPluginMapper;

    @Resource
    private DatasourceService datasourceService;

    @Autowired
    private LoadjarUtil loadjarUtil;

    @Value("${version}")
    private String version;


    public List<MyPlugin> query(BaseGridRequest request) {
        GridExample gridExample = request.convertExample();
        return extSysPluginMapper.query(gridExample);
    }

    /**
     * 从本地安装处插件
     *
     * @param file
     * @return
     */
    public Map<String, Object> localInstall(MultipartFile file) throws Exception{
        //1.上传文件到服务器pluginDir目录下
        File dest = DeFileUtils.upload(file, pluginDir + "temp/");
        //2.解压目标文件dest 得到plugin.json和jar
        String folder = pluginDir + "folder/";
        try {
            ZipUtil.unzip(dest.getAbsolutePath(), folder);
            // ZipUtils.unzip(dest, folder);
        } catch (Exception e) {
            DeFileUtils.deleteFile(pluginDir + "temp/");
            DeFileUtils.deleteFile(folder);
            // 需要删除文件
            // e.printStackTrace();
            LogUtil.error(e.getMessage(), e);
            DEException.throwException(e);
        }
        //3.解析plugin.json 失败则 直接返回错误 删除文件
        File folderFile = new File(folder);
        File[] jsonFiles = folderFile.listFiles(this::isPluginJson);
        if (ArrayUtils.isEmpty(jsonFiles)) {
            DeFileUtils.deleteFile(pluginDir + "temp/");
            DeFileUtils.deleteFile(folder);
            String msg = "缺少插件描述文件【plugin.json】";
            LogUtil.error(msg);
            DEException.throwException(new RuntimeException(msg));
        }
        MyPluginDTO myPlugin = formatJsonFile(jsonFiles[0]);

        if (!versionMatch(myPlugin.getRequire())) {
            String msg = "当前插件要求系统版本最低为：" + myPlugin.getRequire();
            LogUtil.error(msg);
            DEException.throwException(new RuntimeException(msg));
        }
        //4.加载jar包 失败则 直接返回错误 删除文件
        File[] jarFiles = folderFile.listFiles(this::isPluginJar);
        if (ArrayUtils.isEmpty(jarFiles)) {
            DeFileUtils.deleteFile(pluginDir + "temp/");
            DeFileUtils.deleteFile(folder);
            String msg = "缺少插件jar文件";
            LogUtil.error(msg);
            DEException.throwException(new RuntimeException(msg));
        }

        if (pluginExist(myPlugin)) {
            String msg = "插件【"+myPlugin.getName()+"】已存在，请先卸载";
            LogUtil.error(msg);
            DEException.throwException(new RuntimeException(msg));
        }
        String targetDir = null;
        try {
            File jarFile = jarFiles[0];
            targetDir = makeTargetDir(myPlugin);
            String jarPath;
            jarPath = DeFileUtils.copy(jarFile, targetDir);
            if(myPlugin.getCategory().equalsIgnoreCase("datasource")){
                DeFileUtils.copyFolder(folder + "/" + myPlugin.getDsType() + "Driver", targetDir + myPlugin.getDsType() + "Driver");
            }
            loadJar(jarPath, myPlugin);
            myPluginMapper.insert(myPlugin);

            CacheUtils.removeAll(AuthConstants.USER_CACHE_NAME);
            CacheUtils.removeAll(AuthConstants.USER_ROLE_CACHE_NAME);
            CacheUtils.removeAll(AuthConstants.USER_PERMISSION_CACHE_NAME);
        } catch (Exception e) {
            if (StringUtils.isNotEmpty(targetDir)) {
                deleteJarFile(myPlugin);
            }
            LogUtil.error(e.getMessage(), e);
            DEException.throwException(e);
        } finally {
            DeFileUtils.deleteFile(pluginDir + "temp/");
            DeFileUtils.deleteFile(folder);
        }
        return null;
    }

    public void loadJar(String jarPath, MyPlugin myPlugin) throws Exception {
        loadjarUtil.loadJar(jarPath, myPlugin);
    }

    private String makeTargetDir(MyPlugin myPlugin) {
        String store = myPlugin.getStore();
        String dir = pluginDir + store + "/";
        File fileDir = new File(dir);
        if (!fileDir.exists()) {
            fileDir.mkdirs();
        }
        return dir;
    }

    /**
     * 检测插件是否已存在
     * @param myPlugin
     * @return
     */
    public boolean pluginExist(MyPlugin myPlugin) {
        GridExample gridExample = new GridExample();
        List<MyPlugin> plugins = extSysPluginMapper.query(gridExample);
        return plugins.stream().anyMatch(plugin -> {
            return StringUtils.equals(myPlugin.getName(), plugin.getName()) || StringUtils.equals(myPlugin.getModuleName(), plugin.getModuleName());
        });
    }

    /**
     * 卸载插件
     *
     * @param pluginId
     * @return
     */
    public Boolean uninstall(Long pluginId) {
        MyPlugin myPlugin = myPluginMapper.selectByPrimaryKey(pluginId);
        if (ObjectUtils.isEmpty(myPlugin)) {
            String msg = "当前插件不存在";
            LogUtil.error(msg);
            DEException.throwException(new RuntimeException(msg));
        }
        deleteJarFile(myPlugin);
        CacheUtils.removeAll(AuthConstants.USER_CACHE_NAME);
        CacheUtils.removeAll(AuthConstants.USER_ROLE_CACHE_NAME);
        CacheUtils.removeAll(AuthConstants.USER_PERMISSION_CACHE_NAME);

        if(myPlugin.getCategory().equalsIgnoreCase("datasource")){
            if(CollectionUtils.isNotEmpty(datasourceService.selectByType(myPlugin.getDsType()))){
                throw new RuntimeException(Translator.get("i18n_plugin_not_allow_delete"));
            }
            loadjarUtil.deleteModule(myPlugin.getModuleName() + "-" + myPlugin.getVersion());
        }
        myPluginMapper.deleteByPrimaryKey(pluginId);
        return true;
    }

    private void deleteJarFile(MyPlugin plugin) {
        String version = plugin.getVersion();
        String moduleName = plugin.getModuleName();
        String fileName = moduleName + "-" + version + ".jar";
        String path = pluginDir + plugin.getStore() + "/" + fileName;
        File jarFile = new File(path);
        FileUtil.del(jarFile);

        if(plugin.getCategory().equalsIgnoreCase("datasource")){
            File driverFile = new File(pluginDir + plugin.getStore() + "/" + plugin.getDsType() + "Driver");
            FileUtil.del(driverFile);
        }
    }

    /**
     * 改变插件状态
     *
     * @param pluginId
     * @param status   true ？ 使用状态 : 禁用状态
     * @return
     */
    public Boolean changeStatus(Long pluginId, Boolean status) {
        CacheUtils.removeAll(AuthConstants.USER_CACHE_NAME);
        CacheUtils.removeAll(AuthConstants.USER_ROLE_CACHE_NAME);
        CacheUtils.removeAll(AuthConstants.USER_PERMISSION_CACHE_NAME);
        return false;
    }

    //判断当前文件是否实插件描述文件
    //文件名称必须plugin.json
    private boolean isPluginJson(File file) {
        return StringUtils.equals(file.getName(), pluginJsonName);
    }

    private boolean isPluginJar(File file) {
        String name = file.getName();
        return StringUtils.equals(DeFileUtils.getExtensionName(name), "jar");
    }

    /**
     * 从plugin.json文件反序列化为MyPlugin实例对象
     *
     * @return
     */
    private MyPluginDTO formatJsonFile(File file) {
        String str = DeFileUtils.readJson(file);
        Gson gson = new Gson();
        Map<String, Object> myPlugin = gson.fromJson(str, Map.class);
        myPlugin.put("free", (Double) myPlugin.get("free") > 0.0);
        myPlugin.put("loadMybatis", myPlugin.get("loadMybatis") == null ? false : (Double) myPlugin.get("loadMybatis") > 0.0);
        MyPluginDTO result = new MyPluginDTO();
        try {
            org.apache.commons.beanutils.BeanUtils.populate(result, myPlugin);
            result.setInstallTime(System.currentTimeMillis());
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }
        //BeanUtils.copyBean(result, myPlugin);
        if(result.getCategory().equalsIgnoreCase("datasource") && (StringUtils.isEmpty(result.getStore()) || !result.getStore().equalsIgnoreCase("default"))){
            result.setStore("thirdpart");
        }

        return result;
    }

    /**
     * 从插件商城远程安装插件
     * 2.0版本实现
     *
     * @param params
     * @return
     */
    public Map<String, Object> remoteInstall(Map<String, Object> params) {
        return null;
    }

    public boolean versionMatch(String pluginVersion) {
        List<Integer> versionLists = Arrays.stream(version.split("\\.")).map(CodingUtil::string2Integer).collect(Collectors.toList());
        List<Integer> requireVersionLists = Arrays.stream(pluginVersion.split("\\.")).map(CodingUtil::string2Integer).collect(Collectors.toList());
        int maxSize = Math.max(versionLists.size(), requireVersionLists.size());
        for (int i = 0; i < maxSize; i++) {
            Integer currentV = versionLists.size() == i ? 0 : versionLists.get(i);
            Integer requireV = requireVersionLists.size() == i ? 0 : requireVersionLists.get(i);
            if (requireV > currentV) return false;
        }
        return true;
    }
}