woodpecker/plugin/notify/gitter.go
2014-10-02 18:26:47 +04:00

78 lines
2 KiB
Go

package notify
import (
"encoding/json"
"fmt"
"github.com/drone/drone/shared/model"
)
const (
gitterEndpoint = "https://api.gitter.im/v1/rooms/%s/chatMessages"
gitterStartedMessage = "*Building* %s, commit [%s](%s), author %s"
gitterSuccessMessage = "*Success* %s, commit [%s](%s), author %s"
gitterFailureMessage = "*Failed* %s, commit [%s](%s), author %s"
)
type Gitter struct {
RoomID string `yaml:"room_id,omitempty"`
Token string `yaml:"token,omitempty"`
Started bool `yaml:"on_started,omitempty"`
Success bool `yaml:"on_success,omitempty"`
Failure bool `yaml:"on_failure,omitempty"`
}
func (g *Gitter) Send(context *model.Request) error {
switch {
case context.Commit.Status == "Started" && g.Started:
return g.sendStarted(context)
case context.Commit.Status == "Success" && g.Success:
return g.sendSuccess(context)
case context.Commit.Status == "Failure" && g.Failure:
return g.sendFailure(context)
}
return nil
}
func (g *Gitter) getMessage(context *model.Request, message string) string {
url := getBuildUrl(context)
return fmt.Sprintf(message, context.Repo.Name, context.Commit.ShaShort(), url, context.Commit.Author)
}
func (g *Gitter) sendStarted(context *model.Request) error {
return g.send(g.getMessage(context, gitterStartedMessage))
}
func (g *Gitter) sendSuccess(context *model.Request) error {
return g.send(g.getMessage(context, gitterSuccessMessage))
}
func (g *Gitter) sendFailure(context *model.Request) error {
return g.send(g.getMessage(context, gitterFailureMessage))
}
// helper function to send HTTP requests
func (g *Gitter) send(msg string) error {
// data will get posted in this format
data := struct {
Text string `json:"text"`
}{msg}
// data json encoded
payload, err := json.Marshal(data)
if err != nil {
return err
}
// send payload
url := fmt.Sprintf(gitterEndpoint, g.RoomID)
// create headers
headers := make(map[string]string)
headers["Authorization"] = fmt.Sprintf("Bearer %s", g.Token)
go sendJson(url, payload, headers)
return nil
}