78 lines
2 KiB
Go
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
|
|
}
|