Initial functional commit
This commit is contained in:
Options Vamp
188
customer.go
Normal file
188
customer.go
Normal file
@@ -0,0 +1,188 @@
|
||||
package tastytrade
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
type Address struct {
|
||||
StreetOne string `json:"street-one"`
|
||||
City string `json:"city"`
|
||||
StateRegion string `json:"state-region"`
|
||||
PostalCode string `json:"postal-code"`
|
||||
Country string `json:"country"`
|
||||
IsForeign bool `json:"is-foreign"`
|
||||
IsDomestic bool `json:"is-domestic"`
|
||||
}
|
||||
|
||||
type CustomerSuitability struct {
|
||||
ID int `json:"id"`
|
||||
MaritalStatus string `json:"marital-status"`
|
||||
NumberOfDependents int `json:"number-of-dependents"`
|
||||
EmploymentStatus string `json:"employment-status"`
|
||||
Occupation string `json:"occupation"`
|
||||
EmployerName string `json:"employer-name"`
|
||||
JobTitle string `json:"job-title"`
|
||||
AnnualNetIncome int `json:"annual-net-income"`
|
||||
NetWorth int `json:"net-worth"`
|
||||
LiquidNetWorth int `json:"liquid-net-worth"`
|
||||
StockTradingExperience string `json:"stock-trading-experience"`
|
||||
CoveredOptionsTradingExperience string `json:"covered-options-trading-experience"`
|
||||
UncoveredOptionsTradingExperience string `json:"uncovered-options-trading-experience"`
|
||||
FuturesTradingExperience string `json:"futures-trading-experience"`
|
||||
}
|
||||
|
||||
type Person struct {
|
||||
ExternalID string `json:"external-id"`
|
||||
FirstName string `json:"first-name"`
|
||||
LastName string `json:"last-name"`
|
||||
BirthDate string `json:"birth-date"`
|
||||
CitizenshipCountry string `json:"citizenship-country"`
|
||||
USACitizenshipType string `json:"usa-citizenship-type"`
|
||||
MaritalStatus string `json:"marital-status"`
|
||||
NumberOfDependents int `json:"number-of-dependents"`
|
||||
EmploymentStatus string `json:"employment-status"`
|
||||
Occupation string `json:"occupation"`
|
||||
EmployerName string `json:"employer-name"`
|
||||
JobTitle string `json:"job-title"`
|
||||
}
|
||||
|
||||
type CustomerData struct {
|
||||
ID string `json:"id"`
|
||||
FirstName string `json:"first-name"`
|
||||
LastName string `json:"last-name"`
|
||||
Address Address `json:"address"`
|
||||
MailingAddress Address `json:"mailing-address"`
|
||||
CustomerSuitability CustomerSuitability `json:"customer-suitability"`
|
||||
USACitizenshipType string `json:"usa-citizenship-type"`
|
||||
IsForeign bool `json:"is-foreign"`
|
||||
MobilePhoneNumber string `json:"mobile-phone-number"`
|
||||
Email string `json:"email"`
|
||||
TaxNumberType string `json:"tax-number-type"`
|
||||
TaxNumber string `json:"tax-number"`
|
||||
BirthDate string `json:"birth-date"`
|
||||
ExternalID string `json:"external-id"`
|
||||
CitizenshipCountry string `json:"citizenship-country"`
|
||||
SubjectToTaxWithholding bool `json:"subject-to-tax-withholding"`
|
||||
AgreedToMargining bool `json:"agreed-to-margining"`
|
||||
AgreedToTerms bool `json:"agreed-to-terms"`
|
||||
HasIndustryAffiliation bool `json:"has-industry-affiliation"`
|
||||
HasPoliticalAffiliation bool `json:"has-political-affiliation"`
|
||||
HasListedAffiliation bool `json:"has-listed-affiliation"`
|
||||
IsProfessional bool `json:"is-professional"`
|
||||
HasDelayedQuotes bool `json:"has-delayed-quotes"`
|
||||
HasPendingOrApprovedApplication bool `json:"has-pending-or-approved-application"`
|
||||
IdentifiableType string `json:"identifiable-type"`
|
||||
Person Person `json:"person"`
|
||||
}
|
||||
|
||||
type CustomerResponse struct {
|
||||
Context string `json:"context"`
|
||||
Data CustomerData `json:"data"`
|
||||
}
|
||||
|
||||
type Account struct {
|
||||
AccountNumber string `json:"account-number"`
|
||||
ExternalID string `json:"external-id"`
|
||||
OpenedAt string `json:"opened-at"`
|
||||
Nickname string `json:"nickname"`
|
||||
AccountTypeName string `json:"account-type-name"`
|
||||
DayTraderStatus bool `json:"day-trader-status"`
|
||||
IsClosed bool `json:"is-closed"`
|
||||
IsFirmError bool `json:"is-firm-error"`
|
||||
IsFirmProprietary bool `json:"is-firm-proprietary"`
|
||||
IsFuturesApproved bool `json:"is-futures-approved"`
|
||||
IsTestDrive bool `json:"is-test-drive"`
|
||||
MarginOrCash string `json:"margin-or-cash"`
|
||||
IsForeign bool `json:"is-foreign"`
|
||||
FundingDate string `json:"funding-date"`
|
||||
InvestmentObjective string `json:"investment-objective"`
|
||||
FuturesAccountPurpose string `json:"futures-account-purpose"`
|
||||
SuitableOptionsLevel string `json:"suitable-options-level"`
|
||||
CreatedAt string `json:"created-at"`
|
||||
}
|
||||
|
||||
type AccountContainer struct {
|
||||
Account Account `json:"account"`
|
||||
AuthorityLevel string `json:"authority-level"`
|
||||
}
|
||||
|
||||
type AccountData struct {
|
||||
Items []AccountContainer `json:"items"`
|
||||
}
|
||||
|
||||
type AccountResponse struct {
|
||||
Context string `json:"context"`
|
||||
Data Account `json:"data"`
|
||||
}
|
||||
|
||||
type AccountsResponse struct {
|
||||
Context string `json:"context"`
|
||||
Data AccountData `json:"data"`
|
||||
}
|
||||
|
||||
// GetCustomerInfo retrieves customer information
|
||||
func (api *TastytradeAPI) GetCustomerInfo() (CustomerResponse, error) {
|
||||
url := fmt.Sprintf("%s/customers/me", api.host)
|
||||
data, err := api.fetchData(url)
|
||||
if err != nil {
|
||||
return CustomerResponse{}, err
|
||||
}
|
||||
|
||||
var response CustomerResponse
|
||||
jsonData, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return CustomerResponse{}, err
|
||||
}
|
||||
|
||||
err = json.Unmarshal(jsonData, &response)
|
||||
if err != nil {
|
||||
return CustomerResponse{}, err
|
||||
}
|
||||
|
||||
return response, nil
|
||||
}
|
||||
|
||||
// ListCustomerAccounts retrieves a list of customer accounts
|
||||
func (api *TastytradeAPI) ListCustomerAccounts() (AccountsResponse, error) {
|
||||
url := fmt.Sprintf("%s/customers/me/accounts", api.host)
|
||||
data, err := api.fetchData(url)
|
||||
if err != nil {
|
||||
return AccountsResponse{}, err
|
||||
}
|
||||
|
||||
var response AccountsResponse
|
||||
jsonData, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return AccountsResponse{}, err
|
||||
}
|
||||
|
||||
err = json.Unmarshal(jsonData, &response)
|
||||
if err != nil {
|
||||
return AccountsResponse{}, err
|
||||
}
|
||||
|
||||
return response, nil
|
||||
}
|
||||
|
||||
// GetAccount retrieves account information for a specific account
|
||||
func (api *TastytradeAPI) GetAccount(accountNumber string) (AccountResponse, error) {
|
||||
url := fmt.Sprintf("%s/customers/me/accounts/%s", api.host, accountNumber)
|
||||
data, err := api.fetchData(url)
|
||||
if err != nil {
|
||||
return AccountResponse{}, err
|
||||
}
|
||||
|
||||
var response AccountResponse
|
||||
jsonData, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return AccountResponse{}, err
|
||||
}
|
||||
|
||||
err = json.Unmarshal(jsonData, &response)
|
||||
if err != nil {
|
||||
return AccountResponse{}, err
|
||||
}
|
||||
|
||||
return response, nil
|
||||
}
|
||||
Reference in New Issue
Block a user