package com.ruoyi.system.utils.wx;

import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson2.JSONObject;
import com.icbc.api.internal.util.internal.util.fastjson.JSON;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

import java.text.MessageFormat;
import java.util.Base64;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Component
public class WxQrCodeUtil {

    @Value("${wx.url}")
    private String ACCESS_TOKEN_URL;

    @Value("${wx.conf.secretId}")
    private String memberAppSecret;

    @Value("${wx.conf.appId}")
    private String wxAppId;
    @Autowired
    private RestTemplate restTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    private String getAccessToken() {
        Object o = redisTemplate.opsForValue().get("accessToken");
        if (o != null) {
            return o.toString();
        }
        String requestUrl = MessageFormat.format(ACCESS_TOKEN_URL, wxAppId, memberAppSecret);
        String respBody = restTemplate.getForEntity(requestUrl, String.class).getBody();
        JSONObject jsonObject = JSONObject.parseObject(respBody);
        String string = jsonObject.getString("access_token");
        redisTemplate.opsForValue().set("accessToken", string, 5000, TimeUnit.SECONDS);
        return string;
    }

    /**
     *
     * @param code 推荐码
     * @param type 推荐人类型 1:股东 2:营销员/合伙人
     * @param cacheKey 缓存key
     * @return
     */

    public String getQrCodeBase64(String code, Integer type, String cacheKey){
        String accessToken = getAccessToken();
        String url = "https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=" + accessToken;
        Map<String, Object> param = new HashMap<>();
        param.put("page", "pages/home/home");
        param.put("width", 430); // 二维码尺寸
        param.put("scene", "code=" + code + "&type=" + type); // 二维码尺寸
        param.put("env_version", "trial"); // 二维码尺寸
        param.put("check_path", false); //二维码尺寸
        HttpRequest post = HttpUtil.createPost(url);
        post.body(JSON.toJSONString(param));

        byte[] result = post.execute().bodyBytes();
        // base64
        String img = Base64.getEncoder().encodeToString(result);
        // 后台用户的小程序码
        redisTemplate.opsForValue().set(cacheKey, img);
        return img;
    }
}