Aaron Bieber
d6673169be
Expand sms to run Process for each plugin, add a "homestead" plugin while here! \o/
114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package plugins
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/matrix-org/gomatrix"
|
|
)
|
|
|
|
// HomesteadResp is the json returned from our api
|
|
type HomesteadResp struct {
|
|
Status string `json:"status"`
|
|
Data struct {
|
|
Resulttype string `json:"resultType"`
|
|
Result []struct {
|
|
Metric struct {
|
|
Instance string `json:"instance"`
|
|
Job string `json:"job"`
|
|
Name string `json:"name"`
|
|
} `json:"metric"`
|
|
Value []interface{} `json:"value"`
|
|
} `json:"result"`
|
|
} `json:"data"`
|
|
}
|
|
|
|
// Homestead is our plugin type
|
|
type Homestead struct {
|
|
db PluginStore
|
|
}
|
|
|
|
// SetStore is the setup function for a plugin
|
|
func (h *Homestead) SetStore(s PluginStore) {
|
|
h.db = s
|
|
}
|
|
|
|
func (h *Homestead) get(loc string) (*HomesteadResp, error) {
|
|
u := "https://graph.tapenet.org/pub"
|
|
resp, err := http.Get(u)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var w = &HomesteadResp{}
|
|
err = json.Unmarshal(body, w)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return w, nil
|
|
}
|
|
|
|
// Descr describes this plugin
|
|
func (h *Homestead) Descr() string {
|
|
return "Display weather information for the Homestead"
|
|
}
|
|
|
|
// Re is what our weather matches
|
|
func (h *Homestead) Re() string {
|
|
return `(?i)^home:|^homestead:\s?(\w+)?$`
|
|
}
|
|
|
|
// Match checks for "home: name?" messages
|
|
func (h *Homestead) Match(_, msg string) bool {
|
|
re := regexp.MustCompile(h.Re())
|
|
return re.MatchString(msg)
|
|
}
|
|
|
|
func (h *Homestead) fix(msg string) string {
|
|
re := regexp.MustCompile(h.Re())
|
|
return re.ReplaceAllString(msg, "$1")
|
|
}
|
|
|
|
func (h *Homestead) Process(from, post string) string {
|
|
weather := h.fix(post)
|
|
var s []string
|
|
wd, err := h.get(weather)
|
|
if err != nil {
|
|
return fmt.Sprintf("sorry %s, I can't connect to the homestead. %q", from, err)
|
|
}
|
|
|
|
for _, e := range wd.Data.Result {
|
|
if temp, err := strconv.ParseFloat(e.Value[1].(string), 64); err == nil {
|
|
s = append(s, fmt.Sprintf("%s: %.2fC (%.2fF)", e.Metric.Name, temp, (temp*1.8000)+32.00))
|
|
} else {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
return strings.Join(s, ", ")
|
|
}
|
|
|
|
// RespondText to looking up of weather lookup requests
|
|
func (h *Homestead) RespondText(c *gomatrix.Client, ev *gomatrix.Event, _, post string) error {
|
|
return SendText(c, ev.RoomID, h.Process(ev.Sender, post))
|
|
}
|
|
|
|
// Name Homestead!
|
|
func (h *Homestead) Name() string {
|
|
return "Homestead"
|
|
}
|