Steven Polley
4c01dc4ea2
All checks were successful
continuous-integration/drone/push Build is passing
71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
package bitcoin
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
type Provider struct {
|
|
bitcoinAddresses []string // Slice of bitcoin addresses this provider monitors
|
|
ynabAccountID string // YNAB account ID this provider updates - all bitcoin addresses are summed up and mapped to this YNAB account
|
|
client *client // HTTP client for interacting with Questrade API
|
|
}
|
|
|
|
func (p *Provider) Name() string {
|
|
return "Bitcoin - Blockstream.info"
|
|
}
|
|
|
|
// Configures the provider for usage via environment variables and persistentData
|
|
// If an error is returned, the provider will not be used
|
|
func (p *Provider) Configure() error {
|
|
var err error
|
|
|
|
// Load environment variables in continous series with suffix starting at 0
|
|
// Multiple addresses can be configured, (eg _1, _2)
|
|
// As soon as the series is interrupted, we assume we're done
|
|
p.bitcoinAddresses = make([]string, 0)
|
|
for i := 0; true; i++ {
|
|
bitcoinAddress := os.Getenv(fmt.Sprintf("bitcoin_address_%d", i))
|
|
if bitcoinAddress == "" {
|
|
if i == 0 {
|
|
return fmt.Errorf("this account provider is not configured")
|
|
}
|
|
break
|
|
}
|
|
p.bitcoinAddresses = append(p.bitcoinAddresses, bitcoinAddress)
|
|
}
|
|
p.ynabAccountID = os.Getenv("bitcoin_ynab_account")
|
|
|
|
// Create new HTTP client
|
|
p.client, err = newClient()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create new bitcoin client: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Returns slices of account balances and mapped YNAB account IDs, along with an error
|
|
func (p *Provider) GetBalances() ([]int, []string, error) {
|
|
|
|
balances := make([]int, 0)
|
|
ynabAccountIDs := make([]string, 0)
|
|
var satoshiBalance int
|
|
for _, bitcoinAddress := range p.bitcoinAddresses {
|
|
addressResponse, err := p.client.GetAddress(bitcoinAddress)
|
|
if err != nil {
|
|
return balances, ynabAccountIDs, fmt.Errorf("failed to get bitcoin address '%s': %v", bitcoinAddress, err)
|
|
}
|
|
|
|
satoshiBalance += addressResponse.ChainStats.FundedTxoSum - addressResponse.ChainStats.SpentTxoSum
|
|
}
|
|
|
|
fiatBalance, err := p.client.ConvertBTCToCAD(satoshiBalance)
|
|
if err != nil {
|
|
return balances, ynabAccountIDs, fmt.Errorf("failed to convert satoshi balance to fiat balance: %v", err)
|
|
}
|
|
|
|
balances = append(balances, fiatBalance)
|
|
ynabAccountIDs = append(ynabAccountIDs, p.ynabAccountID)
|
|
return balances, ynabAccountIDs, nil
|
|
}
|