bridge.cpp 34.8 KB
Newer Older
1
#include <bridge/bridge.hpp>
2
#include <bridge/list_element.hpp>
3
#include <xmpp/biboumi_component.hpp>
4
#include <network/poller.hpp>
5
#include <utils/empty_if_fixed_server.hpp>
6
#include <utils/encoding.hpp>
7
#include <utils/tolower.hpp>
louiz’'s avatar
louiz’ committed
8
#include <logger/logger.hpp>
9
#include <utils/revstr.hpp>
louiz’'s avatar
louiz’ committed
10
#include <utils/split.hpp>
11
#include <xmpp/jid.hpp>
12
#include <database/database.hpp>
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 19 20 21 22 23 24 25 26 27 28 29

static std::string in_encoding_for(const Bridge& bridge, const Iid& iid)
{
#ifdef USE_DATABASE
  const auto jid = bridge.get_bare_jid();
  auto options = Database::get_irc_channel_options_with_server_default(jid, iid.get_server(), iid.get_local());
  return options.encodingIn.value();
#else
  return {"ISO-8859-1"};
#endif
}

30
Bridge::Bridge(const std::string& user_jid, BiboumiComponent& xmpp, std::shared_ptr<Poller> poller):
31 32 33 34 35 36
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

37
/**
38 39
 * Return the role and affiliation, corresponding to the given irc mode
 */
40 41
static std::tuple<std::string, std::string> get_role_affiliation_from_irc_mode(const char mode)
{
42 43
  if (mode == 'a' || mode == 'q'){
    return std::make_tuple("moderator", "owner");}
44 45 46 47 48 49 50 51 52 53
  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");
}

54
void Bridge::shutdown(const std::string& exit_message)
louiz’'s avatar
louiz’ committed
55 56 57
{
  for (auto it = this->irc_clients.begin(); it != this->irc_clients.end(); ++it)
  {
louiz’'s avatar
louiz’ committed
58 59
    it->second->send_quit_command(exit_message);
    it->second->leave_dummy_channel(exit_message);
louiz’'s avatar
louiz’ committed
60 61 62
  }
}

louiz’'s avatar
louiz’ committed
63 64 65 66 67 68
void Bridge::clean()
{
  auto it = this->irc_clients.begin();
  while (it != this->irc_clients.end())
  {
    IrcClient* client = it->second.get();
louiz’'s avatar
louiz’ committed
69 70
    if (!client->is_connected() && !client->is_connecting() &&
        !client->get_resolver().is_resolving())
louiz’'s avatar
louiz’ committed
71 72 73 74 75 76
      it = this->irc_clients.erase(it);
    else
      ++it;
  }
}

77 78 79 80 81
const std::string& Bridge::get_jid() const
{
  return this->user_jid;
}

louiz’'s avatar
louiz’ committed
82 83 84 85 86 87
std::string Bridge::get_bare_jid() const
{
  Jid jid(this->user_jid);
  return jid.local + "@" + jid.domain;
}

88
Xmpp::body Bridge::make_xmpp_body(const std::string& str, const std::string& encoding)
89 90 91 92 93
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
94
    res = utils::convert_to_utf8(str, encoding.data());
95
  return irc_format_to_xhtmlim(res);
96 97
}

98
IrcClient* Bridge::make_irc_client(const std::string& hostname, const std::string& nickname)
99 100 101 102 103 104 105
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
106 107
      auto username = nickname;
      auto realname = nickname;
louiz’'s avatar
louiz’ committed
108
      Jid jid(this->user_jid);
109 110 111 112 113 114 115 116
      if (Config::get("realname_from_jid", "false") == "true")
        {
          username = jid.local;
          realname = this->get_bare_jid();
        }
      this->irc_clients.emplace(hostname,
                                std::make_shared<IrcClient>(this->poller, hostname,
                                                            nickname, username,
louiz’'s avatar
louiz’ committed
117
                                                            realname, jid.domain,
118
                                                            *this));
119 120 121 122 123
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
124
IrcClient* Bridge::get_irc_client(const std::string& hostname)
125 126 127 128 129 130 131 132 133 134 135
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      throw IRCNotConnected(hostname);
    }
}

