message-pusher/channel/custom.go

package channel

import (
	"bytes"
	"errors"
	"net/http"
	"os"
	"strings"

	"message-pusher/common"
	"message-pusher/model"
)

func SendCustomMessage(message *model.Message, user *model.User, channel_ *model.Channel) error {
	url := channel_.URL
	if strings.HasPrefix(url, "http:") && os.Getenv("CHANNEL_URL_ALLOW_NON_HTTPS") != "true" {
		return errors.New("自定义通道必须使用 HTTPS 协议")
	}
	if strings.HasPrefix(url, common.ServerAddress) {
		return errors.New("自定义通道不能使用本服务地址")
	}
	template := channel_.Other
	template = common.Replace(template, "$url", message.URL, -1)
	template = common.Replace(template, "$to", message.To, -1)
	template = common.Replace(template, "$title", message.Title, -1)
	template = common.Replace(template, "$description", message.Description, -1)
	template = common.Replace(template, "$content", message.Content, -1)
	reqBody := []byte(template)
	resp, err := http.Post(url, "application/json", bytes.NewReader(reqBody))
	if err != nil {
		return err
	}
	if resp.StatusCode != 200 {
		return errors.New(resp.Status)
	}
	return nil
}