From 8705f6a4259abb75c8612a350b16b5dd45ffc811 Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Sat, 3 May 2025 08:06:26 +0300
Subject: [PATCH] refactor: add UserTeam and UserRole types, update
 StrToUserTeam

---
 models/state.go | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/models/state.go b/models/state.go
index a7a8bb0..1dd2733 100644
--- a/models/state.go
+++ b/models/state.go
@@ -1,26 +1,36 @@
 package models
 
-type UserTeam string
+type (
+	UserTeam string
+	UserRole string
+)
 
 const (
+	// UserTeam
 	UserTeamBlue = "blue"
 	UserTeamRed  = "red"
 	UserTeamNone = "none"
+	//UserRole
+	UserRoleMime    = "mime"
+	UserRoleGuesser = "guesser"
+	UserRoleNone    = "none"
 )
 
-func StrToUserTeam(s string) WordColor {
+func StrToUserTeam(s string) UserTeam {
 	switch s {
 	case "blue":
 		return UserTeamBlue
 	case "red":
 		return UserTeamRed
-	case "none":
+	default:
 		return UserTeamNone
 	}
-	return WordColorUserTeamNone
 }
 
+// write StrToUserRole func; ai!
+
 type UserState struct {
 	Username string
 	Room     RoomPublic
+	Team     UserTeam
 }