rosetta-lbry/services/account_service.go

// Copyright 2020 Coinbase, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package services

import (
	"context"

	"github.com/coinbase/rosetta-bitcoin/configuration"

	"github.com/coinbase/rosetta-sdk-go/server"
	"github.com/coinbase/rosetta-sdk-go/types"
)

// AccountAPIService implements the server.AccountAPIServicer interface.
type AccountAPIService struct {
	config *configuration.Configuration
	i      Indexer
}

// NewAccountAPIService returns a new *AccountAPIService.
func NewAccountAPIService(
	config *configuration.Configuration,
	i Indexer,
) server.AccountAPIServicer {
	return &AccountAPIService{
		config: config,
		i:      i,
	}
}

// AccountBalance implements /account/balance.
func (s *AccountAPIService) AccountBalance(
	ctx context.Context,
	request *types.AccountBalanceRequest,
) (*types.AccountBalanceResponse, *types.Error) {
	if s.config.Mode != configuration.Online {
		return nil, wrapErr(ErrUnavailableOffline, nil)
	}

	// If we are fetching the current balance,
	// return all coins for an address and calculate
	// the balance from those coins.
	if request.BlockIdentifier == nil {
		coins, block, err := s.i.GetCoins(ctx, request.AccountIdentifier)
		if err != nil {
			return nil, wrapErr(ErrUnableToGetCoins, err)
		}

		balance := "0"
		for _, coin := range coins {
			balance, err = types.AddValues(balance, coin.Amount.Value)
			if err != nil {
				return nil, wrapErr(ErrUnableToParseIntermediateResult, err)
			}
		}

		return &types.AccountBalanceResponse{
			BlockIdentifier: block,
			Balances: []*types.Amount{
				{
					Value:    balance,
					Currency: s.config.Currency,
				},
			},
		}, nil
	}

	// If we are fetching a historical balance,
	// use balance storage and don't return coins.
	amount, block, err := s.i.GetBalance(
		ctx,
		request.AccountIdentifier,
		s.config.Currency,
		request.BlockIdentifier,
	)
	if err != nil {
		return nil, wrapErr(ErrUnableToGetBalance, err)
	}

	return &types.AccountBalanceResponse{
		BlockIdentifier: block,
		Balances: []*types.Amount{
			amount,
		},
	}, nil
}

// AccountCoins implements /account/coins.
func (s *AccountAPIService) AccountCoins(
	ctx context.Context,
	request *types.AccountCoinsRequest,
) (*types.AccountCoinsResponse, *types.Error) {
	if s.config.Mode != configuration.Online {
		return nil, wrapErr(ErrUnavailableOffline, nil)
	}

	coins, block, err := s.i.GetCoins(ctx, request.AccountIdentifier)
	if err != nil {
		return nil, wrapErr(ErrUnableToGetCoins, err)
	}

	result := &types.AccountCoinsResponse{
		BlockIdentifier: block,
		Coins:           coins,
	}
	//@Todo include_mempool query

	if len(request.Currencies) > 0 {
		filtered := []*types.Coin{}
		for _, curr := range request.Currencies {
			if curr == nil {
				continue
			}
			for _, coin := range coins {
				if coin.Amount.Currency.Symbol == curr.Symbol {
					filtered = append(filtered, coin)
				}
			}
		}
		result.Coins = filtered
	}

	return result, nil
}