package com.common.core.service;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.common.core.constant.GlobalConst;
import com.common.core.exception.BizException;
import com.common.core.utils.JsonUtils;
import com.common.core.utils.RestUtil;
import com.common.core.utils.SecretsManagerUtils;
import com.common.core.utils.StringUtils;
import com.common.redis.util.RedisUtil;
import com.deloitte.system.request.SFTokenDto;
import lombok.extern.slf4j.Slf4j;
import okhttp3.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.env.Environment;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Service;

import java.util.concurrent.TimeUnit;


@Service
@Slf4j
public class SFService {
    @Value("${salesforce.tokenUrl}")
    private String tokenUrl;

    private OkHttpClient okHttpClient;
    @Autowired
    private Environment env;
    @Autowired
    private SecretsManagerUtils secretsManagerUtils;
    @Autowired
    private RedisUtil redisUtil;
    @Value("${salesforce.baseUrl}")
    private String baseUrl;

    /**
     * 获取salesforce token
     * @return
     */
    public SFTokenDto getToken() {
        SFTokenDto sfTokenDto = redisUtil.getObject(GlobalConst.SF_TOKEN_KEY,SFTokenDto.class);
        okHttpClient= new OkHttpClient.Builder()
                .connectTimeout(60*5, TimeUnit.SECONDS)
                .readTimeout(60*5, TimeUnit.SECONDS)
                .build();
        JSONObject object = secretsManagerUtils.getSecret(env.getProperty("aws.secrets.systemauth"));
        String sfGrantType = object.getString("sf_grant_type");
        String sfClientId = object.getString("sf_client_id");
        String sfClientSecret = object.getString("sf_client_secret");
        String sfUsername = object.getString("sf_username");
        String sfPassword = object.getString("sf_password");
        // 设置请求体
        RequestBody body = new FormBody.Builder()
                .add("grant_type",sfGrantType )
                .add("client_id",sfClientId )
                .add("username", sfUsername)
                .add("password", sfPassword)
                .add("client_secret", sfClientSecret).build();
        Request.Builder reqBuilder = new Request.Builder()
                .url(tokenUrl)
                .post(body)
                .header("Accept", "application/json");
        Call call = okHttpClient.newCall(reqBuilder.build());
        Response response=null;
        String responseBody =null;
        try{
            response = call.execute();
            responseBody=response.body().string();
            log.info("salesforce token ->respData: {}", responseBody);
            if (StringUtils.isBlank(JsonUtils.string2Object(responseBody, SFTokenDto.class).getAccessToken())) {
                log.error("获取Salesforce token失败, AccessToken为空");
            }else {
                sfTokenDto = JsonUtils.string2Object(responseBody, SFTokenDto.class);
                redisUtil.set(GlobalConst.SF_TOKEN_KEY, sfTokenDto, GlobalConst.SF_TOKEN_EXPIRE);
            }
        }catch (Exception e){
            log.error("获取Salesforce token失败: ",e);
        }
        return sfTokenDto;
    }

    public JSONArray querySFData(String sql) {
        SFTokenDto tokenDto = getToken();
        String accessToken = tokenDto.getAccessToken();
        HttpHeaders header = new HttpHeaders();
        header.add("Authorization", "Bearer " + accessToken);
        StringBuilder idString = new StringBuilder();
        // 拼接URL
        StringBuilder queryUrl = idString.append(baseUrl).append("services/data/v54.0/query/?q=").append(sql);
        // 请求数据
        JSONArray jsonArray = null;
        try {
            JSONObject jsonObject = RestUtil.get(queryUrl.toString(),header);
            Object totalSize = jsonObject.get("totalSize");
            Boolean done =(Boolean) jsonObject.get("done");
            String nextRecordsUrl = jsonObject.getString("nextRecordsUrl");
            jsonArray = jsonObject.getJSONArray("records");
            if (totalSize == null) {
                log.warn("请求Salesforce数据失败,result:{}",jsonObject);
                throw new BizException("请求Salesforce数据失败");
            }
        }catch (Exception e){
            log.error("请求Salesforce数据失败",e);
            throw  e;
        }
        return jsonArray;
    }

    /**
     * 请求SF更新数据接口
     *
     * @param tableName 表名
     * @param recordId
     * @param params
     * @return 更新是否成功
     */
    public void updateSFData(String tableName,String recordId,JSONObject params) {
        SFTokenDto tokenDto = getToken();
        String accessToken = tokenDto.getAccessToken();
        HttpHeaders header = new HttpHeaders();
        header.add("Authorization", "Bearer " + accessToken);
        header.add("Content-Type","application/json");
        StringBuilder idString = new StringBuilder();
        // 拼接URL
        StringBuilder url = idString.append(baseUrl).append("services/data/v54.0/sobjects/").append(tableName).append("/").append(recordId);
        log.info("updateSFData 请求地址：{}",url.toString());
        // 请求数据
        ResponseEntity sfRes = RestUtil.patchNative(url.toString(), header, null, params.toJSONString());
        log.info("更新Salesforce数据,result:{}",sfRes.toString());
        int statusCodeValue = sfRes.getStatusCodeValue();
        //请求成功 http status:204
        if(statusCodeValue!=204){
            log.error("更新Salesforce数据失败,result:{}",sfRes.toString());
            throw new BizException("更新Salesforce数据失败");
        }
    }
}