Refactor project structure: replace User class with ChatUser, integrate Database class, and update CMake configuration for new files

This commit is contained in:
Torsten Schulz (local)
2025-08-11 14:48:45 +02:00
parent 89956bd01a
commit b81f2de10f
14 changed files with 644 additions and 44 deletions

16
.vscode/c_cpp_properties.json vendored Normal file
View File

@@ -0,0 +1,16 @@
{
"configurations": [
{
"name": "Linux",
"includePath": [
"${workspaceFolder}/**"
],
"defines": [],
"compilerPath": "/usr/bin/g++",
"cStandard": "c11",
"cppStandard": "c++23",
"intelliSenseMode": "linux-gcc-x64"
}
],
"version": 4
}

78
.vscode/settings.json vendored Normal file
View File

@@ -0,0 +1,78 @@
{
"files.associations": {
"array": "cpp",
"atomic": "cpp",
"bit": "cpp",
"bitset": "cpp",
"cctype": "cpp",
"charconv": "cpp",
"chrono": "cpp",
"clocale": "cpp",
"cmath": "cpp",
"codecvt": "cpp",
"compare": "cpp",
"concepts": "cpp",
"condition_variable": "cpp",
"cstdarg": "cpp",
"cstddef": "cpp",
"cstdint": "cpp",
"cstdio": "cpp",
"cstdlib": "cpp",
"cstring": "cpp",
"ctime": "cpp",
"cwchar": "cpp",
"cwctype": "cpp",
"deque": "cpp",
"forward_list": "cpp",
"list": "cpp",
"map": "cpp",
"set": "cpp",
"string": "cpp",
"unordered_map": "cpp",
"unordered_set": "cpp",
"vector": "cpp",
"exception": "cpp",
"algorithm": "cpp",
"functional": "cpp",
"iterator": "cpp",
"memory": "cpp",
"memory_resource": "cpp",
"numeric": "cpp",
"optional": "cpp",
"random": "cpp",
"ratio": "cpp",
"string_view": "cpp",
"system_error": "cpp",
"tuple": "cpp",
"type_traits": "cpp",
"utility": "cpp",
"format": "cpp",
"fstream": "cpp",
"future": "cpp",
"initializer_list": "cpp",
"iomanip": "cpp",
"iosfwd": "cpp",
"iostream": "cpp",
"istream": "cpp",
"limits": "cpp",
"mutex": "cpp",
"new": "cpp",
"numbers": "cpp",
"ostream": "cpp",
"queue": "cpp",
"ranges": "cpp",
"semaphore": "cpp",
"span": "cpp",
"sstream": "cpp",
"stack": "cpp",
"stdexcept": "cpp",
"stdfloat": "cpp",
"stop_token": "cpp",
"streambuf": "cpp",
"text_encoding": "cpp",
"thread": "cpp",
"cinttypes": "cpp",
"typeinfo": "cpp",
"variant": "cpp"
}
}

View File

@@ -5,5 +5,6 @@ project(YourChat VERSION 0.1)
set(CMAKE_CXX_STANDARD 23) set(CMAKE_CXX_STANDARD 23)
set(CMAKE_CXX_STANDARD_REQUIRED True) set(CMAKE_CXX_STANDARD_REQUIRED True)
include_directories(${PROJECT_SOURCE_DIR}) include_directories(${PROJECT_SOURCE_DIR})
add_executable(yourchat main.cpp base.cpp config.cpp server.cpp room.cpp tools.cpp user.cpp) add_executable(yourchat main.cpp base.cpp config.cpp server.cpp room.cpp tools.cpp chat_user.cpp database.cpp)
target_link_libraries(yourchat jsoncpp pthread) # Removed chatroom.cpp, chatuser.cpp, chatroom.h, chatuser.h from the build
target_link_libraries(yourchat jsoncpp pthread pqxx)

175
chat_user.cpp Normal file
View File

