Merge branch 'weixin_message' into 'CommonTenant'

feat:企业微信通知

See merge request yunteng/thingskit!386

+package org.thingsboard.server.common.data.yunteng.config.enterpriseWeChat;
+
+import lombok.Data;
+import org.thingsboard.server.common.data.yunteng.config.message.MessageProviderConfiguration;
+import org.thingsboard.server.common.data.yunteng.enums.MessageProviderTypeEnum;
+
+@Data
+public class EnterpriseWeChatConfiguration implements MessageProviderConfiguration {
+
+  /** 企业id */
+  private String corpId;
+
+  /** 应用id */
+  private Long agentId;
+
+  /** 企业微信应用Secret */
+  private String corpSecret;
+
+  @Override
+  public MessageProviderTypeEnum getType() {
+    return MessageProviderTypeEnum.ENTERPRISE_WECHAT;
+  }
+}

+package org.thingsboard.server.common.data.yunteng.config.enterpriseWeChat;
+import com.fasterxml.jackson.databind.JsonNode;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.thingsboard.server.common.data.yunteng.config.message.AbstractMessageSender;
+import org.thingsboard.server.common.data.yunteng.core.exception.TkDataValidationException;
+import org.thingsboard.server.common.data.yunteng.core.message.ErrorMessage;
+import org.thingsboard.server.common.data.yunteng.enums.ResponseCodeEnum;
+import org.thingsboard.server.common.data.yunteng.utils.HttpClientUtils;
+import org.thingsboard.server.common.data.yunteng.utils.JacksonUtil;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.LinkedHashMap;
+import java.util.List;
+
+@Slf4j
+public class EnterpriseWeChatSmsSender extends AbstractMessageSender {
+
+  /** 钉钉参数配置 */
+  private final EnterpriseWeChatConfiguration config;
+
+  private String  accessToken;
+
+  private  String url = "https://qyapi.weixin.qq.com/cgi-bin";
+
+  private Long expirationTime;
+//
+//  private static final String smsText="{\n" +
+//          "  \"touser\": \"%s\",\n" +
+//          "  \"msgtype\": \"textcard\",\n" +
+//          "  \"agentid\": %s,\n" +
+//          "  \"textcard\": {\n" +
+//          "\"title\": \"平台告警通知\",\n" +
+//          "\"description\": \"<div class=\\\"normal\\\">设备所属组织：%s</div>" +
+//          "<div class=\\\"normal\\\">设备名称：%s</div>" +
+//          "<div class=\\\"normal\\\">告警等级：%s</div>" +
+//          "<div class=\\\"normal\\\">告警类型：%s</div>" +
+//          "<div class=\\\"normal\\\">触发值：%s</div>" +
+//          "<div class=\\\"normal\\\">告警时间：%s</div>\",\n" +
+//          "    \"url\": \" \",\n" +
+//          "    \"btntxt\": \"详情\"\n" +
+//          "  },\n" +
+//          "  \"enable_id_trans\": 0,\n" +
+//          "  \"enable_duplicate_check\": 0,\n" +
+//          "  \"duplicate_check_interval\": 1800\n" +
+//          "}";
+
+  private static final String smsText="{\n" +
+          "   \"touser\" : \"%s\",\n" +
+          "   \"msgtype\" : \"text\",\n" +
+          "   \"agentid\" : %s,\n" +
+          "   \"text\" : {\n" +
+          "       \"content\" : \"平台告警通知\n设备所属组织：%s\n设备名称：%s\n告警等级：%s\n告警类型：%s\n触发值：%s\n告警时间：%s\n\"\n" +
+          "   },\n" +
+          "   \"safe\":0,\n" +
+          "   \"enable_id_trans\": 0,\n" +
+          "   \"enable_duplicate_check\": 0,\n" +
+          "   \"duplicate_check_interval\": 1800\n" +
+          "}";
+
+
+
+  public EnterpriseWeChatSmsSender(EnterpriseWeChatConfiguration config)   {
+    if (StringUtils.isEmpty(config.getCorpId())
+        || StringUtils.isEmpty(config.getCorpSecret())) {
+      throw new TkDataValidationException(ErrorMessage.INVALID_PARAMETER.getMessage());
+    }
+    this.config = config;
+    getAccessToken();
+  }
+
+  /** 初始化token */
+  private void getAccessToken()  {
+    if(expirationTime!=null){
+      Date now = new Date();
+      if(now.getTime()>expirationTime){
+        getAccessTokenByUrl();
+      }
+    }else{
+      getAccessTokenByUrl();
+    }
+  }
+  private void getAccessTokenByUrl()   {
+    try {
+      String  response = HttpClientUtils.sendGet(url+"/gettoken?"+"corpid="+config.getCorpId()+"&corpsecret="+config.getCorpSecret());
+      JsonNode jsonNode = JacksonUtil.toJsonNode(response);
+      this.accessToken = jsonNode.path("access_token").asText();
+      this.expirationTime = new Date().getTime()+5400*1000;
+    } catch (Exception e) {
+      e.printStackTrace();
+      throw new TkDataValidationException(ErrorMessage.OPERATION_FAILED.getMessage());
+    }
+  }
+
+
+  private String getUserId(String phone)   {
+    List<String> numbers = new ArrayList<>();
+    String [] phones = phone.split(",");
+    for(String onePhone: phones){
+      try {
+        String payload ="{\"mobile\":\""+onePhone+"\"}";
+        String  response = HttpClientUtils.sendPOST(url+"/user/getuserid?access_token="+accessToken,payload);
+        JsonNode jsonNode = JacksonUtil.toJsonNode(response);
+        if(jsonNode.path("userid").asText().isEmpty()){
+          continue;
+        }
+        numbers.add(jsonNode.path("userid").asText());
+      } catch (Exception e) {
+        e.printStackTrace();
+      }
+    }
+    return String.join("|",numbers);
+  }
+
+  @Override
+  public String sendSms(
+       String phone, String templateCode, LinkedHashMap<String, Object> param, String signName) {
+    try{
+      getAccessToken();
+      String payload =  String.format(smsText,getUserId(phone),config.getAgentId(),param.get("organization"),param.get("deviceName"),
+              param.get("severity"),param.get("type"),param.get("triggerValue"),param.get("createTime"));
+      String  response =  HttpClientUtils.sendPOST(
+              url +"/message/send?"+"access_token="+accessToken,payload);
+      JsonNode jsonNode = JacksonUtil.toJsonNode(response);
+      return  jsonNode.path("error-msg").asText().equalsIgnoreCase(ResponseCodeEnum.OK.name())
+         ? ResponseCodeEnum.SUCCESS.name()
+          : jsonNode.path("error-msg").asText();
+    } catch (Exception e) {
+      e.printStackTrace();
+      return e.getMessage();
+    }
+  }
+}

