forked from TrueCloudLab/neoneo-go
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
package rpcevent
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/state"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/neorpc"
|
|
"github.com/nspcc-dev/neo-go/pkg/neorpc/result"
|
|
)
|
|
|
|
type (
|
|
// Comparator is an interface required from notification event filter to be able to
|
|
// filter notifications.
|
|
Comparator interface {
|
|
EventID() neorpc.EventID
|
|
Filter() interface{}
|
|
}
|
|
// Container is an interface required from notification event to be able to
|
|
// pass filter.
|
|
Container interface {
|
|
EventID() neorpc.EventID
|
|
EventPayload() interface{}
|
|
}
|
|
)
|
|
|
|
// Matches filters our given Container against Comparator filter.
|
|
func Matches(f Comparator, r Container) bool {
|
|
expectedEvent := f.EventID()
|
|
filter := f.Filter()
|
|
if r.EventID() != expectedEvent {
|
|
return false
|
|
}
|
|
if filter == nil {
|
|
return true
|
|
}
|
|
switch f.EventID() {
|
|
case neorpc.BlockEventID:
|
|
filt := filter.(neorpc.BlockFilter)
|
|
b := r.EventPayload().(*block.Block)
|
|
return int(b.PrimaryIndex) == filt.Primary
|
|
case neorpc.TransactionEventID:
|
|
filt := filter.(neorpc.TxFilter)
|
|
tx := r.EventPayload().(*transaction.Transaction)
|
|
senderOK := filt.Sender == nil || tx.Sender().Equals(*filt.Sender)
|
|
signerOK := true
|
|
if filt.Signer != nil {
|
|
signerOK = false
|
|
for i := range tx.Signers {
|
|
if tx.Signers[i].Account.Equals(*filt.Signer) {
|
|
signerOK = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
return senderOK && signerOK
|
|
case neorpc.NotificationEventID:
|
|
filt := filter.(neorpc.NotificationFilter)
|
|
notification := r.EventPayload().(*state.ContainedNotificationEvent)
|
|
hashOk := filt.Contract == nil || notification.ScriptHash.Equals(*filt.Contract)
|
|
nameOk := filt.Name == nil || notification.Name == *filt.Name
|
|
return hashOk && nameOk
|
|
case neorpc.ExecutionEventID:
|
|
filt := filter.(neorpc.ExecutionFilter)
|
|
applog := r.EventPayload().(*state.AppExecResult)
|
|
return applog.VMState.String() == filt.State
|
|
case neorpc.NotaryRequestEventID:
|
|
filt := filter.(neorpc.TxFilter)
|
|
req := r.EventPayload().(*result.NotaryRequestEvent)
|
|
senderOk := filt.Sender == nil || req.NotaryRequest.FallbackTransaction.Signers[1].Account == *filt.Sender
|
|
signerOK := true
|
|
if filt.Signer != nil {
|
|
signerOK = false
|
|
for _, signer := range req.NotaryRequest.MainTransaction.Signers {
|
|
if signer.Account.Equals(*filt.Signer) {
|
|
signerOK = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
return senderOk && signerOK
|
|
}
|
|
return false
|
|
}
|