drone-matrix/plugin.go

package main

import (
	"fmt"
	"strings"

	"github.com/matrix-org/gomatrix"
)

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 err
	}

	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 err
		}

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

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

	if err != nil {
		return err
	}

	message := message(p.Repo, p.Build)

	if p.Config.Template != "" {
		if message, err = RenderTrim(p.Config.Template, p); err != nil {
			return err
		}

		if err != nil {
			return err
		}
	}

	if _, err := m.SendNotice(joined.RoomID, message); err != nil {
		return err
	}

	return nil
}

func message(repo Repo, build Build) string {
	return fmt.Sprintf(
		"Build %s <%s|%s/%s#%s> (%s) by %s",
		build.Status,
		build.Link,
		repo.Owner,
		repo.Name,
		build.Commit[:8],
		build.Branch,
		build.Author,
	)
}

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

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

	return prefix + s
}