FT_IRC/INVITE_command.cpp at master · iariar/FT_IRC · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include "msg_parse.hpp"
#include "Server.hpp"

int	Server::is_real_user(std::string nickname)
{
	for (std::list<User>::iterator it = __users.begin() ; it != __users.end(); it++)
	{
		if ((*it).get_nickname() == nickname)
			return (1);
	}
	return (0);
}

User *Server::find_user_in_channel_by_nick(std::string nickname, Channel &chan)
{
	std::list<User *>::iterator it = chan.get_users().begin();
	for ( ; it != chan.get_users().end() ; it++)
	{
		if ((*it)->get_nickname() == nickname)
			return (*it);
	}
	return (*it);
}

int		Server::is_operator_on_channel(User &user, Channel &chan)
{
	for (std::vector<User *>::iterator it = chan.get_operators().begin() ; it != chan.get_operators().end(); it++)
	{
		if ((*it)->get_nickname() == user.get_nickname())
			return (1);
	}
	return (0);
}

void	Channel::add_to_invited_list(User &user)
{
	__invited_list.push_back(&user);
}

std::vector<User *> &Channel::get_invited_list(void)
{
	return (__invited_list);
}

User	&Server::find_user_in_server(std::string &nickname)
{
	std::list<User>::iterator it = __users.begin();
	for (; it != __users.end(); it++)
	{
		if (nickname == (*it).get_nickname())
			return (*it);
	}
	return (*it);
}

void	Server::INVITE_handler(User &user, msg_parse &command)
{
	if (command.get_cmd_params().size() > 1)
	{
		if (command.get_cmd_params().size() > 2)
		{
			write_socket(user.get_fd(), user.full_id() + ": more params than necessary\n");
			return ;
		}
		if (!is_real_user(command.get_cmd_params()[0]))
		{
			write_reply(user, ERR_NOSUCHNICK, command);
			return ;
		}
		std::string channel_without_prefix = command.get_cmd_params()[1];
		std::list<Channel>::iterator chan;
		if (find_channel(command.get_cmd_params()[1][0], channel_without_prefix.substr(1, channel_without_prefix.length() - 1)) == get_channels().end())
		{
			write_reply(user, ERR_NOSUCHCHANNEL, command);
			return ;
		}
		chan = find_channel(command.get_cmd_params()[1][0], channel_without_prefix.substr(1, channel_without_prefix.length() - 1));
		if (find_user_in_channel(user, *chan) == *(*chan).get_users().end())
		{
			write_reply(user, ERR_NOTONCHANNEL, command);
			return ;
		}
		if (find_user_in_channel_by_nick(command.get_cmd_params()[0], *chan) != *(*chan).get_users().end())
		{
			write_reply(user, ERR_USERONCHANNEL, command);
			return ;
		}
		std::string nickname = command.get_cmd_params()[0];
		User &invited_user = find_user_in_server(nickname);
		if (invited_user.get_modes().get_a())
		{
			write_reply(user, RPL_AWAY, command);
			return ;
		}
		if ((*chan).get_modes().get_i() && !is_operator_on_channel(user, *chan) && !user.get_modes().get_o())
		{
			write_reply(user, ERR_CHANOPRIVSNEEDED, command);
			return ;
		}
		(*chan).add_to_invited_list(invited_user);
		std::string	full_msg = ":" + this->__name + " " + command.get_cmd() + " 341 " + command.get_cmd_params()[1] + " :" + user.get_nickname() + "\n"/* + user.get_nickname() + "!" + user.get_username() + "@" + user.get_hostname() + "\n"*/;
		send(invited_user.get_fd(), full_msg.c_str(), full_msg.size(), 0);
	}
	else
		write_reply(user, ERR_NEEDMOREPARAMS, command);
}