add webserver for refresh hook
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Steven Polley 2023-11-13 14:07:57 -07:00
parent 7a2e79682c
commit a95df7c42b
2 changed files with 52 additions and 20 deletions

45
main.go
View File

@ -50,28 +50,33 @@ func init() {
} }
} }
// Main program loop
func main() { func main() {
for { go webServer()
// Loop through each configured account provider and attempt to get the account balances, and update YNAB
for _, p := range configuredProviders { for { // Main program loop
balances, accountIDs, err := p.GetBalances() refreshData()
if err != nil {
log.Printf("failed to get balances with provider '%s': %v", p.Name(), err)
continue
}
if len(balances) != len(accountIDs) {
log.Printf("'%s' provider data validation error: mismatched balance and accountID slice lengths - expected the same: balances length = %d, accountIDs length = %d", p.Name(), len(balances), len(accountIDs))
continue
}
for i := range balances {
err = ynabClient.SetAccountBalance(accountIDs[i], balances[i])
if err != nil {
log.Printf("failed to update ynab account '%s' balance: %v", accountIDs[i], err)
}
}
}
log.Print("Sleeping for 6 hours...") log.Print("Sleeping for 6 hours...")
time.Sleep(time.Hour * 6) time.Sleep(time.Hour * 6)
} }
} }
func refreshData() {
// Loop through each configured account provider and attempt to get the account balances, and update YNAB
for _, p := range configuredProviders {
balances, accountIDs, err := p.GetBalances()
if err != nil {
log.Printf("failed to get balances with provider '%s': %v", p.Name(), err)
continue
}
if len(balances) != len(accountIDs) {
log.Printf("'%s' provider data validation error: mismatched balance and accountID slice lengths - expected the same: balances length = %d, accountIDs length = %d", p.Name(), len(balances), len(accountIDs))
continue
}
for i := range balances {
err = ynabClient.SetAccountBalance(accountIDs[i], balances[i])
if err != nil {
log.Printf("failed to update ynab account '%s' balance: %v", accountIDs[i], err)
}
}
}
}

27
webServer.go Normal file
View File

@ -0,0 +1,27 @@
package main
import (
"fmt"
"log"
"net/http"
)
func webServer() {
// Start web server
//Public static files
http.Handle("/public/", http.StripPrefix("/public/", http.FileServer(http.Dir("public"))))
// Page Handlers
// Anything that is responsible for the base elements of a viewable web page
http.HandleFunc("/", homePageHandler)
log.Print("Service listening on :8080")
log.Printf("Web server unexpectedly exiting!: %v", http.ListenAndServe(":8080", nil))
}
func homePageHandler(w http.ResponseWriter, r *http.Request) {
refreshData()
http.Redirect(w, r, fmt.Sprintf("https://app.ynab.com/%s", ynabClient.BudgetID), http.StatusSeeOther)
}