migrate to met norway api

This commit is contained in:
2025-05-11 11:14:38 +01:00
parent 77891fd7fe
commit 2f0aad59cd
2 changed files with 45 additions and 34 deletions

View File

@@ -7,7 +7,8 @@ services:
gid: $GID gid: $GID
environment: environment:
GEMINI_API_KEY: $GEMINI_API_KEY GEMINI_API_KEY: $GEMINI_API_KEY
OPEN_WEATHER_MAP_API_KEY: $OPEN_WEATHER_MAP_API_KEY MET_API_USER_AGENT: $MET_API_USER_AGENT
VAPID_SUBJECT: $VAPID_SUBJECT
VAPID_PRIVATE_KEY_BASE64: $VAPID_PRIVATE_KEY_BASE64 VAPID_PRIVATE_KEY_BASE64: $VAPID_PRIVATE_KEY_BASE64
VAPID_PUBLIC_KEY_BASE64: $VAPID_PUBLIC_KEY_BASE64 VAPID_PUBLIC_KEY_BASE64: $VAPID_PUBLIC_KEY_BASE64
ports: ports:

76
main.go
View File

@@ -28,14 +28,12 @@ import (
"time" "time"
) )
type apiKey struct {
accuWeather string
}
type location struct { type location struct {
displayName string tz *time.Location
lat float32
lon float32
ianaName string ianaName string
key string displayName string
} }
// pageTemplate stores all pre-compiled HTML templates for the application // pageTemplate stores all pre-compiled HTML templates for the application
@@ -70,11 +68,11 @@ type webpushNotificationPayload struct {
} }
type state struct { type state struct {
ctx context.Context ctx context.Context
db *sql.DB db *sql.DB
genai *genai.Client metAPIUserAgent string
apiKey apiKey genai *genai.Client
template pageTemplate template pageTemplate
usePlaceholder bool usePlaceholder bool
@@ -100,10 +98,13 @@ type state struct {
//go:embed web //go:embed web
var webDir embed.FS var webDir embed.FS
var envKeys = []string{"GEMINI_API_KEY", "ACCU_WEATHER_API_KEY", "VAPID_SUBJECT", "VAPID_PRIVATE_KEY_BASE64", "VAPID_PUBLIC_KEY_BASE64"} var envKeys = []string{"GEMINI_API_KEY", "MET_API_USER_AGENT", "VAPID_SUBJECT", "VAPID_PRIVATE_KEY_BASE64", "VAPID_PUBLIC_KEY_BASE64"}
var prompt = `The current time is 5pm. Provide a short summary of the weather forecast in JSON in %v below. Keep it concise. Suggest how to deal with the weather, such as how to dress for the weather, and whether they need an umbrella. var prompt = `The current date and time is %v 7:00am. Provide a short summary of the weather forecast only for today in JSON in %v below.
Use celsius and fahrenheit but not Kelvin for temperature. Mention %v in the summary, but don't add anything else, as the summary will be displayed on a website. Keep it concise. Suggest how to deal with the weather, such as how to dress for the weather, and whether they need an umbrella, but err on the side of caution.
Use celsius and fahrenheit but not Kelvin for temperature.
Mention %v in the summary, but don't add anything else, as the summary will be displayed on a website.
Do not mention today's date or time in the summary.
The summary should be in plaintext for humans. Do not output in JSON.` The summary should be in plaintext for humans. Do not output in JSON.`
var placeholderWeather = map[string]string{ var placeholderWeather = map[string]string{
@@ -120,16 +121,16 @@ var placeholderWeather = map[string]string{
} }
var supportedLocations = map[string]location{ var supportedLocations = map[string]location{
"london": {"London", "Europe/London", "328328"}, "london": {nil, 51.507351, -0.127758, "Europe/London", "London"},
"sf": {"San Francisco", "America/Los_Angeles", "347629"}, "sf": {nil, 37.774929, -122.419418, "America/Los_Angeles", "San Francisco"},
"sj": {"San Jose", "America/Los_Angeles", "347630"}, "sj": {nil, 37.338207, -121.886330, "America/Los_Angeles", "San Jose"},
"la": {"Los Angeles", "America/Los_Angeles", "347625"}, "la": {nil, 34.052235, -118.243683, "America/Los_Angeles", "Los Angeles"},
"nyc": {"New York City", "America/New_York", "14-349727_1_AL"}, "nyc": {nil, 40.712776, -74.005974, "America/New_York", "New York City"},
"tokyo": {"Tokyo", "Asia/Tokyo", "226396"}, "tokyo": {nil, 35.689487, 139.691711, "Asia/Tokyo", "Tokyo"},
"warsaw": {"Warsaw", "Europe/Warsaw", "274663"}, "warsaw": {nil, 52.229675, 21.012230, "Europe/Warsaw", "Warsaw"},
"zurich": {"Zurich", "Europe/Zurich", "316622"}, "zurich": {nil, 47.369019, 8.538030, "Europe/Zurich", "Zurich"},
"berlin": {"Berlin", "Europe/Berlin", "178087"}, "berlin": {nil, 52.520008, 13.404954, "Europe/Berlin", "Berlin"},
"dubai": {"Dubai", "Asia/Dubai", "323091"}, "dubai": {nil, 25.204849, 55.270782, "Asia/Dubai", "Dubai"},
} }
func main() { func main() {
@@ -170,11 +171,9 @@ func main() {
summaryPageTemplate, _ := template.New("summary.html").Parse(string(summaryHTML)) summaryPageTemplate, _ := template.New("summary.html").Parse(string(summaryHTML))
state := state{ state := state{
ctx: ctx, ctx: ctx,
db: db, db: db,
apiKey: apiKey{ metAPIUserAgent: os.Getenv("MET_API_USER_AGENT"),
accuWeather: os.Getenv("ACCU_WEATHER_API_KEY"),
},
template: pageTemplate{ template: pageTemplate{
summary: summaryPageTemplate, summary: summaryPageTemplate,
}, },
@@ -200,6 +199,8 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
loc.tz = l
s, err := gocron.NewScheduler(gocron.WithLocation(l)) s, err := gocron.NewScheduler(gocron.WithLocation(l))
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
@@ -207,7 +208,7 @@ func main() {
_, err = s.NewJob( _, err = s.NewJob(
gocron.DailyJob(1, gocron.NewAtTimes(gocron.NewAtTime(7, 0, 0))), gocron.DailyJob(1, gocron.NewAtTimes(gocron.NewAtTime(7, 0, 0))),
gocron.NewTask(updateSummaries, &state, locKey, &loc), gocron.NewTask(updateSummary, &state, locKey, &loc),
gocron.WithStartAt(gocron.WithStartImmediately()), gocron.WithStartAt(gocron.WithStartImmediately()),
) )
if err != nil { if err != nil {
@@ -551,14 +552,21 @@ func deleteSubscription(state *state, regID uuid.UUID) error {
return err return err
} }
func updateSummaries(state *state, locKey string, loc *location) { func updateSummary(state *state, locKey string, loc *location) {
log.Printf("updating summary for %v...\n", locKey) log.Printf("updating summary for %v...\n", locKey)
var weatherJSON string var weatherJSON string
if state.usePlaceholder { if state.usePlaceholder {
weatherJSON = placeholderWeather[locKey] weatherJSON = placeholderWeather[locKey]
} else { } else {
resp, err := http.Get(fmt.Sprintf("http://dataservice.accuweather.com/forecasts/v1/daily/1day/%v?apikey=%v&detail=true", loc.key, state.apiKey.accuWeather)) req, err := http.NewRequest("GET", fmt.Sprintf("https://api.met.no/weatherapi/locationforecast/2.0/compact?lat=%v&lon=%v", loc.lat, loc.lon), nil)
if err != nil {
log.Printf("error querying weather data for %s: %e\n", locKey, err)
return
}
req.Header.Set("User-Agent", state.metAPIUserAgent)
resp, err := http.DefaultClient.Do(req)
if err != nil { if err != nil {
log.Printf("error updating summaries for %s: %e\n", locKey, err) log.Printf("error updating summaries for %s: %e\n", locKey, err)
return return
@@ -574,9 +582,11 @@ func updateSummaries(state *state, locKey string, loc *location) {
weatherJSON = string(b) weatherJSON = string(b)
} }
date := time.Now().In(loc.tz).Format("2006-02-01")
result, err := state.genai.Models.GenerateContent(state.ctx, "gemini-2.0-flash", []*genai.Content{{ result, err := state.genai.Models.GenerateContent(state.ctx, "gemini-2.0-flash", []*genai.Content{{
Parts: []*genai.Part{ Parts: []*genai.Part{
{Text: fmt.Sprintf(prompt, loc.displayName, loc.displayName)}, {Text: fmt.Sprintf(prompt, date, loc.displayName, loc.displayName)},
{Text: weatherJSON}, {Text: weatherJSON},
}, },
}}, nil) }}, nil)