yourchat/user.cpp

#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() {
            _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);
                if (selectResult == 1 && FD_ISSET(_socket, &readSd) == 1) {
                    _parent->removeUser(_token);
                    _stop = true;
                    delete this;
                    return;
                }
                std::string msg = readSocket(_socket);
                if (msg == "") {
                    continue;
                }
                handleMessage(msg);
            }
        }

        void User::stop() {
            _stop = true;
        }

        std::string User::color() const {
            return _color;
        }

        void User::setParent(Room *parent) {
            _parent = parent;
        }

        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["token"].asString() != _token) {
                std::cout << "token wrong:" << jsonTree["token"].asString() << "/" << _token << std::endl;
                return;
            }
            if (jsonTree["type"].asString() == "message") {
                checkString(jsonTree["message"].asString());
            } 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);
            } else if (jsonTree["type"].asString() == "join") {
                changeRoom(jsonTree["newroom"].asString(), jsonTree["password"].asString());
            }
        }

        void User::checkString(std::string message) {
            if (message.substr(0, 6) == "/join ") {
                message = message.substr(6);
                if (message.find(" ") == std::string::npos) {
                    changeRoom(message, "");
                } else {
                    std::string room = message.substr(0, message.find(" "));
                    std::string password = message.substr(message.find(" ") + 1);
                    changeRoom(room, password);
                }
            } else if (message == "/dice") {
                doDice();
            } else {
                _parent->addMessage(User::message, message, _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;
            }
        }

        void User::changeRoom(std::string newRoom, std::string password) {
            if (!_parent->userToNewRoom(this, newRoom, password)) {
                sendMsg(User::system, "room_not_possible", "", "");
            }
        }
    } // namespace Lib
} // namespace Yc