@@ -0,0 +1,175 @@
// entfernt: #include "user.h"
#include "chat_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 {
ChatUser::ChatUser(std::shared_ptr<Room> parent, std::string name, std::string color, int socket) :
_parent(std::move(parent)),
_name(std::move(name)),
_color(std::move(color)),
_socket(socket),
_stop(false) {
_token = Yc::Lib::Tools::generateRandomString(32);
sendMsg(token, _token, "", "");
thread = std::make_unique<std::thread>(&ChatUser::checkerTask, this);
}
ChatUser::~ChatUser() {
_parent->addMessage(ChatUser::system, std::string("leaved_chat"), std::string(_name), std::string(_color));
}
std::string ChatUser::name() const {
return _name;
}
bool ChatUser::validateToken(std::string token) {
return (token == _token);
}
bool ChatUser::isUser(std::shared_ptr<ChatUser> toValidate) {
return (toValidate.get() == this);
}
void ChatUser::sendMsg(MsgType type, const char *message, std::string userName, std::string color) {
sendMsg(type, std::string(message), userName, color);
}
void ChatUser::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 ChatUser::sendMsg(ChatUser::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 ChatUser::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 ChatUser::stop() {
_stop = true;
}
std::string ChatUser::color() const {
return _color;
}
void ChatUser::setParent(std::shared_ptr<Room> parent) {
_parent = std::move(parent);
}
void ChatUser::send(std::string out) {
Base::send(_socket, out);
}
void ChatUser::send(Json::Value out) {
Base::send(_socket, out);
}
void ChatUser::handleMessage(std::string message) {
Json::Value jsonTree = getJsonTree(message);
if (jsonTree["token"].asString() != _token) {
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(ChatUser::scream, jsonTree["message"].asString(), std::string(_name), std::string(_color));
} else if (jsonTree["type"].asString() == "do") {
_parent->addMessage(ChatUser::dosomething, jsonTree["message"].asString(), std::string(_name), std::string(_color));
} else if (jsonTree["type"].asString() == "join") {
changeRoom(jsonTree["newroom"].asString(), jsonTree["password"].asString());
} else if (jsonTree["type"].asString() == "userlist") {
sendUserList();
}
}
void ChatUser::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(ChatUser::message, std::string(message), std::string(_name), std::string(_color));
}
}
void ChatUser::sendUserList() {
Json::Value userList = _parent->userList();
Json::Value msg = Json::objectValue;
msg["userlist"] = userList;
sendMsg(userListe, msg, "", "");
}
void ChatUser::doDice() {
switch (_parent->addDice(shared_from_this(), (rand() % 6) + 1)) {
case 1:
sendMsg(system, "dice_not_possible", "", "");
break;
case 2:
sendMsg(system, "dice_allready_done", "", "");
break;
default:
break;
}
}
void ChatUser::changeRoom(std::string newRoom, std::string password) {
if (!_parent->userToNewRoom(shared_from_this(), newRoom, password)) {
sendMsg(ChatUser::system, "room_not_possible", "", "");
}
}
} // namespace Lib
} // namespace Yc

68
chat_user.h Normal file
View File

@@ -0,0 +1,68 @@
#ifndef YC_LIB_CHAT_USER_H
#define YC_LIB_CHAT_USER_H
#include <memory>
#include <string>
#include <set>
#include <json/json.h>
#include <thread>
#include "base.h"
// Forward declaration to break cyclic dependency
namespace Yc { namespace Lib { class Room; } }
namespace Yc {
namespace Lib {
class ChatUser: public Base, public std::enable_shared_from_this<ChatUser> {
public:
enum MsgType {
error = -1,
token = 1,
userListe = 2,
roomList = 3,
message = 4,
system = 5,
scream = 6,
dosomething = 7,
dice = 8,
result = 9
};
ChatUser(std::shared_ptr<Room> parent, std::string name, std::string color, int socket);
~ChatUser();
std::string name() const;
bool validateToken(std::string token);
bool isUser(std::shared_ptr<ChatUser> toValidate);
void sendMsg(MsgType type, std::string message, std::string userName, std::string color);
void sendMsg(MsgType type, const char *message, std::string userName, std::string color);
void sendMsg(MsgType type, Json::Value message, std::string userName, std::string color);
void checkerTask();
void stop();
std::string color() const;
void setParent(std::shared_ptr<Room> parent);
private:
std::shared_ptr<Room> _parent;
std::string _name;
std::string _color;
int _socket;
std::string _token;
bool _stop;
std::unique_ptr<std::thread> thread;
void send(std::string out);
void send(Json::Value out);
void handleMessage(std::string message);
void doDice();
void changeRoom(std::string newRoom, std::string password);
void checkString(std::string message);
void sendUserList();
};
} // namespace Lib
} // namespace Yc
#endif // YC_LIB_CHAT_USER_H

