88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package types
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
)
|
|
|
|
// NewCapability returns a reference to a new Capability to be used as an
|
|
// actual capability.
|
|
func NewCapability(index uint64) *Capability {
|
|
return &Capability{Index: index}
|
|
}
|
|
|
|
// String returns the string representation of a Capability. The string contains
|
|
// the Capability's memory reference as the string is to be used in a composite
|
|
// key and to authenticate capabilities.
|
|
func (ck *Capability) String() string {
|
|
return fmt.Sprintf("Capability{%p, %d}", ck, ck.Index)
|
|
}
|
|
|
|
func NewOwner(module, name string) Owner {
|
|
return Owner{Module: module, Name: name}
|
|
}
|
|
|
|
// Key returns a composite key for an Owner.
|
|
func (o Owner) Key() string {
|
|
return fmt.Sprintf("%s/%s", o.Module, o.Name)
|
|
}
|
|
|
|
func (o Owner) String() string {
|
|
bz, _ := yaml.Marshal(o)
|
|
return string(bz)
|
|
}
|
|
|
|
func NewCapabilityOwners() *CapabilityOwners {
|
|
return &CapabilityOwners{Owners: make([]Owner, 0)}
|
|
}
|
|
|
|
// Set attempts to add a given owner to the CapabilityOwners. If the owner
|
|
// already exists, an error will be returned. Set runs in O(log n) average time
|
|
// and O(n) in the worst case.
|
|
func (co *CapabilityOwners) Set(owner Owner) error {
|
|
i, ok := co.Get(owner)
|
|
if ok {
|
|
// owner already exists at co.Owners[i]
|
|
return sdkerrors.Wrapf(ErrOwnerClaimed, owner.String())
|
|
}
|
|
|
|
// owner does not exist in the set of owners, so we insert at position i
|
|
co.Owners = append(co.Owners, Owner{}) // expand by 1 in amortized O(1) / O(n) worst case
|
|
copy(co.Owners[i+1:], co.Owners[i:])
|
|
co.Owners[i] = owner
|
|
|
|
return nil
|
|
}
|
|
|
|
// Remove removes a provided owner from the CapabilityOwners if it exists. If the
|
|
// owner does not exist, Remove is considered a no-op.
|
|
func (co *CapabilityOwners) Remove(owner Owner) {
|
|
if len(co.Owners) == 0 {
|
|
return
|
|
}
|
|
|
|
i, ok := co.Get(owner)
|
|
if ok {
|
|
// owner exists at co.Owners[i]
|
|
co.Owners = append(co.Owners[:i], co.Owners[i+1:]...)
|
|
}
|
|
}
|
|
|
|
// Get returns (i, true) of the provided owner in the CapabilityOwners if the
|
|
// owner exists, where i indicates the owner's index in the set. Otherwise
|
|
// (i, false) where i indicates where in the set the owner should be added.
|
|
func (co *CapabilityOwners) Get(owner Owner) (int, bool) {
|
|
// find smallest index s.t. co.Owners[i] >= owner in O(log n) time
|
|
i := sort.Search(len(co.Owners), func(i int) bool { return co.Owners[i].Key() >= owner.Key() })
|
|
if i < len(co.Owners) && co.Owners[i].Key() == owner.Key() {
|
|
// owner exists at co.Owners[i]
|
|
return i, true
|
|
}
|
|
|
|
return i, false
|
|
}
|