package com.objecteye.handle.viid;

import com.objecteye.config.Gat1400Properties;
import com.objecteye.content.VIIDApi;
import com.objecteye.pojo.viid.common.KeepaliveRequestObject;
import com.objecteye.pojo.viid.common.RegisterObject;
import com.objecteye.pojo.viid.common.RegisterRequestObject;
import com.objecteye.pojo.viid.common.UnRegisterRequestObject;
import com.objecteye.utils.DigestUtils;
import com.objecteye.utils.JSONUtils;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

/**
 * @author: liuhaoyu
 * @date: 2023/6/28
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class CommonHandle {
    private final RestTemplate restTemplate;
    private final Gat1400Properties gat1400Properties;
    private static final String HTTP = "http://";

    /**
     * 注册
     *
     * @param deviceId 注册目标设备
     * @return 注册结果
     */
    public boolean register(String deviceId) {
        // 请求参数设置
        RegisterRequestObject registerRequestObject = new RegisterRequestObject();
        RegisterObject registerObject = new RegisterObject();
        registerObject.setDeviceId(deviceId);
        registerRequestObject.setRegisterObject(registerObject);
        return twoStepCheck(VIIDApi.REGISTER, deviceId, JSONUtils.toJsonString(registerRequestObject));
    }

    /**
     * 保活
     *
     * @param deviceId 心跳目标设备
     * @return 发送结果
     */
    public boolean keepalive(String deviceId) {
        String keepaliveUrl = getAddress() + VIIDApi.KEEPALIVE;
        // 请求参数设置
        KeepaliveRequestObject keepaliveRequestObject = new KeepaliveRequestObject();
        RegisterObject keepaliveObject = new RegisterObject();
        keepaliveObject.setDeviceId(deviceId);
        keepaliveRequestObject.setKeepaliveObject(keepaliveObject);
        if (log.isDebugEnabled()) {
            log.debug("保活请求 url:{} ,参数：{}", keepaliveUrl, keepaliveRequestObject);
        }

        HttpEntity<String> httpEntity = new HttpEntity<>(JSONUtils.toJsonString(keepaliveRequestObject),
                buildRequestHttpHeader(deviceId));
        // 请求执行
        ResponseEntity<String> responseEntity = postForEntity(keepaliveUrl, httpEntity);
        return org.apache.http.HttpStatus.SC_OK == responseEntity.getStatusCode().value();
    }

    /**
     * 注销
     *
     * @param deviceId 注销目标设备
     * @return 注销结果
     */
    public boolean unregister(String deviceId) {
        // 请求参数设置
        UnRegisterRequestObject unRegisterRequestObject = new UnRegisterRequestObject();
        RegisterObject unRegisterObject = new RegisterObject();
        unRegisterObject.setDeviceId(deviceId);
        unRegisterRequestObject.setUnRegisterObject(unRegisterObject);
        return twoStepCheck(VIIDApi.UNREGISTER, deviceId, JSONUtils.toJsonString(unRegisterRequestObject));
    }

    private String getAddress() {
        return HTTP + gat1400Properties.getServerAddress() + ":" + gat1400Properties.getPort();
    }

    private HttpHeaders buildRequestHttpHeader(String deviceId) {
        // 请求头设置
        HttpHeaders headers = new HttpHeaders();
        headers.add("Content-Type", "application/json;charset=UTF-8");
        headers.set("User-Identify", deviceId);
        headers.setConnection("keepalive");
        return headers;
    }

    private ResponseEntity<String> postForEntity(String url, HttpEntity<String> httpEntity) {
        return restTemplate.exchange(url, HttpMethod.POST, httpEntity, String.class);
    }

    private boolean twoStepCheck(String uri, String deviceId, String requestJson) {
        String url = getAddress() + uri;
        // 请求头设置
        HttpHeaders headers = buildRequestHttpHeader(deviceId);
        // 第一次请求
        ResponseEntity<String> responseEntity = postForEntity(url, new HttpEntity<>(requestJson, headers));
        if (org.apache.http.HttpStatus.SC_UNAUTHORIZED == responseEntity.getStatusCode().value()) {
            headers.set("Authorization", getAuthorization(responseEntity, url));
            // 第二次请求
            responseEntity = postForEntity(url, new HttpEntity<>(requestJson, headers));
            return org.apache.http.HttpStatus.SC_OK == responseEntity.getStatusCode().value();
        }
        return false;
    }

    private String getAuthorization(ResponseEntity<String> responseEntity, String url) {
        HttpHeaders responseEntityHeaders = responseEntity.getHeaders();
        String authenticate = responseEntityHeaders.get("WWW-Authenticate").get(0);
        String[] children = authenticate.split(",");
        String realm = null, qop = null, nonce = null, opaque = null, method = "POST";
        for (String item : children) {
            String[] itemEntry = item.split("=");
            if ("Digest realm".equals(itemEntry[0])) {
                realm = itemEntry[1].replaceAll("\"", "");
            } else if ("qop".equals(itemEntry[0])) {
                qop = itemEntry[1].replaceAll("\"", "");
            } else if ("nonce".equals(itemEntry[0])) {
                nonce = itemEntry[1].replaceAll("\"", "");
            }
        }
        String nc = "00000001";
        String cnonce = DigestUtils.generateSalt2(8);
        String response = DigestUtils.getResponse(gat1400Properties.getAccessKey(), realm, gat1400Properties.getSecurityKey(),
                nonce, nc, cnonce, qop, method, url);
        return DigestUtils.getAuthorization(gat1400Properties.getAccessKey(), realm, nonce,
                url, qop, nc, cnonce, response, opaque);
    }
}