request.go

package chatbot

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"time"

	"github.com/tidwall/gjson"
)

const (
	defaultTimeOut     = time.Second * 30                     // 默认超时
	urlSendText        = "/api/v1/chat/sendText"              // 发送文本
	urlSendPic         = "/api/v1/chat/sendPic"               // 发送图片
	urlSendEmoji       = "/api/v1/chat/sendEmoji"             // 发送表情
	urlSendVideo       = "/api/v1/chat/sendVideo"             // 发送视频
	urlSendVoice       = "/api/v1/chat/sendVoice"             // 发送语音
	urlSendMiniProgram = "/api/v1/chat/sendSmallApp"          // 发送语音
	urlDownloadImage   = "/api/v1/chat/downloadImage"         // 下载图片
	urlDownloadVideo   = "/api/v1/chat/downloadVideo"         // 下载视频
	urlDownloadVoice   = "/api/v1/chat/downloadVoice"         // 下载音频
	urlDelGroupMember  = "/api/v1/chatroom/delChatRoomMember" // 删除群成员
)

// BotServer 调用机器人http接口的服务
// 主要用于基本的消息发送
type BotServer struct {
	host  string
	token string
}

func newBotServer(host, token string) *BotServer {
	return &BotServer{
		host:  host,
		token: token,
	}
}

// baseRequest 拼接请求
// 接口都为post,token需要在url中携带
func (bs *BotServer) baseRequest(addr string, body []byte, duration time.Duration, APIRsp interface{}) error {
	u := url.URL{
		Scheme: "http",
		Host:   bs.host,
		Path:   addr,
	}
	query, _ := url.ParseQuery("token=" + bs.token)
	u.RawQuery = query.Encode()
	log.Println("request:", u.String())
	req, err := http.NewRequest(http.MethodPost, u.String(), bytes.NewReader(body))
	if err != nil {
		return err
	}
	req.Header.Set("Content-Type", "application/json; charset=utf-8")

	client := http.Client{Timeout: duration}
	rsp, err := client.Do(req)
	if err != nil {
		return err
	}
	if rsp.StatusCode != http.StatusOK {
		return errors.New(fmt.Sprintf("response status code err:%d", rsp.StatusCode))
	}

	rspBody, err := ioutil.ReadAll(rsp.Body)
	if err != nil {
		return err
	}
	_ = rsp.Body.Close()
	if rspBody == nil {
		return errors.New("body is nill")
	}
	bodyJson := gjson.ParseBytes(rspBody)
	rspData := bodyJson.Get("data").String()
	success := bodyJson.Get("code").Int() == 0
	message := bodyJson.Get("msg").String()
	if success {
		return json.Unmarshal([]byte(rspData), APIRsp)
	}
	return errors.New(message)
}

func (bs *BotServer) toJson(req interface{}) []byte {
	j, _ := json.Marshal(req)
	return j
}

// sendTextMessage 发送文本消息
func (bs *BotServer) sendTextMessage(req *SendTextRequest) (*SendTextResponse, error) {
	rsp := &SendTextResponse{}
	err := bs.baseRequest(urlSendText, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// sendEmojiMessage 发送图片
func (bs *BotServer) sendPicMessage(req *SendPicRequest) (*SendPicResponse, error) {
	rsp := &SendPicResponse{}
	err := bs.baseRequest(urlSendPic, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// sendEmojiMessage 发送表情
func (bs *BotServer) sendEmojiMessage(req *SendEmojiRequest) (*SendEmojiResponse, error) {
	rsp := &SendEmojiResponse{}
	err := bs.baseRequest(urlSendEmoji, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// sendVideoMessage 发送视频
func (bs *BotServer) sendVideoMessage(req *SendVideoRequest) (*SendVideoResponse, error) {
	rsp := &SendVideoResponse{}
	err := bs.baseRequest(urlSendVideo, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// sendVoiceMessage 发送音频
func (bs *BotServer) sendVoiceMessage(req *SendVoiceRequest) (*SendVoiceResponse, error) {
	rsp := &SendVoiceResponse{}
	err := bs.baseRequest(urlSendVoice, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// sendMiniProgramMessage 发送小程序
func (bs *BotServer) sendMiniProgramMessage(req *SendMiniProgramRequest) (*SendMiniProgramResponse, error) {
	rsp := &SendMiniProgramResponse{}
	err := bs.baseRequest(urlSendMiniProgram, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// downloadPic 下载图片消息的图片
func (bs *BotServer) downloadPic(req *DownloadImageRequest) (*DownloadImageResponse, error) {
	rsp := &DownloadImageResponse{}
	err := bs.baseRequest(urlDownloadImage, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// downloadPic 下载视频消息的视频
func (bs *BotServer) downloadVideo(req *DownloadVideoRequest) (*DownloadVideoResponse, error) {
	rsp := &DownloadVideoResponse{}
	err := bs.baseRequest(urlDownloadVideo, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// downloadPic 下载语音消息的语音
func (bs *BotServer) downloadVoice(req *DownloadVoiceRequest) (*DownloadVoiceResponse, error) {
	rsp := &DownloadVoiceResponse{}
	err := bs.baseRequest(urlDownloadVoice, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}

// DelGroupRequest 踢出群用户
func (bs *BotServer) delGroupMembers(req *DelGroupRequest) (*DelGroupResponse, error) {
	rsp := &DelGroupResponse{}
	err := bs.baseRequest(urlDelGroupMember, bs.toJson(req), defaultTimeOut, rsp)
	return rsp, err
}