136
IrcClient* Bridge::find_irc_client(const std::string& hostname) const
louiz’'s avatar
louiz’ committed
137 138 139 140 141 142 143 144 145 146 147
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      return nullptr;
    }
}

148 149 150 151 152 153 154 155 156
bool Bridge::join_irc_channel(const Iid& iid, const std::string& nickname, const std::string& password,
                              const std::string& resource)
{
  const auto hostname = iid.get_server();
  IrcClient* irc = this->make_irc_client(hostname, nickname);
  this->add_resource_to_server(hostname, resource);
  auto res_in_chan = this->is_resource_in_chan(ChannelKey{iid.get_local(), hostname}, resource);
  if (!res_in_chan)
    this->add_resource_to_chan(ChannelKey{iid.get_local(), hostname}, resource);
louiz’'s avatar
louiz’ committed
157
  if (iid.get_local().empty())
louiz’'s avatar
louiz’ committed
158 159 160 161 162 163 164 165 166
    { // 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
167
          const IrcMessage end_join_message(std::string(iid.get_server()), "366",
louiz’'s avatar
louiz’ committed
168 169 170 171 172
                                            {irc->get_nick(),
                                                "", "End of NAMES list"});
          irc->on_channel_completely_joined(end_join_message);
        }
      else
173
        {
louiz’'s avatar
louiz’ committed
174
          irc->get_dummy_channel().joining = true;
175 176
          irc->start();
        }
louiz’'s avatar
louiz’ committed
177 178
      return true;
    }
louiz’'s avatar
louiz’ committed
179
  if (irc->is_channel_joined(iid.get_local()) == false)
louiz’'s avatar
louiz’ committed
180
    {
181
      irc->send_join_command(iid.get_local(), password);
louiz’'s avatar
louiz’ committed
182
      return true;
183 184
    } else if (!res_in_chan) {
      this->generate_channel_join_for_resource(iid, resource);
louiz’'s avatar
louiz’ committed
185 186
    }
  return false;
187 188
}

louiz’'s avatar
louiz’ committed
189 190
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
191
  if (iid.get_server().empty())
louiz’'s avatar
louiz’ committed
192
    {
193
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
194 195 196 197 198
        this->xmpp.send_stanza_error("message", this->user_jid + "/" + resource, std::to_string(iid), "",
                                     "cancel", "remote-server-not-found",
                                     std::to_string(iid) + " is not a valid channel name. "
                                         "A correct room jid is of the form: #<chan>%<server>",
                                     false);
louiz’'s avatar
louiz’ committed
199 200
      return;
    }
louiz’'s avatar
louiz’ committed
201
  IrcClient* irc = this->get_irc_client(iid.get_server());
202

203 204 205 206 207 208 209 210 211 212 213 214
  // 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
