ogb912/micromdm
1
0
Fork 0
mirror of https://github.com/micromdm/micromdm/ synced 2026-08-08 02:25:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle unique constraint for user registration.

Browse Source 
Returns an error if the same username or email already exists.
This commit is contained in:
Victor Vrantchan
2021-02-20 18:24:39 -05:00
parent 6caa73b637
commit 695856b205
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/data/user/user.go
View File

@@ -183,8 +183,15 @@ func (err Error) Error() string {
}
}
var constraints = map[string]map[string]string{
"chk_email_not_empty": map[string]string{"email": "You must provide an email address."},
"chk_username_not_empty": map[string]string{"username": "You must provide a username."},
"chk_password_not_empty": map[string]string{"password": "You must provide a password."},
}
var (
constraints = map[string]map[string]string{
"chk_email_not_empty": map[string]string{"email": "You must provide an email address."},
"chk_username_not_empty": map[string]string{"username": "You must provide a username."},
"chk_password_not_empty": map[string]string{"password": "You must provide a password."},
}
constraintsUnique = map[string]map[string]string{
"users.username": map[string]string{"username": "This username already exists."},
"users.email": map[string]string{"email": "This email already exists."},
}
)

5
internal/data/user/user_sqlite.go
View File

@@ -151,6 +151,11 @@ func checkSqlite(err error) error {
}
switch sErr.Code {
case sqlite.SQLITE_CONSTRAINT_UNIQUE:
c := strings.Split(sErr.Msg, ": ")
if kv, ok := constraintsUnique[c[len(c)-1]]; ok {
return Error{invalid: kv}
}
case sqlite.SQLITE_CONSTRAINT_CHECK:
c := strings.Split(sErr.Msg, ": ")
if kv, ok := constraints[c[len(c)-1]]; ok {