MsgPushUtil.java

package cn.xanderye.util;

import com.alibaba.fastjson.JSONObject;
import org.apache.commons.lang3.StringUtils;

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.Base64;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created on 2021/1/23.
 * 服务器消息推送工具类，依赖HttpUtil
 *
 * @author XanderYe
 */
public class MsgPushUtil {

    private static final String SERVER_CHAN_URL = "https://sc.ftqq.com/${scKey}.send";

    private static final String SERVER_CHAN3_URL = " https://${sendkey}.push.ft07.com/send";

    private static final String DB_BOT_URL = "https://oapi.dingtalk.com/robot/send?access_token=${token}";

    private static final String BARK_URL = "https://api.day.app/${deviceId}/${content}";

    private static final String QMSG_PRIVATE_URL = "https://qmsg.zendee.cn/send/${key}";

    private static final String QMSG_GROUP_URL = "https://qmsg.zendee.cn/group/${key}";

    /**
     * Server酱推送
     *
     * @see <a href="http://sc.ftqq.com/3.version">Server酱推送</a>
     * @param scKey
     * @param title
     * @param content
     * @return java.lang.String
     * @author XanderYe
     * @date 2021/1/23
     */
    public static String serverChanPush(String scKey, String title, String content) throws IOException {
        String url = SERVER_CHAN_URL.replace("${scKey}", scKey);
        Map<String, Object> params = new HashMap<>(16);
        params.put("text", title);
        params.put("desp", content);
        return HttpUtil.doPost(url, params).getResponse();
    }

    /**
     *
     * @param sendKey
     * @param tags 标签列表，多个标签使用竖线分隔
     * @param title 推送的标题，如果未提供则使用 text 的内容
     * @param content 推送的内容
     * @return java.lang.String
     * @author XanderYe
     * @date 2024/9/29
     */
    public static String serverChan3Push(String sendKey, String tags, String title, String content) throws IOException {
        String url = SERVER_CHAN3_URL.replace("${sendkey}", sendKey).trim();
        Map<String, Object> params = new HashMap<>(16);
        params.put("tags", tags);
        params.put("title", title);
        params.put("desp", content);
        return HttpUtil.doPost(url, params).getResponse();
    }


    /**
     * 钉钉机器人推送
     *
     * @see <a href="https://developers.dingtalk.com/document/app/custom-robot-access">钉钉机器人开发文档</a>
     * @param token
     * @param secret
     * @param content
     * @param isAtAll      是否@全体
     * @param atMobileList @的手机号列表
     * @return java.lang.String
     * @author XanderYe
     * @date 2021/1/23
     */
    public static String dingTalkBotPush(String token, String secret, String content, boolean isAtAll, List<String> atMobileList) throws IOException, InvalidKeyException, NoSuchAlgorithmException {
        String webhook = DB_BOT_URL.replace("${token}", token);
        if (StringUtils.isNotEmpty(secret)) {
            long timestamp = System.currentTimeMillis();
            String sign = getSign(timestamp, secret);
            webhook = webhook + "&timestamp=" + timestamp + "&sign=" + sign;
        }
        JSONObject atJson = new JSONObject();
        // 是否通知所有人
        atJson.put("isAtAll", isAtAll);
        // 通知人列表
        if (!isAtAll) {
            atJson.put("atMobiles", atMobileList);
        }
        // 消息内容
        JSONObject contentJson = new JSONObject();
        contentJson.put("content", content);
        // 请求体
        JSONObject params = new JSONObject();
        params.put("msgtype", "text");
        params.put("text", contentJson);
        params.put("at", atJson);
        return HttpUtil.doPostJSON(webhook, params.toJSONString()).getResponse();
    }

    /**
     * 钉钉机器人推送
     *
     * @see <a href="https://developers.dingtalk.com/document/app/custom-robot-access">钉钉机器人开发文档</a>
     * @param token
     * @param secret
     * @param title
     * @param content
     * @param isAtAll      是否@全体
     * @param atMobileList @的手机号列表
     * @return java.lang.String
     * @author XanderYe
     * @date 2021/1/23
     */
    public static String dingTalkBotPushMarkdown(String token, String secret, String title, String content, boolean isAtAll, List<String> atMobileList) throws IOException, InvalidKeyException, NoSuchAlgorithmException {
        String webhook = DB_BOT_URL.replace("${token}", token);
        long timestamp = System.currentTimeMillis();
        String sign = getSign(timestamp, secret);
        webhook = webhook + "&timestamp=" + timestamp + "&sign=" + sign;
        JSONObject atJson = new JSONObject();
        // 是否通知所有人
        atJson.put("isAtAll", isAtAll);
        // 通知人列表
        if (!isAtAll) {
            atJson.put("atMobiles", atMobileList);
        }
        // 消息内容
        JSONObject markdown = new JSONObject();
        markdown.put("title", title);
        markdown.put("text", content);
        // 请求体
        JSONObject params = new JSONObject();
        params.put("msgtype", "markdown");
        params.put("markdown", markdown);
        params.put("at", atJson);
        return HttpUtil.doPostJSON(webhook, params.toJSONString()).getResponse();
    }

    /**
     * ios bark推送
     *
     * @see <a href="https://apps.apple.com/cn/app/bark-customed-notifications/id1403753865">Bark</a>
     * @param deviceId
     * @param content
     * @return java.lang.String
     * @author XanderYe
     * @date 2021/1/23
     */
    public static String barkPush(String deviceId, String content, String sound) throws IOException {
        try {
            content = URLEncoder.encode(content, "UTF-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        String url = BARK_URL.replace("${deviceId}", deviceId).replace("${content}", content);
        Map<String, Object> params = null;
        if (null != sound && !"".equals(sound)) {
            params = new HashMap<>(16);
            params.put("sound", sound);
        }
        return HttpUtil.doGet(url, params).getResponse();
    }

    /**
     * QMsg酱推送
     *
     * @see <a href="https://qmsg.zendee.cn/api.html">Qmsg酱接口文档</a>
     * @param key
     * @param msg 消息 (表情：@face=表情ID@，群@用法：@个人@at=群员QQ号@、@全体@at=-1@)
     * @param qq qq号或群号，逗号隔开
     * @param isGroup 是否是群组
     * @return java.lang.String
     * @author XanderYe
     * @date 2021/2/2
     */
    public static String qMsgPush(String key, String msg, String qq, boolean isGroup) throws IOException {
        String url = isGroup ? QMSG_GROUP_URL : QMSG_PRIVATE_URL;
        url = url.replace("${key}", key);
        Map<String, Object> params = new HashMap<>(16);
        params.put("msg", msg);
        params.put("qq", qq);
        return HttpUtil.doPost(url, params).getResponse();
    }

    /**
     * 钉钉机器人签名
     *
     * @param timestamp
     * @param secret
     * @return java.lang.String
     * @author XanderYe
     * @date 2021/1/23
     */
    private static String getSign(long timestamp, String secret) throws NoSuchAlgorithmException, UnsupportedEncodingException, InvalidKeyException {
        String stringToSign = timestamp + "\n" + secret;
        Mac mac = Mac.getInstance("HmacSHA256");
        mac.init(new SecretKeySpec(secret.getBytes(StandardCharsets.UTF_8), "HmacSHA256"));
        byte[] signData = mac.doFinal(stringToSign.getBytes(StandardCharsets.UTF_8));
        return URLEncoder.encode(new String(Base64.getEncoder().encode(signData)), "UTF-8");
    }
}