90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
|
package marketman
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"math"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
const cacheFilePath = "data/marketman.json"
|
||
|
|
const cacheDuration = 6 * time.Hour
|
||
|
|
|
||
|
|
type walletData struct {
|
||
|
|
Current struct {
|
||
|
|
TotalValueCad float64 `json:"total_value_cad"`
|
||
|
|
} `json:"current"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Provider struct {
|
||
|
|
serverURL string
|
||
|
|
ynabAccount string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Provider) Name() string {
|
||
|
|
return "marketman"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Provider) Configure() error {
|
||
|
|
p.serverURL = os.Getenv("marketman_server")
|
||
|
|
if p.serverURL == "" {
|
||
|
|
return fmt.Errorf("marketman_server environment variable is not set")
|
||
|
|
}
|
||
|
|
|
||
|
|
p.ynabAccount = os.Getenv("marketman_ynab_account")
|
||
|
|
if p.ynabAccount == "" {
|
||
|
|
return fmt.Errorf("marketman_ynab_account environment variable is not set")
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Provider) GetBalances() ([]int, []string, error) {
|
||
|
|
var data walletData
|
||
|
|
|
||
|
|
// Check if cache file exists and is recent enough
|
||
|
|
if stat, err := os.Stat(cacheFilePath); err == nil {
|
||
|
|
if time.Since(stat.ModTime()) < cacheDuration {
|
||
|
|
fileData, err := os.ReadFile(cacheFilePath)
|
||
|
|
if err == nil {
|
||
|
|
if err := json.Unmarshal(fileData, &data); err == nil {
|
||
|
|
cents := int(math.Round(data.Current.TotalValueCad * 100))
|
||
|
|
return []int{cents}, []string{p.ynabAccount}, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fetch fresh data
|
||
|
|
resp, err := http.Get(fmt.Sprintf("%s/api/wallet/data", p.serverURL))
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil, fmt.Errorf("failed to fetch wallet data: %v", err)
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
if resp.StatusCode != http.StatusOK {
|
||
|
|
return nil, nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
|
||
|
|
}
|
||
|
|
|
||
|
|
bodyBytes, err := io.ReadAll(resp.Body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil, fmt.Errorf("failed to read response body: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := json.Unmarshal(bodyBytes, &data); err != nil {
|
||
|
|
return nil, nil, fmt.Errorf("failed to decode wallet data: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save raw response to cache file
|
||
|
|
os.MkdirAll(filepath.Dir(cacheFilePath), 0755)
|
||
|
|
_ = os.WriteFile(cacheFilePath, bodyBytes, 0644)
|
||
|
|
|
||
|
|
cents := int(math.Round(data.Current.TotalValueCad * 100))
|
||
|
|
|
||
|
|
return []int{cents}, []string{p.ynabAccount}, nil
|
||
|
|
}
|