From 58d4c92bd42e7fff78b32dd23659be6e4fc0d843 Mon Sep 17 00:00:00 2001
From: anarcher <anarcher@gmail.com>
Date: Sat, 5 Nov 2022 01:11:47 +0900
Subject: [PATCH] feat: reName and HasPerm

---
 examples/gno.land/r/system/names/names.gno | 44 +++++++++++++++++++++-
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/examples/gno.land/r/system/names/names.gno b/examples/gno.land/r/system/names/names.gno
index de0a6f5aa4a..e6e3703bfb4 100644
--- a/examples/gno.land/r/system/names/names.gno
+++ b/examples/gno.land/r/system/names/names.gno
@@ -2,6 +2,8 @@
 package names
 
 import (
+	"errors"
+	"regexp"
 	"std"
 
 	"gno.land/p/demo/avl"
@@ -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
@@ -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},
@@ -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) {
@@ -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) {
@@ -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.
 
@@ -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
+}