bridge.cpp 32.7 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 30 31 32 33 34 35 36 37 38 39 40

static std::string out_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.encodingOut.value();
#else
  return {"ISO-8859-1"};
#endif
}

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
}

41
Bridge::Bridge(const std::string& user_jid, BiboumiComponent& xmpp, std::shared_ptr<Poller> poller):
42 43 44 45 46 47
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

48
/**
49 50
 * Return the role and affiliation, corresponding to the given irc mode
 */
51 52
static std::tuple<std::string, std::string> get_role_affiliation_from_irc_mode(const char mode)
{
53 54
  if (mode == 'a' || mode == 'q'){
    return std::make_tuple("moderator", "owner");}
55 56 57 58 59 60 61 62 63 64
  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");
}

65
void Bridge::shutdown(const std::string& exit_message)
louiz’'s avatar
louiz’ committed
66 67 68
{
  for (auto it = this->irc_clients.begin(); it != this->irc_clients.end(); ++it)
  {
louiz’'s avatar
louiz’ committed
69 70
    it->second->send_quit_command(exit_message);
    it->second->leave_dummy_channel(exit_message);
louiz’'s avatar
louiz’ committed
71 72 73
  }
}

louiz’'s avatar
louiz’ committed
74 75 76 77 78 79
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
80 81
    if (!client->is_connected() && !client->is_connecting() &&
        !client->get_resolver().is_resolving())
louiz’'s avatar
louiz’ committed
82 83 84 85 86 87
      it = this->irc_clients.erase(it);
    else
      ++it;
  }
}

88 89 90 91 92
const std::string& Bridge::get_jid() const
{
  return this->user_jid;
}

louiz’'s avatar
louiz’ committed
93 94 95 96 97 98
std::string Bridge::get_bare_jid() const
{
  Jid jid(this->user_jid);
  return jid.local + "@" + jid.domain;
}

99
Xmpp::body Bridge::make_xmpp_body(const std::string& str, const std::string& encoding)
100 101 102 103 104
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
105
    res = utils::convert_to_utf8(str, encoding.data());
106
  return irc_format_to_xhtmlim(res);
107 108
}

109
IrcClient* Bridge::make_irc_client(const std::string& hostname, const std::string& nickname)
110 111 112 113 114 115 116
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
117 118
      auto username = nickname;
      auto realname = nickname;
louiz’'s avatar
louiz’ committed
119
      Jid jid(this->user_jid);
120 121 122 123 124 125 126 127
      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
128
                                                            realname, jid.domain,
129
                                                            *this));
130 131 132 133 134
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
135
IrcClient* Bridge::get_irc_client(const std::string& hostname)
136 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)
    {
      throw IRCNotConnected(hostname);
    }
}

IrcClient* Bridge::find_irc_client(const std::string& hostname)
louiz’'s avatar
louiz’ committed
148 149 150 151 152 153 154 155 156 157 158
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      return nullptr;
    }
}

159 160 161 162 163 164 165 166 167
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
168
  if (iid.get_local().empty())
louiz’'s avatar
louiz’ committed
169 170 171 172 173 174 175 176 177
    { // 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
178
          const IrcMessage end_join_message(std::string(iid.get_server()), "366",
louiz’'s avatar
louiz’ committed
179 180 181 182 183
                                            {irc->get_nick(),
                                                "", "End of NAMES list"});
          irc->on_channel_completely_joined(end_join_message);
        }
      else
184
        {
louiz’'s avatar
louiz’ committed
185
          irc->get_dummy_channel().joining = true;
186 187
          irc->start();
        }
louiz’'s avatar
louiz’ committed
188 189
      return true;
    }
louiz’'s avatar
louiz’ committed
190
  if (irc->is_channel_joined(iid.get_local()) == false)
louiz’'s avatar
louiz’ committed
191
    {
192
      irc->send_join_command(iid.get_local(), password);
louiz’'s avatar
louiz’ committed
193
      return true;
194 195
    } else if (!res_in_chan) {
      this->generate_channel_join_for_resource(iid, resource);
louiz’'s avatar
louiz’ committed
196 197
    }
  return false;
198 199
}

louiz’'s avatar
louiz’ committed
200 201
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
202
  if (iid.get_server().empty())
louiz’'s avatar
louiz’ committed
203
    {
204
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
205 206 207 208 209
        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
210 211
      return;
    }
louiz’'s avatar
louiz’ committed
212
  IrcClient* irc = this->get_irc_client(iid.get_server());
213

214 215 216 217 218 219 220 221 222 223 224 225
  // 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
