Merge pull request #2 from samshadwell/update-config

Add new config field.

README.md

@@ -22,6 +22,9 @@ This integration makes the bookkeeping easier. Our setup is:
 
 ## Configuration
 
+!! FIXME !!
+Update the config.yml.example, document it
+
 Configuration is assumed to live in a `config.yml` file at the root of the application. For security reasons, I don't
 check this file into Git, but you can see the general structure in `config.yml.example`. To use this yourself, first
 run:

config.go

@@ -7,26 +7,36 @@ import (
 	"gopkg.in/yaml.v2"
 )
 
-type Config struct {
-	YnabToken string `yaml:"ynabToken"`
-	BudgetId  string `yaml:"budgetId"`
+type config struct {
+	YnabToken       string   `yaml:"ynabToken"`
+	BudgetId        string   `yaml:"budgetId"`
+	SplitAccountIds []string `yaml:"splitAccountIds"`
 }
 
-func NewConfig() *Config {
+func LoadConfig() (*config, error) {
 	f, err := os.Open("config.yml")
 	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error opening config file. Did you create a config.yml?\n\t%v\n", err)
-		os.Exit(1)
+		return nil, fmt.Errorf("Error opening config file. Did you create a config.yml?\n\t%w\n", err)
 	}
 	defer f.Close()
 
-	var cfg Config
 	decoder := yaml.NewDecoder(f)
+	var cfg config
 	err = decoder.Decode(&cfg)
 	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error decoding config file. Make sure it has the correct format. See config.yml.example for an example\n\t%v\n", err)
-		os.Exit(1)
+		return nil, fmt.Errorf(
+			"Error decoding config file. Make sure it has the correct format. See config.yml.example for an example\n\t%w\n",
+			err,
+		)
 	}
 
-	return &cfg
+	if len(cfg.YnabToken) == 0 {
+		return nil, fmt.Errorf("Error: config file is missing ynabToken")
+	}
+
+	if len(cfg.BudgetId) == 0 {
+		return nil, fmt.Errorf("Error: config file is missing budgetId")
+	}
+
+	return &cfg, nil
 }

main.go

@@ -2,16 +2,24 @@ package main
 
 import (
 	"fmt"
-	"reflect"
+	"os"
 
 	"github.com/brunomvsouza/ynab.go"
 )
 
 func main() {
-	config := NewConfig()
-	c := ynab.NewClient(config.YnabToken)
+	config, err := LoadConfig()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "%v\n", err)
+		os.Exit(1)
+	}
+	client := ynab.NewClient(config.YnabToken)
 
-	transactions, _ := c.Transaction().GetTransactions(config.BudgetId, nil)
-	fmt.Println(reflect.TypeOf(transactions))
-	fmt.Println(len(transactions))
+	transactions, err := client.Transaction().GetTransactionsByAccount(config.BudgetId, config.SplitAccountIds[0], nil)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error while fetching transactions from YNAB:\n\t%v\n", err)
+		os.Exit(1)
+	}
+
+	fmt.Println("Server knowledge: ", transactions[0])
 }