215
    {
louiz’'s avatar
louiz’ committed
216
      if (line.substr(0, 5) == "/mode")
217
        {
louiz’'s avatar
louiz’ committed
218
          std::vector<std::string> args = utils::split(line.substr(5), ' ', false);
louiz’'s avatar
louiz’ committed
219
          irc->send_mode_command(iid.get_local(), args);
220 221 222 223
          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
224
        irc->send_channel_message(iid.get_local(), action_prefix + line.substr(4) + "\01");
225
      else
louiz’'s avatar
louiz’ committed
226
        irc->send_channel_message(iid.get_local(), line);
227

228
#ifdef USE_DATABASE
229 230 231
      const auto xmpp_body = this->make_xmpp_body(line);
      Database::store_muc_message(this->get_bare_jid(), iid, std::chrono::system_clock::now(),
                                  std::get<0>(xmpp_body), irc->get_own_nick());
232
#endif
233
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
234 235
        this->xmpp.send_muc_message(std::to_string(iid), irc->get_own_nick(),
                                    this->make_xmpp_body(line), this->user_jid + "/" + resource);
louiz’'s avatar
louiz’ committed
236
    }
louiz’'s avatar
louiz’ committed
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 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302
void Bridge::forward_affiliation_role_change(const Iid& iid, const std::string& nick,
                                             const std::string& affiliation,
                                             const std::string& role)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  IrcChannel* chan = irc->get_channel(iid.get_local());
  if (!chan || !chan->joined)
    return;
  IrcUser* user = chan->find_user(nick);
  if (!user)
    return;
  // For each affiliation or role, we have a “maximal” mode that we want to
  // set. We must remove any superior mode at the same time. For example if
  // the user already has +o mode, and we set its affiliation to member, we
  // remove the +o mode, and add +v.  For each “superior” mode (for example,
  // for +v, the superior modes are 'h', 'a', 'o' and 'q') we check if that
  // user has it, and if yes we remove that mode

  std::size_t nb = 1;               // the number of times the nick must be
                                    // repeated in the argument list
  std::string modes;                // The string of modes to
                                    // add/remove. For example "+v-aoh"
  std::vector<char> modes_to_remove; // List of modes to check for removal
  if (affiliation == "none")
    {
      modes = "";
      nb = 0;
      modes_to_remove = {'v', 'h', 'o', 'a', 'q'};
    }
  else if (affiliation == "member")
    {
      modes = "+v";
      modes_to_remove = {'h', 'o', 'a', 'q'};
    }
  else if (role == "moderator")
    {
      modes = "+h";
      modes_to_remove = {'o', 'a', 'q'};
    }
  else if (affiliation == "admin")
    {
      modes = "+o";
      modes_to_remove = {'a', 'q'};
    }
  else if (affiliation == "owner")
    {
      modes = "+a";
      modes_to_remove = {'q'};
    }
  else
    return;
  for (const char mode: modes_to_remove)
    if (user->modes.find(mode) != user->modes.end())
      {
        modes += "-"s + mode;
        nb++;
      }
  if (modes.empty())
    return;
  std::vector<std::string> args(nb, nick);
  args.insert(args.begin(), modes);
  irc->send_mode_command(iid.get_local(), args);
}

louiz’'s avatar
louiz’ committed
303
void Bridge::send_private_message(const Iid& iid, const std::string& body, const std::string& type)
louiz’'s avatar
louiz’ committed
304
{
louiz’'s avatar
louiz’ committed
305
  if (iid.get_local().empty() || iid.get_server().empty())
306
    {
307
      this->xmpp.send_stanza_error("message", this->user_jid, std::to_string(iid), "",
308 309 310 311
                                    "cancel", "remote-server-not-found",
                                    std::to_string(iid) + " is not a valid channel name. "
                                    "A correct room jid is of the form: #<chan>%<server>",
                                    false);
312 313
      return;
    }
314
  IrcClient* irc = this->get_irc_client(iid.get_server());
315 316 317 318 319 320
  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
321
        irc->send_private_message(iid.get_local(), action_prefix + line.substr(4) + "\01", type);
322
      else
louiz’'s avatar
louiz’ committed
323
        irc->send_private_message(iid.get_local(), line, type);
324
    }
louiz’'s avatar
louiz’ committed
325 326
}

louiz’'s avatar
louiz’ committed
327 328 329 330 331 332
void Bridge::send_raw_message(const std::string& hostname, const std::string& body)
{
  IrcClient* irc = this->get_irc_client(hostname);
  irc->send_raw(body);
}

333
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message, const std::string& resource)
louiz’'s avatar
louiz’ committed
334
{
louiz’'s avatar
louiz’ committed
335
  IrcClient* irc = this->get_irc_client(iid.get_server());
336 337 338 339 340 341
  const auto key = iid.to_tuple();
  if (!this->is_resource_in_chan(key, resource))
    return ;

  const auto resources = this->number_of_resources_in_chan(key);
  if (resources == 1)
342 343 344 345 346 347
    {
      irc->send_part_command(iid.get_local(), status_message);
      // Since there are no resources left in that channel, we don't
      // want to receive private messages using this room's JID
      this->remove_all_preferred_from_jid_of_room(iid.get_local());
    }
348 349 350 351 352 353 354 355 356 357
  else
    {
      IrcChannel* chan = irc->get_channel(iid.get_local());
      if (chan)
        {
          auto nick = chan->get_self()->nick;
          this->remove_resource_from_chan(key, resource);
          this->send_muc_leave(std::move(iid), std::move(nick),
                               "Biboumi note: "s + std::to_string(resources - 1) + " resources are still in this channel.",
          true, resource);
358 359
          if (this->number_of_channels_the_resource_is_in(iid.get_server(), resource) == 0)
            this->remove_resource_from_server(iid.get_server(), resource);
360 361
        }
    }
362
}
louiz’'s avatar
louiz’ committed
363

