133 lines
4.3 KiB
C++
133 lines
4.3 KiB
C++
#include "user.h"
|
|
|
|
#include <tools.h>
|
|
#include <json/json.h>
|
|
#include <unistd.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include "room.h"
|
|
#include <iostream>
|
|
#include <netinet/in.h>
|
|
#include <netinet/tcp.h>
|
|
#include <sstream>
|
|
|
|
namespace Yc {
|
|
namespace Lib {
|
|
|
|
User::User(Room *parent, std::string name, std::string color, int socket) :
|
|
_parent(parent),
|
|
_name(name),
|
|
_color(color),
|
|
_socket(socket),
|
|
_stop(false) {
|
|
_token = Yc::Lib::Tools::generateRandomString(32);
|
|
sendMsg(token, _token, "", "");
|
|
thread = new std::thread(&User::checkerTask, this);
|
|
}
|
|
|
|
User::~User() {
|
|
delete thread;
|
|
_parent->addMessage(User::system, std::string("leaved_chat"), _name, _color);
|
|
}
|
|
|
|
std::string User::name() const {
|
|
return _name;
|
|
}
|
|
|
|
bool User::validateToken(std::string token) {
|
|
return (token == _token);
|
|
}
|
|
|
|
bool User::isUser(User *toValidate) {
|
|
return (toValidate == this);
|
|
}
|
|
|
|
void User::sendMsg(MsgType type, const char *message, std::string userName, std::string color) {
|
|
sendMsg(type, std::string(message), userName, color);
|
|
}
|
|
|
|
void User::sendMsg(MsgType type, std::string message , std::string userName, std::string color) {
|
|
Json::Value sendMessage;
|
|
sendMessage["type"] = type;
|
|
sendMessage["message"] = message;
|
|
sendMessage["userName"] = userName;
|
|
sendMessage["color"] = color;
|
|
send(sendMessage);
|
|
}
|
|
|
|
void User::sendMsg(User::MsgType type, Json::Value message, std::string userName, std::string color) {
|
|
Json::Value sendMessage;
|
|
sendMessage["type"] = type;
|
|
sendMessage["message"] = message;
|
|
sendMessage["userName"] = userName;
|
|
sendMessage["color"] = color;
|
|
send(sendMessage);
|
|
}
|
|
|
|
void User::checkerTask() {
|
|
while (!_stop) {
|
|
fd_set readSd;
|
|
FD_ZERO(&readSd);
|
|
FD_SET(_socket, &readSd);
|
|
timeval tv;
|
|
tv.tv_sec = 0;
|
|
tv.tv_usec = 500;
|
|
int selectResult = select(_socket + 1, &readSd, NULL, NULL, &tv);
|
|
std::cout << selectResult << "/" << FD_ISSET(_socket, &readSd) << std::endl;
|
|
if (selectResult == 1 && FD_ISSET(_socket, &readSd) == 1) {
|
|
_parent->removeUser(_token);
|
|
_stop = true;
|
|
}
|
|
std::string msg = readSocket(_socket);
|
|
if (msg == "") {
|
|
continue;
|
|
}
|
|
std::cout << msg << std::endl;
|
|
handleMessage(msg);
|
|
}
|
|
}
|
|
|
|
void User::stop() {
|
|
_stop = true;
|
|
}
|
|
|
|
std::string User::color() const {
|
|
return _color;
|
|
}
|
|
|
|
void User::send(std::string out) {
|
|
Base::send(_socket, out);
|
|
}
|
|
|
|
void User::send(Json::Value out) {
|
|
Base::send(_socket, out);
|
|
}
|
|
|
|
void User::handleMessage(std::string message) {
|
|
Json::Value jsonTree = getJsonTree(message);
|
|
if (jsonTree["type"].asString() == "message") {
|
|
_parent->addMessage(User::message, jsonTree["message"].asString(), _name, _color);
|
|
} else if (jsonTree["type"].asString() == "dice") {
|
|
doDice();
|
|
} else if (jsonTree["type"].asString() == "scream") {
|
|
_parent->addMessage(User::scream, jsonTree["message"].asString(), _name, _color);
|
|
} else if (jsonTree["type"].asString() == "do") {
|
|
_parent->addMessage(User::dosomething, jsonTree["message"].asString(), _name, _color);
|
|
}
|
|
}
|
|
|
|
void User::doDice() {
|
|
switch (_parent->addDice(this, (rand() % 6) + 1)) {
|
|
case 1:
|
|
sendMsg(system, "dice_not_possible", "", "");
|
|
break;
|
|
case 2:
|
|
sendMsg(system, "dice_allready_done", "", "");
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
} // namespace Lib
|
|
} // namespace Yc
|