finish regex

backend/global/regex.go

@@ -34,36 +34,32 @@ func GetDefaultRegex() (reg Regex) {
 }
 
 // SetUserRegex set Regex.name = re
-func (u *UserDatabase) SetUserRegex(id int, name, re string) error {
-	if name == "" || name == "ID" {
-		return ErrInvalidFieldName
-	}
-	if re == "" {
+func (u *UserDatabase) SetUserRegex(id int, reg *Regex) error {
+	if reg == nil {
 		return ErrEmptyRegex
 	}
 	user, err := UserDB.GetUserByID(id)
 	if err != nil {
 		return err
 	}
-	if !user.IsFileManager() {
+	if !user.IsSuper() && id != *user.ID {
 		return ErrInvalidRole
 	}
-	_, err = regexp.Compile(re)
-	if err != nil {
-		return err
-	}
-	reg := GetDefaultRegex()
-	rreg := reflect.ValueOf(&reg).Elem()
-	f := rreg.FieldByName(name)
-	if !f.IsValid() {
-		return ErrNoSuchFieldName
+	defaultrf := reflect.ValueOf(GetDefaultRegex())
+	rreg := reflect.ValueOf(reg).Elem()
+	for i := 1; i < rreg.NumField(); i++ {
+		if rreg.Field(i).Equal(defaultrf.Field(i)) {
+			rreg.Field(i).SetString("")
+		} else {
+			_, err = regexp.Compile(rreg.Field(i).String())
+			if err != nil {
+				return err
+			}
+		}
 	}
 	u.mu.Lock()
 	defer u.mu.Unlock()
-	_ = u.db.Find(UserTableRegex, &reg, "WHERE ID="+strconv.Itoa(id))
-	reg.ID = id
-	f.SetString(re)
-	return u.db.Insert(UserTableRegex, &reg)
+	return u.db.Insert(UserTableRegex, reg)
 }
 
 // GetUserRegex default newRegex()
@@ -72,7 +68,7 @@ func (u *UserDatabase) GetUserRegex(id int) (*Regex, error) {
 	if err != nil {
 		return nil, err
 	}
-	if !user.IsFileManager() {
+	if !user.IsSuper() || id != *user.ID {
 		return nil, ErrInvalidRole
 	}
 	u.mu.RLock()

backend/global/user.go

@@ -30,7 +30,6 @@ var (
 	ErrUsernameExists    = errors.New("username exists")
 	ErrInvalidName       = errors.New("invalid name")
 	ErrInvalidContact    = errors.New("invalid contact")
-	ErrInvalidFieldName  = errors.New("invalid field name")
 	ErrNoSuchFieldName   = errors.New("no such field name")
 	ErrEmptyRegex        = errors.New("empty regex")
 )

backend/regex.go

@@ -1,6 +1,7 @@
 package backend
 
 import (
+	"encoding/json"
 	"net/http"
 
 	"github.com/fumiama/paper-manager/backend/global"
@@ -23,4 +24,26 @@ func init() {
 		}
 		writeresult(w, codeSuccess, reg, messageOk, typeSuccess)
 	}}
+
+	apimap["/api/setUserRegex"] = &apihandler{"POST", func(w http.ResponseWriter, r *http.Request) {
+		token := r.Header.Get("Authorization")
+		user := usertokens.Get(token)
+		if user == nil {
+			writeresult(w, codeError, nil, errInvalidToken.Error(), typeError)
+			return
+		}
+		defer r.Body.Close()
+		reg := &global.Regex{}
+		err := json.NewDecoder(r.Body).Decode(reg)
+		if err != nil {
+			writeresult(w, codeError, nil, err.Error(), typeError)
+			return
+		}
+		err = global.UserDB.SetUserRegex(*user.ID, reg)
+		if err != nil {
+			writeresult(w, codeError, nil, err.Error(), typeError)
+			return
+		}
+		writeresult(w, codeSuccess, "成功", messageOk, typeSuccess)
+	}}
 }