user.go

package goroomlib

import "encoding/json"

type User struct {
	id             int
	userId         int
	name           string
	joinedRooms    map[string](*Room)
	userProperties UserProperties
	isConnected    bool
}

type UserProperties struct {
	IsAdmin bool `json:"isAdmin"`
}

func (u *User) Init(userProperties map[string]interface{}) {
	u.joinedRooms = make(map[string]*Room)
	if userProperties == nil || len(userProperties) == 0 {
		userProperties = make(map[string]interface{})
		userProperties["IsAdmin"] = false
	}
	jsonbody, err := json.Marshal(userProperties)
	if err != nil {
		return
	}

	u.userProperties = UserProperties{}
	json.Unmarshal(jsonbody, &(u.userProperties))
}

func (u *User) GetId() int {
	return u.id
}

func (u *User) GetUserId() int {
	return u.id
}

func (u *User) GetName() string {
	return u.name
}

func (u *User) SetIsConnected(isConnected bool) {
	u.isConnected = isConnected
}

func (u *User) GetIsConnected() bool {
	return u.isConnected
}

func (u *User) GetJoinedRooms() map[string]*Room {
	return u.joinedRooms
}

func (u *User) AddRoom(r Room) map[string]*Room {
	u.joinedRooms[r.GetRoomName()] = &r
	return u.joinedRooms
}

func (u *User) RemoveRoom(r Room) map[string]*Room {
	delete(u.GetJoinedRooms(), r.GetRoomName())
	return u.joinedRooms
}

func (u *User) GetJoinedRoomByName(roomName string) (Room, bool) {
	room, ok := u.GetJoinedRooms()[roomName]
	return *room, ok
}

func (u *User) Remove() {
	u.DisconnectUser()
}

func (u *User) DisconnectUser() {
	for _, room := range u.joinedRooms {
		(*room).RemoveUserFromRoom(*u)
	}
}