louiz’'s avatar
louiz’ committed
364 365
void Bridge::send_irc_nick_change(const Iid& iid, const std::string& new_nick)
{
louiz’'s avatar
louiz’ committed
366
  IrcClient* irc = this->get_irc_client(iid.get_server());
367
  irc->send_nick_command(new_nick);
louiz’'s avatar
louiz’ committed
368 369
}

370 371 372 373 374 375
void Bridge::send_irc_channel_list_request(const Iid& iid, const std::string& iq_id,
                                           const std::string& to_jid)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());

  irc->send_list_command();
376

377
  std::vector<ListElement> list;
378

379 380 381
  irc_responder_callback_t cb = [this, iid, iq_id, to_jid, list=std::move(list)](const std::string& irc_hostname,
                                                           const IrcMessage& message) mutable -> bool
    {
382 383
      if (irc_hostname != iid.get_server())
        return false;
384 385 386 387 388 389
      if (message.command == "263" || message.command == "RPL_TRYAGAIN" ||
          message.command == "ERR_TOOMANYMATCHES" || message.command == "ERR_NOSUCHSERVER")
        {
          std::string text;
          if (message.arguments.size() >= 2)
            text = message.arguments[1];
390
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id,
391
                                        "wait", "service-unavailable", text, false);
392 393 394 395 396 397 398 399 400 401 402
          return true;
        }
      else if (message.command == "322" || message.command == "RPL_LIST")
        { // Add element to list
          if (message.arguments.size() == 4)
            list.emplace_back(message.arguments[1], message.arguments[2],
                              message.arguments[3]);
          return false;
        }
      else if (message.command == "323" || message.command == "RPL_LISTEND")
        { // Send the iq response with the content of the list
403
          this->xmpp.send_iq_room_list_result(iq_id, to_jid, std::to_string(iid), list);
404 405 406 407 408 409 410
          return true;
        }
      return false;
    };
  this->add_waiting_irc(std::move(cb));
}

411 412
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)
413
{
louiz’'s avatar
louiz’ committed
414
  IrcClient* irc = this->get_irc_client(iid.get_server());
415 416 417 418

  irc->send_kick_command(iid.get_local(), target, reason);
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, iid](const std::string& irc_hostname,
                                                                   const IrcMessage& message) -> bool
419
    {
420 421 422 423 424 425 426
      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())
427
            return false;
428
          this->xmpp.send_iq_result(iq_id, to_jid, std::to_string(iid));
429 430 431 432 433 434 435 436 437
        }
      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];
438
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "item-not-found",
439 440 441 442 443 444 445 446 447 448
                                        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];
449
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "not-allowed",
450 451 452 453 454
                                        error_message, false);
        }
      return true;
    };
  this->add_waiting_irc(std::move(cb));
455 456
}

457 458
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
louiz’'s avatar
louiz’ committed
459
  IrcClient* irc = this->get_irc_client(iid.get_server());
460
  irc->send_topic_command(iid.get_local(), subject);
461 462
}

louiz’'s avatar
louiz’ committed
463 464 465 466 467 468 469
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");
}

470 471
void Bridge::send_irc_ping_result(const Iid& iid, const std::string& id)
{
472
  this->send_private_message(iid, "\01PING "s + utils::revstr(id) + "\01", "NOTICE");
473 474 475 476 477 478
}

