859 lines
27 KiB
Go
859 lines
27 KiB
Go
package collectibles
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"errors"
|
|
"math/big"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/afex/hystrix-go/hystrix"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/event"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/status-im/status-go/contracts/community-tokens/collectibles"
|
|
"github.com/status-im/status-go/rpc"
|
|
"github.com/status-im/status-go/server"
|
|
"github.com/status-im/status-go/services/wallet/async"
|
|
"github.com/status-im/status-go/services/wallet/bigint"
|
|
walletCommon "github.com/status-im/status-go/services/wallet/common"
|
|
"github.com/status-im/status-go/services/wallet/community"
|
|
"github.com/status-im/status-go/services/wallet/connection"
|
|
"github.com/status-im/status-go/services/wallet/thirdparty"
|
|
"github.com/status-im/status-go/services/wallet/walletevent"
|
|
)
|
|
|
|
const requestTimeout = 5 * time.Second
|
|
const signalUpdatedCollectiblesDataPageSize = 10
|
|
|
|
const hystrixContractOwnershipClientName = "contractOwnershipClient"
|
|
|
|
const EventCollectiblesConnectionStatusChanged walletevent.EventType = "wallet-collectible-status-changed"
|
|
|
|
// ERC721 does not support function "TokenURI" if call
|
|
// returns error starting with one of these strings
|
|
var noTokenURIErrorPrefixes = []string{
|
|
"execution reverted",
|
|
"abi: attempting to unmarshall",
|
|
}
|
|
|
|
var (
|
|
ErrAllProvidersFailedForChainID = errors.New("all providers failed for chainID")
|
|
ErrNoProvidersAvailableForChainID = errors.New("no providers available for chainID")
|
|
)
|
|
|
|
type ManagerInterface interface {
|
|
FetchAssetsByCollectibleUniqueID(ctx context.Context, uniqueIDs []thirdparty.CollectibleUniqueID, asyncFetch bool) ([]thirdparty.FullCollectibleData, error)
|
|
}
|
|
|
|
type Manager struct {
|
|
rpcClient *rpc.Client
|
|
contractOwnershipProviders []thirdparty.CollectibleContractOwnershipProvider
|
|
accountOwnershipProviders []thirdparty.CollectibleAccountOwnershipProvider
|
|
collectibleDataProviders []thirdparty.CollectibleDataProvider
|
|
collectionDataProviders []thirdparty.CollectionDataProvider
|
|
collectibleProviders []thirdparty.CollectibleProvider
|
|
|
|
httpClient *http.Client
|
|
|
|
collectiblesDataDB *CollectibleDataDB
|
|
collectionsDataDB *CollectionDataDB
|
|
communityManager *community.Manager
|
|
ownershipDB *OwnershipDB
|
|
|
|
mediaServer *server.MediaServer
|
|
|
|
statuses map[string]*connection.Status
|
|
statusNotifier *connection.StatusNotifier
|
|
feed *event.Feed
|
|
}
|
|
|
|
func NewManager(
|
|
db *sql.DB,
|
|
rpcClient *rpc.Client,
|
|
communityManager *community.Manager,
|
|
contractOwnershipProviders []thirdparty.CollectibleContractOwnershipProvider,
|
|
accountOwnershipProviders []thirdparty.CollectibleAccountOwnershipProvider,
|
|
collectibleDataProviders []thirdparty.CollectibleDataProvider,
|
|
collectionDataProviders []thirdparty.CollectionDataProvider,
|
|
mediaServer *server.MediaServer,
|
|
feed *event.Feed) *Manager {
|
|
hystrix.ConfigureCommand(hystrixContractOwnershipClientName, hystrix.CommandConfig{
|
|
Timeout: 10000,
|
|
MaxConcurrentRequests: 100,
|
|
SleepWindow: 300000,
|
|
ErrorPercentThreshold: 25,
|
|
})
|
|
|
|
ownershipDB := NewOwnershipDB(db)
|
|
|
|
statuses := make(map[string]*connection.Status)
|
|
|
|
allChainIDs := walletCommon.AllChainIDs()
|
|
for _, chainID := range allChainIDs {
|
|
status := connection.NewStatus()
|
|
state := status.GetState()
|
|
latestUpdateTimestamp, err := ownershipDB.GetLatestOwnershipUpdateTimestamp(chainID)
|
|
if err == nil {
|
|
state.LastSuccessAt = latestUpdateTimestamp
|
|
status.SetState(state)
|
|
}
|
|
statuses[chainID.String()] = status
|
|
}
|
|
|
|
statusNotifier := connection.NewStatusNotifier(
|
|
statuses,
|
|
EventCollectiblesConnectionStatusChanged,
|
|
feed,
|
|
)
|
|
|
|
// Get list of all providers
|
|
collectibleProvidersMap := make(map[string]thirdparty.CollectibleProvider)
|
|
collectibleProviders := make([]thirdparty.CollectibleProvider, 0)
|
|
for _, provider := range contractOwnershipProviders {
|
|
collectibleProvidersMap[provider.ID()] = provider
|
|
}
|
|
for _, provider := range accountOwnershipProviders {
|
|
collectibleProvidersMap[provider.ID()] = provider
|
|
}
|
|
for _, provider := range collectibleDataProviders {
|
|
collectibleProvidersMap[provider.ID()] = provider
|
|
}
|
|
for _, provider := range collectionDataProviders {
|
|
collectibleProvidersMap[provider.ID()] = provider
|
|
}
|
|
for _, provider := range collectibleProvidersMap {
|
|
collectibleProviders = append(collectibleProviders, provider)
|
|
}
|
|
|
|
return &Manager{
|
|
rpcClient: rpcClient,
|
|
contractOwnershipProviders: contractOwnershipProviders,
|
|
accountOwnershipProviders: accountOwnershipProviders,
|
|
collectibleDataProviders: collectibleDataProviders,
|
|
collectionDataProviders: collectionDataProviders,
|
|
collectibleProviders: collectibleProviders,
|
|
httpClient: &http.Client{
|
|
Timeout: requestTimeout,
|
|
},
|
|
collectiblesDataDB: NewCollectibleDataDB(db),
|
|
collectionsDataDB: NewCollectionDataDB(db),
|
|
communityManager: communityManager,
|
|
ownershipDB: ownershipDB,
|
|
mediaServer: mediaServer,
|
|
statuses: statuses,
|
|
statusNotifier: statusNotifier,
|
|
feed: feed,
|
|
}
|
|
}
|
|
|
|
func mapToList[K comparable, T any](m map[K]T) []T {
|
|
list := make([]T, 0, len(m))
|
|
for _, v := range m {
|
|
list = append(list, v)
|
|
}
|
|
return list
|
|
}
|
|
|
|
func makeContractOwnershipCall(main func() (any, error), fallback func() (any, error)) (any, error) {
|
|
resultChan := make(chan any, 1)
|
|
errChan := hystrix.Go(hystrixContractOwnershipClientName, func() error {
|
|
res, err := main()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resultChan <- res
|
|
return nil
|
|
}, func(err error) error {
|
|
if fallback == nil {
|
|
return err
|
|
}
|
|
|
|
res, err := fallback()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resultChan <- res
|
|
return nil
|
|
})
|
|
select {
|
|
case result := <-resultChan:
|
|
return result, nil
|
|
case err := <-errChan:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func (o *Manager) doContentTypeRequest(ctx context.Context, url string) (string, error) {
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodHead, url, nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
resp, err := o.httpClient.Do(req)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer func() {
|
|
if err := resp.Body.Close(); err != nil {
|
|
log.Error("failed to close head request body", "err", err)
|
|
}
|
|
}()
|
|
|
|
return resp.Header.Get("Content-Type"), nil
|
|
}
|
|
|
|
// Need to combine different providers to support all needed ChainIDs
|
|
func (o *Manager) FetchBalancesByOwnerAndContractAddress(ctx context.Context, chainID walletCommon.ChainID, ownerAddress common.Address, contractAddresses []common.Address) (thirdparty.TokenBalancesPerContractAddress, error) {
|
|
ret := make(thirdparty.TokenBalancesPerContractAddress)
|
|
|
|
for _, contractAddress := range contractAddresses {
|
|
ret[contractAddress] = make([]thirdparty.TokenBalance, 0)
|
|
}
|
|
|
|
// Try with account ownership providers first
|
|
assetsContainer, err := o.FetchAllAssetsByOwnerAndContractAddress(ctx, chainID, ownerAddress, contractAddresses, thirdparty.FetchFromStartCursor, thirdparty.FetchNoLimit, thirdparty.FetchFromAnyProvider)
|
|
if err == ErrNoProvidersAvailableForChainID {
|
|
// Use contract ownership providers
|
|
for _, contractAddress := range contractAddresses {
|
|
ownership, err := o.FetchCollectibleOwnersByContractAddress(ctx, chainID, contractAddress)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, nftOwner := range ownership.Owners {
|
|
if nftOwner.OwnerAddress == ownerAddress {
|
|
ret[contractAddress] = nftOwner.TokenBalances
|
|
break
|
|
}
|
|
}
|
|
}
|
|
} else if err == nil {
|
|
// Account ownership providers succeeded
|
|
for _, fullData := range assetsContainer.Items {
|
|
contractAddress := fullData.CollectibleData.ID.ContractID.Address
|
|
balance := thirdparty.TokenBalance{
|
|
TokenID: fullData.CollectibleData.ID.TokenID,
|
|
Balance: &bigint.BigInt{Int: big.NewInt(1)},
|
|
}
|
|
ret[contractAddress] = append(ret[contractAddress], balance)
|
|
}
|
|
} else {
|
|
// OpenSea could have provided, but returned error
|
|
return nil, err
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func (o *Manager) FetchAllAssetsByOwnerAndContractAddress(ctx context.Context, chainID walletCommon.ChainID, owner common.Address, contractAddresses []common.Address, cursor string, limit int, providerID string) (*thirdparty.FullCollectibleDataContainer, error) {
|
|
defer o.checkConnectionStatus(chainID)
|
|
|
|
anyProviderAvailable := false
|
|
for _, provider := range o.accountOwnershipProviders {
|
|
if !provider.IsChainSupported(chainID) {
|
|
continue
|
|
}
|
|
anyProviderAvailable = true
|
|
if providerID != thirdparty.FetchFromAnyProvider && providerID != provider.ID() {
|
|
continue
|
|
}
|
|
|
|
assetContainer, err := provider.FetchAllAssetsByOwnerAndContractAddress(ctx, chainID, owner, contractAddresses, cursor, limit)
|
|
if err != nil {
|
|
log.Error("FetchAllAssetsByOwnerAndContractAddress failed for", "provider", provider.ID(), "chainID", chainID, "err", err)
|
|
continue
|
|
}
|
|
|
|
_, err = o.processFullCollectibleData(ctx, assetContainer.Items, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return assetContainer, nil
|
|
}
|
|
|
|
if anyProviderAvailable {
|
|
return nil, ErrAllProvidersFailedForChainID
|
|
}
|
|
return nil, ErrNoProvidersAvailableForChainID
|
|
}
|
|
|
|
func (o *Manager) FetchAllAssetsByOwner(ctx context.Context, chainID walletCommon.ChainID, owner common.Address, cursor string, limit int, providerID string) (*thirdparty.FullCollectibleDataContainer, error) {
|
|
defer o.checkConnectionStatus(chainID)
|
|
|
|
anyProviderAvailable := false
|
|
for _, provider := range o.accountOwnershipProviders {
|
|
if !provider.IsChainSupported(chainID) {
|
|
continue
|
|
}
|
|
anyProviderAvailable = true
|
|
if providerID != thirdparty.FetchFromAnyProvider && providerID != provider.ID() {
|
|
continue
|
|
}
|
|
|
|
assetContainer, err := provider.FetchAllAssetsByOwner(ctx, chainID, owner, cursor, limit)
|
|
if err != nil {
|
|
log.Error("FetchAllAssetsByOwner failed for", "provider", provider.ID(), "chainID", chainID, "err", err)
|
|
continue
|
|
}
|
|
|
|
_, err = o.processFullCollectibleData(ctx, assetContainer.Items, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return assetContainer, nil
|
|
}
|
|
|
|
if anyProviderAvailable {
|
|
return nil, ErrAllProvidersFailedForChainID
|
|
}
|
|
return nil, ErrNoProvidersAvailableForChainID
|
|
}
|
|
|
|
func (o *Manager) FetchCollectibleOwnershipByOwner(ctx context.Context, chainID walletCommon.ChainID, owner common.Address, cursor string, limit int, providerID string) (*thirdparty.CollectibleOwnershipContainer, error) {
|
|
// We don't yet have an API that will return only Ownership data
|
|
// Use the full Ownership + Metadata endpoint and use the data we need
|
|
assetContainer, err := o.FetchAllAssetsByOwner(ctx, chainID, owner, cursor, limit, providerID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ret := assetContainer.ToOwnershipContainer()
|
|
return &ret, nil
|
|
}
|
|
|
|
// Returns collectible metadata for the given unique IDs.
|
|
// If asyncFetch is true, empty metadata will be returned for any missing collectibles and an EventCollectiblesDataUpdated will be sent when the data is ready.
|
|
// If asyncFetch is false, it will wait for all collectibles' metadata to be retrieved before returning.
|
|
func (o *Manager) FetchAssetsByCollectibleUniqueID(ctx context.Context, uniqueIDs []thirdparty.CollectibleUniqueID, asyncFetch bool) ([]thirdparty.FullCollectibleData, error) {
|
|
missingIDs, err := o.collectiblesDataDB.GetIDsNotInDB(uniqueIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
missingIDsPerChainID := thirdparty.GroupCollectibleUIDsByChainID(missingIDs)
|
|
|
|
group := async.NewGroup(ctx)
|
|
group.Add(func(ctx context.Context) error {
|
|
for chainID, idsToFetch := range missingIDsPerChainID {
|
|
defer o.checkConnectionStatus(chainID)
|
|
|
|
for _, provider := range o.collectibleDataProviders {
|
|
if !provider.IsChainSupported(chainID) {
|
|
continue
|
|
}
|
|
|
|
fetchedAssets, err := provider.FetchAssetsByCollectibleUniqueID(ctx, idsToFetch)
|
|
if err != nil {
|
|
log.Error("FetchAssetsByCollectibleUniqueID failed for", "provider", provider.ID(), "chainID", chainID, "err", err)
|
|
continue
|
|
}
|
|
|
|
updatedCollectibles, err := o.processFullCollectibleData(ctx, fetchedAssets, asyncFetch)
|
|
if err != nil {
|
|
log.Error("processFullCollectibleData failed for", "provider", provider.ID(), "chainID", chainID, "len(fetchedAssets)", len(fetchedAssets), "err", err)
|
|
return err
|
|
}
|
|
|
|
if asyncFetch {
|
|
o.signalUpdatedCollectiblesData(updatedCollectibles)
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if !asyncFetch {
|
|
group.Wait()
|
|
}
|
|
|
|
return o.getCacheFullCollectibleData(uniqueIDs)
|
|
}
|
|
|
|
func (o *Manager) FetchCollectionsDataByContractID(ctx context.Context, ids []thirdparty.ContractID) ([]thirdparty.CollectionData, error) {
|
|
missingIDs, err := o.collectionsDataDB.GetIDsNotInDB(ids)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
missingIDsPerChainID := thirdparty.GroupContractIDsByChainID(missingIDs)
|
|
|
|
for chainID, idsToFetch := range missingIDsPerChainID {
|
|
defer o.checkConnectionStatus(chainID)
|
|
|
|
for _, provider := range o.collectionDataProviders {
|
|
if !provider.IsChainSupported(chainID) {
|
|
continue
|
|
}
|
|
|
|
fetchedCollections, err := provider.FetchCollectionsDataByContractID(ctx, idsToFetch)
|
|
if err != nil {
|
|
log.Error("FetchCollectionsDataByContractID failed for", "provider", provider.ID(), "chainID", chainID, "err", err)
|
|
continue
|
|
}
|
|
|
|
err = o.processCollectionData(ctx, fetchedCollections)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
break
|
|
}
|
|
}
|
|
|
|
data, err := o.collectionsDataDB.GetData(ids)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return mapToList(data), nil
|
|
}
|
|
|
|
func (o *Manager) getContractOwnershipProviders(chainID walletCommon.ChainID) (mainProvider thirdparty.CollectibleContractOwnershipProvider, fallbackProvider thirdparty.CollectibleContractOwnershipProvider) {
|
|
mainProvider = nil
|
|
fallbackProvider = nil
|
|
|
|
for _, provider := range o.contractOwnershipProviders {
|
|
if provider.IsChainSupported(chainID) {
|
|
if mainProvider == nil {
|
|
// First provider found
|
|
mainProvider = provider
|
|
continue
|
|
}
|
|
// Second provider found
|
|
fallbackProvider = provider
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func getCollectibleOwnersByContractAddressFunc(ctx context.Context, chainID walletCommon.ChainID, contractAddress common.Address, provider thirdparty.CollectibleContractOwnershipProvider) func() (any, error) {
|
|
if provider == nil {
|
|
return nil
|
|
}
|
|
return func() (any, error) {
|
|
res, err := provider.FetchCollectibleOwnersByContractAddress(ctx, chainID, contractAddress)
|
|
if err != nil {
|
|
log.Error("FetchCollectibleOwnersByContractAddress failed for", "provider", provider.ID(), "chainID", chainID, "err", err)
|
|
}
|
|
return res, err
|
|
}
|
|
}
|
|
|
|
func (o *Manager) FetchCollectibleOwnersByContractAddress(ctx context.Context, chainID walletCommon.ChainID, contractAddress common.Address) (*thirdparty.CollectibleContractOwnership, error) {
|
|
defer o.checkConnectionStatus(chainID)
|
|
|
|
mainProvider, fallbackProvider := o.getContractOwnershipProviders(chainID)
|
|
if mainProvider == nil {
|
|
return nil, ErrNoProvidersAvailableForChainID
|
|
}
|
|
|
|
mainFn := getCollectibleOwnersByContractAddressFunc(ctx, chainID, contractAddress, mainProvider)
|
|
fallbackFn := getCollectibleOwnersByContractAddressFunc(ctx, chainID, contractAddress, fallbackProvider)
|
|
|
|
owners, err := makeContractOwnershipCall(mainFn, fallbackFn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return owners.(*thirdparty.CollectibleContractOwnership), nil
|
|
}
|
|
|
|
func (o *Manager) fetchTokenURI(ctx context.Context, id thirdparty.CollectibleUniqueID) (string, error) {
|
|
if id.TokenID == nil {
|
|
return "", errors.New("empty token ID")
|
|
}
|
|
backend, err := o.rpcClient.EthClient(uint64(id.ContractID.ChainID))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
caller, err := collectibles.NewCollectiblesCaller(id.ContractID.Address, backend)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
tokenURI, err := caller.TokenURI(&bind.CallOpts{
|
|
Context: ctx,
|
|
}, id.TokenID.Int)
|
|
|
|
if err != nil {
|
|
for _, errorPrefix := range noTokenURIErrorPrefixes {
|
|
if strings.HasPrefix(err.Error(), errorPrefix) {
|
|
// Contract doesn't support "TokenURI" method
|
|
return "", nil
|
|
}
|
|
}
|
|
return "", err
|
|
}
|
|
|
|
return tokenURI, err
|
|
}
|
|
|
|
func isMetadataEmpty(asset thirdparty.CollectibleData) bool {
|
|
return asset.Description == "" &&
|
|
asset.ImageURL == ""
|
|
}
|
|
|
|
// Processes collectible metadata obtained from a provider and ensures any missing data is fetched.
|
|
// If asyncFetch is true, community collectibles metadata will be fetched async and an EventCollectiblesDataUpdated will be sent when the data is ready.
|
|
// If asyncFetch is false, it will wait for all community collectibles' metadata to be retrieved before returning.
|
|
func (o *Manager) processFullCollectibleData(ctx context.Context, assets []thirdparty.FullCollectibleData, asyncFetch bool) ([]thirdparty.CollectibleUniqueID, error) {
|
|
fullyFetchedAssets := make(map[string]*thirdparty.FullCollectibleData)
|
|
communityCollectibles := make(map[string][]*thirdparty.FullCollectibleData)
|
|
processedIDs := make([]thirdparty.CollectibleUniqueID, 0, len(assets))
|
|
|
|
// Start with all assets, remove if any of the fetch steps fail
|
|
for idx := range assets {
|
|
asset := &assets[idx]
|
|
id := asset.CollectibleData.ID
|
|
fullyFetchedAssets[id.HashKey()] = asset
|
|
}
|
|
|
|
// Detect community collectibles
|
|
for _, asset := range fullyFetchedAssets {
|
|
// Only check community ownership if metadata is empty
|
|
if isMetadataEmpty(asset.CollectibleData) {
|
|
// Get TokenURI if not given by provider
|
|
err := o.fillTokenURI(ctx, asset)
|
|
if err != nil {
|
|
log.Error("fillTokenURI failed", "err", err)
|
|
delete(fullyFetchedAssets, asset.CollectibleData.ID.HashKey())
|
|
continue
|
|
}
|
|
|
|
// Get CommunityID if obtainable from TokenURI
|
|
err = o.fillCommunityID(asset)
|
|
if err != nil {
|
|
log.Error("fillCommunityID failed", "err", err)
|
|
delete(fullyFetchedAssets, asset.CollectibleData.ID.HashKey())
|
|
continue
|
|
}
|
|
|
|
// Get metadata from community if community collectible
|
|
communityID := asset.CollectibleData.CommunityID
|
|
if communityID != "" {
|
|
if _, ok := communityCollectibles[communityID]; !ok {
|
|
communityCollectibles[communityID] = make([]*thirdparty.FullCollectibleData, 0)
|
|
}
|
|
communityCollectibles[communityID] = append(communityCollectibles[communityID], asset)
|
|
|
|
// Community collectibles are handled separately, remove from list
|
|
delete(fullyFetchedAssets, asset.CollectibleData.ID.HashKey())
|
|
}
|
|
}
|
|
}
|
|
|
|
// Community collectibles are grouped by community ID
|
|
for communityID, communityAssets := range communityCollectibles {
|
|
if asyncFetch {
|
|
o.fetchCommunityAssetsAsync(ctx, communityID, communityAssets)
|
|
} else {
|
|
err := o.fetchCommunityAssets(communityID, communityAssets)
|
|
if err != nil {
|
|
log.Error("fetchCommunityAssets failed", "communityID", communityID, "err", err)
|
|
continue
|
|
}
|
|
for _, asset := range communityAssets {
|
|
processedIDs = append(processedIDs, asset.CollectibleData.ID)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, asset := range fullyFetchedAssets {
|
|
err := o.fillAnimationMediatype(ctx, asset)
|
|
if err != nil {
|
|
log.Error("fillAnimationMediatype failed", "err", err)
|
|
delete(fullyFetchedAssets, asset.CollectibleData.ID.HashKey())
|
|
continue
|
|
}
|
|
}
|
|
|
|
// Save successfully fetched data to DB
|
|
collectiblesData := make([]thirdparty.CollectibleData, 0, len(assets))
|
|
collectionsData := make([]thirdparty.CollectionData, 0, len(assets))
|
|
missingCollectionIDs := make([]thirdparty.ContractID, 0)
|
|
|
|
for _, asset := range fullyFetchedAssets {
|
|
id := asset.CollectibleData.ID
|
|
processedIDs = append(processedIDs, id)
|
|
|
|
collectiblesData = append(collectiblesData, asset.CollectibleData)
|
|
if asset.CollectionData != nil {
|
|
collectionsData = append(collectionsData, *asset.CollectionData)
|
|
} else {
|
|
missingCollectionIDs = append(missingCollectionIDs, id.ContractID)
|
|
}
|
|
}
|
|
|
|
err := o.collectiblesDataDB.SetData(collectiblesData, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = o.collectionsDataDB.SetData(collectionsData, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(missingCollectionIDs) > 0 {
|
|
// Calling this ensures collection data is fetched and cached (if not already available)
|
|
_, err := o.FetchCollectionsDataByContractID(ctx, missingCollectionIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return processedIDs, nil
|
|
}
|
|
|
|
func (o *Manager) fillTokenURI(ctx context.Context, asset *thirdparty.FullCollectibleData) error {
|
|
id := asset.CollectibleData.ID
|
|
|
|
tokenURI := asset.CollectibleData.TokenURI
|
|
// Only need to fetch it from contract if it was empty
|
|
if tokenURI == "" {
|
|
tokenURI, err := o.fetchTokenURI(ctx, id)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
asset.CollectibleData.TokenURI = tokenURI
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (o *Manager) fillCommunityID(asset *thirdparty.FullCollectibleData) error {
|
|
tokenURI := asset.CollectibleData.TokenURI
|
|
|
|
communityID := ""
|
|
if tokenURI != "" {
|
|
communityID = o.communityManager.GetCommunityID(tokenURI)
|
|
}
|
|
|
|
asset.CollectibleData.CommunityID = communityID
|
|
return nil
|
|
}
|
|
|
|
func (o *Manager) fetchCommunityAssets(communityID string, communityAssets []*thirdparty.FullCollectibleData) error {
|
|
communityInfo, err := o.communityManager.FetchCommunityInfo(communityID)
|
|
|
|
// If the community is found, we update the DB.
|
|
// If the community is not found, we only insert new entries to the DB (don't replace what is already there).
|
|
allowUpdate := false
|
|
if err != nil {
|
|
log.Error("fetchCommunityInfo failed", "communityID", communityID, "err", err)
|
|
} else if communityInfo == nil {
|
|
log.Warn("fetchCommunityAssets community not found", "communityID", communityID)
|
|
} else {
|
|
for _, communityAsset := range communityAssets {
|
|
err := o.communityManager.FillCollectibleMetadata(communityAsset)
|
|
if err != nil {
|
|
log.Error("FillCollectibleMetadata failed", "communityID", communityID, "err", err)
|
|
return err
|
|
}
|
|
}
|
|
allowUpdate = true
|
|
}
|
|
|
|
collectiblesData := make([]thirdparty.CollectibleData, 0, len(communityAssets))
|
|
collectionsData := make([]thirdparty.CollectionData, 0, len(communityAssets))
|
|
|
|
for _, asset := range communityAssets {
|
|
collectiblesData = append(collectiblesData, asset.CollectibleData)
|
|
if asset.CollectionData != nil {
|
|
collectionsData = append(collectionsData, *asset.CollectionData)
|
|
}
|
|
}
|
|
|
|
err = o.collectiblesDataDB.SetData(collectiblesData, allowUpdate)
|
|
if err != nil {
|
|
log.Error("collectiblesDataDB SetData failed", "communityID", communityID, "err", err)
|
|
return err
|
|
}
|
|
|
|
err = o.collectionsDataDB.SetData(collectionsData, allowUpdate)
|
|
if err != nil {
|
|
log.Error("collectionsDataDB SetData failed", "communityID", communityID, "err", err)
|
|
return err
|
|
}
|
|
|
|
for _, asset := range communityAssets {
|
|
if asset.CollectibleCommunityInfo != nil {
|
|
err = o.collectiblesDataDB.SetCommunityInfo(asset.CollectibleData.ID, *asset.CollectibleCommunityInfo)
|
|
if err != nil {
|
|
log.Error("collectiblesDataDB SetCommunityInfo failed", "communityID", communityID, "err", err)
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (o *Manager) fetchCommunityAssetsAsync(ctx context.Context, communityID string, communityAssets []*thirdparty.FullCollectibleData) {
|
|
if len(communityAssets) == 0 {
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
err := o.fetchCommunityAssets(communityID, communityAssets)
|
|
if err != nil {
|
|
log.Error("fetchCommunityAssets failed", "communityID", communityID, "err", err)
|
|
return
|
|
}
|
|
|
|
// Metadata is up to date in db at this point, fetch and send Event.
|
|
ids := make([]thirdparty.CollectibleUniqueID, 0, len(communityAssets))
|
|
for _, asset := range communityAssets {
|
|
ids = append(ids, asset.CollectibleData.ID)
|
|
}
|
|
o.signalUpdatedCollectiblesData(ids)
|
|
}()
|
|
}
|
|
|
|
func (o *Manager) fillAnimationMediatype(ctx context.Context, asset *thirdparty.FullCollectibleData) error {
|
|
if len(asset.CollectibleData.AnimationURL) > 0 {
|
|
contentType, err := o.doContentTypeRequest(ctx, asset.CollectibleData.AnimationURL)
|
|
if err != nil {
|
|
asset.CollectibleData.AnimationURL = ""
|
|
}
|
|
asset.CollectibleData.AnimationMediaType = contentType
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (o *Manager) processCollectionData(ctx context.Context, collections []thirdparty.CollectionData) error {
|
|
return o.collectionsDataDB.SetData(collections, true)
|
|
}
|
|
|
|
func (o *Manager) getCacheFullCollectibleData(uniqueIDs []thirdparty.CollectibleUniqueID) ([]thirdparty.FullCollectibleData, error) {
|
|
ret := make([]thirdparty.FullCollectibleData, 0, len(uniqueIDs))
|
|
|
|
collectiblesData, err := o.collectiblesDataDB.GetData(uniqueIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
contractIDs := make([]thirdparty.ContractID, 0, len(uniqueIDs))
|
|
for _, id := range uniqueIDs {
|
|
contractIDs = append(contractIDs, id.ContractID)
|
|
}
|
|
|
|
collectionsData, err := o.collectionsDataDB.GetData(contractIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, id := range uniqueIDs {
|
|
collectibleData, ok := collectiblesData[id.HashKey()]
|
|
if !ok {
|
|
// Use empty data, set only ID
|
|
collectibleData = thirdparty.CollectibleData{
|
|
ID: id,
|
|
}
|
|
}
|
|
if o.mediaServer != nil && len(collectibleData.ImagePayload) > 0 {
|
|
collectibleData.ImageURL = o.mediaServer.MakeWalletCollectibleImagesURL(collectibleData.ID)
|
|
}
|
|
|
|
collectionData, ok := collectionsData[id.ContractID.HashKey()]
|
|
if !ok {
|
|
// Use empty data, set only ID
|
|
collectionData = thirdparty.CollectionData{
|
|
ID: id.ContractID,
|
|
}
|
|
}
|
|
if o.mediaServer != nil && len(collectionData.ImagePayload) > 0 {
|
|
collectionData.ImageURL = o.mediaServer.MakeWalletCollectionImagesURL(collectionData.ID)
|
|
}
|
|
|
|
communityInfo, _, err := o.communityManager.GetCommunityInfo(collectibleData.CommunityID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
collectibleCommunityInfo, err := o.collectiblesDataDB.GetCommunityInfo(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ownership, err := o.ownershipDB.GetOwnership(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fullData := thirdparty.FullCollectibleData{
|
|
CollectibleData: collectibleData,
|
|
CollectionData: &collectionData,
|
|
CommunityInfo: communityInfo,
|
|
CollectibleCommunityInfo: collectibleCommunityInfo,
|
|
Ownership: ownership,
|
|
}
|
|
ret = append(ret, fullData)
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
// Reset connection status to trigger notifications
|
|
// on the next status update
|
|
func (o *Manager) ResetConnectionStatus() {
|
|
for _, status := range o.statuses {
|
|
status.ResetStateValue()
|
|
}
|
|
}
|
|
|
|
func (o *Manager) checkConnectionStatus(chainID walletCommon.ChainID) {
|
|
for _, provider := range o.collectibleProviders {
|
|
if provider.IsChainSupported(chainID) && provider.IsConnected() {
|
|
o.statuses[chainID.String()].SetIsConnected(true)
|
|
return
|
|
}
|
|
}
|
|
o.statuses[chainID.String()].SetIsConnected(false)
|
|
}
|
|
|
|
func (o *Manager) signalUpdatedCollectiblesData(ids []thirdparty.CollectibleUniqueID) {
|
|
// We limit how much collectibles data we send in each event to avoid problems on the client side
|
|
for startIdx := 0; startIdx < len(ids); startIdx += signalUpdatedCollectiblesDataPageSize {
|
|
endIdx := startIdx + signalUpdatedCollectiblesDataPageSize
|
|
if endIdx > len(ids) {
|
|
endIdx = len(ids)
|
|
}
|
|
pageIDs := ids[startIdx:endIdx]
|
|
|
|
collectibles, err := o.getCacheFullCollectibleData(pageIDs)
|
|
if err != nil {
|
|
log.Error("Error getting FullCollectibleData from cache: %v", err)
|
|
return
|
|
}
|
|
|
|
// Send update event with most complete data type available
|
|
details := fullCollectiblesDataToDetails(collectibles)
|
|
|
|
payload, err := json.Marshal(details)
|
|
if err != nil {
|
|
log.Error("Error marshaling response: %v", err)
|
|
return
|
|
}
|
|
|
|
event := walletevent.Event{
|
|
Type: EventCollectiblesDataUpdated,
|
|
Message: string(payload),
|
|
}
|
|
|
|
o.feed.Send(event)
|
|
}
|
|
}
|