From d7b265b995e3eaf16cddf9178427e5626c548b80 Mon Sep 17 00:00:00 2001 From: Aaron Lindsay Date: Sat, 1 Apr 2017 11:05:40 -0400 Subject: [PATCH] Add open orders to investment statment responses --- invstmt.go | 232 ++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 228 insertions(+), 4 deletions(-) diff --git a/invstmt.go b/invstmt.go index 1f30b41..3543dc7 100644 --- a/invstmt.go +++ b/invstmt.go @@ -688,6 +688,230 @@ type InvBalance struct { BalList []Balance `xml:"BALLIST>BAL,omitempty"` } +type OO struct { + XMLName xml.Name `xml:"OO"` + FiTId String `xml:"FITID"` + SrvrTId String `xml:"SRVRTID,omitempty"` + SecId SecurityId `xml:"SECID"` + DtPlaced Date `xml:"DTPLACED"` // Date the order was placed + Units Amount `xml:"UNITS"` // Quantity of the security the open order is for + SubAcct String `xml:"SUBACCT"` // One of CASH, MARGIN, SHORT, OTHER + Duration String `xml:"DURATION"` // How long the order is good for. One of DAY, GOODTILCANCEL, IMMEDIATE + Restriction String `xml:"RESTRICTION"` // Special restriction on the order: One of ALLORNONE, MINUNITS, NONE + MinUnits Amount `xml:"MINUNITS,omitempty"` // Minimum number of units that must be filled for the order + LimitPrice Amount `xml:"LIMITPRICE,omitempty"` + StopPrice Amount `xml:"STOPPRICE,omitempty"` + Memo String `xml:"MEMO,omitempty"` + Currency *Currency `xml:"CURRENCY,omitempty"` // Overriding currency for UNITPRICE + Inv401kSource String `xml:"INV401KSOURCE,omitempty"` // One of PRETAX, AFTERTAX, MATCH, PROFITSHARING, ROLLOVER, OTHERVEST, OTHERNONVEST for 401(k) accounts. Default if not present is OTHERNONVEST. The following cash source types are subject to vesting: MATCH, PROFITSHARING, and OTHERVEST +} + +type OpenOrder interface { + OrderType() string +} + +type OOBuyDebt struct { + XMLName xml.Name `xml:"OOBUYDEBT"` + OO OO `xml:"OO"` + Auction Boolean `xml:"AUCTION"` // whether the debt should be purchased at the auction + DtAuction *Date `xml:"DTAUCTION,omitempty"` +} + +func (o OOBuyDebt) OrderType() string { + return "OOBUYDEBT" +} + +type OOBuyMF struct { + XMLName xml.Name `xml:"OOBUYMF"` + OO OO `xml:"OO"` + BuyType String `xml:"BUYTYPE"` // One of BUY, BUYTOCOVER + UnitType String `xml:"UNITTYPE"` // What the units represent: one of SHARES, CURRENCY +} + +func (o OOBuyMF) OrderType() string { + return "OOBUYMF" +} + +type OOBuyOpt struct { + XMLName xml.Name `xml:"OOBUYOPT"` + OO OO `xml:"OO"` + OptBuyType String `xml:"OPTBUYTYPE"` // One of BUYTOOPEN, BUYTOCLOSE +} + +func (o OOBuyOpt) OrderType() string { + return "OOBUYOPT" +} + +type OOBuyOther struct { + XMLName xml.Name `xml:"OOBUYOTHER"` + OO OO `xml:"OO"` + UnitType String `xml:"UNITTYPE"` // What the units represent: one of SHARES, CURRENCY +} + +func (o OOBuyOther) OrderType() string { + return "OOBUYOTHER" +} + +type OOBuyStock struct { + XMLName xml.Name `xml:"OOBUYSTOCK"` + OO OO `xml:"OO"` + BuyType String `xml:"BUYTYPE"` // One of BUY, BUYTOCOVER +} + +func (o OOBuyStock) OrderType() string { + return "OOBUYSTOCK" +} + +type OOSellDebt struct { + XMLName xml.Name `xml:"OOSELLDEBT"` + OO OO `xml:"OO"` +} + +func (o OOSellDebt) OrderType() string { + return "OOSELLDEBT" +} + +type OOSellMF struct { + XMLName xml.Name `xml:"OOSELLMF"` + OO OO `xml:"OO"` + SellType String `xml:"SELLTYPE"` // One of SELL, SELLSHORT + UnitType String `xml:"UNITTYPE"` // What the units represent: one of SHARES, CURRENCY + SellAll Boolean `xml:"SELLALL"` // Sell entire holding +} + +func (o OOSellMF) OrderType() string { + return "OOSELLMF" +} + +type OOSellOpt struct { + XMLName xml.Name `xml:"OOSELLOPT"` + OO OO `xml:"OO"` + OptSellType String `xml:"OPTSELLTYPE"` // One of SELLTOOPEN, SELLTOCLOSE +} + +func (o OOSellOpt) OrderType() string { + return "OOSELLOPT" +} + +type OOSellOther struct { + XMLName xml.Name `xml:"OOSELLOTHER"` + OO OO `xml:"OO"` + UnitType String `xml:"UNITTYPE"` // What the units represent: one of SHARES, CURRENCY +} + +func (o OOSellOther) OrderType() string { + return "OOSELLOTHER" +} + +type OOSellStock struct { + XMLName xml.Name `xml:"OOSELLSTOCK"` + OO OO `xml:"OO"` + SellType String `xml:"SELLTYPE"` // One of SELL, SELLSHORT +} + +func (o OOSellStock) OrderType() string { + return "OOSELLSTOCK" +} + +type OOSwitchMF struct { + XMLName xml.Name `xml:"SWITCHMF"` + OO OO `xml:"OO"` + SecId SecurityId `xml:"SECID"` // Security ID of the fund to switch to or purchase + UnitType String `xml:"UNITTYPE"` // What the units represent: one of SHARES, CURRENCY + SwitchAll Boolean `xml:"SWITCHALL"` // Switch entire holding +} + +func (o OOSwitchMF) OrderType() string { + return "SWITCHMF" +} + +type OOList []OpenOrder + +func (o *OOList) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error { + for { + tok, err := nextNonWhitespaceToken(d) + if err != nil { + return err + } else if end, ok := tok.(xml.EndElement); ok && end.Name.Local == start.Name.Local { + // If we found the end of our starting element, we're done parsing + return nil + } else if startElement, ok := tok.(xml.StartElement); ok { + switch startElement.Name.Local { + case "OOBUYDEBT": + var oo OOBuyDebt + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOBUYMF": + var oo OOBuyMF + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOBUYOPT": + var oo OOBuyOpt + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOBUYOTHER": + var oo OOBuyOther + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOBUYSTOCK": + var oo OOBuyStock + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOSELLDEBT": + var oo OOSellDebt + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOSELLMF": + var oo OOSellMF + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOSELLOPT": + var oo OOSellOpt + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOSELLOTHER": + var oo OOSellOther + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "OOSELLSTOCK": + var oo OOSellStock + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + case "SWITCHMF": + var oo OOSwitchMF + if err := d.DecodeElement(&oo, &startElement); err != nil { + return err + } + *o = append(*o, OpenOrder(oo)) + default: + return errors.New("Invalid OOList child tag: " + startElement.Name.Local) + } + } else { + return errors.New("Didn't find an opening element") + } + } +} + type ContribSecurity struct { XMLName xml.Name `xml:"CONTRIBSECURITY"` SecId SecurityId `xml:"SECID"` @@ -805,10 +1029,10 @@ type InvStatementResponse struct { InvTranList *InvTranList `xml:"INVSTMTRS>INVTRANLIST,omitempty"` InvPosList PositionList `xml:"INVSTMTRS>INVPOSLIST,omitempty"` InvBal *InvBalance `xml:"INVSTMTRS>INVBAL,omitempty"` - // TODO INVOOLIST - MktgInfo String `xml:"INVSTMTRS>MKTGINFO,omitempty"` // Marketing information - Inv401K *Inv401K `xml:"INVSTMTRS>INV401K,omitempty"` - Inv401KBal *Inv401KBal `xml:"INVSTMTRS>INV401KBAL,omitempty"` + InvOOList OOList `xml:"INVSTMTRS>INVOOLIST,omitempty"` + MktgInfo String `xml:"INVSTMTRS>MKTGINFO,omitempty"` // Marketing information + Inv401K *Inv401K `xml:"INVSTMTRS>INV401K,omitempty"` + Inv401KBal *Inv401KBal `xml:"INVSTMTRS>INV401KBAL,omitempty"` } func (sr *InvStatementResponse) Name() string {