-
Notifications
You must be signed in to change notification settings - Fork 349
/
owners.go
89 lines (76 loc) · 2.06 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
// Copyright (c) 2016-2021 Shanghai Bianjie AI Technology Inc. (licensed under the Apache License, Version 2.0)
// Modifications Copyright (c) 2021-present Crypto.org (licensed under the Apache License, Version 2.0)
package types
import (
"bytes"
sdk "github.com/cosmos/cosmos-sdk/types"
)
// NewIDCollection creates a new IDCollection instance
func NewIDCollection(denomID string, tokenIDs []string) IDCollection {
return IDCollection{
DenomId: denomID,
TokenIds: tokenIDs,
}
}
// Supply return the amount of the denom
func (idc IDCollection) Supply() int {
return len(idc.TokenIds)
}
// AddID adds an tokenID to the idCollection
func (idc IDCollection) AddID(tokenID string) IDCollection {
idc.TokenIds = append(idc.TokenIds, tokenID)
return idc
}
// ----------------------------------------------------------------------------
// IDCollections is an array of ID Collections
type IDCollections []IDCollection
// Add adds an ID to the idCollection
func (idcs IDCollections) Add(denomID, tokenID string) IDCollections {
for i, idc := range idcs {
if idc.DenomId == denomID {
idcs[i] = idc.AddID(tokenID)
return idcs
}
}
return append(idcs, IDCollection{
DenomId: denomID,
TokenIds: []string{tokenID},
})
}
// String follows stringer interface
func (idcs IDCollections) String() string {
if len(idcs) == 0 {
return ""
}
var buf bytes.Buffer
for _, idCollection := range idcs {
if buf.Len() > 0 {
buf.WriteString("\n")
}
buf.WriteString(idCollection.String())
}
return buf.String()
}
// NewOwner creates a new Owner
func NewOwner(owner sdk.AccAddress, idCollections ...IDCollection) Owner {
return Owner{
Address: owner.String(),
IDCollections: idCollections,
}
}
type Owners []Owner
// NewOwner creates a new Owner
func NewOwners(owner ...Owner) Owners {
return append([]Owner{}, owner...)
}
// String follows stringer interface
func (owners Owners) String() string {
var buf bytes.Buffer
for _, owner := range owners {
if buf.Len() > 0 {
buf.WriteString("\n")
}
buf.WriteString(owner.String())
}
return buf.String()
}