@@ -3,6 +3,7 @@ import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.fasterxml.jackson.annotation.JsonSubTypes;
 import com.fasterxml.jackson.annotation.JsonTypeInfo;
 import org.thingsboard.server.common.data.yunteng.config.dingtalk.DingTalkSmsProviderConfiguration;
+import org.thingsboard.server.common.data.yunteng.config.enterpriseWeChat.EnterpriseWeChatConfiguration;
 import org.thingsboard.server.common.data.yunteng.config.message.ali.AliSmsProviderConfiguration;
 import org.thingsboard.server.common.data.yunteng.config.message.tencent.TencentSmsProviderConfiguration;
 import org.thingsboard.server.common.data.yunteng.config.voice.AliVoiceSmsProviderConfiguration;
@@ -13,7 +14,8 @@ import org.thingsboard.server.common.data.yunteng.enums.MessageProviderTypeEnum;
         @JsonSubTypes.Type(value = AliSmsProviderConfiguration.class, name = "ALI_CLOUD"),
         @JsonSubTypes.Type(value = TencentSmsProviderConfiguration.class, name = "TENCENT_CLOUD"),
         @JsonSubTypes.Type(value = DingTalkSmsProviderConfiguration.class, name = "DING_TALK"),
-        @JsonSubTypes.Type(value = AliVoiceSmsProviderConfiguration.class, name = "ALI_VOICE")
+        @JsonSubTypes.Type(value = AliVoiceSmsProviderConfiguration.class, name = "ALI_VOICE"),
+        @JsonSubTypes.Type(value = EnterpriseWeChatConfiguration.class, name = "ENTERPRISE_WECHAT")
 })
 public interface MessageProviderConfiguration {
 

 package org.thingsboard.server.common.data.yunteng.config.message;
-
 import java.util.LinkedHashMap;
 
 public interface MessageSender {

@@ -3,6 +3,8 @@ package org.thingsboard.server.common.data.yunteng.config.message;
 import org.springframework.stereotype.Component;
 import org.thingsboard.server.common.data.yunteng.config.dingtalk.DingTalkSmsProviderConfiguration;
 import org.thingsboard.server.common.data.yunteng.config.dingtalk.DingTalkSmsSender;
+import org.thingsboard.server.common.data.yunteng.config.enterpriseWeChat.EnterpriseWeChatConfiguration;
+import org.thingsboard.server.common.data.yunteng.config.enterpriseWeChat.EnterpriseWeChatSmsSender;
 import org.thingsboard.server.common.data.yunteng.config.message.ali.AliSmsProviderConfiguration;
 import org.thingsboard.server.common.data.yunteng.config.message.ali.AliSmsSender;
 import org.thingsboard.server.common.data.yunteng.config.message.tencent.TencentSmsProviderConfiguration;
@@ -42,6 +44,11 @@ public class TkDefaultMessageSenderFactory implements MessageSenderFactory {
                 (DingTalkSmsProviderConfiguration) config);
         smsSenderMap.put(key,dingTalkSmsSender);
         return dingTalkSmsSender;
+      case ENTERPRISE_WECHAT:
+        EnterpriseWeChatSmsSender enterpriseWeChatSmsSender = new EnterpriseWeChatSmsSender(
+                (EnterpriseWeChatConfiguration) config);
+        smsSenderMap.put(key,enterpriseWeChatSmsSender);
+        return enterpriseWeChatSmsSender;
       case ALI_VOICE:
         return new AliVoiceSmsSender(
                 (AliVoiceSmsProviderConfiguration) config);

@@ -5,5 +5,6 @@ public enum MessageProviderTypeEnum {
   ALI_CLOUD,
   TENCENT_CLOUD,
   DING_TALK,
+  ENTERPRISE_WECHAT,
   ALI_VOICE
 }

@@ -6,5 +6,6 @@ public enum MessageTypeEnum {
   PHONE_MESSAGE,
   DING_TALK_MESSAGE,
   WECHAT_MESSAGE,
+  ENTERPRISE_WECHAT_MESSAGE,
   VOICE_MESSAGE
 }

@@ -48,14 +48,5 @@ public class HttpClientUtils {
 
     }
 
-    public static void main(String[] args) throws IOException {
-//        String url  = "https://open.ys7.com/api/lapp/token/get?appKey=0cf3f1753e6941e3b5dfe3c00162924e&appSecret=cbe2f41bebee529fde568143109a9903";
-//        String param  = "{\"appKey\":\"0cf3f1753e6941e3b5dfe3c00162924e\",\"appSecret\":\"cbe2f41bebee529fde568143109a9903\"}";
-//        String response = sendPOST(url,param);System.out.println(response);
-        String url2  = "https://open.ys7.com/api/lapp/v2/live/address/get?accessToken=at.cm82uxq261a7ji0xcf32z4su2f5vc066-2krkymy3is-1of7jb5-v1jamwzxw" +
-                "&deviceSerial=E53165567&protocol=4";
-        String response = sendPOST(url2,"");
-        System.out.println(response);
-    }
 
 }

@@ -102,6 +102,12 @@ public class TkMessageServiceImpl implements TkMessageService {
             || null == configJsonNode.get("clientSecret"))) {
       isMatch = false;
     }
