mcchunkie/plugins/federation.go

package plugins

import (
	"fmt"
	"net/url"
	"regexp"
	"time"

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

// ServiceInfo represents the version info from a response
type ServiceInfo struct {
	Name    string `json:"name"`
	Version string `json:"version"`
	Error   string `json:"error"`
}

// FedResp represents a federation statuse response
type FedResp struct {
	Status bool        `json:"FederationOK"`
	Info   ServiceInfo `json:"Version"`
}

// Feder responds to federation check requests
type Feder struct {
}

// Descr describes this plugin
func (h *Feder) Descr() string {
	return "check the Matrix federation status of a given URL."
}

// Re returns the federation check matching string
func (h *Feder) Re() string {
	return `(?i)^(?:feder: |tayshame: |danshame: )(.*)$`
}

func (h *Feder) fix(msg string) string {
	re := regexp.MustCompile(h.Re())
	return re.ReplaceAllString(msg, "$1")
}

// Match determines if we should call the response for Feder
func (h *Feder) Match(_, msg string) bool {
	re := regexp.MustCompile(h.Re())
	return re.MatchString(msg)
}

// SetStore we don't need a store here.
func (h *Feder) SetStore(_ PluginStore) {}

func (h *Feder) Process(from, post string) string {
	homeServer := h.fix(post)
	if homeServer != "" {
		u, err := url.Parse(fmt.Sprintf("https://%s", homeServer))
		if err != nil {
			return fmt.Sprintf("that's not a real host name: %q", homeServer)
		}

		homeServer = u.Hostname()

		furl := fmt.Sprintf("%s%s",
			"https://federationtester.matrix.org/api/report?server_name=",
			url.PathEscape(homeServer),
		)
		var fed = &FedResp{}

		var req = HTTPRequest{
			Timeout: 5 * time.Second,
			URL:     furl,
			Method:  "GET",
			ResBody: fed,
		}
		err = req.DoJSON()

		if err != nil {
			return fmt.Sprintf("sorry %s, I can't look up the federation status (%s)", from, err)
		}

		stat := "broken"
		if fed.Status {
			stat = "OK"
		}

		if fed.Info.Error != "" {
			return fmt.Sprintf("%s seems to be broken, maybe it isn't a homeserver?\n%s", homeServer, fed.Info.Error)
		} else {
			return fmt.Sprintf("%s is running %s (%s) and is %s.", homeServer, fed.Info.Name, fed.Info.Version, stat)
		}
	}
	return "invalid hostname"
}

// RespondText to looking up of federation check requests
func (h *Feder) RespondText(c *gomatrix.Client, ev *gomatrix.Event, _, post string) error {
	return SendText(c, ev.RoomID, h.Process(ev.Sender, post))
}

// Name Feder!
func (h *Feder) Name() string {
	return "Feder"
}