package handlers import ( "errors" "fmt" "github.com/aclindsa/moneygo/internal/models" "github.com/aclindsa/ofxgo" "io" "math/big" ) type OFXImport struct { Securities []models.Security Accounts []models.Account Transactions []models.Transaction // Balances map[int64]string // map AccountIDs to ending balances } func (i *OFXImport) GetSecurity(ofxsecurityid int64) (*models.Security, error) { if ofxsecurityid < 0 || ofxsecurityid > int64(len(i.Securities)) { return nil, errors.New("OFXImport.GetSecurity: SecurityID out of range") } return &i.Securities[ofxsecurityid], nil } func (i *OFXImport) GetSecurityAlternateId(alternateid string, securityType models.SecurityType) (*models.Security, error) { for _, security := range i.Securities { if alternateid == security.AlternateId && securityType == security.Type { return &security, nil } } return nil, errors.New("OFXImport.FindSecurity: Unable to find security") } func (i *OFXImport) GetAddCurrency(isoname string) (*models.Security, error) { for _, security := range i.Securities { if isoname == security.Name && models.Currency == security.Type { return &security, nil } } template := FindSecurityTemplate(isoname, models.Currency) if template == nil { return nil, fmt.Errorf("Failed to find Security for \"%s\"", isoname) } var security models.Security = *template security.SecurityId = int64(len(i.Securities) + 1) i.Securities = append(i.Securities, security) return &security, nil } func (i *OFXImport) AddTransaction(tran *ofxgo.Transaction, account *models.Account) error { var t models.Transaction t.Date = tran.DtPosted.UTC() // Construct the description from whichever of the descriptive OFX fields are present if len(tran.Name) > 0 { t.Description = string(tran.Name) } else if tran.Payee != nil { t.Description = string(tran.Payee.Name) } if len(tran.Memo) > 0 { if len(t.Description) > 0 { t.Description = t.Description + " - " + string(tran.Memo) } else { t.Description = string(tran.Memo) } } var s1, s2 models.Split if len(tran.ExtdName) > 0 { s1.Memo = tran.ExtdName.String() } if len(tran.CheckNum) > 0 { s1.Number = tran.CheckNum.String() } else if len(tran.RefNum) > 0 { s1.Number = tran.RefNum.String() } amt := big.NewRat(0, 1) // Convert TrnAmt to account's currency if Currency is set if ok, _ := tran.Currency.Valid(); ok { amt.Mul(&tran.Currency.CurRate.Rat, &tran.TrnAmt.Rat) } else { amt.Set(&tran.TrnAmt.Rat) } if account.SecurityId < 1 || account.SecurityId > int64(len(i.Securities)) { return errors.New("Internal error: security index not found in OFX import\n") } s1.RemoteId = "ofx:" + tran.FiTID.String() // TODO CorrectFiTID/CorrectAction? s1.ImportSplitType = models.ImportAccount s2.ImportSplitType = models.ExternalAccount security := i.Securities[account.SecurityId-1] s1.Amount = amt.FloatString(security.Precision) s2.Amount = amt.Neg(amt).FloatString(security.Precision) s1.Status = models.Imported s2.Status = models.Imported s1.AccountId = account.AccountId s2.AccountId = -1 s1.SecurityId = -1 s2.SecurityId = security.SecurityId t.Splits = append(t.Splits, &s1) t.Splits = append(t.Splits, &s2) i.Transactions = append(i.Transactions, t) return nil } func (i *OFXImport) importOFXBank(stmt *ofxgo.StatementResponse) error { security, err := i.GetAddCurrency(stmt.CurDef.String()) if err != nil { return err } account := models.Account{ AccountId: int64(len(i.Accounts) + 1), ExternalAccountId: stmt.BankAcctFrom.AcctID.String(), SecurityId: security.SecurityId, ParentAccountId: -1, Type: models.Bank, } if stmt.BankTranList != nil { for _, tran := range stmt.BankTranList.Transactions { if err := i.AddTransaction(&tran, &account); err != nil { return err } } } i.Accounts = append(i.Accounts, account) return nil } func (i *OFXImport) importOFXCC(stmt *ofxgo.CCStatementResponse) error { security, err := i.GetAddCurrency(stmt.CurDef.String()) if err != nil { return err } account := models.Account{ AccountId: int64(len(i.Accounts) + 1), ExternalAccountId: stmt.CCAcctFrom.AcctID.String(), SecurityId: security.SecurityId, ParentAccountId: -1, Type: models.Liability, } i.Accounts = append(i.Accounts, account) if stmt.BankTranList != nil { for _, tran := range stmt.BankTranList.Transactions { if err := i.AddTransaction(&tran, &account); err != nil { return err } } } // TODO balance(s) return nil } func (i *OFXImport) importSecurities(seclist *ofxgo.SecurityList) error { for _, security := range seclist.Securities { var si ofxgo.SecInfo if sec, ok := (security).(ofxgo.DebtInfo); ok { si = sec.SecInfo } else if sec, ok := (security).(ofxgo.MFInfo); ok { si = sec.SecInfo } else if sec, ok := (security).(ofxgo.OptInfo); ok { si = sec.SecInfo } else if sec, ok := (security).(ofxgo.OtherInfo); ok { si = sec.SecInfo } else if sec, ok := (security).(ofxgo.StockInfo); ok { si = sec.SecInfo } else { return errors.New("Can't import unrecognized type satisfying ofxgo.Security interface") } s := models.Security{ SecurityId: int64(len(i.Securities) + 1), Name: string(si.SecName), Description: string(si.Memo), Symbol: string(si.Ticker), Precision: 5, // TODO How to actually determine this? Type: models.Stock, AlternateId: string(si.SecID.UniqueID), } if len(s.Description) == 0 { s.Description = s.Name } if len(s.Symbol) == 0 { s.Symbol = s.Name } i.Securities = append(i.Securities, s) } return nil } func (i *OFXImport) GetInvTran(invtran *ofxgo.InvTran) models.Transaction { var t models.Transaction t.Description = string(invtran.Memo) t.Date = invtran.DtTrade.UTC() return t } func (i *OFXImport) GetInvBuyTran(buy *ofxgo.InvBuy, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&buy.InvTran) security, err := i.GetSecurityAlternateId(string(buy.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(buy.InvTran.Memo) if len(memo) > 0 { memo += " " } var commission, taxes, fees, load, total, tradingTotal big.Rat commission.Abs(&buy.Commission.Rat) taxes.Abs(&buy.Taxes.Rat) fees.Abs(&buy.Fees.Rat) load.Abs(&buy.Load.Rat) total.Abs(&buy.Total.Rat) total.Neg(&total) tradingTotal.Neg(&total) tradingTotal.Sub(&tradingTotal, &commission) tradingTotal.Sub(&tradingTotal, &taxes) tradingTotal.Sub(&tradingTotal, &fees) tradingTotal.Sub(&tradingTotal, &load) // Convert amounts to account's currency if Currency is set if ok, _ := buy.Currency.Valid(); ok { commission.Mul(&commission, &buy.Currency.CurRate.Rat) taxes.Mul(&taxes, &buy.Currency.CurRate.Rat) fees.Mul(&fees, &buy.Currency.CurRate.Rat) load.Mul(&load, &buy.Currency.CurRate.Rat) total.Mul(&total, &buy.Currency.CurRate.Rat) tradingTotal.Mul(&tradingTotal, &buy.Currency.CurRate.Rat) } if num := commission.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Commission, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo + "(commission)", Amount: commission.FloatString(curdef.Precision), }) } if num := taxes.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Taxes, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo + "(taxes)", Amount: taxes.FloatString(curdef.Precision), }) } if num := fees.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Fees, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo + "(fees)", Amount: fees.FloatString(curdef.Precision), }) } if num := load.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Load, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo + "(load)", Amount: load.FloatString(curdef.Precision), }) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.TradingAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo, Amount: tradingTotal.FloatString(curdef.Precision), }) var units big.Rat units.Abs(&buy.Units.Rat) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.SubAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) units.Neg(&units) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.TradingAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + buy.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) return &t, nil } func (i *OFXImport) GetIncomeTran(income *ofxgo.Income, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&income.InvTran) security, err := i.GetSecurityAlternateId(string(income.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(income.InvTran.Memo) if len(memo) > 0 { memo += " " } else { memo = income.IncomeType.String() + " on " + security.Symbol } var total big.Rat total.Set(&income.Total.Rat) if ok, _ := income.Currency.Valid(); ok { total.Mul(&total, &income.Currency.CurRate.Rat) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + income.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) total.Neg(&total) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.IncomeAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + income.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) return &t, nil } func (i *OFXImport) GetInvExpenseTran(expense *ofxgo.InvExpense, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&expense.InvTran) security, err := i.GetSecurityAlternateId(string(expense.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(expense.InvTran.Memo) if len(memo) == 0 { memo = "INVEXPENSE" } memo += " (" + security.Symbol + ")" var total big.Rat total.Set(&expense.Total.Rat) if ok, _ := expense.Currency.Valid(); ok { total.Mul(&total, &expense.Currency.CurRate.Rat) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + expense.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) total.Neg(&total) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ExpenseAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + expense.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) return &t, nil } func (i *OFXImport) GetMarginInterestTran(marginint *ofxgo.MarginInterest, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&marginint.InvTran) memo := string(marginint.InvTran.Memo) if len(memo) == 0 { memo = "MARGININTEREST" } var total big.Rat total.Set(&marginint.Total.Rat) if ok, _ := marginint.Currency.Valid(); ok { total.Mul(&total, &marginint.Currency.CurRate.Rat) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + marginint.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) total.Neg(&total) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.IncomeAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + marginint.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) return &t, nil } func (i *OFXImport) GetReinvestTran(reinvest *ofxgo.Reinvest, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&reinvest.InvTran) security, err := i.GetSecurityAlternateId(string(reinvest.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(reinvest.InvTran.Memo) if len(memo) > 0 { memo += " " } var commission, taxes, fees, load, total, tradingTotal big.Rat commission.Abs(&reinvest.Commission.Rat) taxes.Abs(&reinvest.Taxes.Rat) fees.Abs(&reinvest.Fees.Rat) load.Abs(&reinvest.Load.Rat) total.Abs(&reinvest.Total.Rat) total.Neg(&total) tradingTotal.Neg(&total) tradingTotal.Sub(&tradingTotal, &commission) tradingTotal.Sub(&tradingTotal, &taxes) tradingTotal.Sub(&tradingTotal, &fees) tradingTotal.Sub(&tradingTotal, &load) // Convert amounts to account's currency if Currency is set if ok, _ := reinvest.Currency.Valid(); ok { commission.Mul(&commission, &reinvest.Currency.CurRate.Rat) taxes.Mul(&taxes, &reinvest.Currency.CurRate.Rat) fees.Mul(&fees, &reinvest.Currency.CurRate.Rat) load.Mul(&load, &reinvest.Currency.CurRate.Rat) total.Mul(&total, &reinvest.Currency.CurRate.Rat) tradingTotal.Mul(&tradingTotal, &reinvest.Currency.CurRate.Rat) } if num := commission.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Commission, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo + "(commission)", Amount: commission.FloatString(curdef.Precision), }) } if num := taxes.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Taxes, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo + "(taxes)", Amount: taxes.FloatString(curdef.Precision), }) } if num := fees.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Fees, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo + "(fees)", Amount: fees.FloatString(curdef.Precision), }) } if num := load.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Load, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo + "(load)", Amount: load.FloatString(curdef.Precision), }) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.IncomeAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) total.Neg(&total) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.TradingAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo, Amount: tradingTotal.FloatString(curdef.Precision), }) var units big.Rat units.Abs(&reinvest.Units.Rat) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.SubAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) units.Neg(&units) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.TradingAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + reinvest.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) return &t, nil } func (i *OFXImport) GetRetOfCapTran(retofcap *ofxgo.RetOfCap, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&retofcap.InvTran) security, err := i.GetSecurityAlternateId(string(retofcap.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(retofcap.InvTran.Memo) if len(memo) == 0 { memo = "RETOFCAP" } memo += " (" + security.Symbol + ")" var total big.Rat total.Set(&retofcap.Total.Rat) if ok, _ := retofcap.Currency.Valid(); ok { total.Mul(&total, &retofcap.Currency.CurRate.Rat) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + retofcap.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) total.Neg(&total) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.IncomeAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + retofcap.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) return &t, nil } func (i *OFXImport) GetInvSellTran(sell *ofxgo.InvSell, curdef *models.Security, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&sell.InvTran) security, err := i.GetSecurityAlternateId(string(sell.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(sell.InvTran.Memo) if len(memo) > 0 { memo += " " } var commission, taxes, fees, load, total, tradingTotal big.Rat commission.Abs(&sell.Commission.Rat) taxes.Abs(&sell.Taxes.Rat) fees.Abs(&sell.Fees.Rat) load.Abs(&sell.Load.Rat) total.Abs(&sell.Total.Rat) commission.Neg(&commission) taxes.Neg(&taxes) fees.Neg(&fees) load.Neg(&load) tradingTotal.Neg(&total) tradingTotal.Add(&tradingTotal, &commission) tradingTotal.Add(&tradingTotal, &taxes) tradingTotal.Add(&tradingTotal, &fees) tradingTotal.Add(&tradingTotal, &load) // Convert amounts to account's currency if Currency is set if ok, _ := sell.Currency.Valid(); ok { commission.Mul(&commission, &sell.Currency.CurRate.Rat) taxes.Mul(&taxes, &sell.Currency.CurRate.Rat) fees.Mul(&fees, &sell.Currency.CurRate.Rat) load.Mul(&load, &sell.Currency.CurRate.Rat) total.Mul(&total, &sell.Currency.CurRate.Rat) tradingTotal.Mul(&tradingTotal, &sell.Currency.CurRate.Rat) } if num := commission.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Commission, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo + "(commission)", Amount: commission.FloatString(curdef.Precision), }) } if num := taxes.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Taxes, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo + "(taxes)", Amount: taxes.FloatString(curdef.Precision), }) } if num := fees.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Fees, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo + "(fees)", Amount: fees.FloatString(curdef.Precision), }) } if num := load.Num(); !num.IsInt64() || num.Int64() != 0 { t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.Load, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo + "(load)", Amount: load.FloatString(curdef.Precision), }) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ImportAccount, AccountId: account.AccountId, SecurityId: -1, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo, Amount: total.FloatString(curdef.Precision), }) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.TradingAccount, AccountId: -1, SecurityId: curdef.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo, Amount: tradingTotal.FloatString(curdef.Precision), }) var units big.Rat units.Abs(&sell.Units.Rat) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.TradingAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) units.Neg(&units) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.SubAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + sell.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) return &t, nil } func (i *OFXImport) GetTransferTran(transfer *ofxgo.Transfer, account *models.Account) (*models.Transaction, error) { t := i.GetInvTran(&transfer.InvTran) security, err := i.GetSecurityAlternateId(string(transfer.SecID.UniqueID), models.Stock) if err != nil { return nil, err } memo := string(transfer.InvTran.Memo) var units big.Rat if transfer.TferAction == ofxgo.TferActionIn { units.Set(&transfer.Units.Rat) } else { units.Neg(&transfer.Units.Rat) } t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.SubAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + transfer.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) units.Neg(&units) t.Splits = append(t.Splits, &models.Split{ // TODO ReversalFiTID? Status: models.Imported, ImportSplitType: models.ExternalAccount, AccountId: -1, SecurityId: security.SecurityId, RemoteId: "ofx:" + transfer.InvTran.FiTID.String(), Memo: memo, Amount: units.FloatString(security.Precision), }) return &t, nil } func (i *OFXImport) AddInvTransaction(invtran *ofxgo.InvTransaction, account *models.Account, curdef *models.Security) error { if curdef.SecurityId < 1 || curdef.SecurityId > int64(len(i.Securities)) { return errors.New("Internal error: security index not found in OFX import\n") } var t *models.Transaction var err error if tran, ok := (*invtran).(ofxgo.BuyDebt); ok { t, err = i.GetInvBuyTran(&tran.InvBuy, curdef, account) } else if tran, ok := (*invtran).(ofxgo.BuyMF); ok { t, err = i.GetInvBuyTran(&tran.InvBuy, curdef, account) } else if tran, ok := (*invtran).(ofxgo.BuyOpt); ok { t, err = i.GetInvBuyTran(&tran.InvBuy, curdef, account) } else if tran, ok := (*invtran).(ofxgo.BuyOther); ok { t, err = i.GetInvBuyTran(&tran.InvBuy, curdef, account) } else if tran, ok := (*invtran).(ofxgo.BuyStock); ok { t, err = i.GetInvBuyTran(&tran.InvBuy, curdef, account) // } else if tran, ok := (*invtran).(ofxgo.ClosureOpt); ok { // TODO implementme } else if tran, ok := (*invtran).(ofxgo.Income); ok { t, err = i.GetIncomeTran(&tran, curdef, account) } else if tran, ok := (*invtran).(ofxgo.InvExpense); ok { t, err = i.GetInvExpenseTran(&tran, curdef, account) // } else if tran, ok := (*invtran).(ofxgo.JrnlFund); ok { // TODO implementme // } else if tran, ok := (*invtran).(ofxgo.JrnlSec); ok { // TODO implementme } else if tran, ok := (*invtran).(ofxgo.MarginInterest); ok { t, err = i.GetMarginInterestTran(&tran, curdef, account) } else if tran, ok := (*invtran).(ofxgo.Reinvest); ok { t, err = i.GetReinvestTran(&tran, curdef, account) } else if tran, ok := (*invtran).(ofxgo.RetOfCap); ok { t, err = i.GetRetOfCapTran(&tran, curdef, account) } else if tran, ok := (*invtran).(ofxgo.SellDebt); ok { t, err = i.GetInvSellTran(&tran.InvSell, curdef, account) } else if tran, ok := (*invtran).(ofxgo.SellMF); ok { t, err = i.GetInvSellTran(&tran.InvSell, curdef, account) } else if tran, ok := (*invtran).(ofxgo.SellOpt); ok { t, err = i.GetInvSellTran(&tran.InvSell, curdef, account) } else if tran, ok := (*invtran).(ofxgo.SellOther); ok { t, err = i.GetInvSellTran(&tran.InvSell, curdef, account) } else if tran, ok := (*invtran).(ofxgo.SellStock); ok { t, err = i.GetInvSellTran(&tran.InvSell, curdef, account) // } else if tran, ok := (*invtran).(ofxgo.Split); ok { // TODO implementme } else if tran, ok := (*invtran).(ofxgo.Transfer); ok { t, err = i.GetTransferTran(&tran, account) } else { return errors.New("Unrecognized type satisfying ofxgo.InvTransaction interface: " + (*invtran).TransactionType()) } if err != nil { return err } i.Transactions = append(i.Transactions, *t) return nil } func (i *OFXImport) importOFXInv(stmt *ofxgo.InvStatementResponse) error { security, err := i.GetAddCurrency(stmt.CurDef.String()) if err != nil { return err } account := models.Account{ AccountId: int64(len(i.Accounts) + 1), ExternalAccountId: stmt.InvAcctFrom.AcctID.String(), SecurityId: security.SecurityId, ParentAccountId: -1, Type: models.Investment, } i.Accounts = append(i.Accounts, account) if stmt.InvTranList != nil { for _, invtran := range stmt.InvTranList.InvTransactions { if err := i.AddInvTransaction(&invtran, &account, security); err != nil { return err } } for _, bt := range stmt.InvTranList.BankTransactions { // TODO Should we do something different for the value of // bt.SubAcctFund? for _, tran := range bt.Transactions { if err := i.AddTransaction(&tran, &account); err != nil { return err } } } } // TODO InvPosList // TODO InvBal // TODO Inv401K and INV401kBal??? return nil } func ImportOFX(r io.Reader) (*OFXImport, error) { var i OFXImport response, err := ofxgo.ParseResponse(r) if err != nil { return nil, fmt.Errorf("Unexpected error parsing OFX response: %s\n", err) } if response.Signon.Status.Code != 0 { meaning, _ := response.Signon.Status.CodeMeaning() return nil, fmt.Errorf("Nonzero signon status (%d: %s) with message: %s\n", response.Signon.Status.Code, meaning, response.Signon.Status.Message) } for _, bank := range response.Bank { if stmt, ok := bank.(*ofxgo.StatementResponse); ok { err = i.importOFXBank(stmt) if err != nil { return nil, err } return &i, nil } } for _, cc := range response.CreditCard { if stmt, ok := cc.(*ofxgo.CCStatementResponse); ok { err = i.importOFXCC(stmt) if err != nil { return nil, err } return &i, nil } } for _, seclist := range response.SecList { if securitylist, ok := seclist.(*ofxgo.SecurityList); ok { err = i.importSecurities(securitylist) if err != nil { return nil, err } } } for _, inv := range response.InvStmt { if stmt, ok := inv.(*ofxgo.InvStatementResponse); ok { err = i.importOFXInv(stmt) if err != nil { return nil, err } return &i, nil } } return nil, errors.New("No OFX statement found") }