package com.qianwen.core.sms;

import com.yunpian.sdk.YunpianClient;
import com.yunpian.sdk.model.Result;
import com.yunpian.sdk.model.SmsBatchSend;
import java.time.Duration;
import java.util.Collection;
import java.util.Map;
import com.qianwen.core.redis.cache.BladeRedis;
import com.qianwen.core.sms.model.SmsCode;
import com.qianwen.core.sms.model.SmsData;
import com.qianwen.core.sms.model.SmsResponse;
import com.qianwen.core.sms.props.SmsProperties;
import com.qianwen.core.tool.support.Kv;
import com.qianwen.core.tool.utils.PlaceholderUtil;
import com.qianwen.core.tool.utils.StringUtil;


public class YunpianSmsTemplate implements SmsTemplate {
    private final SmsProperties smsProperties;
    private final YunpianClient client;
    private final BladeRedis bladeRedis;

    public YunpianSmsTemplate(final SmsProperties smsProperties, final YunpianClient client, final BladeRedis bladeRedis) {
        this.smsProperties = smsProperties;
        this.client = client;
        this.bladeRedis = bladeRedis;
    }

    @Override // com.qianwen.core.sms.SmsTemplate
    public SmsResponse sendMessage(SmsData smsData, Collection<String> phones) {
        String templateId = this.smsProperties.getTemplateId();
        String templateText = PlaceholderUtil.getResolver("#", "#").resolveByMap(templateId, Kv.create().setAll(smsData.getParams()));
        Map<String, String> param = this.client.newParam(2);
        param.put("mobile", StringUtil.join(phones));
        param.put("text", templateText);
        Result<SmsBatchSend> result = this.client.sms().multi_send(param);
        return new SmsResponse(result.getCode().intValue() == 0, result.getCode(), result.toString());
    }

    @Override // com.qianwen.core.sms.SmsTemplate
    public SmsCode sendValidate(SmsData smsData, String phone) {
        SmsCode smsCode = new SmsCode();
        boolean temp = sendSingle(smsData, phone);
        if (temp && StringUtil.isNotBlank(smsData.getKey())) {
            String id = StringUtil.randomUUID();
            String value = smsData.getParams().get(smsData.getKey());
            this.bladeRedis.setEx(cacheKey(phone, id), value, Duration.ofMinutes(30L));
            smsCode.setId(id).setValue(value);
        } else {
            smsCode.setSuccess(Boolean.FALSE.booleanValue());
        }
        return smsCode;
    }

    @Override // com.qianwen.core.sms.SmsTemplate
    public boolean validateMessage(SmsCode smsCode) {
        String id = smsCode.getId();
        String value = smsCode.getValue();
        String phone = smsCode.getPhone();
        String cache = (String) this.bladeRedis.get(cacheKey(phone, id));
        if (StringUtil.isNotBlank(value) && StringUtil.equalsIgnoreCase(cache, value)) {
            this.bladeRedis.del(cacheKey(phone, id));
            return true;
        }
        return false;
    }
}