package opensea import ( "encoding/json" "fmt" "io/ioutil" "net/http" "net/url" "strconv" "strings" "sync" "time" "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/log" "github.com/status-im/status-go/services/wallet/bigint" "github.com/status-im/status-go/services/wallet/thirdparty" ) const AssetLimit = 200 const CollectionLimit = 300 const RequestTimeout = 5 * time.Second const GetRequestRetryMaxCount = 15 const GetRequestWaitTime = 300 * time.Millisecond const ChainIDRequiringAPIKey = 1 func getbaseURL(chainID uint64) (string, error) { switch chainID { case 1: return "https://api.opensea.io/api/v1", nil case 4: return "https://rinkeby-api.opensea.io/api/v1", nil case 5: return "https://testnets-api.opensea.io/api/v1", nil } return "", fmt.Errorf("chainID not supported") } var OpenseaClientInstances = make(map[uint64]*Client) var OpenseaHTTPClient *HTTPClient = nil type TraitValue string func (st *TraitValue) UnmarshalJSON(b []byte) error { var item interface{} if err := json.Unmarshal(b, &item); err != nil { return err } switch v := item.(type) { case float64: *st = TraitValue(strconv.FormatFloat(v, 'f', 2, 64)) case int: *st = TraitValue(strconv.Itoa(v)) case string: *st = TraitValue(v) } return nil } type AssetContainer struct { Assets []Asset `json:"assets"` NextCursor string `json:"next"` PreviousCursor string `json:"previous"` } type Contract struct { Address string `json:"address"` } type Trait struct { TraitType string `json:"trait_type"` Value TraitValue `json:"value"` DisplayType string `json:"display_type"` MaxValue string `json:"max_value"` } type PaymentToken struct { ID int `json:"id"` Symbol string `json:"symbol"` Address string `json:"address"` ImageURL string `json:"image_url"` Name string `json:"name"` Decimals int `json:"decimals"` EthPrice string `json:"eth_price"` UsdPrice string `json:"usd_price"` } type LastSale struct { PaymentToken PaymentToken `json:"payment_token"` } type SellOrder struct { CurrentPrice string `json:"current_price"` } type Asset struct { ID int `json:"id"` TokenID *bigint.BigInt `json:"token_id"` Name string `json:"name"` Description string `json:"description"` Permalink string `json:"permalink"` ImageThumbnailURL string `json:"image_thumbnail_url"` ImageURL string `json:"image_url"` Contract Contract `json:"asset_contract"` Collection Collection `json:"collection"` Traits []Trait `json:"traits"` LastSale LastSale `json:"last_sale"` SellOrders []SellOrder `json:"sell_orders"` BackgroundColor string `json:"background_color"` TokenURI string `json:"token_metadata"` } type CollectionTrait struct { Min float64 `json:"min"` Max float64 `json:"max"` } type Collection struct { Name string `json:"name"` Slug string `json:"slug"` ImageURL string `json:"image_url"` Traits map[string]CollectionTrait `json:"traits"` } type OwnedCollection struct { Collection OwnedAssetCount *bigint.BigInt `json:"owned_asset_count"` } type HTTPClient struct { client *http.Client getRequestLock sync.RWMutex } func newHTTPClient() *HTTPClient { return &HTTPClient{ client: &http.Client{ Timeout: RequestTimeout, }, } } func (o *HTTPClient) doGetRequest(url string, apiKey string) ([]byte, error) { // Ensure only one thread makes a request at a time o.getRequestLock.Lock() defer o.getRequestLock.Unlock() retryCount := 0 statusCode := http.StatusOK // Try to do the request without an apiKey first tmpAPIKey := "" for { req, err := http.NewRequest(http.MethodGet, url, nil) if err != nil { return nil, err } req.Header.Set("Content-Type", "application/json") req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:96.0) Gecko/20100101 Firefox/96.0") if len(tmpAPIKey) > 0 { req.Header.Set("X-API-KEY", tmpAPIKey) } resp, err := o.client.Do(req) if err != nil { return nil, err } defer func() { if err := resp.Body.Close(); err != nil { log.Error("failed to close opensea request body", "err", err) } }() statusCode = resp.StatusCode switch resp.StatusCode { case http.StatusOK: body, err := ioutil.ReadAll(resp.Body) return body, err case http.StatusTooManyRequests: if retryCount < GetRequestRetryMaxCount { // sleep and retry time.Sleep(GetRequestWaitTime) retryCount++ continue } // break and error case http.StatusForbidden: // Request requires an apiKey, set it and retry if tmpAPIKey == "" && apiKey != "" { tmpAPIKey = apiKey // sleep and retry time.Sleep(GetRequestWaitTime) continue } // break and error default: // break and error } break } return nil, fmt.Errorf("unsuccessful request: %d %s", statusCode, http.StatusText(statusCode)) } type Client struct { client *HTTPClient url string apiKey string IsConnected bool LastCheckedAt int64 IsConnectedLock sync.RWMutex } // new opensea client. func NewOpenseaClient(chainID uint64, apiKey string) (*Client, error) { if OpenseaHTTPClient == nil { OpenseaHTTPClient = newHTTPClient() } var tmpAPIKey string = "" if chainID == ChainIDRequiringAPIKey { tmpAPIKey = apiKey } if client, ok := OpenseaClientInstances[chainID]; ok { if client.apiKey == tmpAPIKey { return client, nil } } baseURL, err := getbaseURL(chainID) if err != nil { return nil, err } openseaClient := &Client{ client: OpenseaHTTPClient, url: baseURL, apiKey: tmpAPIKey, IsConnected: true, LastCheckedAt: time.Now().Unix(), } OpenseaClientInstances[chainID] = openseaClient return openseaClient, nil } func (o *Client) setConnected(value bool) { o.IsConnectedLock.Lock() defer o.IsConnectedLock.Unlock() o.IsConnected = value o.LastCheckedAt = time.Now().Unix() } func (o *Client) FetchAllCollectionsByOwner(owner common.Address) ([]OwnedCollection, error) { offset := 0 var collections []OwnedCollection for { url := fmt.Sprintf("%s/collections?asset_owner=%s&offset=%d&limit=%d", o.url, owner, offset, CollectionLimit) body, err := o.client.doGetRequest(url, o.apiKey) if err != nil { o.setConnected(false) return nil, err } // if Json is not returned there must be an error if !json.Valid(body) { return nil, fmt.Errorf("invalid json: %s", string(body)) } var tmp []OwnedCollection err = json.Unmarshal(body, &tmp) if err != nil { o.setConnected(false) return nil, err } collections = append(collections, tmp...) if len(tmp) < CollectionLimit { break } } o.setConnected(true) return collections, nil } func (o *Client) FetchAllAssetsByOwnerAndCollection(owner common.Address, collectionSlug string, cursor string, limit int) (*AssetContainer, error) { queryParams := url.Values{ "owner": {owner.String()}, "collection": {collectionSlug}, } if len(cursor) > 0 { queryParams["cursor"] = []string{cursor} } return o.fetchAssets(queryParams, limit) } func (o *Client) FetchAllAssetsByOwnerAndContractAddress(owner common.Address, contractAddresses []common.Address, cursor string, limit int) (*AssetContainer, error) { queryParams := url.Values{ "owner": {owner.String()}, } for _, contractAddress := range contractAddresses { queryParams.Add("asset_contract_addresses", contractAddress.String()) } if len(cursor) > 0 { queryParams["cursor"] = []string{cursor} } return o.fetchAssets(queryParams, limit) } func (o *Client) FetchAllAssetsByOwner(owner common.Address, cursor string, limit int) (*AssetContainer, error) { queryParams := url.Values{ "owner": {owner.String()}, } if len(cursor) > 0 { queryParams["cursor"] = []string{cursor} } return o.fetchAssets(queryParams, limit) } func (o *Client) FetchAssetsByNFTUniqueID(uniqueIDs []thirdparty.NFTUniqueID, limit int) (*AssetContainer, error) { queryParams := url.Values{} for _, uniqueID := range uniqueIDs { queryParams.Add("token_ids", uniqueID.TokenID.String()) queryParams.Add("asset_contract_addresses", uniqueID.ContractAddress.String()) } return o.fetchAssets(queryParams, limit) } func (o *Client) fetchAssets(queryParams url.Values, limit int) (*AssetContainer, error) { assets := new(AssetContainer) if len(queryParams["cursor"]) > 0 { assets.PreviousCursor = queryParams["cursor"][0] } tmpLimit := AssetLimit if limit > 0 && limit < tmpLimit { tmpLimit = AssetLimit } queryParams["limit"] = []string{strconv.Itoa(tmpLimit)} for { url := o.url + "/assets?" + queryParams.Encode() body, err := o.client.doGetRequest(url, o.apiKey) if err != nil { o.setConnected(false) return nil, err } // if Json is not returned there must be an error if !json.Valid(body) { return nil, fmt.Errorf("invalid json: %s", string(body)) } container := AssetContainer{} err = json.Unmarshal(body, &container) if err != nil { o.setConnected(false) return nil, err } for _, asset := range container.Assets { for i := range asset.Traits { asset.Traits[i].TraitType = strings.Replace(asset.Traits[i].TraitType, "_", " ", 1) asset.Traits[i].Value = TraitValue(strings.Title(string(asset.Traits[i].Value))) } assets.Assets = append(assets.Assets, asset) } assets.NextCursor = container.NextCursor if len(assets.NextCursor) == 0 { break } queryParams["cursor"] = []string{assets.NextCursor} if limit > 0 && len(assets.Assets) >= limit { break } } o.setConnected(true) return assets, nil }