86 lines
1.4 KiB
Go
86 lines
1.4 KiB
Go
package claims
|
|
|
|
import (
|
|
"encoding/json"
|
|
"gopkg.in/yaml.v3"
|
|
"sort"
|
|
)
|
|
|
|
type PermStorage struct {
|
|
values map[string]struct{}
|
|
}
|
|
|
|
func NewPermStorage() *PermStorage {
|
|
return new(PermStorage).setup()
|
|
}
|
|
|
|
func (p *PermStorage) setup() *PermStorage {
|
|
if p.values == nil {
|
|
p.values = make(map[string]struct{})
|
|
}
|
|
return p
|
|
}
|
|
|
|
func (p *PermStorage) Set(perm string) {
|
|
p.values[perm] = struct{}{}
|
|
}
|
|
|
|
func (p *PermStorage) Clear(perm string) {
|
|
delete(p.values, perm)
|
|
}
|
|
|
|
func (p *PermStorage) Has(perm string) bool {
|
|
_, ok := p.values[perm]
|
|
return ok
|
|
}
|
|
|
|
func (p *PermStorage) OneOf(o *PermStorage) bool {
|
|
for i := range o.values {
|
|
if o.Has(i) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (p *PermStorage) dump() []string {
|
|
var a []string
|
|
for i := range p.values {
|
|
a = append(a, i)
|
|
}
|
|
sort.Strings(a)
|
|
return a
|
|
}
|
|
|
|
func (p *PermStorage) prepare(a []string) {
|
|
for _, i := range a {
|
|
p.Set(i)
|
|
}
|
|
}
|
|
|
|
func (p *PermStorage) MarshalJSON() ([]byte, error) { return json.Marshal(p.dump()) }
|
|
|
|
func (p *PermStorage) UnmarshalJSON(bytes []byte) error {
|
|
p.setup()
|
|
var a []string
|
|
err := json.Unmarshal(bytes, &a)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
p.prepare(a)
|
|
return nil
|
|
}
|
|
|
|
func (p *PermStorage) MarshalYAML() (interface{}, error) { return yaml.Marshal(p.dump()) }
|
|
|
|
func (p *PermStorage) UnmarshalYAML(value *yaml.Node) error {
|
|
p.setup()
|
|
var a []string
|
|
err := value.Decode(&a)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
p.prepare(a)
|
|
return nil
|
|
}
|