102 lines
2.3 KiB
Go
102 lines
2.3 KiB
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: )(.*)$`
|
|
}
|
|
|
|
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) {}
|
|
|
|
// RespondText to looking up of federation check requests
|
|
func (h *Feder) RespondText(c *gomatrix.Client, ev *gomatrix.Event, _, post string) error {
|
|
homeServer := h.fix(post)
|
|
if homeServer != "" {
|
|
u, err := url.Parse(fmt.Sprintf("https://%s", homeServer))
|
|
if err != nil {
|
|
return SendText(c, ev.RoomID, "that's not a real host name.")
|
|
}
|
|
|
|
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 SendText(c, ev.RoomID, fmt.Sprintf("sorry %s, I can't look up the federation status (%s)", ev.Sender, err))
|
|
}
|
|
|
|
stat := "broken"
|
|
if fed.Status {
|
|
stat = "OK"
|
|
}
|
|
|
|
if fed.Info.Error != "" {
|
|
return SendText(c, ev.RoomID, fmt.Sprintf("%s seems to be broken, maybe it isn't a homeserver?\n%s",
|
|
homeServer, fed.Info.Error))
|
|
} else {
|
|
return SendText(c, ev.RoomID, fmt.Sprintf("%s is running %s (%s) and is %s.",
|
|
homeServer, fed.Info.Name, fed.Info.Version, stat))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Name Feder!
|
|
func (h *Feder) Name() string {
|
|
return "Feder"
|
|
}
|