61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
|
package alert
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.dev.m-and-m.ovh/mderasse/tesla/api"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
var apiClient *api.Client
|
||
|
|
|
||
|
|
// Init will initialize telegram bot.
|
||
|
|
func Init(alertChan chan api.Availability) {
|
||
|
|
log.Info("Starting alert ticker")
|
||
|
|
|
||
|
|
var err error
|
||
|
|
apiClient, err = api.NewClient()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Fail to instantiate the HTTP Client. Error: %s", err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
priceAlertStr := os.Getenv("PRICE_ALERT")
|
||
|
|
if priceAlertStr != "" {
|
||
|
|
PriceAlert, err = strconv.ParseInt(priceAlertStr, 10, 64)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("invalid configuration. PRICE_ALERT environement variable should be an integer.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// define an interval and the ticker for this interval
|
||
|
|
interval := time.Duration(1) * time.Minute
|
||
|
|
// create a new Ticker
|
||
|
|
tk := time.NewTicker(interval)
|
||
|
|
|
||
|
|
for range tk.C {
|
||
|
|
checkPrice(alertChan)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func checkPrice(alertChan chan api.Availability) {
|
||
|
|
log.Info("Ticker - Checking price")
|
||
|
|
|
||
|
|
availabilities, err := apiClient.GetAvailabilities(context.Background(), &api.AvailabilityParams{
|
||
|
|
Query: carFilter,
|
||
|
|
Count: 1,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
log.Warnf("Fail to contact API. Error: %s", err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if availabilities.Results[0].Price < PriceAlert {
|
||
|
|
log.Info("Launching an alert !")
|
||
|
|
alertChan <- availabilities.Results[0]
|
||
|
|
}
|
||
|
|
}
|