// Copyright © 2022 siddharth ravikumar <s@ricketyspace.net>
// SPDX-License-Identifier: ISC
package main
import (
"embed"
"flag"
"fmt"
"html/template"
"log"
"net/http"
"regexp"
"strconv"
"strings"
"time"
"ricketyspace.net/peach/nws"
"ricketyspace.net/peach/photon"
"ricketyspace.net/peach/version"
)
// Peach port. Defaults to 8151
var peachPort = flag.Int("p", 8151, "Port to run peach on")
// Peach listen address. Set during init.
var peachAddr = ""
// Holds static content.
//go:embed templates static
var peachFS embed.FS
// HTML templates.
var peachTemplates = template.Must(template.ParseFS(peachFS, "templates/*.tmpl"))
// Lat,Long regex.
var latLngRegex = regexp.MustCompile(`/(-?[0-9]+\.?[0-9]+?),(-?[0-9]+\.?[0-9]+)`)
type Weather struct {
Title string
Version string
Location string
Now WeatherNow
Q2HTimeline WeatherTimeline // Q2H forecast of the next 12 hours.
BiDailyTimeline WeatherTimeline // BiDaily forecast for the next 3 days.
}
type WeatherNow struct {
Temperature int
TemperatureUnit string
Forecast string
WindSpeed string
WindDirection string
}
type WeatherPeriod struct {
Name string
Forecast string
Hour int
Temperature int
TemperatureUnit string
}
type WeatherTimeline struct {
Periods []WeatherPeriod
}
type Search struct {
Title string
Version string
Location string
Message string
MatchingCoords []photon.Coordinates
}
func init() {
flag.Parse()
if *peachPort < 80 {
log.Fatalf("port number is invalid: %v", *peachPort)
}
peachAddr = fmt.Sprintf(":%d", *peachPort)
}
func main() {
// Search handler.
http.HandleFunc("/search", showSearch)
// Default handler.
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
logRequest(r)
if r.URL.Path == "/" {
http.Redirect(w, r, "/41.115,-83.177", 302)
return
}
m := latLngRegex.FindStringSubmatch(r.URL.Path)
if len(m) != 3 || m[0] != r.URL.Path {
http.NotFound(w, r)
return
}
lat, err := strconv.ParseFloat(m[1], 32)
if err != nil {
http.Error(w, err.Error(), 400)
}
lng, err := strconv.ParseFloat(m[2], 32)
if err != nil {
http.Error(w, err.Error(), 400)
}
showWeather(w, float32(lat), float32(lng))
})
// Static files handler.
http.HandleFunc("/static/", serveStaticFile)
// Start server
log.Fatal(http.ListenAndServe(peachAddr, nil))
}
func showWeather(w http.ResponseWriter, lat, lng float32) {
point, err := nws.Points(lat, lng)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
// Get forecast
f, err := nws.GetForecast(point)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
fh, err := nws.GetForecastHourly(point)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
// Make weather
weather, err := NewWeather(point, f, fh)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
// Render.
err = peachTemplates.ExecuteTemplate(w, "weather.tmpl", weather)
if err != nil {
log.Printf("weather: template: %v", err)
return
}
}
func showSearch(w http.ResponseWriter, r *http.Request) {
logRequest(r)
// Search is disabled if photon is not enabled.
if !photon.Enabled() {
http.NotFound(w, r)
return
}
search, err := NewSearch(r)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
err = peachTemplates.ExecuteTemplate(w, "search.tmpl", search)
if err != nil {
log.Printf("search: template: %v", err)
return
}
}
func serveStaticFile(w http.ResponseWriter, r *http.Request) {
logRequest(r)
// Add Cache-Control header
w.Header().Set("Cache-Control", "max-age=604800")
// Serve.
server := http.FileServer(http.FS(peachFS))
server.ServeHTTP(w, r)
}
func NewWeather(point *nws.Point, f, fh *nws.Forecast) (*Weather, error) {
w := new(Weather)
w.Location = fmt.Sprintf("%s, %s",
strings.ToLower(point.Properties.RelativeLocation.Properties.City),
strings.ToLower(point.Properties.RelativeLocation.Properties.State),
)
w.Title = w.Location
w.Version = version.Version
w.Now = WeatherNow{
Temperature: fh.Properties.Periods[0].Temperature,
TemperatureUnit: fh.Properties.Periods[0].TemperatureUnit,
Forecast: fh.Properties.Periods[0].ShortForecast,
WindSpeed: fh.Properties.Periods[0].WindSpeed,
WindDirection: fh.Properties.Periods[0].WindDirection,
}
// Build Q2H timeline for the 12 hours.
q2hPeriods := []WeatherPeriod{}
max := 6
for i, period := range fh.Properties.Periods {
if i%2 != 0 {
continue // Take every other period
}
t, err := time.Parse(time.RFC3339, period.StartTime)
if err != nil {
return nil, err
}
p := WeatherPeriod{
Forecast: period.DetailedForecast,
Hour: t.Hour(),
Temperature: period.Temperature,
TemperatureUnit: period.TemperatureUnit,
}
q2hPeriods = append(q2hPeriods, p)
if len(q2hPeriods) == max {
break
}
}
w.Q2HTimeline = WeatherTimeline{
Periods: q2hPeriods,
}
// Build BiDaily timeline for the next 3 days.
bdPeriods := []WeatherPeriod{}
max = 8
for _, period := range f.Properties.Periods {
p := WeatherPeriod{
Name: period.Name,
Forecast: period.DetailedForecast,
Temperature: period.Temperature,
TemperatureUnit: period.TemperatureUnit,
}
bdPeriods = append(bdPeriods, p)
if len(bdPeriods) == max {
break
}
}
w.BiDailyTimeline = WeatherTimeline{
Periods: bdPeriods,
}
return w, nil
}
func NewSearch(r *http.Request) (*Search, error) {
s := new(Search)
s.Title = "search"
s.Version = version.Version
if r.Method == "GET" {
return s, nil
}
// Get location.
err := r.ParseForm()
if err != nil {
return s, fmt.Errorf("form: %v", err)
}
location := strings.TrimSpace(r.PostForm.Get("location"))
s.Location = location
if len(location) < 2 {
s.Message = "location invalid"
}
// Try to fetch matching coordinates.
s.MatchingCoords, err = photon.Geocode(location)
if err != nil {
log.Printf("search: geocode: %v", err)
s.Message = "unable to lookup location"
return s, nil
}
if len(s.MatchingCoords) < 1 {
s.Message = "location not found"
return s, nil
}
return s, nil
}
func logRequest(r *http.Request) {
addr := r.RemoteAddr
if len(r.Header.Get("X-Forwarded-For")) > 0 {
addr = r.Header.Get("X-Forwarded-For")
}
log.Printf("%v - %v - %v", addr, r.URL, r.Header.Get("User-agent"))
}
