drone-matrix/plugin.go

package main

import (
	"strings"

	"github.com/drone/drone-template-lib/template"
	"github.com/matrix-org/gomatrix"
	"github.com/pkg/errors"
	"github.com/microcosm-cc/bluemonday"
	"gopkg.in/russross/blackfriday.v2"
)

type (
	Repo struct {
		Owner string
		Name  string
	}

	Build struct {
		Tag      string
		Event    string
		Number   int
		Commit   string
		Ref      string
		Branch   string
		Author   string
		Message  string
		DeployTo string
		Status   string
		Link     string
		Started  int64
		Created  int64
	}

	Job struct {
		Started int64
	}

	Config struct {
		Username    string
		Password    string
		UserID      string
		AccessToken string
		Homeserver  string
		RoomID      string
		Template    string
	}

	Plugin struct {
		Repo   Repo
		Build  Build
		Job    Job
		Config Config
	}
)

func (p Plugin) Exec() error {
	m, err := gomatrix.NewClient(p.Config.Homeserver, prepend("@", p.Config.UserID), p.Config.AccessToken)

	if err != nil {
		return errors.Wrap(err, "failed to initialize client")
	}

	if p.Config.UserID == "" || p.Config.AccessToken == "" {
		r, err := m.Login(&gomatrix.ReqLogin{
			Type:                     "m.login.password",
			User:                     p.Config.Username,
			Password:                 p.Config.Password,
			InitialDeviceDisplayName: "Drone",
		})

		if err != nil {
			return errors.Wrap(err, "failed to authenticate user")
		}

		m.SetCredentials(r.UserID, r.AccessToken)
	}

	joined, err := m.JoinRoom(prepend("!", p.Config.RoomID), "", nil)

	if err != nil {
		return errors.Wrap(err, "failed to join room")
	}

	message, err := template.RenderTrim(p.Config.Template, p)

	if err != nil {
		return errors.Wrap(err, "failed to render template")
	}

	formatted := bluemonday.UGCPolicy().SanitizeBytes(
		blackfriday.Run([]byte(message)),
	)

	content := gomatrix.HTMLMessage{
		Body: message,
		MsgType: "m.notice",
		Format: "org.matrix.custom.html",
		FormattedBody: string(formatted),
	}

	if _, err := m.SendMessageEvent(joined.RoomID, "m.room.message", content); err != nil {
		return errors.Wrap(err, "failed to submit message")
	}

	return nil
}

func prepend(prefix, s string) string {
	if s == "" {
		return s
	}

	if strings.HasPrefix(s, prefix) {
		return s
	}

	return prefix + s
}