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

louiz’'s avatar
louiz’ committed
14 15
using namespace std::string_literals;

16 17
static const char* action_prefix = "\01ACTION ";

18
Bridge::Bridge(const std::string& user_jid, XmppComponent* xmpp, std::shared_ptr<Poller> poller):
19 20 21 22 23 24 25 26 27 28
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

Bridge::~Bridge()
{
}

29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
/**
 * 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");
}

45
void Bridge::shutdown(const std::string& exit_message)
louiz’'s avatar
louiz’ committed
46 47 48
{
  for (auto it = this->irc_clients.begin(); it != this->irc_clients.end(); ++it)
  {
louiz’'s avatar
louiz’ committed
49 50
    it->second->send_quit_command(exit_message);
    it->second->leave_dummy_channel(exit_message);
louiz’'s avatar
louiz’ committed
51 52 53
  }
}

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

67 68 69 70 71
const std::string& Bridge::get_jid() const
{
  return this->user_jid;
}

72
Xmpp::body Bridge::make_xmpp_body(const std::string& str)
73 74 75 76 77 78
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
    res = utils::convert_to_utf8(str, "ISO-8859-1");
79
  return irc_format_to_xhtmlim(res);
80 81
}

82 83 84 85 86 87 88 89
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)
    {
90
      this->irc_clients.emplace(hostname, std::make_shared<IrcClient>(this->poller, hostname, username, this));
91 92 93 94 95
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
96 97 98 99 100 101 102 103 104 105 106 107
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
108
bool Bridge::join_irc_channel(const Iid& iid, const std::string& username)
109
{
louiz’'s avatar
louiz’ committed
110 111
  IrcClient* irc = this->get_irc_client(iid.get_server(), username);
  if (iid.get_local().empty())
louiz’'s avatar
louiz’ committed
112 113 114 115 116 117 118 119 120
    { // Join the dummy channel
      if (irc->is_welcomed())
        {
          if (irc->get_dummy_channel().joined)
            return false;
          // Immediately simulate a message coming from the IRC server saying that we
          // joined the channel
          const IrcMessage join_message(irc->get_nick(), "JOIN", {""});
          irc->on_channel_join(join_message);
louiz’'s avatar
louiz’ committed
121
          const IrcMessage end_join_message(std::string(iid.get_server()), "366",
louiz’'s avatar
louiz’ committed
122 123 124 125 126
                                            {irc->get_nick(),
                                                "", "End of NAMES list"});
          irc->on_channel_completely_joined(end_join_message);
        }
      else
127
        {
louiz’'s avatar
louiz’ committed
128
          irc->get_dummy_channel().joining = true;
129 130
          irc->start();
        }
louiz’'s avatar
louiz’ committed
131 132
      return true;
    }
louiz’'s avatar
louiz’ committed
133
  if (irc->is_channel_joined(iid.get_local()) == false)
louiz’'s avatar
louiz’ committed
134
    {
louiz’'s avatar
louiz’ committed
135
      irc->send_join_command(iid.get_local());
louiz’'s avatar
louiz’ committed
136 137 138
      return true;
    }
  return false;
139 140
}

louiz’'s avatar
louiz’ committed
141 142
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
louiz’'s avatar
louiz’ committed
143
  if (iid.get_local().empty() || iid.get_server().empty())
louiz’'s avatar
louiz’ committed
144
    {
louiz’'s avatar
louiz’ committed
145
      log_warning("Cannot send message to channel: [" << iid.get_local() << "] on server [" << iid.get_server() << "]");
louiz’'s avatar
louiz’ committed
146 147
      return;
    }
louiz’'s avatar
louiz’ committed
148
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
149 150
  if (!irc)
    {
louiz’'s avatar
louiz’ committed
151
      log_warning("Cannot send message: no client exist for server " << iid.get_server());
louiz’'s avatar
louiz’ committed
152 153
      return;
    }
154 155 156 157 158 159 160 161 162 163 164 165
  // Because an IRC message cannot contain \n, we need to convert each line
  // of text into a separate IRC message. For conveniance, we also cut the
  // message into submessages on the XMPP side, this way the user of the
  // gateway sees what was actually sent over IRC.  For example if an user
  // sends “hello\n/me waves”, two messages will be generated: “hello” and
  // “/me waves”. Note that the “command” handling (messages starting with
  // /me, /mode, etc) is done for each message generated this way. In the
  // above example, the /me will be interpreted.
  std::vector<std::string> lines = utils::split(body, '\n', true);
  if (lines.empty())
    return ;
  for (const std::string& line: lines)
louiz’'s avatar
louiz’ committed
166
    {
167 168 169
      if (line.substr(0, 6) == "/mode ")
        {
          std::vector<std::string> args = utils::split(line.substr(6), ' ', false);
louiz’'s avatar
louiz’ committed
170
          irc->send_mode_command(iid.get_local(), args);
171 172 173 174
          continue;             // We do not want to send that back to the
                                // XMPP user, that’s not a textual message.
        }
      else if (line.substr(0, 4) == "/me ")
louiz’'s avatar
louiz’ committed
175
        irc->send_channel_message(iid.get_local(), action_prefix + line.substr(4) + "\01");
176
      else
louiz’'s avatar
louiz’ committed
177 178
        irc->send_channel_message(iid.get_local(), line);
      this->xmpp->send_muc_message(std::to_string(iid), irc->get_own_nick(),
179
                                   this->make_xmpp_body(line), this->user_jid);
louiz’'s avatar
louiz’ committed
180
    }
louiz’'s avatar
louiz’ committed
181 182
}

louiz’'s avatar
louiz’ committed
183
void Bridge::send_private_message(const Iid& iid, const std::string& body, const std::string& type)
louiz’'s avatar
louiz’ committed
184
{
louiz’'s avatar
louiz’ committed
185
  if (iid.get_local().empty() || iid.get_server().empty())
louiz’'s avatar
louiz’ committed
186
    return ;
louiz’'s avatar
louiz’ committed
187
  IrcClient* irc = this->get_irc_client(iid.get_server());
188 189
  if (!irc)
    {
louiz’'s avatar
louiz’ committed
190
      log_warning("Cannot send message: no client exist for server " << iid.get_server());
191 192 193 194 195 196 197 198
      return;
    }
  std::vector<std::string> lines = utils::split(body, '\n', true);
  if (lines.empty())
    return ;
  for (const std::string& line: lines)
    {
      if (line.substr(0, 4) == "/me ")
louiz’'s avatar
louiz’ committed
199
        irc->send_private_message(iid.get_local(), action_prefix + line.substr(4) + "\01", type);
200
      else
louiz’'s avatar
louiz’ committed
201
        irc->send_private_message(iid.get_local(), line, type);
202
    }
louiz’'s avatar
louiz’ committed
203 204
}

louiz’'s avatar
louiz’ committed
205 206
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message)
{
louiz’'s avatar
louiz’ committed
207
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
208
  if (irc)
louiz’'s avatar
louiz’ committed
209
    irc->send_part_command(iid.get_local(), status_message);
louiz’'s avatar
louiz’ committed
210 211
}

louiz’'s avatar
louiz’ committed
212 213
void Bridge::send_irc_nick_change(const Iid& iid, const std::string& new_nick)
{
louiz’'s avatar
louiz’ committed
214
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
215 216 217 218
  if (irc)
    irc->send_nick_command(new_nick);
}

219 220
void Bridge::send_irc_kick(const Iid& iid, const std::string& target, const std::string& reason,
                           const std::string& iq_id, const std::string& to_jid)
221
{
louiz’'s avatar
louiz’ committed
222
  IrcClient* irc = this->get_irc_client(iid.get_server());
223
  if (irc)
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261
    {
      irc->send_kick_command(iid.get_local(), target, reason);
      this->add_waiting_iq([this, target, iq_id, to_jid, iid](const std::string& irc_hostname, const IrcMessage& message){
          if (irc_hostname != iid.get_server())
            return false;
          if (message.command == "KICK" && message.arguments.size() >= 2)
            {
              const std::string target_later = message.arguments[1];
              const std::string chan_name_later = utils::tolower(message.arguments[0]);
              if (target_later != target || chan_name_later != iid.get_local())
                return false;
              this->xmpp->send_iq_result(iq_id, to_jid, std::to_string(iid));
            }
          else if (message.command == "401" && message.arguments.size() >= 2)
            {
              const std::string target_later = message.arguments[1];
              if (target_later != target)
                return false;
              std::string error_message = "No such nick";
              if (message.arguments.size() >= 3)
                error_message = message.arguments[2];
              this->xmpp->send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "item-not-found",
                                            error_message, false);
            }
          else if (message.command == "482" && message.arguments.size() >= 2)
            {
              const std::string chan_name_later = utils::tolower(message.arguments[1]);
              if (chan_name_later != iid.get_local())
                return false;
              std::string error_message = "You're not channel operator";
              if (message.arguments.size() >= 3)
                error_message = message.arguments[2];
              this->xmpp->send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "not-allowed",
                                            error_message, false);
            }
          return true;
        });
    }
262 263
}

264 265
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
louiz’'s avatar
louiz’ committed
266
  IrcClient* irc = this->get_irc_client(iid.get_server());
267
  if (irc)
louiz’'s avatar
louiz’ committed
268
    irc->send_topic_command(iid.get_local(), subject);
269 270
}

louiz’'s avatar
louiz’ committed
271 272 273 274 275 276 277
void Bridge::send_xmpp_version_to_irc(const Iid& iid, const std::string& name, const std::string& version, const std::string& os)
{
  std::string result(name + " " + version + " " + os);

  this->send_private_message(iid, "\01VERSION "s + result + "\01", "NOTICE");
}

louiz’'s avatar
louiz’ committed
278
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
279
{
louiz’'s avatar
louiz’ committed
280
  if (muc)
louiz’'s avatar
louiz’ committed
281
    this->xmpp->send_muc_message(std::to_string(iid), nick,
282
                                 this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
283
  else
284 285 286 287 288 289 290 291 292 293 294 295
    {
      std::string target = std::to_string(iid);
      bool fulljid = false;
      auto it = this->preferred_user_from.find(iid.get_local());
      if (it != this->preferred_user_from.end())
        {
          target = it->second;
          fulljid = true;
        }
      this->xmpp->send_message(target, this->make_xmpp_body(body),
                               this->user_jid, "chat", fulljid);
    }
louiz’'s avatar
louiz’ committed
296 297
}

298 299
void Bridge::send_join_failed(const Iid& iid, const std::string& nick, const std::string& type, const std::string& condition, const std::string& text)
{
louiz’'s avatar
louiz’ committed
300
  this->xmpp->send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, text);
301 302
}

louiz’'s avatar
louiz’ committed
303
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
304
{
louiz’'s avatar
louiz’ committed
305 306
  this->xmpp->send_muc_leave(std::to_string(iid), std::move(nick), this->make_xmpp_body(message), this->user_jid, self);
  IrcClient* irc = this->get_irc_client(iid.get_server());
307 308
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
309 310
}

311 312 313 314 315
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
316
{
317 318 319 320
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

louiz’'s avatar
louiz’ committed
321
  this->xmpp->send_nick_change(std::to_string(iid),
322
                               old_nick, new_nick, affiliation, role, this->user_jid, self);
louiz’'s avatar
louiz’ committed
323 324
}

325 326
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
327 328
  std::string body;
  if (!author.empty())
329 330
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
331
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
332 333
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
334
    }
louiz’'s avatar
louiz’ committed
335 336
  else
    body = msg;
337
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid, "chat");
338 339
}

340 341
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
342
                            const IrcUser* user,
343
                            const char user_mode,
344
                            const bool self)
345
{
346 347 348 349
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

350 351
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
352 353 354 355
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
356
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
357
}
louiz’'s avatar
louiz’ committed
358 359 360

std::string Bridge::get_own_nick(const Iid& iid)
{
louiz’'s avatar
louiz’ committed
361
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
362 363 364 365
  if (irc)
    return irc->get_own_nick();
  return "";
}
366

367
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
368 369 370 371
{
  return this->irc_clients.size();
}

372 373
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
louiz’'s avatar
louiz’ committed
374
  this->xmpp->kick_user(std::to_string(iid), target, reason, author, this->user_jid);
375
}
376 377 378

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
louiz’'s avatar
louiz’ committed
379
  this->xmpp->send_nickname_conflict_error(std::to_string(iid), nickname, this->user_jid);
380
}
381 382 383 384 385

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
louiz’'s avatar
louiz’ committed
388
  this->xmpp->send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid);
389
}
390 391 392

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
louiz’'s avatar
louiz’ committed
393
  this->xmpp->send_iq_version_request(nick + "!" + hostname, this->user_jid);
394
}
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410

void Bridge::set_preferred_from_jid(const std::string& nick, const std::string& full_jid)
{
  auto it = this->preferred_user_from.find(nick);
  if (it == this->preferred_user_from.end())
    this->preferred_user_from.emplace(nick, full_jid);
  else
    this->preferred_user_from[nick] = full_jid;
}

void Bridge::remove_preferred_from_jid(const std::string& nick)
{
  auto it = this->preferred_user_from.find(nick);
  if (it != this->preferred_user_from.end())
    this->preferred_user_from.erase(it);
}
411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427

void Bridge::add_waiting_iq(iq_responder_callback_t&& callback)
{
  this->waiting_iq.emplace_back(std::move(callback));
}

void Bridge::trigger_response_iq(const std::string& irc_hostname, const IrcMessage& message)
{
  auto it = this->waiting_iq.begin();
  while (it != this->waiting_iq.end())
    {
      if ((*it)(irc_hostname, message) == true)
        it = this->waiting_iq.erase(it);
      else
        ++it;
    }
}