harness-drone/plugin/notify/gitter.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 == model.StatusStarted && g.Started:
		return g.sendStarted(context)
	case context.Commit.Status == model.StatusSuccess && g.Success:
		return g.sendSuccess(context)
	case context.Commit.Status == model.StatusFailure && 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["Accept"] = "application/json"
	headers["Authorization"] = fmt.Sprintf("Bearer %s", g.Token)

	return sendJson(url, payload, headers)
}