package com.dobbinsoft.fw.support.component.open;

import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.crypto.asymmetric.KeyType;
import com.alibaba.fastjson.JSONObject;
import com.dobbinsoft.fw.support.component.open.exception.OpenPlatformException;
import com.dobbinsoft.fw.support.component.open.model.*;
import lombok.extern.slf4j.Slf4j;
import okhttp3.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.stream.Collectors;

/**
 * ClassName: AbstractOpenPlatform
 * Description: 一个抽象的开放平台，定义了抽象的功能，持久层方案，由业务层决定
 *
 * @author: e-weichaozheng
 * @date: 2021-04-23
 */
@Slf4j
public class OpenPlatform {

    @Autowired
    protected OpenPlatformStorageStrategy openPlatformStorageStrategy;

    private OkHttpClient okHttpClient = new OkHttpClient();

    private ThreadLocal<String> clientCodeThreadLocal = new ThreadLocal<>();


    public void setClientCode(String clientCode) {
        clientCodeThreadLocal.set(clientCode);
    }

    public String getClientCode() {
        return clientCodeThreadLocal.get();
    }

    public void removeClientCode() {
        clientCodeThreadLocal.remove();
    }

    /**
     * 初始化客户端
     *
     * @param client
     * @return
     */
    public OPClient init(OPClient client) throws OpenPlatformException {
        // 1. 校验Code是否重复
        if (StrUtil.isEmpty(client.getCode())) {
            throw new OpenPlatformException("客户代号不能为空");
        }
        if (openPlatformStorageStrategy.checkClientExists(client.getCode())) {
            throw new OpenPlatformException("客户代号重复，请换一个");
        }
        // 2. 持久化
        OPClient opClient = openPlatformStorageStrategy.initClient(client.getCode(), client.getNotifyUrl(), client.getPublicKey2());
        opClient.setPrivateKey1(null);
        return opClient;
    }

    public Boolean setPublicKey(String clientCode, String publicKey) {
        if (StrUtil.isEmpty(clientCode)) {
            throw new OpenPlatformException("客户代号不能为空");
        }
        if (StrUtil.isEmpty(publicKey)) {
            throw new OpenPlatformException("公钥不能为空");
        }
        return openPlatformStorageStrategy.setClientPublicKey(clientCode, publicKey);
    }

    public Boolean setNotifyUrl(String clientCode, String notifyUrl) {
        if (StrUtil.isEmpty(clientCode)) {
            throw new OpenPlatformException("客户代号不能为空");
        }
        if (StrUtil.isEmpty(notifyUrl)) {
            throw new OpenPlatformException("回调地址不能为空");
        }
        return openPlatformStorageStrategy.setClientNotifyUrl(clientCode, notifyUrl);
    }

    public OPClient getClient(String clientCode) {
        OPClient client = openPlatformStorageStrategy.getClient(clientCode);
        if (client == null) {
            throw new OpenPlatformException("客户代号并不存在");
        }
        return client;
    }

    /**
     * 获取客户权限列表
     *
     * @param clientCode
     * @return
     */
    public Boolean setPermissionList(String clientCode, List<OPClientPermission> permissions) {
        if (StrUtil.isEmpty(clientCode)) {
            throw new OpenPlatformException("客户代号不能为空");
        }
        return openPlatformStorageStrategy.setClientPermissionList(clientCode, permissions);
    }

    public Boolean checkApiPermission(String clientCode, String group, String method) {
        if (StrUtil.isEmpty(clientCode)) {
            throw new OpenPlatformException("客户代号不能为空");
        }
        return openPlatformStorageStrategy.checkApiPermission(clientCode, group, method);
    }

    public Map<String, String[]> decrypt(String clientCode, String ciphertext) {
        String publicKey2 = openPlatformStorageStrategy.getClient(clientCode).getPublicKey2();
        String parameterMapStr = new String(SecureUtil.rsa(null, publicKey2).decrypt(ciphertext, KeyType.PublicKey), StandardCharsets.UTF_8);
        String[] pairs = parameterMapStr.split("&");
        Map<String, String[]> map = new HashMap<>();
        for (String v : pairs) {
            if (!StringUtils.isEmpty(v)) {
                String[] pair = v.split("=");
                if (pair.length > 1 && !StringUtils.isEmpty(pair[0])) {
                    map.put(pair[0], new String[] {pair[1]});
                }
            }
        }
        return map;
    }


    /**
     * 向客户回调地址发送请求
     *
     * @param clientCode
     * @param params     请求参数
     */
    public void sendNotify(String clientCode, Map<String, String> params) {
        params.put("opversion", UUID.randomUUID().toString());
        List<String> paramsRawList = params.keySet().stream().map(item -> (item + "=" + params.get(item))).collect(Collectors.toList());
        OPNotify opNotify = new OPNotify();
        opNotify.setParams(paramsRawList);
        opNotify.setClientCode(clientCode);
        opNotify.setTimes(1);
        int status = this.sendNotify(opNotify);
        opNotify.setStatus(status);
        if (status == 0) {
            opNotify.setNextNotify(new Date(System.currentTimeMillis() + 3000l));
        } else {
            opNotify.setNextNotify(new Date());
        }
        openPlatformStorageStrategy.storeNotify(opNotify);

    }

    public int sendNotify(OPNotify opNotify) {
        try {
            opNotify.getParams().add("timestamp=" + System.currentTimeMillis());
            opNotify.getParams().sort(String::compareTo);
            // 发送一次Http请求
            String rawBody = opNotify.getParams().stream().collect(Collectors.joining("&"));
            OPData opData = new OPData();
            opData.setClientCode(opNotify.getClientCode());
            OPClient clientWithPK = openPlatformStorageStrategy.getClientWithPK(opNotify.getClientCode());
            opData.setCiphertext(SecureUtil.rsa(clientWithPK.getPrivateKey1(),
                    clientWithPK.getPublicKey1()).encryptBase64(rawBody, KeyType.PrivateKey));
            MediaType mediaType = MediaType.parse("application/json");
            RequestBody body = RequestBody.create(mediaType, JSONObject.toJSONString(opData));
            Response responseBody = okHttpClient.newCall(new Request
                    .Builder()
                    .url(clientWithPK.getNotifyUrl())
                    .post(body).build())
                    .execute();
            if (responseBody.code() == 200) {
                return OPNotifyStatusType.OK.getCode();
            } else {
                return OPNotifyStatusType.FAIL.getCode();
            }
        } catch (IOException e) {
            return OPNotifyStatusType.FAIL.getCode();
        } catch (Exception e) {
            log.error("[发送通知] 异常", e);
            return OPNotifyStatusType.FAIL.getCode();
        }
    }

}