54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package message
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"github.com/songquanpeng/one-api/common/config"
|
|
"net/http"
|
|
)
|
|
|
|
type request struct {
|
|
Title string `json:"title"`
|
|
Description string `json:"description"`
|
|
Content string `json:"content"`
|
|
URL string `json:"url"`
|
|
Channel string `json:"channel"`
|
|
Token string `json:"token"`
|
|
}
|
|
|
|
type response struct {
|
|
Success bool `json:"success"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
func SendMessage(title string, description string, content string) error {
|
|
if config.MessagePusherAddress == "" {
|
|
return errors.New("message pusher address is not set")
|
|
}
|
|
req := request{
|
|
Title: title,
|
|
Description: description,
|
|
Content: content,
|
|
Token: config.MessagePusherToken,
|
|
}
|
|
data, err := json.Marshal(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resp, err := http.Post(config.MessagePusherAddress,
|
|
"application/json", bytes.NewBuffer(data))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var res response
|
|
err = json.NewDecoder(resp.Body).Decode(&res)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !res.Success {
|
|
return errors.New(res.Message)
|
|
}
|
|
return nil
|
|
}
|