42
database.cpp Normal file
View File

@@ -0,0 +1,42 @@
#include "database.h"
#include <pqxx/pqxx>
#include <stdexcept>
namespace Yc {
namespace Lib {
Database::Database(std::shared_ptr<Config> config)
{
// Hole Verbindungsdaten aus der Config
std::string dbname = config->value("database", "database").asString();
std::string user = config->value("database", "user").asString();
std::string password = config->value("database", "password").asString();
std::string host = config->value("database", "host").asString();
std::string conninfo =
"dbname=" + dbname +
" user=" + user +
" password=" + password +
" host=" + host;
try {
_connection = std::make_unique<pqxx::connection>(conninfo);
if (!_connection->is_open()) {
throw std::runtime_error("Failed to open database connection");
}
} catch (const std::exception& e) {
throw std::runtime_error(std::string("Database connection error: ") + e.what());
}
}
// Beispielmethode für eine Abfrage
pqxx::result Database::exec(const std::string& query)
{
pqxx::work txn(*_connection);
pqxx::result r = txn.exec(query);
txn.commit();
return r;
}
} // namespace Lib
} // namespace Yc

23
database.h Normal file
View File

@@ -0,0 +1,23 @@
#ifndef YC_LIB_DATABASE_H
#define YC_LIB_DATABASE_H
#include <memory>
#include <pqxx/pqxx>
#include "config.h"
namespace Yc {
namespace Lib {
class Database {
public:
Database(std::shared_ptr<Config> config);
~Database() = default;
pqxx::result exec(const std::string& query);
private:
std::unique_ptr<pqxx::connection> _connection;
};
} // namespace Lib
} // namespace Yc
#endif // YC_LIB_DATABASE_H

17
db_example.cpp Normal file
View File

@@ -0,0 +1,17 @@
#include <iostream>
#include <pqxx/pqxx>
int main() {
try {
pqxx::connection c{"dbname=your_db user=your_user password=your_pass host=localhost"};
pqxx::work txn{c};
pqxx::result r = txn.exec("SELECT version();");
for (auto row : r) {
std::cout << row[0].c_str() << std::endl;
}
} catch (const std::exception &e) {
std::cerr << "PostgreSQL error: " << e.what() << std::endl;
return 1;
}
return 0;
}

View File

