add AQI lookup #4

Manually merged
sneak merged 3 commits from next into master 2020-09-09 05:13:45 +00:00
3 changed files with 107 additions and 1 deletions

View File

@ -2,6 +2,23 @@
Mattermost bot.
# Environment Variables
## for bot comms to mattermost server
* `SCO_API_URL`
* `SCO_WEBSOCKET_URL`
* `SCO_DEBUG_CHANNEL`
* `SCO_ACCOUNT_EMAIL`
* `SCO_ACCOUNT_PASSWORD`
* `SCO_TEAM_NAME`
* `SCO_ACCOUNT_USERNAME`
## remote stuff
* `METAR_API_TOKEN` for https://avwx.rest
* `AIRNOW_API_KEY` for http://www.airnowapi.org
# status
[![Build Status](https://drone.datavi.be/api/badges/sneak/sco/status.svg)](https://drone.datavi.be/sneak/sco)

84
bot/aqi.go Normal file
View File

@ -0,0 +1,84 @@
package bot
//import "github.com/kr/pretty"
import (
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"os"
"regexp"
"time"
"github.com/rs/zerolog/log"
)
type AQIResponse []struct {
DateObserved string `json:"DateObserved"`
HourObserved int `json:"HourObserved"`
LocalTimeZone string `json:"LocalTimeZone"`
ReportingArea string `json:"ReportingArea"`
StateCode string `json:"StateCode"`
Latitude float64 `json:"Latitude"`
Longitude float64 `json:"Longitude"`
ParameterName string `json:"ParameterName"`
AQI int `json:"AQI"`
Category struct {
Number int `json:"Number"`
Name string `json:"Name"`
} `json:"Category"`
}
func (b *Bot) HandleAirQualityRequest(channelid string, postid string, message string) {
// we are using a very bare image with no CA cert bundle
// actually if you docker bind mount the ca cert bundle in the right
// place, golang will find it and use it.
//http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
log.Info().Msgf("aqi request received: `%s`", message)
r := regexp.MustCompile(`aqi\s+([0-9]{5})`)
matches := r.FindStringSubmatch(message)
if len(matches) < 2 {
b.SendMsgToChannel("error, sorry", postid, channelid)
}
zip4 := matches[1]
apikey := os.Getenv("AIRNOW_API_KEY")
url := fmt.Sprintf("http://www.airnowapi.org/aq/observation/zipCode/current/?format=application/json&zipCode=%s&distance=25&API_KEY=%s", zip4, apikey)
log.Info().Msgf("calculated url: `%s`", url)
client := http.Client{
Timeout: 5 * time.Second,
}
req, err := http.NewRequest("GET", url, nil)
resp, err := client.Do(req)
if err != nil {
b.SendMsgToChannel(fmt.Sprintf("aqi fetch error: %s", err), postid, channelid)
return
}
if resp.StatusCode != http.StatusOK {
b.SendMsgToChannel(fmt.Sprintf("aqi fetch error: http status %d", resp.StatusCode), postid, channelid)
return
}
data, _ := ioutil.ReadAll(resp.Body)
var parsedAQIResponse AQIResponse
log.Info().Msgf("aqi %s: %s", zip4, data)
err = nil
err = json.Unmarshal([]byte(data), &parsedAQIResponse)
if err != nil {
b.SendMsgToChannel("error deserializing AQI data", postid, channelid)
return
}
b.SendMsgToChannel(fmt.Sprintf("AQI for `%s`: \n```\n%+v\n```\n", zip4, parsedAQIResponse), postid, channelid)
}

View File

@ -247,11 +247,16 @@ func (b *Bot) HandleMsgFromChannel(event *model.WebSocketEvent) {
return
}
if matched, _ := regexp.MatchString(`(?:^|\W)metar(?:$|\W)`, post.Message); matched {
if matched, _ := regexp.MatchString(`metar\s+[^\s]+`, post.Message); matched {
b.HandleWeatherRequest(post.ChannelId, post.Id, post.Message)
return
}
if matched, _ := regexp.MatchString(`aqi\s+[^\s]+`, post.Message); matched {
b.HandleAirQualityRequest(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)
return