feat: account for holdings that do not exist from source

Author: will7200

.gitignore

@@ -2,3 +2,5 @@
 /.idea/
 /api/templates/go-experimental/
 /cookies.json
+*.gob
+*.exe

cmd/crypto-sync/sync.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"errors"
+	"log"
 
 	"github.com/pelletier/go-toml"
 	"github.com/pelletier/go-toml/query"
@@ -33,6 +34,7 @@ func (s *SyncCmd) Run(ctx *Context) error {
 	)
 	allHoldings := make([]holdings.Holding, 0, 2)
 	for _, holding := range s.Holdings {
+		log.Println("Fetching holdings from ", holding)
 		holdingsProvider, err := holdings.GetProvider(holding)
 		if err != nil {
 			return err

internal/common/compare.go

@@ -0,0 +1,12 @@
+package common
+
+import "github.com/shopspring/decimal"
+
+func MustParseFloatStringFloat64(val string) float64 {
+	quantity, err := decimal.NewFromString(val)
+	if err != nil {
+		panic(err)
+	}
+	quantityFloat, _ := quantity.Float64()
+	return quantityFloat
+}

internal/common/compare_test.go

@@ -0,0 +1,50 @@
+package common
+
+import (
+	"bytes"
+	"encoding/gob"
+	"io/ioutil"
+	"testing"
+
+	"github.com/davecgh/go-spew/spew"
+
+	"github.com/will7200/go-crypto-sync/internal/holdings"
+	"github.com/will7200/go-crypto-sync/pkg/personalcapital"
+)
+
+func ReadGlob(p interface{}, filename string) {
+	b, err := ioutil.ReadFile(filename)
+	if err != nil {
+		panic(err)
+	}
+	err = gob.NewDecoder(bytes.NewBuffer(b)).Decode(p)
+	if err != nil {
+		panic(err)
+	}
+}
+
+func TestAlgo(t *testing.T) {
+	var pcHoldings personalcapital.GetHoldingsResponse
+	var _holdings holdings.Holdings
+	ReadGlob(&pcHoldings, "pc_holdings.gob")
+	ReadGlob(&_holdings, "coinbase_holdings.gob")
+	//for _, pcHolding := range pcHoldings.Holdings {
+	//	spew.Dump(_holdings.HasCurrencyName(pcHolding.Ticker))
+	//}
+	spew.Dump(pcHoldings.Holdings)
+	spew.Dump(_holdings)
+	spew.Dump(_holdings.HasCurrencyMap(func(l holdings.IHolding) string {
+		return l.CurrencyName()
+	}, func(r holdings.IHolding) string {
+		return r.CurrencyName()
+	}, PCHoldingsToIHoldings(pcHoldings.Holdings)...))
+
+}
+
+func PCHoldingsToIHoldings(h []personalcapital.HoldingsType) []holdings.IHolding {
+	a := make([]holdings.IHolding, len(h))
+	for i := 0; i < len(h); i++ {
+		a[i] = h[i]
+	}
+	return a
+}

internal/holdings/account.go

@@ -51,29 +51,45 @@ func (h Holdings) HasCurrencyName(name string) bool {
 }
 
 type HasInfo struct {
-	Exists bool
-	LPos   int
-	RPos   int
+	LPos int
+	RPos int
+}
+
+func (hi HasInfo) FoundBoth() bool {
+	return hi.LPos != -1 && hi.RPos != -1
+}
+
+func (hi HasInfo) LeftOnly() bool {
+	return hi.LPos != -1 && hi.RPos == -1
+}
+
+func (hi HasInfo) RightOnly() bool {
+	return hi.LPos == -1 && hi.RPos != -1
 }
 
 func (h Holdings) HasCurrencyMap(left func(l IHolding) string, right func(r IHolding) string, ih ...IHolding) map[string]HasInfo {
 	mb := make(map[string]HasInfo, len(h))
 	for index, v := range h {
 		mb[strings.ToLower(left(v))] = HasInfo{
-			Exists: false,
-			LPos:   index,
-			RPos:   -1,
+			LPos: index,
+			RPos: -1,
 		}
 	}
 	for index, v := range ih {
 		key := strings.ToLower(right(v))
 		if val, ok := mb[key]; ok {
-			if val.Exists {
+			if val.RPos != -1 {
 				panic("conflicting entries")
 			}
-			val.Exists = true
 			val.RPos = index
 			mb[key] = val
+		} else {
+			if len(key) > 0 {
+				mb[key] = HasInfo{
+					LPos: -1,
+					RPos: index,
+				}
+			}
 		}
 	}
 	return mb

internal/pc/client.go

@@ -23,10 +23,12 @@ var (
 	_ = spew.Dump
 )
 
+// Sync source holdings to personal capital account
+// Currently cookies will be saved in the working directory of where this command is run
 func Sync(email, password string, holds holdings.Holdings, pricing holdings.Price) {
 
 	// Get saved cookies if available
-	var cookies []*(http.Cookie)
+	var cookies []*http.Cookie
 	personalcapital.LoadSession(&cookies, "cookies.json")
 
 	// Setup a cookie jar so the session can be saved
@@ -85,6 +87,7 @@ func Sync(email, password string, holds holdings.Holdings, pricing holdings.Pric
 		log.Println(key, value)
 		// Handle special case if is us dollar
 		if key == "us dollar" {
+			log.Printf("updating PC cash amount\n")
 			left := holds[value.LPos]
 			quantity, err := decimal.NewFromString(left.TotalSharesString())
 			if err != nil {
@@ -97,7 +100,8 @@ func Sync(email, password string, holds holdings.Holdings, pricing holdings.Pric
 			}
 			continue
 		}
-		if value.Exists {
+		if value.FoundBoth() {
+			log.Printf("%s found in pc, updating holding\n", holds[value.LPos].CurrencySymbolName())
 			left, right := holds[value.LPos], pcHoldings.Holdings[value.RPos]
 			quantity, err := decimal.NewFromString(left.TotalSharesString())
 			if err != nil {
@@ -125,7 +129,8 @@ func Sync(email, password string, holds holdings.Holdings, pricing holdings.Pric
 			if err != nil {
 				panic(err)
 			}
-		} else {
+		} else if value.LeftOnly() {
+			log.Printf("%s not found in pc, create new holding\n", holds[value.LPos].CurrencySymbolName())
 			quantity, err := decimal.NewFromString(holds[value.LPos].TotalSharesString())
 			if err != nil {
 				panic(err)
@@ -159,6 +164,35 @@ func Sync(email, password string, holds holdings.Holdings, pricing holdings.Pric
 			if err != nil {
 				panic(err)
 			}
+		} else if value.RightOnly() {
+			right := pcHoldings.Holdings[value.RPos]
+			currencySymbol := strings.TrimSpace(strings.Split(right.CurrencySymbolName(), "-")[0])
+			log.Printf("%s not found from source, setting quantity to zero\n", currencySymbol)
+			quantity, err := decimal.NewFromString("0.00")
+			if err != nil {
+				panic(err)
+			}
+			quantityFloat, _ := quantity.Float64()
+			p, err := pricing.GetExchange(currencySymbol, "USD")
+			if err != nil {
+				panic(err)
+			}
+			pf, err := decimal.NewFromString(p)
+			if err != nil {
+				panic(err)
+			}
+			priceFloat, _ := pf.Float64()
+			v := pf.Mul(quantity)
+			valueFloat, _ := v.Float64()
+			d := HoldingTypeToHoldingRequest(right)
+			d.PriceSource = "COINBASE"
+			d.Quantity = quantityFloat
+			d.Value = valueFloat
+			d.Price = priceFloat
+			_, err = apiClient.Holdings.UpdateHoldings(context.Background(), d)
+			if err != nil {
+				panic(err)
+			}
 		}
 	}
 	personalcapital.SaveSession(client, cfg.Host, "cookies.json")