sjc
46dbff0530
Renamed Authenticate to Authenticate2 and added remember-me support Created Authenticate as wrapper for Authenticate2 for backwards compat.
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package tastytrade
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
baseURL = "https://api.tastytrade.com"
|
|
)
|
|
|
|
// TastytradeAPI represents the Tastytrade API client
|
|
type TastytradeAPI struct {
|
|
httpClient *http.Client
|
|
authToken string
|
|
remToken string
|
|
host string
|
|
}
|
|
|
|
// NewTastytradeAPI creates a new instance of TastytradeAPI
|
|
func NewTastytradeAPI(hosts ...string) *TastytradeAPI {
|
|
host := baseURL
|
|
if len(hosts) > 0 {
|
|
host = hosts[0]
|
|
}
|
|
return &TastytradeAPI{
|
|
httpClient: &http.Client{Timeout: 10 * time.Second},
|
|
host: host,
|
|
}
|
|
}
|
|
|
|
// fetchData sends a GET request to the specified URL with authorization
|
|
func (api *TastytradeAPI) fetchData(url string) (map[string]interface{}, error) {
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Set("Authorization", api.authToken)
|
|
|
|
resp, err := api.httpClient.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode >= 400 && resp.StatusCode < 500 {
|
|
return nil, fmt.Errorf("client error occurred: status code %d", resp.StatusCode)
|
|
} else if resp.StatusCode >= 500 {
|
|
return nil, fmt.Errorf("server error occurred: status code %d", resp.StatusCode)
|
|
}
|
|
|
|
var data map[string]interface{}
|
|
if err := json.NewDecoder(resp.Body).Decode(&data); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
// Helper function to fetch and unmarshal data
|
|
func (api *TastytradeAPI) fetchDataAndUnmarshal(urlVal string, v interface{}) error {
|
|
req, err := http.NewRequest("GET", urlVal, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.Header.Set("Authorization", api.authToken)
|
|
|
|
resp, err := api.httpClient.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode >= 400 && resp.StatusCode < 500 {
|
|
return fmt.Errorf("client error occurred: status code %d", resp.StatusCode)
|
|
} else if resp.StatusCode >= 500 {
|
|
return fmt.Errorf("server error occurred: status code %d", resp.StatusCode)
|
|
}
|
|
|
|
if err := json.NewDecoder(resp.Body).Decode(&v); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|