From a38926062fb3e5adcd1d766813907f25064a3a0f Mon Sep 17 00:00:00 2001 From: sneak Date: Tue, 8 Sep 2020 20:20:52 -0700 Subject: [PATCH] rough weather fetcher --- bot/bot.go | 39 ++++++++++++++++++++++++++++++++++++++- 1 file changed, 38 insertions(+), 1 deletion(-) diff --git a/bot/bot.go b/bot/bot.go index 0c3504e..ea0333c 100644 --- a/bot/bot.go +++ b/bot/bot.go @@ -4,6 +4,7 @@ package bot import ( "fmt" "github.com/mattermost/mattermost-server/v5/model" + "net/http" "os" "os/signal" "regexp" @@ -71,7 +72,7 @@ func (b *Bot) Main() int { // Lets create a bot channel for logging debug messages into b.CreateBotDebuggingChannelIfNeeded() - msg := fmt.Sprintf("_**%s** has started up_\n\nrunning on version %s", b.BotName, b.Version) + msg := fmt.Sprintf("_**%s** (version `%s`) is now starting up", b.BotName, b.Version) b.SendMsgToDebuggingChannel(msg, "") // Lets start listening to some channels via the websocket! @@ -229,6 +230,38 @@ func (b *Bot) Shutdown() { syscall.Kill(syscall.Getpid(), syscall.SIGINT) } +func (b *Bot) HandleWeatherRequest(channelid string, postid string, message string) { + msg := fmt.Sprintf("weather request received: `%s`", message) + b.SendMsgToChannel(msg, postid, channelid) + + r := regexp.MustCompile(`metar\s+([A-Za-z]{4})`) + loc := r.FindString(message) + if loc == "" { + b.SendMsgToChannel("error, sorry", postid, channelid) + } + + token := os.Getenv("METAR_API_TOKEN") + url := fmt.Sprintf("https://avwx.rest/api/metar/%s?options=&airport=true&reporting=true&format=json&onfail=cache", loc) + + msg = fmt.Sprintf("calculated url: `%s`", url) + b.SendMsgToChannel(msg, postid, channelid) + + client := http.Client{ + Timeout: 5 * time.Second, + } + req, err := http.NewRequest("GET", url, nil) + req.Header.Add("Authorization", `Token `+token) + resp, err := client.Do(req) + + if err != nil { + b.SendMsgToChannel(fmt.Sprintf("weather fetch error: %s", err), postid, channelid) + return + } + + b.SendMsgToChannel(fmt.Sprintf("weather %s: %s", loc, resp), postid, channelid) + +} + func (b *Bot) HandleMsgFromChannel(event *model.WebSocketEvent) { post := model.PostFromJson(strings.NewReader(event.Data["post"].(string))) @@ -237,6 +270,10 @@ func (b *Bot) HandleMsgFromChannel(event *model.WebSocketEvent) { } //pretty.Print(post) + if matched, _ := regexp.MatchString(`(?:^|\W)metar(?:$|\W)`, post.Message); matched { + b.HandleWeatherRequest(post.ChannelId, post.Id, post.Message) + return + } if matched, _ := regexp.MatchString(`(?:^|\W)alive(?:$|\W)`, post.Message); matched { b.SendMsgToChannel("yes I'm running", post.Id, post.ChannelId)