119 lines
3.5 KiB
C++
119 lines
3.5 KiB
C++
/******************************************************************************/
|
|
/* */
|
|
/* ::: :::::::: */
|
|
/* bot.cpp :+: :+: :+: */
|
|
/* +:+ +:+ +:+ */
|
|
/* By: vvaas <vvaas@student.42angouleme.fr> +#+ +:+ +#+ */
|
|
/* +#+#+#+#+#+ +#+ */
|
|
/* Created: 2024/01/30 01:54:56 by vvaas #+# #+# */
|
|
/* Updated: 2024/01/30 21:45:18 by vvaas ### ########.fr */
|
|
/* */
|
|
/******************************************************************************/
|
|
|
|
#include <Bot.hpp>
|
|
#include <logs.hpp>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <cstring>
|
|
#include <cstdlib>
|
|
#include <signal.h>
|
|
|
|
bool active = true;
|
|
|
|
Bot::Bot() : _channel_created(false), _logged(false), _fd(-1)
|
|
{}
|
|
|
|
Bot::~Bot()
|
|
{
|
|
if(_fd >= 0)
|
|
close(_fd);
|
|
}
|
|
|
|
void signalsHandler(int foo)
|
|
{
|
|
(void)foo;
|
|
std::cout << "\b\b \b\b" << std::flush;
|
|
irc::logs::report(irc::log_message, "Shutting down...");
|
|
active = false;
|
|
}
|
|
|
|
bool Bot::init()
|
|
{
|
|
_connect_commands.push_back("PASS " PASSWORD "\r\n");
|
|
_connect_commands.push_back("NICK greg\r\n");
|
|
_connect_commands.push_back("USER greg_Bot 0 * :Botrealname\r\n");
|
|
_connect_commands.push_back("JOIN #greg\r\n");
|
|
_fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
if(_fd == -1)
|
|
irc::logs::report(irc::log_fatal_error, "FD error");
|
|
_serv_addr.sin_family = AF_INET;
|
|
_serv_addr.sin_port = htons(PORT);
|
|
_serv_addr.sin_addr.s_addr = inet_addr(IP);
|
|
if(connect(_fd, (struct sockaddr*)&_serv_addr, sizeof(_serv_addr)) < 0)
|
|
{
|
|
irc::logs::report(irc::log_error, "connect error");
|
|
return (false);
|
|
}
|
|
if(fcntl(_fd, F_SETFL, O_NONBLOCK) < 0)
|
|
{
|
|
irc::logs::report(irc::log_error, "fcntl error");
|
|
return (false);
|
|
}
|
|
signal(SIGINT, signalsHandler);
|
|
signal(SIGQUIT, signalsHandler);
|
|
return (true);
|
|
}
|
|
|
|
void Bot::send_message(const std::string &content)
|
|
{
|
|
if(send(_fd, content.c_str(), content.length(), 0) < 0)
|
|
irc::logs::report(irc::log_fatal_error, "send error");
|
|
}
|
|
|
|
void Bot::handle_response(std::string buffer)
|
|
{
|
|
std::cout << buffer << std::flush;
|
|
|
|
if(!_logged && buffer == ":yipirc 001 greg :Welcome to yipirc :), greg\r\n")
|
|
{
|
|
_logged = true;
|
|
irc::logs::report(irc::log_message, "Logged in succesfully");
|
|
}
|
|
else if(!_logged)
|
|
return ;
|
|
if(!_channel_created && buffer == ":greg JOIN :#greg\r\n")
|
|
{
|
|
_channel_created = true;
|
|
irc::logs::report(irc::log_message, "Created the channel succesfully");
|
|
}
|
|
else if(!_channel_created)
|
|
return ;
|
|
if(buffer.find("KICK #greg greg :") != std::string::npos || buffer.find("explose") != std::string::npos)
|
|
{
|
|
send_message("QUIT: Explose\r\n");
|
|
std::exit(0);
|
|
}
|
|
if(buffer.find("quoi") != std::string::npos)
|
|
send_message("PRIVMSG #greg :feur\r\n");
|
|
}
|
|
|
|
void Bot::connect_to_server()
|
|
{
|
|
char buffer[1024];
|
|
for(std::vector<std::string>::iterator it = _connect_commands.begin(); it != _connect_commands.end(); ++it)
|
|
{
|
|
if(send(_fd, (*it).c_str(), (*it).size(), 0) < 0)
|
|
irc::logs::report(irc::log_fatal_error, "send error");
|
|
if(recv(_fd, buffer, 1024, 0) > 0)
|
|
handle_response(buffer);
|
|
}
|
|
while (active)
|
|
{
|
|
if(recv(_fd, buffer, 1024, 0) > 0)
|
|
handle_response(buffer);
|
|
std::memset(buffer, 0, sizeof(buffer));
|
|
}
|
|
}
|