bridge.cpp 8.45 KB
Newer Older
1
#include <bridge/bridge.hpp>
2
#include <bridge/colors.hpp>
3
4
#include <xmpp/xmpp_component.hpp>
#include <network/poller.hpp>
5
#include <utils/encoding.hpp>
louiz’'s avatar
louiz’ committed
6
#include <logger/logger.hpp>
louiz’'s avatar
louiz’ committed
7
#include <utils/split.hpp>
louiz’'s avatar
louiz’ committed
8
#include <stdexcept>
louiz’'s avatar
louiz’ committed
9
#include <iostream>
10
#include <tuple>
louiz’'s avatar
louiz’ committed
11

12
13
14
static const char* action_prefix = "\01ACTION ";
static const size_t action_prefix_len = 8;

15
16
17
18
19
20
21
22
23
24
25
Bridge::Bridge(const std::string& user_jid, XmppComponent* xmpp, Poller* poller):
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

Bridge::~Bridge()
{
}

26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
/**
 * Return the role and affiliation, corresponding to the given irc mode */
static std::tuple<std::string, std::string> get_role_affiliation_from_irc_mode(const char mode)
{
  if (mode == 'a' || mode == 'q')
    return std::make_tuple("moderator", "owner");
  else if (mode == 'o')
    return std::make_tuple("moderator", "admin");
  else if (mode == 'h')
    return std::make_tuple("moderator", "member");
  else if (mode == 'v')
    return std::make_tuple("participant", "member");
  else
    return std::make_tuple("participant", "none");
}

louiz’'s avatar
louiz’ committed
42
43
44
45
void Bridge::shutdown()
{
  for (auto it = this->irc_clients.begin(); it != this->irc_clients.end(); ++it)
  {
46
    it->second->send_quit_command("Gateway shutdown");
louiz’'s avatar
louiz’ committed
47
48
49
  }
}

louiz’'s avatar
louiz’ committed
50
51
52
53
54
55
void Bridge::clean()
{
  auto it = this->irc_clients.begin();
  while (it != this->irc_clients.end())
  {
    IrcClient* client = it->second.get();
56
    if (!client->is_connected() && !client->is_connecting())
louiz’'s avatar
louiz’ committed
57
58
59
60
61
62
      it = this->irc_clients.erase(it);
    else
      ++it;
  }
}

63
Xmpp::body Bridge::make_xmpp_body(const std::string& str)
64
65
66
67
68
69
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
    res = utils::convert_to_utf8(str, "ISO-8859-1");
70
  return irc_format_to_xhtmlim(res);
71
72
}

73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
IrcClient* Bridge::get_irc_client(const std::string& hostname, const std::string& username)
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      this->irc_clients.emplace(hostname, std::make_shared<IrcClient>(hostname, username, this));
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      this->poller->add_socket_handler(irc);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
88
89
90
91
92
93
94
95
96
97
98
99
IrcClient* Bridge::get_irc_client(const std::string& hostname)
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      return nullptr;
    }
}

louiz’'s avatar
louiz’ committed
100
bool Bridge::join_irc_channel(const Iid& iid, const std::string& username)
101
102
{
  IrcClient* irc = this->get_irc_client(iid.server, username);
louiz’'s avatar
louiz’ committed
103
104
105
106
107
108
  if (irc->is_channel_joined(iid.chan) == false)
    {
      irc->send_join_command(iid.chan);
      return true;
    }
  return false;
109
110
}

louiz’'s avatar
louiz’ committed
111
112
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
113
114
115
116
  std::vector<std::string> lines = utils::split(body, '\n', true);
  if (lines.empty())
    return ;
  const std::string first_line = lines[0];
louiz’'s avatar
louiz’ committed
117
118
  if (iid.chan.empty() || iid.server.empty())
    {
louiz’'s avatar
louiz’ committed
119
      log_warning("Cannot send message to channel: [" << iid.chan << "] on server [" << iid.server << "]");
louiz’'s avatar
louiz’ committed
120
121
122
123
124
      return;
    }
  IrcClient* irc = this->get_irc_client(iid.server);
  if (!irc)
    {
louiz’'s avatar
louiz’ committed
125
      log_warning("Cannot send message: no client exist for server " << iid.server);
louiz’'s avatar
louiz’ committed
126
127
      return;
    }
128
  if (first_line.substr(0, 6) == "/mode ")
