Merge branch 'next' of sneak/sco into master

README.md

@@ -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)

bot/aqi.go

@@ -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)
+}

bot/bot.go

@@ -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