-
Notifications
You must be signed in to change notification settings - Fork 12
/
owners.go
122 lines (108 loc) · 2.84 KB
/
owners.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package types
import (
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
)
// type SortedStringArray []string
// Owner defines key(address) -> value(items) for query
type Owner struct {
Address sdk.AccAddress `json:"address" yaml:"address"`
Items Items `json:"items" yaml:"items"`
}
// NewOwner is a constructor function for Owner
func NewOwner(owner sdk.AccAddress, items Items) Owner {
return Owner{
Address: owner,
Items: items,
}
}
// GetItem gets a Item from the owner
func (owner Owner) GetItem(id string) (item Item, err error) {
item, found := owner.Items.Find(id)
if found {
return item, nil
}
return Item{}, sdkerrors.Wrap(ErrUnknownItem,
fmt.Sprintf("Item #%s doesn't exist in owner %s", id, owner.Address),
)
}
// ContainsItem returns whether or not a owner contains an Item
func (owner Owner) ContainsItem(id string) bool {
_, err := owner.GetItem(id)
return err == nil
}
// AddItem adds an Item to the owner
func (owner Owner) AddItem(item Item) (Owner, error) {
id := item.GetID()
exists := owner.ContainsItem(id)
if exists {
return owner, sdkerrors.Wrap(ErrItemAlreadyExists,
fmt.Sprintf("Item #%s already exists in owner %s", id, owner.Address),
)
}
owner.Items = owner.Items.Append(item)
return owner, nil
}
// UpdateItem updates an Item from a owner
func (owner Owner) UpdateItem(item Item) (Owner, error) {
items, ok := owner.Items.Update(item.GetID(), item)
if !ok {
return owner, sdkerrors.Wrap(ErrUnknownItem,
fmt.Sprintf("Item #%s doesn't exist on owner %s", item.GetID(), owner.Address),
)
}
owner.Items = items
return owner, nil
}
// DeleteItem deletes an Item from a owner
func (owner Owner) DeleteItem(item Item) (Owner, error) {
items, ok := owner.Items.Remove(item.GetID())
if !ok {
return owner, sdkerrors.Wrap(ErrUnknownItem,
fmt.Sprintf("Item #%s doesn't exist on owner %s", item.GetID(), owner.Address),
)
}
owner.Items = items
return owner, nil
}
// Supply return total item
func (owner Owner) Supply() int {
return len(owner.Items)
}
func (owner Owner) String() string {
return fmt.Sprintf(`Address:%s
Items:%s`,
owner.Address,
owner.Items,
)
}
//// ========== SortedStringArray implementation ==========
//func (sa SortedStringArray) String() string {
// return strings.Join(sa[:], ",")
//}
//
//func (sa SortedStringArray) ElAtIndex(index int) string {
// return sa[index]
//}
//
//func (sa SortedStringArray) find(el string) (idx int) {
// return FindUtil(sa, el)
//}
//
//func (sa SortedStringArray) Len() int {
// return len(sa)
//}
//
//func (sa SortedStringArray) Less(i, j int) bool {
// return strings.Compare(sa[i], sa[j]) == -1
//}
//
//func (sa SortedStringArray) Swap(i, j int) {
// sa[i], sa[j] = sa[j], sa[i]
//}
//
//func (sa SortedStringArray) Sort() SortedStringArray {
// sort.Sort(sa)
// return sa
//}