mirror of
https://github.com/songquanpeng/message-pusher.git
synced 2024-05-03 08:45:04 +08:00
87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package channel
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"message-pusher/model"
|
|
"net/http"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
var TelegramMaxMessageLength = 4096
|
|
|
|
type telegramMessageRequest struct {
|
|
ChatId string `json:"chat_id"`
|
|
Text string `json:"text"`
|
|
ParseMode string `json:"parse_mode"`
|
|
}
|
|
|
|
type telegramMessageResponse struct {
|
|
Ok bool `json:"ok"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
func SendTelegramMessage(message *model.Message, user *model.User, channel_ *model.Channel) error {
|
|
// https://core.telegram.org/bots/api#sendmessage
|
|
messageRequest := telegramMessageRequest{
|
|
ChatId: channel_.AccountId,
|
|
}
|
|
if message.To != "" {
|
|
messageRequest.ChatId = message.To
|
|
}
|
|
if message.Content == "" {
|
|
messageRequest.Text = message.Description
|
|
} else {
|
|
messageRequest.Text = message.Content
|
|
messageRequest.ParseMode = "markdown"
|
|
}
|
|
text := messageRequest.Text
|
|
idx := 0
|
|
for idx < len(text) {
|
|
nextIdx := idx + TelegramMaxMessageLength
|
|
if nextIdx > len(text) {
|
|
// we have reach the end, must be valid
|
|
nextIdx = len(text)
|
|
} else {
|
|
nextIdx = getNearestValidSplit(text, nextIdx)
|
|
}
|
|
messageRequest.Text = text[idx:nextIdx]
|
|
idx = nextIdx
|
|
jsonData, err := json.Marshal(messageRequest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resp, err := http.Post(fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", channel_.Secret), "application/json",
|
|
bytes.NewBuffer(jsonData))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var res telegramMessageResponse
|
|
err = json.NewDecoder(resp.Body).Decode(&res)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !res.Ok {
|
|
return errors.New(res.Description)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getNearestValidSplit(s string, idx int) int {
|
|
if idx >= len(s) {
|
|
return idx
|
|
}
|
|
if idx == 0 {
|
|
return 0
|
|
}
|
|
isStartByte := utf8.RuneStart(s[idx])
|
|
if isStartByte {
|
|
return idx
|
|
} else {
|
|
return getNearestValidSplit(s, idx-1)
|
|
}
|
|
}
|