com.feilong.net.wxwork.bot.DefaultWxworkBot Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of feilong Show documentation
feilong is a suite of core and expanded libraries that include utility classes, http, excel,cvs, io classes, and much much more.
There is a newer version: 4.0.8
Show newest version
/*
 * Copyright (C) 2008 feilong
 *
 * 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.feilong.net.wxwork.bot;

import static com.feilong.core.TimeInterval.MILLISECOND_PER_MINUTE;
import static com.feilong.core.bean.ConvertUtil.toList;
import static com.feilong.net.http.HttpClientUtil.getResponseBodyAsString;
import static com.feilong.net.http.HttpMethodType.POST;

import com.feilong.core.Validate;
import com.feilong.json.JsonUtil;
import com.feilong.net.http.ConnectionConfig;
import com.feilong.net.http.HttpRequest;
import com.feilong.net.wxwork.bot.message.BotMessage;
import com.feilong.net.wxwork.bot.message.WxworkResponse;
import com.feilong.net.wxwork.bot.message.markdown.Markdown;
import com.feilong.net.wxwork.bot.message.markdown.WxworkMarkdownMessage;
import com.feilong.net.wxwork.bot.message.news.Article;
import com.feilong.net.wxwork.bot.message.news.News;
import com.feilong.net.wxwork.bot.message.news.WxworkNewsMessage;

/**
 * 默认的微信机器人.
 *
 * @author feilong
 * @since 3.0.9
 */
public class DefaultWxworkBot implements WxworkBot{

    private static final String BOT_WEBHOOK_URL = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send";

    /** The key. */
    private String              key;

    //---------------------------------------------------------------

    /**
     * Instantiates a new default wxwork bot.
     */
    public DefaultWxworkBot(){
        super();
    }

    /**
     * Instantiates a new default wxwork bot.
     *
     * @param key
     *            the key
     */
    public DefaultWxworkBot(String key){
        super();
        this.key = key;
    }

    //---------------------------------------------------------------

    @Override
    public WxworkResponse sendMessage(String content){
        Validate.notBlank(content, "content can't be blank!");
        return pushMessage(new WxworkMarkdownMessage(new Markdown(content)));
    }

    @Override
    public WxworkResponse sendNewsMessage(Article...articles){
        Validate.notEmpty(articles, "articles can't be null/empty!");

        News news = new News(toList(articles));
        return pushMessage(new WxworkNewsMessage(news));
    }

    //---------------------------------------------------------------

    private  WxworkResponse pushMessage(T botMessage){
        String msgtype = botMessage.getMsgtype();
        Validate.notBlank(msgtype, "msgtype can't be blank!");

        //---------------------------------------------------------------
        String url = BOT_WEBHOOK_URL + "?key=" + key;

        HttpRequest httpRequest = new HttpRequest(url, POST);
        httpRequest.setRequestBody(JsonUtil.format(botMessage));

        String json = getResponseBodyAsString(httpRequest, new ConnectionConfig(2 * MILLISECOND_PER_MINUTE));
        return JsonUtil.toBean(json, WxworkResponse.class);
    }

    //---------------------------------------------------------------

    /**
     * Sets the key.
     *
     * @param key
     *            the key to set
     */
    public void setKey(String key){
        this.key = key;
    }

}