engigu
42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package channels
|
|
|
|
import "github.com/engigu/baihu-panel/internal/sdk/message"
|
|
|
|
type DtalkChannel struct{ *BaseChannel }
|
|
|
|
func NewDtalkChannel() Channel {
|
|
return &DtalkChannel{NewBaseChannel(ChannelDtalk, []string{FormatTypeMarkdown, FormatTypeText})}
|
|
}
|
|
|
|
func (c *DtalkChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
|
|
accessToken := config.GetString("access_token")
|
|
secret := config.GetString("secret")
|
|
|
|
if accessToken == "" {
|
|
return SendError("dtalk config missing: access_token is required"), nil
|
|
}
|
|
|
|
contentType, formattedContent := c.FormatContent(msg)
|
|
atMobiles := msg.GetAtMobiles()
|
|
if msg.AtAll {
|
|
atMobiles = append(atMobiles, "all")
|
|
}
|
|
|
|
cli := message.Dtalk{AccessToken: accessToken, Secret: secret}
|
|
var res []byte
|
|
var err error
|
|
|
|
if contentType == FormatTypeText {
|
|
res, err = cli.SendMessageText(formattedContent, atMobiles...)
|
|
} else if contentType == FormatTypeMarkdown {
|
|
res, err = cli.SendMessageMarkdown(msg.Title, formattedContent, atMobiles...)
|
|
} else {
|
|
return SendError("未知的钉钉发送内容类型：%s", contentType), nil
|
|
}
|
|
|
|
if err != nil {
|
|
return ErrorResult(string(res), err), nil
|
|
}
|
|
return SuccessResult(string(res)), nil
|
|
}
|