void Bridge::send_irc_user_ping_request(const std::string& irc_hostname, const std::string& nick,
                                        const std::string& iq_id, const std::string& to_jid,
                                        const std::string& from_jid)
{
479
  Iid iid(nick, irc_hostname, Iid::Type::User);
480 481
  this->send_private_message(iid, "\01PING " + iq_id + "\01");

482
  irc_responder_callback_t cb = [this, nick=utils::tolower(nick), iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
483
    {
louiz’'s avatar
louiz’ committed
484
      if (irc_hostname != hostname || message.arguments.size() < 2)
485 486 487
        return false;
      IrcUser user(message.prefix);
      const std::string body = message.arguments[1];
louiz’'s avatar
louiz’ committed
488 489
      if (message.command == "NOTICE" && utils::tolower(user.nick) == nick
          && body.substr(0, 6) == "\01PING ")
490
        {
491
          const std::string id = body.substr(6, body.size() - 7);
492 493
          if (id != iq_id)
            return false;
494
          this->xmpp.send_iq_result_full_jid(iq_id, to_jid, from_jid);
495 496
          return true;
        }
louiz’'s avatar
louiz’ committed
497
      if (message.command == "401" && message.arguments[1] == nick)
498 499 500 501
        {
          std::string error_message = "No such nick";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
502
          this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "service-unavailable",
503 504 505 506 507 508 509 510 511
                                        error_message, true);
          return true;
        }

      return false;
    };
  this->add_waiting_irc(std::move(cb));
}

512 513 514 515 516 517 518 519
void Bridge::send_irc_participant_ping_request(const Iid& iid, const std::string& nick,
                                               const std::string& iq_id, const std::string& to_jid,
                                               const std::string& from_jid)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  IrcChannel* chan = irc->get_channel(iid.get_local());
  if (!chan->joined)
    {
520
      this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "not-allowed",
521 522 523 524 525
                                    "", true);
      return;
    }
  if (chan->get_self()->nick != nick && !chan->find_user(nick))
    {
526
      this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
527 528 529 530 531 532 533 534
                                    "Recipient not in room", true);
      return;
    }

  // The user is in the room, send it a direct PING
  this->send_irc_user_ping_request(iid.get_server(), nick, iq_id, to_jid, from_jid);
}

535 536 537 538
void Bridge::on_gateway_ping(const std::string& irc_hostname, const std::string& iq_id, const std::string& to_jid,
                             const std::string& from_jid)
{
  Jid jid(from_jid);
539
  if (irc_hostname.empty() || this->find_irc_client(irc_hostname))
540
    this->xmpp.send_iq_result(iq_id, to_jid, jid.local);
541
  else
542
    this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "service-unavailable",
543 544 545
                                  "", true);
}

546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561
void Bridge::send_irc_invitation(const Iid& iid, const std::string to)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  Jid to_jid(to);
  std::string target_nick;
  // Many ways to address a nick:
  // A jid (ANY jid…) with a resource
  if (!to_jid.resource.empty())
    target_nick = to_jid.resource;
  else if (!to_jid.local.empty()) // A jid with a iid with a local part
    target_nick = Iid(to_jid.local, {}).get_local();
  else
    target_nick = to; // Not a jid, just the nick
  irc->send_invitation(iid.get_local(), target_nick);
}

562 563 564 565
void Bridge::send_irc_version_request(const std::string& irc_hostname, const std::string& target,
                                      const std::string& iq_id, const std::string& to_jid,
                                      const std::string& from_jid)
{
566
  Iid iid(target, irc_hostname, Iid::Type::User);
567 568 569
  this->send_private_message(iid, "\01VERSION\01");
  // TODO, add a timer to remove that waiting iq if the server does not
  // respond with a matching command before n seconds
570 571
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
    {
572 573 574 575 576 577 578 579
      if (irc_hostname != hostname)
        return false;
      IrcUser user(message.prefix);
      if (message.command == "NOTICE" && user.nick == target &&
          message.arguments.size() >= 2 && message.arguments[1].substr(0, 9) == "\01VERSION ")
        {
          // remove the "\01VERSION " and the "\01" parts from the string
          const std::string version = message.arguments[1].substr(9, message.arguments[1].size() - 10);
580
          this->xmpp.send_version(iq_id, to_jid, from_jid, version);
581 582 583 584 585 586 587 588
          return true;
        }
      if (message.command == "401" && message.arguments.size() >= 2
          && message.arguments[1] == target)
        {
          std::string error_message = "No such nick";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
589
          this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
590 591 592 593
                                        error_message, true);
          return true;
        }
      return false;
594 595
    };
  this->add_waiting_irc(std::move(cb));
