mjwt/claims/perms.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 p.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
}