package ofxgo import ( "errors" "github.com/aclindsa/go/src/encoding/xml" ) type SecurityId struct { XMLName xml.Name `xml:"SECID"` UniqueId String `xml:"UNIQUEID"` // CUSIP for US FI's UniqueIdType String `xml:"UNIQUEIDTYPE"` // Should always be "CUSIP" for US FI's } type SecurityRequest struct { XMLName xml.Name `xml:"SECRQ"` // Only one of the next three should be present SecId SecurityId `xml:"SECID,omitempty"` Ticker String `xml:"TICKER,omitempty"` FiId String `xml:"FIID,omitempty"` } type SecListRequest struct { XMLName xml.Name `xml:"SECLISTTRNRQ"` TrnUID UID `xml:"TRNUID"` CltCookie String `xml:"CLTCOOKIE,omitempty"` TAN String `xml:"TAN,omitempty"` // Transaction authorization number Securities []SecurityRequest `xml:"SECLISTRQ>SECRQ,omitempty"` } func (r *SecListRequest) Name() string { return "SECLISTTRNRQ" } func (r *SecListRequest) Valid() (bool, error) { if ok, err := r.TrnUID.Valid(); !ok { return false, err } return true, nil } type SecListResponse struct { XMLName xml.Name `xml:"SECLISTTRNRS"` TrnUID UID `xml:"TRNUID"` Status Status `xml:"STATUS"` CltCookie String `xml:"CLTCOOKIE,omitempty"` // SECLISTRS is always empty, so we don't parse it here. The actual securities list will be in a top-level element parallel to SECLISTTRNRS } func (r SecListResponse) Name() string { return "SECLISTTRNRS" } func (r SecListResponse) Valid() (bool, error) { if ok, err := r.TrnUID.Valid(); !ok { return false, err } return true, nil } type Security interface { SecurityType() string } type SecInfo struct { XMLName xml.Name `xml:"SECINFO"` SecId SecurityId `xml:"SECID"` SecName String `xml:"SECNAME"` // Full name of security Ticker String `xml:"TICKER,omitempty"` // Ticker symbol FiId String `xml:"FIID,omitempty"` Rating String `xml:"RATING,omitempty"` UnitPrice Amount `xml:"UNITPRICE,omitempty"` // Current price, as of DTASOF DtAsOf Date `xml:"DTASOF,omitempty"` // Date UNITPRICE was for Currency Currency `xml:"CURRENCY,omitempty"` // Overriding currency for UNITPRICE Memo String `xml:"MEMO,omitempty"` } type DebtInfo struct { XMLName xml.Name `xml:"DEBTINFO"` SecInfo SecInfo `xml:"SECINFO"` ParValue Amount `xml:"PARVALUE"` DebtType String `xml:"DEBTTYPE"` // One of COUPON, ZERO (zero coupon) DebtClass String `xml:"DEBTCLASS,omitempty"` // One of TREASURY, MUNICIPAL, CORPORATE, OTHER CouponRate Amount `xml:"COUPONRT,omitempty"` // Bond coupon rate for next closest call date DtCoupon Date `xml:"DTCOUPON,omitempty"` // Maturity date for next coupon CouponFreq String `xml:"COUPONFREQ,omitempty"` // When coupons mature - one of MONTHLY, QUARTERLY, SEMIANNUAL, ANNUAL, or OTHER CallPrice Amount `xml:"CALLPRICE,omitempty"` // Bond call price YieldToCall Amount `xml:"YIELDTOCALL,omitempty"` // Yield to next call DtCall Date `xml:"DTCALL,omitempty"` // Next call date CallType String `xml:"CALLTYPE,omitempt"` // Type of next call. One of CALL, PUT, PREFUND, MATURITY YieldToMat Amount `xml:"YIELDTOMAT,omitempty"` // Yield to maturity DtMat Date `xml:"DTMAT,omitempty"` // Debt maturity date AssetClass String `xml:"ASSETCLASS,omitempty"` // One of DOMESTICBOND, INTLBOND, LARGESTOCK, SMALLSTOCK, INTLSTOCK, MONEYMRKT, OTHER FiAssetClass String `xml:"FIASSETCLASS,omitempty"` // FI-defined asset class } func (i DebtInfo) SecurityType() string { return "DEBTINFO" } type AssetPortion struct { XMLName xml.Name `xml:"PORTION"` AssetClass String `xml:"ASSETCLASS"` // One of DOMESTICBOND, INTLBOND, LARGESTOCK, SMALLSTOCK, INTLSTOCK, MONEYMRKT, OTHER Percent Amount `xml:"PERCENT"` // Percentage of the fund that falls under this asset class } type FiAssetPortion struct { XMLName xml.Name `xml:"FIPORTION"` FiAssetClass String `xml:"FIASSETCLASS,omitempty"` // FI-defined asset class Percent Amount `xml:"PERCENT"` // Percentage of the fund that falls under this asset class } type MFInfo struct { XMLName xml.Name `xml:"MFINFO"` SecInfo SecInfo `xml:"SECINFO"` MfType String `xml:"MFTYPE"` // One of OPEN, END, CLOSEEND, OTHER Yield Amount `xml:"YIELD,omitempty"` // Current yield reported as the dividend expressed as a portion of the current stock price DtYieldAsOf Date `xml:"DTYIELDASOF,omitempty"` // Date YIELD is valid for AssetClasses []AssetPortion `xml:"MFASSETCLASS>PORTION"` FiAssetClasses []FiAssetPortion `xml:"FIMFASSETCLASS>FIPORTION"` } func (i MFInfo) SecurityType() string { return "MFINFO" } type OptInfo struct { XMLName xml.Name `xml:"OPTINFO"` SecInfo SecInfo `xml:"SECINFO"` OptType String `xml:"OPTTYPE"` // One of PUT, CALL StrikePrice Amount `xml:"STRIKEPRICE"` DtExpire Date `xml"DTEXPIRE"` // Expiration date ShPerCtrct Int `xml:"SHPERCTRCT"` // Shares per contract SecId SecurityId `xml:"SECID,omitempty"` // Security ID of the underlying security AssetClass String `xml:"ASSETCLASS,omitempty"` // One of DOMESTICBOND, INTLBOND, LARGESTOCK, SMALLSTOCK, INTLSTOCK, MONEYMRKT, OTHER FiAssetClass String `xml:"FIASSETCLASS,omitempty"` // FI-defined asset class } func (i OptInfo) SecurityType() string { return "OPTINFO" } type OtherInfo struct { XMLName xml.Name `xml:"OTHERINFO"` SecInfo SecInfo `xml:"SECINFO"` TypeDesc String `xml:"TYPEDESC,omitempty"` // Description of security type AssetClass String `xml:"ASSETCLASS,omitempty"` // One of DOMESTICBOND, INTLBOND, LARGESTOCK, SMALLSTOCK, INTLSTOCK, MONEYMRKT, OTHER FiAssetClass String `xml:"FIASSETCLASS,omitempty"` // FI-defined asset class } func (i OtherInfo) SecurityType() string { return "OTHERINFO" } type StockInfo struct { XMLName xml.Name `xml:"STOCKINFO"` SecInfo SecInfo `xml:"SECINFO"` StockType String `xml:"STOCKTYPE,omitempty"` // One of COMMON, PREFERRED, CONVERTIBLE, OTHER Yield Amount `xml:"YIELD,omitempty"` // Current yield reported as the dividend expressed as a portion of the current stock price DtYieldAsOf Date `xml:"DTYIELDASOF,omitempty"` // Date YIELD is valid for AssetClass String `xml:"ASSETCLASS,omitempty"` // One of DOMESTICBOND, INTLBOND, LARGESTOCK, SMALLSTOCK, INTLSTOCK, MONEYMRKT, OTHER FiAssetClass String `xml:"FIASSETCLASS,omitempty"` // FI-defined asset class } func (i StockInfo) SecurityType() string { return "STOCKINFO" } type SecurityList struct { Securities []Security } func (r SecurityList) Name() string { return "SECLIST" } func (r SecurityList) Valid() (bool, error) { // TODO implement return true, nil } func (r *SecurityList) 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 "DEBTINFO": var info DebtInfo if err := d.DecodeElement(&info, &startElement); err != nil { return err } r.Securities = append(r.Securities, Security(info)) case "MFINFO": var info MFInfo if err := d.DecodeElement(&info, &startElement); err != nil { return err } r.Securities = append(r.Securities, Security(info)) case "OPTINFO": var info OptInfo if err := d.DecodeElement(&info, &startElement); err != nil { return err } r.Securities = append(r.Securities, Security(info)) case "OTHERINFO": var info OtherInfo if err := d.DecodeElement(&info, &startElement); err != nil { return err } r.Securities = append(r.Securities, Security(info)) case "STOCKINFO": var info StockInfo if err := d.DecodeElement(&info, &startElement); err != nil { return err } r.Securities = append(r.Securities, Security(info)) default: return errors.New("Invalid SECLIST child tag: " + startElement.Name.Local) } } else { return errors.New("Didn't find an opening element") } } } func DecodeSecuritiesMessageSet(d *xml.Decoder, start xml.StartElement) ([]Message, error) { var msgs []Message for { tok, err := nextNonWhitespaceToken(d) if err != nil { return nil, 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 msgs, nil } else if startElement, ok := tok.(xml.StartElement); ok { switch startElement.Name.Local { case "SECLISTTRNRS": var info SecListResponse if err := d.DecodeElement(&info, &startElement); err != nil { return nil, err } msgs = append(msgs, Message(info)) case "SECLIST": var info SecurityList if err := d.DecodeElement(&info, &startElement); err != nil { return nil, err } msgs = append(msgs, Message(info)) default: return nil, errors.New("Unsupported securities response tag: " + startElement.Name.Local) } } else { return nil, errors.New("Didn't find an opening element") } } }