Initial functional commit

2024-05-14 21:50:57 -04:00
parent 70888184fa
commit 461b8cef7b
23 changed files with 2704 additions and 1 deletions
--- a/status.go
+++ b/status.go
@@ -0,0 +1,67 @@
+package tastytrade
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+type TradingStatusData struct {
+	AccountNumber                            string `json:"account-number"`
+	DayTradeCount                            int    `json:"day-trade-count"`
+	EquitiesMarginCalculationType            string `json:"equities-margin-calculation-type"`
+	FeeScheduleName                          string `json:"fee-schedule-name"`
+	FuturesMarginRateMultiplier              string `json:"futures-margin-rate-multiplier"`
+	HasIntradayEquitiesMargin                bool   `json:"has-intraday-equities-margin"`
+	ID                                       int    `json:"id"`
+	IsAggregatedAtClearing                   bool   `json:"is-aggregated-at-clearing"`
+	IsClosed                                 bool   `json:"is-closed"`
+	IsClosingOnly                            bool   `json:"is-closing-only"`
+	IsCryptocurrencyClosingOnly              bool   `json:"is-cryptocurrency-closing-only"`
+	IsCryptocurrencyEnabled                  bool   `json:"is-cryptocurrency-enabled"`
+	IsFrozen                                 bool   `json:"is-frozen"`
+	IsFullEquityMarginRequired               bool   `json:"is-full-equity-margin-required"`
+	IsFuturesClosingOnly                     bool   `json:"is-futures-closing-only"`
+	IsFuturesIntraDayEnabled                 bool   `json:"is-futures-intra-day-enabled"`
+	IsFuturesEnabled                         bool   `json:"is-futures-enabled"`
+	IsInDayTradeEquityMaintenanceCall        bool   `json:"is-in-day-trade-equity-maintenance-call"`
+	IsInMarginCall                           bool   `json:"is-in-margin-call"`
+	IsPatternDayTrader                       bool   `json:"is-pattern-day-trader"`
+	IsRiskReducingOnly                       bool   `json:"is-risk-reducing-only"`
+	IsSmallNotionalFuturesIntraDayEnabled    bool   `json:"is-small-notional-futures-intra-day-enabled"`
+	IsRollTheDayForwardEnabled               bool   `json:"is-roll-the-day-forward-enabled"`
+	AreFarOtmNetOptionsRestricted            bool   `json:"are-far-otm-net-options-restricted"`
+	OptionsLevel                             string `json:"options-level"`
+	ShortCallsEnabled                        bool   `json:"short-calls-enabled"`
+	SmallNotionalFuturesMarginRateMultiplier string `json:"small-notional-futures-margin-rate-multiplier"`
+	IsEquityOfferingEnabled                  bool   `json:"is-equity-offering-enabled"`
+	IsEquityOfferingClosingOnly              bool   `json:"is-equity-offering-closing-only"`
+	EnhancedFraudSafeguardsEnabledAt         string `json:"enhanced-fraud-safeguards-enabled-at"`
+	UpdatedAt                                string `json:"updated-at"`
+}
+
+type TradingStatusResponse struct {
+	Context string            `json:"context"`
+	Data    TradingStatusData `json:"data"`
+}
+
+// GetAccountTradingStatus retrieves trading status for a specific account
+func (api *TastytradeAPI) GetAccountTradingStatus(accountNumber string) (TradingStatusResponse, error) {
+	url := fmt.Sprintf("%s/accounts/%s/trading-status", api.host, accountNumber)
+	data, err := api.fetchData(url)
+	if err != nil {
+		return TradingStatusResponse{}, err
+	}
+
+	var response TradingStatusResponse
+	jsonData, err := json.Marshal(data)
+	if err != nil {
+		return TradingStatusResponse{}, err
+	}
+
+	err = json.Unmarshal(jsonData, &response)
+	if err != nil {
+		return TradingStatusResponse{}, err
+	}
+
+	return response, nil
+}