flesh out some struct action

This commit is contained in:
Aaron Bieber 2020-02-01 15:31:56 -07:00
parent 608806becb
commit ca8a9d0b19

View File

@ -1,6 +1,7 @@
package plugins
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
@ -15,28 +16,58 @@ import (
type Beer struct {
}
func (h *Beer) match(msg string) string {
// BeerRecord is the parent beer type
type BeerRecord struct {
DatasetID string `json:"datasetid"`
RecordID string `json:"recordid"`
Fields BeerFields `json:"fields"`
}
// BeerFields ar the fields we care about for any given beer
type BeerFields struct {
Website string `json:"website"`
City string `json:"city"`
Name string `json:"name"`
Country string `json:"country"`
Descr string `json:"descript"`
Style string `json:"style_name"`
IBU int `json:"ibu"`
BreweryName string `json:"name_breweries"`
}
// BeerRecords are a collection of responses
type BeerRecords []BeerRecord
func (h *Beer) fix(msg string) string {
re := regexp.MustCompile(`(?i)^beer: `)
return re.ReplaceAllString(msg, "$1")
}
func (h *Beer) get(beer string) (string, error) {
func (h *Beer) match(msg string) bool {
re := regexp.MustCompile(`(?i)^beer: `)
return re.MatchString(msg)
}
func (h *Beer) get(beer string) (*BeerRecords, error) {
u := "https://data.opendatasoft.com/api/records/1.0/search?dataset=open-beer-database%40public-us&q="
u = fmt.Sprintf("%s%s", u, url.PathEscape(beer))
log.Println(u)
resp, err := http.Get(u)
if err != nil {
return "", err
return nil, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return "", err
return nil, err
}
return string(body), nil
beers := &BeerRecords{}
_ = json.Unmarshal([]byte(body), beers)
return beers, nil
}
// Respond to hi events
@ -45,14 +76,16 @@ func (h *Beer) Respond(c *gomatrix.Client, ev *gomatrix.Event, user string) {
switch mtype {
case "m.text":
if post, ok := ev.Body(); ok {
beer := h.match(post)
if beer != "" {
log.Printf("%s: responding to '%s'", h.Name(), ev.Sender)
json, err := h.get(beer)
if err != nil {
SendMessage(c, ev.RoomID, fmt.Sprintf("sorry %s, I can't look for beer. (%s)", ev.Sender, err))
if h.match(post) {
beer := h.fix(post)
if beer != "" {
log.Printf("%s: responding to '%s'", h.Name(), ev.Sender)
j, err := h.get(beer)
if err != nil {
SendMessage(c, ev.RoomID, fmt.Sprintf("sorry %s, I can't look for beer. (%s)", ev.Sender, err))
}
SendMessage(c, ev.RoomID, fmt.Sprintf("%v", j))
}
SendMessage(c, ev.RoomID, fmt.Sprintf("%s!", json))
}
}
}