mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-13 05:45:02 +00:00
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
package stack
|
||
|
|
||
|
import "errors"
|
||
|
|
||
|
// Invocation embeds a Random Access stack
|
||
|
// Providing helper methods for the context object
|
||
|
type Invocation struct{ RandomAccess }
|
||
|
|
||
|
//NewInvocation will return a new
|
||
|
// Invocation stack
|
||
|
func NewInvocation() *Invocation {
|
||
|
return &Invocation{
|
||
|
RandomAccess{
|
||
|
vals: make([]Item, 0, StackAverageSize),
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (i *Invocation) peekContext(n uint16) (*Context, error) {
|
||
|
item, err := i.Peek(n)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
ctx, err := item.Context()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return ctx, nil
|
||
|
}
|
||
|
|
||
|
// CurrentContext returns the current context on the invocation stack
|
||
|
func (i *Invocation) CurrentContext() (*Context, error) {
|
||
|
return i.peekContext(0)
|
||
|
}
|
||
|
|
||
|
// CallingContext will return the cntext item
|
||
|
// that will be called next.
|
||
|
func (i *Invocation) CallingContext() (*Context, error) {
|
||
|
if i.Len() < 1 {
|
||
|
return nil, errors.New("Length of invocation stack is < 1, no calling context")
|
||
|
}
|
||
|
return i.peekContext(1)
|
||
|
}
|
||
|
|
||
|
// EntryContext will return the context item that
|
||
|
// started the program
|
||
|
func (i *Invocation) EntryContext() (*Context, error) {
|
||
|
|
||
|
// firstItemIndex refers to the first item
|
||
|
// that was popped on the stack
|
||
|
firstItemIndex := uint16(i.Len() - 1) // N.B. if this overflows because len is zero, then an error will be returned
|
||
|
return i.peekContext(firstItemIndex)
|
||
|
}
|