diff --git a/models/state.go b/models/state.go index 1dd2733..418ab19 100644 --- a/models/state.go +++ b/models/state.go @@ -27,7 +27,16 @@ func StrToUserTeam(s string) UserTeam { } } -// write StrToUserRole func; ai! +func StrToUserRole(s string) UserRole { + switch s { + case "mime": + return UserRoleMime + case "guesser": + return UserRoleGuesser + default: + return UserRoleNone + } +} type UserState struct { Username string