596 597
}

louiz’'s avatar
louiz’ committed
598
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
599
{
600
  const auto encoding = in_encoding_for(*this, iid);
louiz’'s avatar
louiz’ committed
601
  if (muc)
602
    {
603
#ifdef USE_DATABASE
604 605 606
      const auto xmpp_body = this->make_xmpp_body(body, encoding);
      Database::store_muc_message(this->get_bare_jid(), iid, std::chrono::system_clock::now(),
                                  std::get<0>(xmpp_body), nick);
607
#endif
608
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
609 610 611
        {
          this->xmpp.send_muc_message(std::to_string(iid), nick,
                                      this->make_xmpp_body(body, encoding), this->user_jid + "/" + resource);
612

613 614
        }
    }
louiz’'s avatar
louiz’ committed
615
  else
616 617
    {
      std::string target = std::to_string(iid);
618
      const auto it = this->preferred_user_from.find(iid.get_local());
619 620
      if (it != this->preferred_user_from.end())
        {
621
          const auto chan_name = Iid(Jid(it->second).local, {}).get_local();
622 623 624
          for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, iid.get_server()}])
            this->xmpp.send_message(it->second, this->make_xmpp_body(body, encoding),
                                    this->user_jid + "/" + resource, "chat", true);
625
        }
626
      else
627
        {
628 629 630
          for (const auto& resource: this->resources_in_server[iid.get_server()])
            this->xmpp.send_message(std::to_string(iid), this->make_xmpp_body(body, encoding),
                                    this->user_jid + "/" + resource, "chat", false);
631
        }
632
    }
louiz’'s avatar
louiz’ committed
633 634
}

635 636 637
void Bridge::send_presence_error(const Iid& iid, const std::string& nick,
                                 const std::string& type, const std::string& condition,
                                 const std::string& error_code, const std::string& text)
638
{
639
  this->xmpp.send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, error_code, text);
640 641
}

642
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self, const std::string& resource)
louiz’'s avatar
louiz’ committed
643
{
644 645 646 647 648 649 650
  if (!resource.empty())
    this->xmpp.send_muc_leave(std::to_string(iid), std::move(nick), this->make_xmpp_body(message), this->user_jid + "/" + resource,
                              self);
  else
    for (const auto& res: this->resources_in_chan[iid.to_tuple()])
      this->xmpp.send_muc_leave(std::to_string(iid), std::move(nick), this->make_xmpp_body(message), this->user_jid + "/" + res,
                                self);
651
  IrcClient* irc = this->find_irc_client(iid.get_server());
652 653
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
654 655
}

656 657 658 659 660
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
661
{
662 663 664 665
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

666
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
667 668
    this->xmpp.send_nick_change(std::to_string(iid),
                                old_nick, new_nick, affiliation, role, this->user_jid + "/" + resource, self);
louiz’'s avatar
louiz’ committed
669 670
}

671 672
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
673 674
  std::string body;
  if (!author.empty())
675 676
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
677
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
678 679
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
680
    }
louiz’'s avatar
louiz’ committed
681 682
  else
    body = msg;
683

684
  const auto encoding = in_encoding_for(*this, {from, this});
685 686 687 688
  for (const auto& resource: this->resources_in_server[from])
    {
        this->xmpp.send_message(from, this->make_xmpp_body(body, encoding), this->user_jid + "/" + resource, "chat");
    }
689 690
}

691 692 693 694 695 696 697 698 699 700
void Bridge::send_user_join(const std::string& hostname, const std::string& chan_name,
                            const IrcUser* user, const char user_mode, const bool self)
{
  for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, hostname}])
    this->send_user_join(hostname, chan_name, user, user_mode, self, resource);
}

void Bridge::send_user_join(const std::string& hostname, const std::string& chan_name,
                       const IrcUser* user, const char user_mode,
                       const bool self, const std::string& resource)