226
    {
louiz’'s avatar
louiz’ committed
227
      if (line.substr(0, 5) == "/mode")
228
        {
louiz’'s avatar
louiz’ committed
229
          std::vector<std::string> args = utils::split(line.substr(5), ' ', false);
louiz’'s avatar
louiz’ committed
230
          irc->send_mode_command(iid.get_local(), args);
231 232 233 234
          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
235
        irc->send_channel_message(iid.get_local(), action_prefix + line.substr(4) + "\01");
236
      else
louiz’'s avatar
louiz’ committed
237
        irc->send_channel_message(iid.get_local(), line);
238
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
239 240
        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
241
    }
louiz’'s avatar
louiz’ committed
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 303 304 305 306 307
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
308
void Bridge::send_private_message(const Iid& iid, const std::string& body, const std::string& type)
louiz’'s avatar
louiz’ committed
309
{
louiz’'s avatar
louiz’ committed
310
  if (iid.get_local().empty() || iid.get_server().empty())
311
    {
312
      this->xmpp.send_stanza_error("message", this->user_jid, std::to_string(iid), "",
313 314 315 316
                                    "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);
317 318
      return;
    }
319
  IrcClient* irc = this->get_irc_client(iid.get_server());
320 321 322 323 324 325
  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
326
        irc->send_private_message(iid.get_local(), action_prefix + line.substr(4) + "\01", type);
327
      else
louiz’'s avatar
louiz’ committed
328
        irc->send_private_message(iid.get_local(), line, type);
329
    }
louiz’'s avatar
louiz’ committed
330 331
}

louiz’'s avatar
louiz’ committed
332 333 334 335 336 337
void Bridge::send_raw_message(const std::string& hostname, const std::string& body)
{
  IrcClient* irc = this->get_irc_client(hostname);
  irc->send_raw(body);
}

338
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message, const std::string& resource)
louiz’'s avatar
louiz’ committed
339
{
louiz’'s avatar
louiz’ committed
340
  IrcClient* irc = this->get_irc_client(iid.get_server());
341 342 343 344 345 346
  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)
347 348 349 350 351 352
    {
      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());
    }
353 354 355 356 357 358 359 360 361 362 363 364 365
  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);
        }
    }
  }
louiz’'s avatar
louiz’ committed
366

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

373 374 375 376 377 378
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();
379

380 381 382 383 384 385 386
  irc_responder_callback_t cb = [this, iid, iq_id, to_jid](const std::string& irc_hostname,
                                                           const IrcMessage& message) -> bool
    {
      static std::vector<ListElement> list;

      if (irc_hostname != iid.get_server())
        return false;
387 388 389 390 391 392
      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];
393
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id,
394
                                        "wait", "service-unavailable", text, false);
395 396 397 398 399 400 401 402 403 404 405
          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
406
          this->xmpp.send_iq_room_list_result(iq_id, to_jid, std::to_string(iid), list);
407 408 409 410 411 412 413
          return true;
        }
      return false;
    };
  this->add_waiting_irc(std::move(cb));
}

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

  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
422
    {
423 424 425 426 427 428 429
      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())
430
            return false;
431
          this->xmpp.send_iq_result(iq_id, to_jid, std::to_string(iid));
432 433 434 435 436 437 438 439 440
        }
      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];
441
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "item-not-found",
442 443 444 445 446 447 448 449 450 451
                                        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];
452
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "not-allowed",
453 454 455 456 457
                                        error_message, false);
        }
      return true;
    };
  this->add_waiting_irc(std::move(cb));
458 459
}

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

louiz’'s avatar
louiz’ committed
466 467 468 469 470 471 472
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");
}

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

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)
{
  Iid iid(nick + "!" + irc_hostname);
  this->send_private_message(iid, "\01PING " + iq_id + "\01");

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

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

516 517 518 519 520 521 522 523
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)
    {
524
      this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "not-allowed",
525 526 527 528 529
                                    "", true);
      return;
    }
  if (chan->get_self()->nick != nick && !chan->find_user(nick))
    {
530
      this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
531 532 533 534 535 536 537 538
                                    "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);
}

539 540 541 542
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);
543
  if (irc_hostname.empty() || this->find_irc_client(irc_hostname))
544
    this->xmpp.send_iq_result(iq_id, to_jid, jid.local);
545
  else
546
    this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "service-unavailable",
547 548 549
                                  "", true);
}

550 551 552 553 554 555 556 557
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)
{
  Iid iid(target + "!" + irc_hostname);
  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
558 559
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
    {
560 561 562 563 564 565 566 567
      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);
568
          this->xmpp.send_version(iq_id, to_jid, from_jid, version);
569 570 571 572 573 574 575 576
          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];
577
          this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
578 579 580 581
                                        error_message, true);
          return true;
        }
      return false;
582 583
    };
  this->add_waiting_irc(std::move(cb));
584 585
}

louiz’'s avatar
louiz’ committed
586
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
587
{
588
  const auto encoding = in_encoding_for(*this, iid);
louiz’'s avatar
louiz’ committed
589
  if (muc)
590
    {
591
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
592 593 594 595 596
        {
          this->xmpp.send_muc_message(std::to_string(iid), nick,
                                      this->make_xmpp_body(body, encoding), this->user_jid + "/" + resource);
        }
    }
louiz’'s avatar
louiz’ committed
597
  else
598 599
    {
      std::string target = std::to_string(iid);
600
      const auto it = this->preferred_user_from.find(iid.get_local());
601 602
      if (it != this->preferred_user_from.end())
        {
603 604 605 606
          const auto chan_name = Iid(Jid(it->second).local).get_local();
          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);
607
        }
608
      else
609
        {
610 611 612
          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);
613
        }