@@ -1,10 +1,13 @@
#include "config.h" #include "config.h"
#include "server.h" #include "server.h"
#include "database.h"
// main function // main function
int main(int, char **) { int main(int, char **) {
auto config = std::make_shared<Yc::Lib::Config>(); auto config = std::make_shared<Yc::Lib::Config>();
auto server = std::make_shared<Yc::Lib::Server>(config); auto database = std::make_shared<Yc::Lib::Database>(config);
auto server = std::make_shared<Yc::Lib::Server>(config, database);
server->createRooms(config->group("rooms")); server->createRooms(config->group("rooms"));
server->run(); server->run();
return 0; return 0;

View File

@@ -55,15 +55,15 @@ namespace Yc {
if (_password != "" && _password == _password && std::find(std::begin(_allowedUsers), std::end(_allowedUsers), _userName) == std::end(_allowedUsers)) { if (_password != "" && _password == _password && std::find(std::begin(_allowedUsers), std::end(_allowedUsers), _userName) == std::end(_allowedUsers)) {
return false; return false;
} }
auto newUser = std::make_shared<User>(shared_from_this(), _userName, color, socket); auto newUser = std::make_shared<ChatUser>(shared_from_this(), _userName, color, socket);
_users.push_back(newUser); _users.push_back(newUser);
newUser->sendMsg(User::roomList, _parent->jsonRoomList(), "", ""); newUser->sendMsg(ChatUser::roomList, _parent->jsonRoomList(), "", "");
addMessage(User::system, "room_entered", newUser->name(), newUser->color()); addMessage(ChatUser::system, "room_entered", newUser->name(), newUser->color());
_initRound(); _initRound();
return true; return true;
} }
bool Room::addUser(std::shared_ptr<User> user, std::string password) { bool Room::addUser(std::shared_ptr<ChatUser> user, std::string password) {
if (password == _password) { if (password == _password) {
_users.push_back(user); _users.push_back(user);
user->setParent(shared_from_this()); user->setParent(shared_from_this());
@@ -86,7 +86,7 @@ namespace Yc {
for (auto it = _users.begin(); it != _users.end(); ++it) { for (auto it = _users.begin(); it != _users.end(); ++it) {
if ((*it)->validateToken(_token)) { if ((*it)->validateToken(_token)) {
if (!silent) { if (!silent) {
addMessage(User::system, "room_exit", (*it)->name(), (*it)->color()); addMessage(ChatUser::system, (*it)->name(), (*it)->color());
} }
_users.erase(it); _users.erase(it);
break; break;
@@ -94,11 +94,11 @@ namespace Yc {
} }
} }
void Room::removeUser(std::shared_ptr<User> userToRemove, bool silent) { void Room::removeUser(std::shared_ptr<ChatUser> userToRemove, bool silent) {
for (auto it = _users.begin(); it != _users.end(); ++it) { for (auto it = _users.begin(); it != _users.end(); ++it) {
if (*it == userToRemove) { if (*it == userToRemove) {
if (!silent) { if (!silent) {
addMessage(User::system, "room_exit", (*it)->name(), (*it)->color()); addMessage(ChatUser::system, (*it)->name(), (*it)->color());
} }
_users.erase(it); _users.erase(it);
break; break;
@@ -110,11 +110,11 @@ namespace Yc {
_stop = true; _stop = true;
} }
void Room::addMessage(User::MsgType type, const char *messageText, std::string userName, std::string color) { void Room::addMessage(ChatUser::MsgType type, const char *messageText, std::string userName, std::string color) {
addMessage(type, (std::string)messageText, userName, color); addMessage(type, (std::string)messageText, userName, color);
} }
void Room::addMessage(User::MsgType type, std::string messageText, std::string userName, std::string color) { void Room::addMessage(ChatUser::MsgType type, std::string messageText, std::string userName, std::string color) {
Message message; Message message;
message.type = type; message.type = type;
message.messageTr = messageText; message.messageTr = messageText;
@@ -123,11 +123,11 @@ namespace Yc {
_msgQueue.push(message); _msgQueue.push(message);
} }
void Room::addMessage(User::MsgType type, Json::Value messageText, std::string userName, std::string color) { void Room::addMessage(ChatUser::MsgType type, Json::Value messageText, std::string userName, std::string color) {
addMessage(type, getJsonString(messageText), userName, color); addMessage(type, getJsonString(messageText), userName, color);
} }
void Room::addUserWhenQueueEmpty(std::shared_ptr<User> user) { void Room::addUserWhenQueueEmpty(std::shared_ptr<ChatUser> user) {
while (_msgQueue.size() > 0) { while (_msgQueue.size() > 0) {
std::this_thread::sleep_for(std::chrono::milliseconds(100)); std::this_thread::sleep_for(std::chrono::milliseconds(100));
} }
@@ -135,7 +135,7 @@ namespace Yc {
user->setParent(shared_from_this()); user->setParent(shared_from_this());
} }
bool Room::userToNewRoom(std::shared_ptr<User> user, std::string newRoom, std::string password) { bool Room::userToNewRoom(std::shared_ptr<ChatUser> user, std::string newRoom, std::string password) {
return _parent->changeRoom(user, newRoom, password); return _parent->changeRoom(user, newRoom, password);
} }
@@ -169,7 +169,7 @@ namespace Yc {
return (_roundRunning || (_type & rounds) == rounds) && (_type & dice) == dice; return (_roundRunning || (_type & rounds) == rounds) && (_type & dice) == dice;
} }
unsigned int Room::addDice(std::shared_ptr<User> user, int diceValue) { unsigned int Room::addDice(std::shared_ptr<ChatUser> user, int diceValue) {
if (!canDice()) { if (!canDice()) {
return 1; return 1;
} }
@@ -179,7 +179,7 @@ namespace Yc {
} }
} }
_diceValues.push_back(std::make_pair(user, diceValue)); _diceValues.push_back(std::make_pair(user, diceValue));
addMessage(User::dice, std::to_string(diceValue), user->name(), user->color()); addMessage(ChatUser::dice, std::to_string(diceValue), user->name(), user->color());
return 0; return 0;
} }
@@ -201,13 +201,13 @@ namespace Yc {
if (_roundRunning && (_users.size() < 2 || _roundStart + _roundLength >= time(NULL))) { if (_roundRunning && (_users.size() < 2 || _roundStart + _roundLength >= time(NULL))) {
_lastRoundEnd = time(NULL); _lastRoundEnd = time(NULL);
_roundRunning = false; _roundRunning = false;
addMessage(User::system, "round_ends"); addMessage(ChatUser::system, "round_ends");
_showDiceRoundResults(); _showDiceRoundResults();
} else if (!_roundRunning && _lastRoundEnd <= time(NULL) - 15 && _users.size() >= 2) { } else if (!_roundRunning && _lastRoundEnd <= time(NULL) - 15 && _users.size() >= 2) {
_roundStart = time(NULL); _roundStart = time(NULL);
_roundRunning = true; _roundRunning = true;
_diceValues.clear(); _diceValues.clear();
addMessage(User::system, "next_round_starts_now"); addMessage(ChatUser::system, "next_round_starts_now");
} }
} }
} }
@@ -215,12 +215,12 @@ namespace Yc {
void Room::_initRound() { void Room::_initRound() {
if (_users.size() == 2) { if (_users.size() == 2) {
_lastRoundEnd = time(NULL); _lastRoundEnd = time(NULL);
addMessage(User::system, "next_round_starts_soon"); addMessage(ChatUser::system, "next_round_starts_soon");
} }
} }
void Room::_showDiceRoundResults() { void Room::_showDiceRoundResults() {
std::sort(_diceValues.begin(), _diceValues.end(), [=](const std::pair<std::shared_ptr<User>, int>& val1, const std::pair<std::shared_ptr<User>, int>& val2) { std::sort(_diceValues.begin(), _diceValues.end(), [=](const std::pair<std::shared_ptr<ChatUser>, int>& val1, const std::pair<std::shared_ptr<ChatUser>, int>& val2) {
return (val1.second > val2.second); return (val1.second > val2.second);
}); });
Json::Value userList = Json::arrayValue; Json::Value userList = Json::arrayValue;
@@ -230,7 +230,7 @@ namespace Yc {
entry["value"] = user.second; entry["value"] = user.second;
userList.append(entry); userList.append(entry);
} }
addMessage(User::result, userList); addMessage(ChatUser::result, userList);
} }
std::string Room::name() { std::string Room::name() {

29
room.h
View File

@@ -1,20 +1,23 @@
#ifndef YC_LIB_ROOM_H #ifndef YC_LIB_ROOM_H
#define YC_LIB_ROOM_H #define YC_LIB_ROOM_H
#include <vector> #include <vector>
#include <string> #include <string>
#include <user.h>
#include <queue> #include <queue>
#include <thread> #include <thread>
#include <future> #include <future>
#include <utility> #include <utility>
#include <vector> #include <vector>
#include <base.h> #include <base.h>
#include "chat_user.h"
namespace Yc { namespace Yc {
namespace Lib { namespace Lib {
class Server;
class Server;
class Room: public Base, public std::enable_shared_from_this<Room> { class Room: public Base, public std::enable_shared_from_this<Room> {
public: public:
@@ -31,27 +34,27 @@ namespace Yc {
void run(); void run();
std::string name(); std::string name();
bool addUser(std::string userName, std::string color, std::string password, int socket); bool addUser(std::string userName, std::string color, std::string password, int socket);
bool addUser(std::shared_ptr<User> user, std::string password); bool addUser(std::shared_ptr<ChatUser> user, std::string password);
bool userNameExists(std::string userName); bool userNameExists(std::string userName);
void removeUser(std::string _token, bool silent = false); void removeUser(std::string _token, bool silent = false);
void removeUser(std::shared_ptr<User> user, bool silent = false); void removeUser(std::shared_ptr<ChatUser> user, bool silent = false);
void setStop(); void setStop();
void addMessage(User::MsgType type, const char* messageText, std::string userName = "", std::string color = ""); void addMessage(ChatUser::MsgType type, const char* messageText, std::string userName = "", std::string color = "");
void addMessage(User::MsgType type, std::string messageText, std::string userName = "", std::string color = ""); void addMessage(ChatUser::MsgType type, std::string messageText, std::string userName = "", std::string color = "");
void addMessage(User::MsgType type, Json::Value messageText, std::string userName = "", std::string color = ""); void addMessage(ChatUser::MsgType type, Json::Value messageText, std::string userName = "", std::string color = "");
RoomType type(); RoomType type();
bool isType(RoomType type); bool isType(RoomType type);
bool canDice(); bool canDice();
unsigned int addDice(std::shared_ptr<User> user, int diceValue); unsigned int addDice(std::shared_ptr<ChatUser> user, int diceValue);
bool accessAllowed(std::string userName, std::string password); bool accessAllowed(std::string userName, std::string password);
bool userIsInRoom(std::string userName); bool userIsInRoom(std::string userName);
void addUserWhenQueueEmpty(std::shared_ptr<User> user); void addUserWhenQueueEmpty(std::shared_ptr<ChatUser> user);
bool userToNewRoom(std::shared_ptr<User> user, std::string newRoom, std::string password); bool userToNewRoom(std::shared_ptr<ChatUser> user, std::string newRoom, std::string password);
unsigned int flags(); unsigned int flags();
Json::Value userList(); Json::Value userList();
private: private:
struct Message { struct Message {
User::MsgType type; ChatUser::MsgType type;
std::string messageTr; std::string messageTr;
std::string userName; std::string userName;
std::string color; std::string color;
@@ -62,7 +65,7 @@ namespace Yc {
std::string _password; std::string _password;
std::vector<std::string> _allowedUsers; std::vector<std::string> _allowedUsers;
RoomType _type; RoomType _type;
std::vector<std::shared_ptr<User>> _users; std::vector<std::shared_ptr<ChatUser>> _users;
bool _blocked; bool _blocked;
bool _stop; bool _stop;
std::queue<Message> _msgQueue; std::queue<Message> _msgQueue;
@@ -71,7 +74,7 @@ namespace Yc {
time_t _roundStart; time_t _roundStart;
time_t _lastRoundEnd; time_t _lastRoundEnd;
int _roundLength; int _roundLength;
std::vector<std::pair<std::shared_ptr<User>, int>> _diceValues; std::vector<std::pair<std::shared_ptr<ChatUser>, int>> _diceValues;
void _handleDice(); void _handleDice();
void _startDiceRound(); void _startDiceRound();
void _endDiceRound(); void _endDiceRound();

View File

@@ -15,8 +15,9 @@
namespace Yc { namespace Yc {
namespace Lib { namespace Lib {
Server::Server(std::shared_ptr<Yc::Lib::Config> config) : Server::Server(std::shared_ptr<Yc::Lib::Config> config, std::shared_ptr<Yc::Lib::Database> database) :
_config(std::move(config)), _config(std::move(config)),
_database(std::move(database)),
_stop(false) { _stop(false) {
struct sockaddr_in serverAddr; struct sockaddr_in serverAddr;
int opt = true; int opt = true;
@@ -82,7 +83,7 @@ namespace Yc {
return false; return false;
} }
bool Server::changeRoom(std::shared_ptr<User> user, std::string newRoom, std::string password) { bool Server::changeRoom(std::shared_ptr<ChatUser> user, std::string newRoom, std::string password) {
if (!roomAllowed(newRoom, user->name(), password)) { if (!roomAllowed(newRoom, user->name(), password)) {
return false; return false;
} }
@@ -96,16 +97,16 @@ namespace Yc {
msg["tr"] = "room_change_to"; msg["tr"] = "room_change_to";
msg["to"] = newRoom; msg["to"] = newRoom;
userMsg["from"] = room->name(); userMsg["from"] = room->name();
room->addMessage(User::system, msg, user->name(), user->color()); room->addMessage(ChatUser::system, msg, user->name(), user->color());
} }
} }
user->sendMsg(User::system, userMsg, "", ""); user->sendMsg(ChatUser::system, userMsg, "", "");
for (auto &room: _rooms) { for (auto &room: _rooms) {
if (room->name() == newRoom) { if (room->name() == newRoom) {
Json::Value msg = Json::objectValue; Json::Value msg = Json::objectValue;
msg["tr"] = "room_change_to"; msg["tr"] = "room_change_to";
msg["from"] = userMsg["from"]; msg["from"] = userMsg["from"];
room->addMessage(User::system, msg, user->name(), user->color()); room->addMessage(ChatUser::system, msg, user->name(), user->color());
room->addUserWhenQueueEmpty(user); room->addUserWhenQueueEmpty(user);
} }
} }
@@ -155,7 +156,7 @@ namespace Yc {
void Server::initUser(int userSocket, Json::Value data) { void Server::initUser(int userSocket, Json::Value data) {
if (userExists(data["name"].asString())) { if (userExists(data["name"].asString())) {
Json::Value errorJson; Json::Value errorJson;
errorJson["type"] = User::error; errorJson["type"] = ChatUser::error;
errorJson["message"] = "loggedin"; errorJson["message"] = "loggedin";
send(userSocket, errorJson); send(userSocket, errorJson);
close(userSocket); close(userSocket);

View File

@@ -2,6 +2,7 @@
#define YP_LIB_SERVER_H #define YP_LIB_SERVER_H
#include "config.h" #include "config.h"
#include "database.h"
#include <vector> #include <vector>
#include "room.h" #include "room.h"
#include "base.h" #include "base.h"
@@ -13,15 +14,15 @@ namespace Yc {
class Server: public Base, public std::enable_shared_from_this<Server> class Server: public Base, public std::enable_shared_from_this<Server>
{ {
public: public:
Server(std::shared_ptr<Yc::Lib::Config> config); Server(std::shared_ptr<Yc::Lib::Config> config, std::shared_ptr<Yc::Lib::Database> database);
void run(); void run();
std::vector<std::string> roomList(); std::vector<std::string> roomList();
Json::Value jsonRoomList(); Json::Value jsonRoomList();
bool roomAllowed(std::string roomName, std::string userName, std::string password); bool roomAllowed(std::string roomName, std::string userName, std::string password);
bool changeRoom(std::shared_ptr<User> user, std::string newRoom, std::string password); bool changeRoom(std::shared_ptr<ChatUser> user, std::string newRoom, std::string password);
public:
int _socket; int _socket;
std::shared_ptr<Yc::Lib::Config> _config; std::shared_ptr<Yc::Lib::Config> _config;
std::shared_ptr<Yc::Lib::Database> _database;
bool _stop; bool _stop;
std::vector<std::shared_ptr<Yc::Lib::Room>> _rooms; std::vector<std::shared_ptr<Yc::Lib::Room>> _rooms;
void createRooms(Json::Value roomList); void createRooms(Json::Value roomList);

172
user.cpp
View File

@@ -170,3 +170,175 @@ namespace Yc {
} }
} // namespace Lib } // namespace Lib
} // namespace Yc } // namespace Yc
#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(std::shared_ptr<Room> parent, std::string name, std::string color, int socket) :
_parent(std::move(parent)),
_name(std::move(name)),
_color(std::move(color)),
_socket(socket),
_stop(false) {
_token = Yc::Lib::Tools::generateRandomString(32);
sendMsg(token, _token, "", "");
thread = std::make_unique<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(std::shared_ptr<User> toValidate) {
return (toValidate.get() == 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(std::shared_ptr<Room> parent) {
_parent = std::move(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) {
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());
} else if (jsonTree["type"].asString() == "userlist") {
sendUserList();
}
}
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::sendUserList() {
Json::Value userList = _parent->userList();
Json::Value msg = Json::objectValue;
msg["userlist"] = userList;
sendMsg(userListe, msg, "", "");
}
void User::doDice() {
switch (_parent->addDice(shared_from_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(shared_from_this(), newRoom, password)) {
sendMsg(User::system, "room_not_possible", "", "");
}
}
} // namespace Lib
} // namespace Yc