52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
package watched_contracts
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/8thlight/vulcanizedb/pkg/core"
|
|
"github.com/8thlight/vulcanizedb/pkg/repositories"
|
|
)
|
|
|
|
type ContractSummary struct {
|
|
ContractHash string
|
|
NumberOfTransactions int
|
|
LastTransaction *core.Transaction
|
|
blockChain core.Blockchain
|
|
}
|
|
|
|
var NewContractNotWatchedErr = func(contractHash string) error {
|
|
return errors.New(fmt.Sprintf("Contract %v not being watched", contractHash))
|
|
}
|
|
|
|
func NewSummary(blockchain core.Blockchain, repository repositories.Repository, contractHash string) (*ContractSummary, error) {
|
|
watchedContract := repository.FindWatchedContract(contractHash)
|
|
if watchedContract != nil {
|
|
return newContractSummary(blockchain, *watchedContract), nil
|
|
} else {
|
|
return nil, NewContractNotWatchedErr(contractHash)
|
|
}
|
|
}
|
|
|
|
func (contractSummary ContractSummary) GetStateAttribute(attributeName string) string {
|
|
result, _ := contractSummary.blockChain.GetContractStateAttribute(contractSummary.ContractHash, attributeName)
|
|
return *result
|
|
}
|
|
|
|
func newContractSummary(blockchain core.Blockchain, watchedContract core.WatchedContract) *ContractSummary {
|
|
return &ContractSummary{
|
|
blockChain: blockchain,
|
|
ContractHash: watchedContract.Hash,
|
|
NumberOfTransactions: len(watchedContract.Transactions),
|
|
LastTransaction: lastTransaction(watchedContract),
|
|
}
|
|
}
|
|
|
|
func lastTransaction(contract core.WatchedContract) *core.Transaction {
|
|
if len(contract.Transactions) > 0 {
|
|
return &contract.Transactions[0]
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|