Steven Polley
7284545571
All checks were successful
continuous-integration/drone/push Build is passing
82 lines
2.9 KiB
Go
82 lines
2.9 KiB
Go
package ynab
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/url"
|
|
)
|
|
|
|
// Reference: https://api.ynab.com/v1#/Accounts/
|
|
|
|
type Accounts struct {
|
|
Data struct {
|
|
Account struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Type string `json:"type"`
|
|
OnBudget bool `json:"on_budget"`
|
|
Closed bool `json:"closed"`
|
|
Note interface{} `json:"note"`
|
|
Balance int `json:"balance"`
|
|
ClearedBalance int `json:"cleared_balance"`
|
|
UnclearedBalance int `json:"uncleared_balance"`
|
|
TransferPayeeID string `json:"transfer_payee_id"`
|
|
DirectImportLinked bool `json:"direct_import_linked"`
|
|
DirectImportInError bool `json:"direct_import_in_error"`
|
|
Deleted bool `json:"deleted"`
|
|
} `json:"account"`
|
|
ServerKnowledge int `json:"server_knowledge"`
|
|
} `json:"data"`
|
|
}
|
|
|
|
func (c *Client) GetAccount(accountID string) (*Accounts, error) {
|
|
response := Accounts{}
|
|
|
|
err := c.get(fmt.Sprintf("/accounts/%s", accountID), &response, url.Values{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get accounts: %v", err)
|
|
}
|
|
|
|
return &response, nil
|
|
}
|
|
|
|
// Creates a "Capital Gains or Losses" adjustment transaction or updates an existing one if it exists to ensure the account balance
|
|
// for the accountID provided equals the balance provided
|
|
func (c *Client) SetAccountBalance(accountID string, balance int) error {
|
|
|
|
ynabTransactionID, ynabTransactionAmount, err := c.getTodayYnabCapitalGainsTransaction(accountID)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get ynab capital gains transaction ID: %v", err)
|
|
}
|
|
|
|
ynabAccount, err := c.GetAccount(accountID)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get ynab account with id '%s': %v", accountID, err)
|
|
}
|
|
|
|
balanceDelta := balance - ynabAccount.Data.Account.Balance
|
|
balanceDelta += ynabTransactionAmount // Take into account the existing transaction
|
|
if balanceDelta == 0 {
|
|
return nil // If balanceDelta is 0 do not create a transaction i.e. market is closed today
|
|
}
|
|
|
|
if ynabTransactionID == "" {
|
|
// there is no transaction - so create a new one
|
|
err = c.createTodayYNABCapitalGainsTransaction(accountID, balanceDelta)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create YNAB capital gains transaction for account ID '%s': %v", accountID, err)
|
|
}
|
|
log.Printf("Creating new capital gains transaction for YNAB account '%s' for amount: %d", accountID, balanceDelta)
|
|
|
|
} else {
|
|
// there is an existing transaction - so update the existing one
|
|
|
|
err = c.updateTodayYNABCapitalGainsTransaction(accountID, ynabTransactionID, balanceDelta)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to update YNAB capital gains transaction for account ID '%s': %v", accountID, err)
|
|
}
|
|
log.Printf("Updating existing capital gains transaction for YNAB account '%s' for amount: %d", accountID, balanceDelta)
|
|
}
|
|
return nil
|
|
}
|