All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.jaemon.dingtalk.DingTalkRobot Maven / Gradle / Ivy

There is a newer version: 3.1.2-RELEASE
Show newest version
/*
 * Copyright 2015-2020 Jaemon([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.jaemon.dingtalk;

import com.alibaba.fastjson.JSON;
import com.jaemon.dingtalk.dinger.DingerConfig;
import com.jaemon.dingtalk.entity.*;
import com.jaemon.dingtalk.entity.enums.ContentTypeEnum;
import com.jaemon.dingtalk.entity.enums.MsgTypeEnum;
import com.jaemon.dingtalk.entity.enums.ResultCode;
import com.jaemon.dingtalk.entity.message.Message;
import com.jaemon.dingtalk.entity.message.MsgType;
import com.jaemon.dingtalk.exception.AsyncCallException;
import com.jaemon.dingtalk.exception.MsgTypeException;
import com.jaemon.dingtalk.exception.SendMsgException;
import com.jaemon.dingtalk.support.CustomMessage;
import com.jaemon.dingtalk.utils.ConfigTools;
import com.jaemon.dingtalk.utils.DingTalkUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * DingTalk Robot
 *
 * @author Jaemon@[email protected]
 * @version 1.0
 */
public class DingTalkRobot extends AbstractDingTalkSender {

    public DingTalkRobot(DingTalkProperties dingTalkProperties, DingTalkManagerBuilder dingTalkManagerBuilder) {
        super(dingTalkProperties, dingTalkManagerBuilder);
    }

    @Override
    public DingTalkResult send(MsgTypeEnum msgType, String keyword, String subTitle, String content) {
        CustomMessage customMessage;
        try {
            customMessage = checkMsgType(msgType);
        } catch (MsgTypeException ex) {
            return exceptionResult(keyword, content, ex);
        }
        Message message = msgType.message(customMessage, keyword, subTitle, content, dingTalkProperties, null);

        return send(keyword, message);
    }


    @Override
    public DingTalkResult send(MsgTypeEnum msgType, String keyword, String subTitle, String content, List phones) {
        CustomMessage customMessage;
        try {
            customMessage = checkMsgType(msgType);
        } catch (MsgTypeException ex) {
            return exceptionResult(keyword, content, ex);
        }
        Message message = msgType.message(customMessage, keyword, subTitle, content, dingTalkProperties, phones);;
        message.setAt(new Message.At(phones));

        return send(keyword, message);
    }


    @Override
    public DingTalkResult sendAll(MsgTypeEnum msgType, String keyword, String subTitle, String content) {
        CustomMessage customMessage;
        try {
            customMessage = checkMsgType(msgType);
        } catch (MsgTypeException ex) {
            return exceptionResult(keyword, content, ex);
        }
        Message message = msgType.message(customMessage, keyword, subTitle, content, dingTalkProperties, null);
        message.setAt(new Message.At(true));

        return send(keyword, message);
    }


    @Override
    public DingTalkResult send(String keyword, Message message) {
        return send(keyword, JSON.toJSONString(message));
    }



    @Override
    public  DingTalkResult send(String keyword, T message) {
        return send(keyword, JSON.toJSONString(message));
    }

    @Override
    public DingTalkResult send(String keyword, String message) {
        String dkid = dingTalkManagerBuilder.dkIdGenerator.dkid();
        if (!dingTalkProperties.isEnabled()) {
            return DingTalkResult.failed(ResultCode.DINGTALK_DISABLED, dkid);
        }

        DingerConfig localDinger = getLocalDinger();
        // dinger is null? use global configuration and check whether dinger send
        boolean dingerConfig = localDinger != null;
        try {
            String tokenId;
            boolean isDecryptProp = dingTalkProperties.isDecrypt();
            String decryptKeyProp = dingTalkProperties.getDecryptKey();
            String tokenIdProp = dingTalkProperties.getTokenId();
            String secretProp = dingTalkProperties.getSecret();
            boolean isAsyncProp = dingerConfig ? localDinger.getAsyncExecute() : dingTalkProperties.isAsync();

            // deal with tokenId
            if (dingerConfig && !DingTalkUtils.isEmpty(localDinger.getTokenId())) {
                if (!DingTalkUtils.isEmpty(localDinger.getDecryptKey())) {
                    tokenId = ConfigTools.decrypt(localDinger.getDecryptKey(), localDinger.getTokenId());
                } else {
                    tokenId = localDinger.getTokenId();
                }
            } else {
                // inner decrypt
                if (isDecryptProp) {
                    tokenId = ConfigTools.decrypt(decryptKeyProp, tokenIdProp);
                } else {
                    tokenId = tokenIdProp;
                }
            }

            StringBuilder webhook = new StringBuilder();
            webhook.append(dingTalkProperties.getRobotUrl()).append("=").append(tokenId);

            if (dingerConfig && !DingTalkUtils.isEmpty(localDinger.getSecret())) {
                secretProp = localDinger.getSecret();
            }

            // 处理签名问题
            if (!DingTalkUtils.isEmpty(secretProp)) {
                SignBase sign = dingTalkManagerBuilder.dkSignAlgorithm.sign(secretProp.trim());
                webhook.append(sign.transfer());
            }

            RequestHeader headers = new RequestHeader();
            RequestHeader.Pairs pairs = new RequestHeader.Pairs("Content-Type", "application/json; charset=utf-8");
            ArrayList list = new ArrayList<>();
            list.add(pairs);
            headers.setData(list);

            // 异步处理, 直接返回标识id
            if (isAsyncProp) {
                dingTalkManagerBuilder.dingTalkExecutor.execute(() -> {
                    try {
                        String result = dingTalkManagerBuilder.httpClient.doPost(webhook.toString(), headers, message, ContentTypeEnum.JSON.mediaType());
                        dingTalkManagerBuilder.dkCallable.execute(dkid, result);
                    } catch (Exception e) {
                        AsyncCallException ex = new AsyncCallException(e);
                        DkExCallable dkExCallable = DkExCallable.builder()
                                .dkid(dkid)
                                .dingTalkProperties(dingTalkProperties)
                                .keyword(keyword)
                                .message(message)
                                .ex(ex).build();
                        dingTalkManagerBuilder.notice.callback(dkExCallable);
                    }
                });
                return DingTalkResult.success(dkid, dkid);
            }
            String response = dingTalkManagerBuilder.httpClient.doPost(webhook.toString(), headers, message, ContentTypeEnum.JSON.mediaType());
            return DingTalkResult.success(dkid, response);
        } catch (Exception e) {
            SendMsgException ex = new SendMsgException(e);
            DkExCallable dkExCallable = DkExCallable.builder()
                    .dkid(dkid)
                    .dingTalkProperties(dingTalkProperties)
                    .keyword(keyword)
                    .message(message)
                    .ex(ex).build();
            dingTalkManagerBuilder.notice.callback(dkExCallable);
            return DingTalkResult.failed(ResultCode.SEND_MESSAGE_FAILED, dkid);
        }
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy