bridge.cpp 10.8 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>
louiz’'s avatar
louiz’ committed
7
#include <logger/logger.hpp>
louiz’'s avatar
louiz’ committed
8
#include <utils/split.hpp>
louiz’'s avatar
louiz’ committed
9
#include <stdexcept>
louiz’'s avatar
louiz’ committed
10
#include <iostream>
11
#include <tuple>
louiz’'s avatar
louiz’ committed
12

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

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

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

Bridge::~Bridge()
{
}

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

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

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

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

76 77 78 79 80 81 82 83
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)
    {
84
      this->irc_clients.emplace(hostname, std::make_shared<IrcClient>(this->poller, hostname, username, this));
85 86 87 88 89
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
90 91 92 93 94 95 96 97 98 99 100 101
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
102
bool Bridge::join_irc_channel(const Iid& iid, const std::string& username)
103 104
{
  IrcClient* irc = this->get_irc_client(iid.server, username);
louiz’'s avatar
louiz’ committed
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120
  if (iid.chan.empty())
    { // 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);
          const IrcMessage end_join_message(std::string(iid.server), "366",
                                            {irc->get_nick(),
                                                "", "End of NAMES list"});
          irc->on_channel_completely_joined(end_join_message);
        }
      else
121
        {
louiz’'s avatar
louiz’ committed
122
          irc->get_dummy_channel().joining = true;
123 124
          irc->start();
        }
louiz’'s avatar
louiz’ committed
125 126
      return true;
    }
louiz’'s avatar
louiz’ committed
127 128 129 130 131 132
  if (irc->is_channel_joined(iid.chan) == false)
    {
      irc->send_join_command(iid.chan);
      return true;
    }
  return false;
133 134
}

louiz’'s avatar
louiz’ committed
135 136 137 138
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
  if (iid.chan.empty() || iid.server.empty())
    {
louiz’'s avatar
louiz’ committed
139
      log_warning("Cannot send message to channel: [" << iid.chan << "] on server [" << iid.server << "]");
louiz’'s avatar
louiz’ committed
140 141 142 143 144
      return;
    }
  IrcClient* irc = this->get_irc_client(iid.server);
  if (!irc)
    {
louiz’'s avatar
louiz’ committed
145
      log_warning("Cannot send message: no client exist for server " << iid.server);
louiz’'s avatar
louiz’ committed
146 147
      return;
    }
148 149 150 151 152 153 154 155 156 157 158 159
  // 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
160
    {
161 162 163 164 165 166 167 168 169 170 171 172 173
      if (line.substr(0, 6) == "/mode ")
        {
          std::vector<std::string> args = utils::split(line.substr(6), ' ', false);
          irc->send_mode_command(iid.chan, args);
          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 ")
        irc->send_channel_message(iid.chan, action_prefix + line.substr(4) + "\01");
      else
        irc->send_channel_message(iid.chan, line);
      this->xmpp->send_muc_message(iid.chan + "%" + iid.server, irc->get_own_nick(),
                                   this->make_xmpp_body(line), this->user_jid);
louiz’'s avatar
louiz’ committed
174
    }
louiz’'s avatar
louiz’ committed
175 176
}

louiz’'s avatar
louiz’ committed
177
void Bridge::send_private_message(const Iid& iid, const std::string& body, const std::string& type)
louiz’'s avatar
louiz’ committed
178 179 180 181
{
  if (iid.chan.empty() || iid.server.empty())
    return ;
  IrcClient* irc = this->get_irc_client(iid.server);
182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
  if (!irc)
    {
      log_warning("Cannot send message: no client exist for server " << iid.server);
      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 ")
        irc->send_private_message(iid.chan, action_prefix + line.substr(4) + "\01", type);
      else
        irc->send_private_message(iid.chan, line, type);
    }
louiz’'s avatar
louiz’ committed
197 198
}

louiz’'s avatar
louiz’ committed
199 200 201 202 203 204 205
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_part_command(iid.chan, status_message);
}

louiz’'s avatar
louiz’ committed
206 207 208 209 210 211 212
void Bridge::send_irc_nick_change(const Iid& iid, const std::string& new_nick)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_nick_command(new_nick);
}

213 214 215 216 217 218 219
void Bridge::send_irc_kick(const Iid& iid, const std::string& target, const std::string& reason)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_kick_command(iid.chan, target, reason);
}

220 221 222 223 224 225 226
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_topic_command(iid.chan, subject);
}

louiz’'s avatar
louiz’ committed
227 228 229 230 231 232 233
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
234
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
235
{
louiz’'s avatar
louiz’ committed
236
  if (muc)
237 238
    this->xmpp->send_muc_message(iid.chan + "%" + iid.server, nick,
                                 this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
239
  else
240
    this->xmpp->send_message(iid.chan + "%" + iid.server,
241
                             this->make_xmpp_body(body), this->user_jid, "chat");
louiz’'s avatar
louiz’ committed
242 243
}

louiz’'s avatar
louiz’ committed
244
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
245
{
246
  this->xmpp->send_muc_leave(std::move(iid.chan) + "%" + std::move(iid.server), std::move(nick), this->make_xmpp_body(message), this->user_jid, self);
247 248 249
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
250 251
}

252 253 254 255 256
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
257
{
258 259 260 261
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

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

266 267
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
268 269
  std::string body;
  if (!author.empty())
270 271
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
272
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
273 274
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
275
    }
louiz’'s avatar
louiz’ committed
276 277
  else
    body = msg;
278
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid, "chat");
279 280
}

281 282
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
283
                            const IrcUser* user,
284
                            const char user_mode,
285
                            const bool self)
286
{
287 288 289 290
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

291 292
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
293 294 295 296
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
297
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
298
}
louiz’'s avatar
louiz’ committed
299 300 301 302 303 304 305 306

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

308
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
309 310 311 312
{
  return this->irc_clients.size();
}

313 314 315 316
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
  this->xmpp->kick_user(iid.chan + "%" + iid.server, target, reason, author, this->user_jid);
}
317 318 319 320 321

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
  this->xmpp->send_nickname_conflict_error(iid.chan + "%" + iid.server, nickname, this->user_jid);
}
322 323 324 325 326

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
329 330
  this->xmpp->send_affiliation_role_change(iid.chan + "%" + iid.server, target, affiliation, role, this->user_jid);
}
331 332 333 334 335

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
  this->xmpp->send_iq_version_request(nick + "%" + hostname, this->user_jid);
}