701
{
702 703 704 705
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

706 707 708 709
  std::string encoded_chan_name(chan_name);
  xep0106::encode(encoded_chan_name);

  this->xmpp.send_user_join(encoded_chan_name + utils::empty_if_fixed_server("%" + hostname), user->nick, user->host,
710
                            affiliation, role, this->user_jid + "/" + resource, self);
711 712
}

713
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string& topic, const std::string& who)
714
{
715 716
  for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, hostname}])
    {
717
      this->send_topic(hostname, chan_name, topic, who, resource);
718
    }
719
}
louiz’'s avatar
louiz’ committed
720

721 722 723 724
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name,
                        const std::string& topic, const std::string& who,
                        const std::string& resource)
{
725 726
  std::string encoded_chan_name(chan_name);
  xep0106::encode(encoded_chan_name);
727
  const auto encoding = in_encoding_for(*this, {encoded_chan_name, hostname, Iid::Type::Channel});
728
  this->xmpp.send_topic(encoded_chan_name + utils::empty_if_fixed_server(
729 730 731 732
      "%" + hostname), this->make_xmpp_body(topic, encoding), this->user_jid + "/" + resource, who);

}

louiz’'s avatar
louiz’ committed
733 734
std::string Bridge::get_own_nick(const Iid& iid)
{
735
  IrcClient* irc = this->find_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
736 737 738 739
  if (irc)
    return irc->get_own_nick();
  return "";
}
740

741
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
742 743 744 745
{
  return this->irc_clients.size();
}

746 747
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
748
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
749
    this->xmpp.kick_user(std::to_string(iid), target, reason, author, this->user_jid + "/" + resource);
750
}
751 752 753

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
754
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
755
    this->xmpp.send_presence_error(std::to_string(iid), nickname, this->user_jid + "/" + resource, "cancel", "conflict", "409", "");
756
}
757 758 759 760 761

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
764
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
765
    this->xmpp.send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid + "/" + resource);
766
}
767 768 769

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
770 771
  const auto resources = this->resources_in_server[hostname];
  if (resources.begin() != resources.end())
772
    this->xmpp.send_iq_version_request(utils::tolower(nick) + "%" + utils::empty_if_fixed_server(hostname), this->user_jid + "/" + *resources.begin());
773
}
774

775 776 777 778 779 780
void Bridge::send_xmpp_ping_request(const std::string& nick, const std::string& hostname,
                                    const std::string& id)
{
  // Use revstr because the forwarded ping to target XMPP user must not be
  // the same that the request iq, but we also need to get it back easily
  // (revstr again)
781 782 783
  // Forward to the first resource (arbitrary, based on the “order” of the std::set) only
  const auto resources = this->resources_in_server[hostname];
  if (resources.begin() != resources.end())
784
    this->xmpp.send_ping_request(utils::tolower(nick) + "%" + utils::empty_if_fixed_server(hostname), this->user_jid + "/" + *resources.begin(), utils::revstr(id));
785 786
}

787 788 789 790 791 792 793 794 795 796 797 798 799 800 801
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);
}
802

803 804 805 806
void Bridge::remove_all_preferred_from_jid_of_room(const std::string& channel_name)
{
  for (auto it = this->preferred_user_from.begin(); it != this->preferred_user_from.end();)
    {
807
      Iid iid(Jid(it->second).local, {});
808 809 810 811 812 813 814
      if (iid.get_local() == channel_name)
        it = this->preferred_user_from.erase(it);
      else
        ++it;
    }
}

815
void Bridge::add_waiting_irc(irc_responder_callback_t&& callback)
816
{
817
  this->waiting_irc.emplace_back(std::move(callback));
818 819
}

820
void Bridge::trigger_on_irc_message(const std::string& irc_hostname, const IrcMessage& message)
821
{
822 823
  auto it = this->waiting_irc.begin();
  while (it != this->waiting_irc.end())
824 825
    {
      if ((*it)(irc_hostname, message) == true)
826
        it = this->waiting_irc.erase(it);
827 828 829 830
      else
        ++it;
    }
}
831 832 833 834 835