louiz’'s avatar
louiz’ committed
129
    {
130
      std::vector<std::string> args = utils::split(first_line.substr(6), ' ', false);
louiz’'s avatar
louiz’ committed
131
132
133
      irc->send_mode_command(iid.chan, args);
      return;
    }
134
135
  if (first_line.substr(0, 4) == "/me ")
    irc->send_channel_message(iid.chan, action_prefix + first_line.substr(4) + "\01");
136
  else
137
138
139
140
    irc->send_channel_message(iid.chan, first_line);
  // Send each of the other lines of the message as a separate IRC message
  for (std::vector<std::string>::const_iterator it = lines.begin() + 1; it != lines.end(); ++it)
    irc->send_channel_message(iid.chan, *it);
141
142
  // We do not need to convert body to utf-8: it comes from our XMPP server,
  // so it's ok to send it back
143
144
  this->xmpp->send_muc_message(iid.chan + "%" + iid.server, irc->get_own_nick(),
                               this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
145
146
}

louiz’'s avatar
louiz’ committed
147
148
149
150
151
152
153
154
155
void Bridge::send_private_message(const Iid& iid, const std::string& body)
{
  if (iid.chan.empty() || iid.server.empty())
    return ;
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_private_message(iid.chan, body);
}

louiz’'s avatar
louiz’ committed
156
157
158
159
160
161
162
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_part_command(iid.chan, status_message);
}

louiz’'s avatar
louiz’ committed
163
164
165
166
167
168
169
void Bridge::send_irc_nick_change(const Iid& iid, const std::string& new_nick)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_nick_command(new_nick);
}

170
171
172
173
174
175
176
void Bridge::send_irc_kick(const Iid& iid, const std::string& target, const std::string& reason)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_kick_command(iid.chan, target, reason);
}

177
178
179
180
181
182
183
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_topic_command(iid.chan, subject);
}

louiz’'s avatar
louiz’ committed
184
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
185
{
louiz’'s avatar
louiz’ committed
186
  if (muc)
187
188
    this->xmpp->send_muc_message(iid.chan + "%" + iid.server, nick,
                                 this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
189
  else
190
191
    this->xmpp->send_message(iid.chan + "%" + iid.server,
                             this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
192
193
}

louiz’'s avatar
louiz’ committed
194
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
195
{
196
  this->xmpp->send_muc_leave(std::move(iid.chan) + "%" + std::move(iid.server), std::move(nick), this->make_xmpp_body(message), this->user_jid, self);
197
198
199
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
200
201
}

202
203
204
205
206
void Bridge::send_nick_change(Iid&& iid,
                              const std::string& old_nick,
                              const std::string& new_nick,
                              const char user_mode,
                              const bool self)
louiz’'s avatar
louiz’ committed
207
{
208
209
210
211
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

louiz’'s avatar
louiz’ committed
212
  this->xmpp->send_nick_change(std::move(iid.chan) + "%" + std::move(iid.server),
213
                               old_nick, new_nick, affiliation, role, this->user_jid, self);
louiz’'s avatar
louiz’ committed
214
215
}

216
217
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
218
219
220
221
222
  std::string body;
  if (!author.empty())
    body = std::string("[") + author + std::string("] ") + msg;
  else
    body = msg;
223
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid);
224
225
}

226
227
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
228
                            const IrcUser* user,
229
                            const char user_mode,
230
                            const bool self)
231
{
232
233
234
235
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

236
237
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
238
239
240
241
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
242
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
243
}
louiz’'s avatar
louiz’ committed
244
245
246
247
248
249
250
251

std::string Bridge::get_own_nick(const Iid& iid)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    return irc->get_own_nick();
  return "";
}
252

253
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
254
255
256
257
{
  return this->irc_clients.size();
}

258
259
260
261
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
  this->xmpp->kick_user(iid.chan + "%" + iid.server, target, reason, author, this->user_jid);
}
262
263
264
265
266

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
  this->xmpp->send_nickname_conflict_error(iid.chan + "%" + iid.server, nickname, this->user_jid);
}
267
268
269
270
271

void Bridge::send_affiliation_role_change(const Iid& iid, const std::string& target, const char mode)
{
  std::string role;
  std::string affiliation;
272
273

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
274
275
  this->xmpp->send_affiliation_role_change(iid.chan + "%" + iid.server, target, affiliation, role, this->user_jid);
}