+    else if (platForm.equals(MessageProviderTypeEnum.ENTERPRISE_WECHAT.name())
+            && (null == configJsonNode.get("corpId")
+            || null == configJsonNode.get("agentId")
+            || null == configJsonNode.get("corpSecret"))) {
+      isMatch = false;
+    }
     else if (platForm.equals(MessageProviderTypeEnum.ALI_VOICE.name())
             && (null == configJsonNode.get("accessKeyId")
             || null == configJsonNode.get("accessKeySecret"))) {
@@ -117,6 +123,12 @@ public class TkMessageServiceImpl implements TkMessageService {
       keyList.add(configJsonNode.get("clientSecret").toString());
       keyList.add(messageConfig.getTenantId());
     }
+    else if(isMatch&&platForm.equals(MessageProviderTypeEnum.ENTERPRISE_WECHAT.name())){
+      keyList.add(configJsonNode.get("corpId").toString());
+      keyList.add(configJsonNode.get("agentId").toString());
+      keyList.add(configJsonNode.get("corpSecret").toString());
+      keyList.add(messageConfig.getTenantId());
+    }
     MessageProviderConfiguration smsProviderConfiguration =
             JacksonUtil.convertValue(configObjectNode, MessageProviderConfiguration.class);
     if (null != smsProviderConfiguration) {

@@ -153,6 +153,14 @@ public class TkNoticeServiceImpl implements TkNoticeService {
                     organization,
                     contacts);
               }
+                if (messageCode.contains(MessageTypeEnum.ENTERPRISE_WECHAT_MESSAGE.name())
+                        && templatesMap.containsKey(MessageTypeEnum.ENTERPRISE_WECHAT_MESSAGE.name())) {
+                    enterpriseWeChat4Alarm(
+                            alarmInfo,
+                            templatesMap.get(MessageTypeEnum.ENTERPRISE_WECHAT_MESSAGE.name()),
+                            organization,
+                            contacts);
+                }
                 if (messageCode.contains(MessageTypeEnum.VOICE_MESSAGE.name())
                         && templatesMap.containsKey(MessageTypeEnum.VOICE_MESSAGE.name())) {
                     aLiVoice4Alarm(
@@ -268,6 +276,37 @@ public class TkNoticeServiceImpl implements TkNoticeService {
     smsService.sendSms(info);
   }
 
+    /**
+     * 企业微信通知设备告警信息
+     *
+     * @param alarmInfo 告警信息
+     * @param templateId 告警模板主键
+     * @param organization 设备所属组织
+     */
+    private void enterpriseWeChat4Alarm(
+            AlarmInfoDTO alarmInfo,
+            String templateId,
+            TkOrganizationEntity organization,
+            List<TkAlarmContactEntity> contacts) {
+        // TODO 推送企业微信通知
+        SmsReqDTO info = new SmsReqDTO();
+        info.setId(templateId);
+        info.setParams(setAliAlarmParams(alarmInfo, organization));
+        info.setTemplatePurpose(MsgTemplatePurposeEnum.FOR_ALARM_NOTICE.name());
+        List<String> phones = new ArrayList<>();
+        contacts.stream()
+                .parallel()
+                .forEach(
+                        item -> {
+                            if (StringUtils.isEmpty(item.getWechat())&&StringUtils.isEmpty(item.getPhone())) {
+                                return;
+                            }
+                            phones.add(item.getWechat()==null?item.getPhone():item.getWechat());
+                        });
+        String phone = String.join(",",phones);
+        info.setPhoneNumbers(phone);
+        smsService.sendSms(info);
+    }
   /**
    * 微信通知设备告警信息
    *