Skip to content

Commit

Permalink
feat: reName and HasPerm
Browse files Browse the repository at this point in the history
  • Loading branch information
anarcher committed Nov 4, 2022
1 parent 238bb11 commit 703644d
Showing 1 changed file with 42 additions and 2 deletions.
44 changes: 42 additions & 2 deletions examples/gno.land/r/system/names/names.gno
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
package names

import (
"errors"
"regexp"
"std"

"gno.land/p/demo/avl"
Expand All @@ -13,6 +15,9 @@ import (
// determine if an address can publish a package or not.
var namespaces avl.MutTree // name(string) -> Space

// TODO: more accurate.
var reNamespace = regexp.MustCompile(`^[a-z][a-z0-9_]{2,30}$`)

type Space struct {
Admins []std.Address
Editors []std.Address
Expand All @@ -28,6 +33,10 @@ func Register(namespace string) {
panic("namespace already exists")
}

if err := validateNamespace(namespace); err != nil {
panic(err)
}

caller := std.GetOrigCaller()
namespaces.Set(namespace, &Space{
Admins: []std.Address{caller},
Expand All @@ -52,12 +61,14 @@ func RemoveAdmin(namespace string, newAdmin std.Address) {
if len(space.Admins) == 1 {
panic("namespace at least needs one admin")
}
var admins []std.Address
for i, admin := range space.Admins {
if admin == newAdmin {
space.Admins = append(space.Admins[:i], space.Admins[i+1:]...)
admins = append(space.Admins[:i], space.Admins[i+1:]...)
break
}
}
space.Admins = admins
}

func AddEditor(namespace string, newEditor std.Address) {
Expand All @@ -74,12 +85,14 @@ func RemoveEditor(namespace string, newEditor std.Address) {

// remove editor
space := getSpace(namespace)
var editors []std.Address
for i, editor := range space.Editors {
if editor == newEditor {
space.Editors = append(space.Editors[:i], space.Editors[i+1:]...)
editors = append(space.Editors[:i], space.Editors[i+1:]...)
break
}
}
space.Editors = editors
}

func SetInPause(namespace string, state bool) {
Expand All @@ -88,6 +101,26 @@ func SetInPause(namespace string, state bool) {
space.InPause = state
}

// HasPerm returns true if the caller has permission of the namespace.
func HasPerm(namespace string) bool {
caller := std.GetOrigCaller()
space := getSpace(namespace)

if space.InPause {
return false
}

if containsAddress(space.Admins, caller) {
return true
}

if containsAddress(space.Editors, caller) {
return true
}

return false
}

func Render(path string) string {
// TODO: by address.

Expand Down Expand Up @@ -141,3 +174,10 @@ func renderAddresses(addresses []std.Address) string {
}
return output
}

func validateNamespace(namespace string) error {
if !reNamespace.MatchString(namespace) {
return errors.New("invalid namespace")
}
return nil
}

0 comments on commit 703644d

Please sign in to comment.