+dmr plugin

plugins/dmr.go

@@ -0,0 +1,161 @@
+package plugins
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"regexp"
+	"strings"
+
+	"github.com/matrix-org/gomatrix"
+)
+
+// DMRUser represents a response from:
+// https://database.radioid.net/api/dmr/user/
+type DMRUser struct {
+	Count   int `json:"count"`
+	Results []struct {
+		Callsign string `json:"callsign"`
+		City     string `json:"city"`
+		Country  string `json:"country"`
+		Fname    string `json:"fname"`
+		ID       int    `json:"id"`
+		Remarks  string `json:"remarks"`
+		State    string `json:"state"`
+		Surname  string `json:"surname"`
+	} `json:"results"`
+}
+
+// DMRRepeater represents a response from:
+// https://database.radioid.net/api/dmr/repeater/
+type DMRRepeater struct {
+	Count   int `json:"count"`
+	Results []struct {
+		Callsign    string `json:"callsign"`
+		City        string `json:"city"`
+		ColorCode   int    `json:"color_code"`
+		Country     string `json:"country"`
+		Details     string `json:"details"`
+		Frequency   string `json:"frequency"`
+		ID          string `json:"id"`
+		IpscNetwork string `json:"ipsc_network"`
+		Offset      string `json:"offset"`
+		Rfinder     string `json:"rfinder"`
+		State       string `json:"state"`
+		Trustee     string `json:"trustee"`
+		TsLinked    string `json:"ts_linked"`
+	} `json:"results"`
+}
+
+// DMR is our plugin type
+type DMR struct {
+}
+
+// SetStore is the setup function for a plugin
+func (p *DMR) SetStore(s PluginStore) {
+}
+
+// Descr describes this plugin
+func (p *DMR) Descr() string {
+	return "Queries radioid.net"
+}
+
+// Re is what our dmr request matches
+func (p *DMR) Re() string {
+	return `(?i)^dmr (user|repeater) (surname|id|callsign|city|county) (.+)$`
+}
+
+// Match checks for "dmr " messages
+func (p *DMR) Match(_, msg string) bool {
+	re := regexp.MustCompile(p.Re())
+	return re.MatchString(msg)
+}
+
+func (p *DMR) param(msg string) string {
+	re := regexp.MustCompile(p.Re())
+	return re.ReplaceAllString(msg, "$2")
+}
+
+func (p *DMR) mode(msg string) string {
+	re := regexp.MustCompile(p.Re())
+	return re.ReplaceAllString(msg, "$1")
+}
+
+func (p *DMR) query(msg string) string {
+	re := regexp.MustCompile(p.Re())
+	return re.ReplaceAllString(msg, "$3")
+}
+
+// RespondText to looking up of DMR info
+func (p *DMR) RespondText(c *gomatrix.Client, ev *gomatrix.Event, _, post string) error {
+	mode := p.mode(post)
+	param := p.param(post)
+	search := p.query(post)
+
+	endpoint := "https://database.radioid.net/api/dmr/%s/?%s"
+
+	params := url.Values{}
+	params.Set(param, search)
+
+	u := fmt.Sprintf(endpoint, mode, params.Encode())
+
+	fmt.Println(u)
+
+	resp, err := http.Get(u)
+	if err != nil {
+		_ = SendText(c, ev.RoomID, fmt.Sprintf("Can't search radioid.net: %s", err))
+		return err
+	}
+
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+
+	switch mode {
+	case "repeater":
+		var res = &DMRRepeater{}
+		err = json.Unmarshal(body, res)
+		if err != nil {
+			return err
+		}
+		if res.Count == 0 {
+			return SendMD(c, ev.RoomID, fmt.Sprintf("nothing found for '%s'", params.Encode()))
+		}
+
+		var s []string
+		s = append(s, fmt.Sprintf("- **Callsign**: %s", res.Results[0].Callsign))
+		s = append(s, fmt.Sprintf("- **ID**: %s", res.Results[0].ID))
+		s = append(s, fmt.Sprintf("- **Frequency**: %s", res.Results[0].Frequency))
+		s = append(s, fmt.Sprintf("- **Offset**: %s", res.Results[0].Offset))
+
+		return SendMD(c, ev.RoomID, strings.Join(s, "\n"))
+
+	case "user":
+		var res = &DMRUser{}
+		err = json.Unmarshal(body, res)
+		if err != nil {
+			return err
+		}
+		if res.Count == 0 {
+			return SendMD(c, ev.RoomID, fmt.Sprintf("nothing found for '%s'", params.Encode()))
+		}
+
+		var s []string
+		s = append(s, fmt.Sprintf("- **Name**: %s %s", res.Results[0].Fname, res.Results[0].Surname))
+		s = append(s, fmt.Sprintf("- **ID**: %d", res.Results[0].ID))
+		s = append(s, fmt.Sprintf("- **Callsign**: %s", res.Results[0].Callsign))
+
+		return SendMD(c, ev.RoomID, strings.Join(s, "\n"))
+	}
+	return nil
+}
+
+// Name DMR!
+func (p *DMR) Name() string {
+	return "DMR"
+}

plugins/plugins.go

@@ -236,6 +236,7 @@ var Plugs = Plugins{
 	&Beer{},
 	&BotSnack{},
 	&Covid{},
+	&DMR{},
 	&Feder{},
 	&Groan{},
 	&Ham{},