X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=db%2Fuser.go;h=24ce491e80ab9e4e3b7ee9135912aa4da1d9c853;hb=4059b21bd00261699ae6c7619adf053a607073e5;hp=7a0a32b37105a1fac3f83b421c986ab684bd80ab;hpb=85545aba62546f219a9c9730945511412a3174ef;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FFront.git

diff --git a/db/user.go b/db/user.go
index 7a0a32b..24ce491 100644
--- a/db/user.go
+++ b/db/user.go
@@ -11,10 +11,16 @@ const (
 	AwaitingConfirmation
 )
 
+type UserRole string
+
+const RoleUser UserRole = "user"
+const RoleAdmin UserRole = "admin"
+
 type User struct {
 	Id           int64
-	Email        string `sql:",unique,notnull"`
-	PasswordHash string `sql:",notnull"`
+	Role         UserRole
+	Email        string
+	PasswordHash string
 	OtpSecret    string
 	IsOtpSetup   bool
 	Status       UserStatus
@@ -80,3 +86,9 @@ func SetPassword(user *User, password string) error {
 
 	return DB.Update(user)
 }
+
+func SetUserStatus(user *User, status UserStatus) error {
+	user.Status = status
+
+	return DB.Update(user)
+}