614
    }
louiz’'s avatar
louiz’ committed
615 616
}

617 618 619
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)
620
{
621
  this->xmpp.send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, error_code, text);
622 623
}

624
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
625
{
626 627 628 629 630 631 632
  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);
633
  IrcClient* irc = this->find_irc_client(iid.get_server());
634 635
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
636 637
}

638 639 640 641 642
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
643
{
644 645 646 647
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

648
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
649 650
    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
651 652
}

653 654
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
655 656
  std::string body;
  if (!author.empty())
657 658
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
659
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
660 661
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
662
    }
louiz’'s avatar
louiz’ committed
663 664
  else
    body = msg;
665 666

  const auto encoding = in_encoding_for(*this, {from});
667 668 669 670
  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");
    }
671 672
}

673 674 675 676 677 678 679 680 681 682
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)
683
{
684 685 686 687
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

688 689
  this->xmpp.send_user_join(chan_name + utils::empty_if_fixed_server("%" + hostname), user->nick, user->host,
                            affiliation, role, this->user_jid + "/" + resource, self);
690 691
}

692
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string& topic, const std::string& who)
693
{
694 695
  for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, hostname}])
    {
696
      this->send_topic(hostname, chan_name, topic, who, resource);
697
    }
698
}
louiz’'s avatar
louiz’ committed
699

700 701 702 703 704 705 706 707 708 709
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)
{
  const auto encoding = in_encoding_for(*this, {chan_name + '%' + hostname});
  this->xmpp.send_topic(chan_name + utils::empty_if_fixed_server(
      "%" + hostname), this->make_xmpp_body(topic, encoding), this->user_jid + "/" + resource, who);

}

louiz’'s avatar
louiz’ committed
710 711
std::string Bridge::get_own_nick(const Iid& iid)
{
712
  IrcClient* irc = this->find_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
713 714 715 716
  if (irc)
    return irc->get_own_nick();
  return "";
}
717

718
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
719 720 721 722
{
  return this->irc_clients.size();
}

723 724
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
725
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
726
    this->xmpp.kick_user(std::to_string(iid), target, reason, author, this->user_jid + "/" + resource);
727
}
728 729 730

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
731
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
732
    this->xmpp.send_presence_error(std::to_string(iid), nickname, this->user_jid + "/" + resource, "cancel", "conflict", "409", "");
733
}
734 735 736 737 738

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
741
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
742
    this->xmpp.send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid + "/" + resource);
743
}
744 745 746

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
747
  this->xmpp.send_iq_version_request(nick + "!" + utils::empty_if_fixed_server(hostname), this->user_jid);
748
}
749

750 751 752 753 754 755
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)
756
  this->xmpp.send_ping_request(nick + "!" + utils::empty_if_fixed_server(hostname), this->user_jid, utils::revstr(id));
757 758
}

759 760 761 762 763 764 765 766 767 768 769 770 771 772 773
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);
}
774

775 776 777 778 779 780 781 782 783 784 785 786
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();)
    {
      Iid iid(Jid(it->second).local);
      if (iid.get_local() == channel_name)
        it = this->preferred_user_from.erase(it);
      else
        ++it;
    }
}

787
void Bridge::add_waiting_irc(irc_responder_callback_t&& callback)
788
{
789
  this->waiting_irc.emplace_back(std::move(callback));
790 791
}

792
void Bridge::trigger_on_irc_message(const std::string& irc_hostname, const IrcMessage& message)
793
{
794 795
  auto it = this->waiting_irc.begin();
  while (it != this->waiting_irc.end())
796 797
    {
      if ((*it)(irc_hostname, message) == true)
798
        it = this->waiting_irc.erase(it);
799 800 801 802
      else
        ++it;
    }
}
803 804 805 806 807

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

809
void Bridge::add_resource_to_chan(const Bridge::ChannelKey& channel, const std::string& resource)
810 811 812 813 814 815 816 817
{
  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);
}

818
void Bridge::remove_resource_from_chan(const Bridge::ChannelKey& channel, const std::string& resource)
819 820 821 822 823 824 825 826 827 828
{
  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);
    }
}

829
bool Bridge::is_resource_in_chan(const Bridge::ChannelKey& channel, const std::string& resource) const
830 831 832 833 834 835 836
{
  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;
}
837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866

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;
}

867 868 869 870 871 872 873 874
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();
}

875 876 877 878
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());
879 880
  const auto self = channel->get_self();

881 882 883
  // Send the occupant list
  for (const auto& user: channel->get_users())
    {
884 885 886 887 888 889 890
      if (user->nick != self->nick)
        {
          log_debug(user->nick);
          this->send_user_join(iid.get_server(), iid.get_local(),
                               user.get(), user->get_most_significant_mode(irc->get_sorted_user_modes()),
                               false, resource);
        }
891
    }
892 893 894 895
  this->send_user_join(iid.get_server(), iid.get_local(),
                       self, self->get_most_significant_mode(irc->get_sorted_user_modes()),
                       true, resource);
  this->send_topic(iid.get_server(), iid.get_local(), channel->topic, channel->topic_author, resource);
896
}