std::unordered_map<std::string, std::shared_ptr<IrcClient>>& Bridge::get_irc_clients()
{
  return this->irc_clients;
}
836

837 838 839 840 841 842 843 844
std::set<char> Bridge::get_chantypes(const std::string& hostname) const
{
  IrcClient* irc = this->find_irc_client(hostname);
  if (!irc)
    return {'#', '&'};
  return irc->get_chantypes();
}

845
void Bridge::add_resource_to_chan(const Bridge::ChannelKey& channel, const std::string& resource)
846 847 848 849 850 851 852 853
{
  auto it = this->resources_in_chan.find(channel);
  if (it == this->resources_in_chan.end())
    this->resources_in_chan[channel] = {resource};
  else
    it->second.insert(resource);
}

854
void Bridge::remove_resource_from_chan(const Bridge::ChannelKey& channel, const std::string& resource)
855 856 857 858 859 860 861 862 863 864
{
  auto it = this->resources_in_chan.find(channel);
  if (it != this->resources_in_chan.end())
    {
      it->second.erase(resource);
      if (it->second.empty())
        this->resources_in_chan.erase(it);
    }
}

865
bool Bridge::is_resource_in_chan(const Bridge::ChannelKey& channel, const std::string& resource) const
866 867 868 869 870 871 872
{
  auto it = this->resources_in_chan.find(channel);
  if (it != this->resources_in_chan.end())
    if (it->second.count(resource) == 1)
      return true;
  return false;
}
873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902

void Bridge::add_resource_to_server(const Bridge::IrcHostname& irc_hostname, const std::string& resource)
{
  auto it = this->resources_in_server.find(irc_hostname);
  if (it == this->resources_in_server.end())
    this->resources_in_server[irc_hostname] = {resource};
  else
    it->second.insert(resource);
}

void Bridge::remove_resource_from_server(const Bridge::IrcHostname& irc_hostname, const std::string& resource)
{
  auto it = this->resources_in_server.find(irc_hostname);
  if (it != this->resources_in_server.end())
    {
      it->second.erase(resource);
      if (it->second.empty())
        this->resources_in_server.erase(it);
    }
}

bool Bridge::is_resource_in_server(const Bridge::IrcHostname& irc_hostname, const std::string& resource) const
{
  auto it = this->resources_in_server.find(irc_hostname);
  if (it != this->resources_in_server.end())
    if (it->second.count(resource) == 1)
      return true;
  return false;
}

903 904 905 906 907 908 909 910
std::size_t Bridge::number_of_resources_in_chan(const Bridge::ChannelKey& channel_key) const
{
  auto it = this->resources_in_chan.find(channel_key);
  if (it == this->resources_in_chan.end())
    return 0;
  return it->second.size();
}

911 912 913 914 915 916 917 918 919 920 921
std::size_t Bridge::number_of_channels_the_resource_is_in(const std::string& irc_hostname, const std::string& resource) const
{
  std::size_t res = 0;
  for (auto pair: this->resources_in_chan)
    {
      if (std::get<0>(pair.first) == irc_hostname && pair.second.count(resource) != 0)
        res++;
    }
  return res;
}

922 923 924 925
void Bridge::generate_channel_join_for_resource(const Iid& iid, const std::string& resource)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  IrcChannel* channel = irc->get_channel(iid.get_local());
926 927
  const auto self = channel->get_self();

928 929 930
  // Send the occupant list
  for (const auto& user: channel->get_users())
    {
931 932 933
      if (user->nick != self->nick)
        {
          log_debug(user->nick);
934
          this->send_user_join(iid.get_server(), iid.get_encoded_local(),
935 936 937
                               user.get(), user->get_most_significant_mode(irc->get_sorted_user_modes()),
                               false, resource);
        }
938
    }
939
  this->send_user_join(iid.get_server(), iid.get_encoded_local(),
940 941
                       self, self->get_most_significant_mode(irc->get_sorted_user_modes()),
                       true, resource);
942
  this->send_topic(iid.get_server(), iid.get_encoded_local(), channel->topic, channel->topic_author, resource);
943
}