134 lines
3.3 KiB
Go
134 lines
3.3 KiB
Go
package stickers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"math/big"
|
|
|
|
"github.com/status-im/status-go/multiaccounts/settings"
|
|
"github.com/status-im/status-go/services/wallet/bigint"
|
|
)
|
|
|
|
func (api *API) AddPending(chainID uint64, packID *bigint.BigInt) error {
|
|
pendingPacks, err := api.pendingStickerPacks()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, exists := pendingPacks[uint(packID.Uint64())]; exists {
|
|
return errors.New("sticker pack is already pending")
|
|
}
|
|
|
|
stickerType, err := api.contractMaker.NewStickerType(chainID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
stickerPack, err := api.fetchPackData(stickerType, packID.Int, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pendingPacks[uint(packID.Uint64())] = *stickerPack
|
|
|
|
return api.accountsDB.SaveSettingField(settings.StickersPacksPending, pendingPacks)
|
|
}
|
|
|
|
func (api *API) pendingStickerPacks() (StickerPackCollection, error) {
|
|
stickerPacks := make(StickerPackCollection)
|
|
|
|
pendingStickersJSON, err := api.accountsDB.GetPendingStickerPacks()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if pendingStickersJSON == nil {
|
|
return stickerPacks, nil
|
|
}
|
|
|
|
err = json.Unmarshal(*pendingStickersJSON, &stickerPacks)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return stickerPacks, nil
|
|
}
|
|
|
|
func (api *API) Pending() (StickerPackCollection, error) {
|
|
stickerPacks, err := api.pendingStickerPacks()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for packID, stickerPack := range stickerPacks {
|
|
stickerPack.Status = statusPending
|
|
stickerPack.Preview = api.hashToURL(stickerPack.Preview)
|
|
stickerPack.Thumbnail = api.hashToURL(stickerPack.Thumbnail)
|
|
for i, sticker := range stickerPack.Stickers {
|
|
sticker.URL = api.hashToURL(sticker.Hash)
|
|
stickerPack.Stickers[i] = sticker
|
|
}
|
|
stickerPacks[packID] = stickerPack
|
|
}
|
|
|
|
return stickerPacks, nil
|
|
}
|
|
|
|
func (api *API) ProcessPending(chainID uint64) (pendingChanged StickerPackCollection, err error) {
|
|
pendingStickerPacks, err := api.pendingStickerPacks()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
accs, err := api.accountsDB.GetAccounts()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
purchasedPacks := make(map[uint]struct{})
|
|
purchasedPackChan := make(chan *big.Int)
|
|
errChan := make(chan error)
|
|
doneChan := make(chan struct{}, 1)
|
|
go api.getAccountsPurchasedPack(chainID, accs, purchasedPackChan, errChan, doneChan)
|
|
for {
|
|
select {
|
|
case err := <-errChan:
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
case packID := <-purchasedPackChan:
|
|
if packID != nil {
|
|
purchasedPacks[uint(packID.Uint64())] = struct{}{}
|
|
}
|
|
case <-doneChan:
|
|
result := make(StickerPackCollection)
|
|
for _, stickerPack := range pendingStickerPacks {
|
|
packID := uint(stickerPack.ID.Uint64())
|
|
if _, exists := purchasedPacks[packID]; !exists {
|
|
continue
|
|
}
|
|
delete(pendingStickerPacks, packID)
|
|
stickerPack.Status = statusPurchased
|
|
result[packID] = stickerPack
|
|
}
|
|
err = api.accountsDB.SaveSettingField(settings.StickersPacksPending, pendingStickerPacks)
|
|
return result, err
|
|
}
|
|
}
|
|
}
|
|
|
|
func (api *API) RemovePending(packID *bigint.BigInt) error {
|
|
pendingPacks, err := api.pendingStickerPacks()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, exists := pendingPacks[uint(packID.Uint64())]; !exists {
|
|
return nil
|
|
}
|
|
|
|
delete(pendingPacks, uint(packID.Uint64()))
|
|
|
|
return api.accountsDB.SaveSettingField(settings.StickersPacksPending, pendingPacks)
|
|
}
|