From 950467bd2effd6b6c85b04f0b30f46cad5cfab6c Mon Sep 17 00:00:00 2001
From: sneak <sneak@sneak.berlin>
Date: Tue, 8 Sep 2020 22:03:39 -0700
Subject: [PATCH] add AQI lookup

---
 bot/aqi.go | 84 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 bot/bot.go |  7 ++++-
 2 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 bot/aqi.go

diff --git a/bot/aqi.go b/bot/aqi.go
new file mode 100644
index 0000000..74f274d
--- /dev/null
+++ b/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=%d&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", zip, 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", loc, parsedAQIResponse), postid, channelid)
+}
diff --git a/bot/bot.go b/bot/bot.go
index 955f8a3..56472ef 100644
--- a/bot/bot.go
+++ b/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