bridge.cpp 6.29 KB
Newer Older
1
#include <bridge/bridge.hpp>
2
#include <bridge/colors.hpp>
3 4
#include <xmpp/xmpp_component.hpp>
#include <network/poller.hpp>
5 6
#include <utils/encoding.hpp>

louiz’'s avatar
louiz’ committed
7
#include <utils/split.hpp>
louiz’'s avatar
louiz’ committed
8 9
#include <iostream>

10 11 12
static const char* action_prefix = "\01ACTION ";
static const size_t action_prefix_len = 8;

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

Bridge::~Bridge()
{
}

24 25 26 27 28 29 30 31 32 33
std::string Bridge::sanitize_for_xmpp(const std::string& str)
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
    res = utils::convert_to_utf8(str, "ISO-8859-1");
  return res;
}

34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
IrcClient* Bridge::get_irc_client(const std::string& hostname, const std::string& username)
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      this->irc_clients.emplace(hostname, std::make_shared<IrcClient>(hostname, username, this));
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      this->poller->add_socket_handler(irc);
      irc->start();
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
50 51 52 53 54 55 56 57 58 59 60 61
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
62
bool Bridge::join_irc_channel(const Iid& iid, const std::string& username)
63 64
{
  IrcClient* irc = this->get_irc_client(iid.server, username);
louiz’'s avatar
louiz’ committed
65 66 67 68 69 70
  if (irc->is_channel_joined(iid.chan) == false)
    {
      irc->send_join_command(iid.chan);
      return true;
    }
  return false;
71 72
}

louiz’'s avatar
louiz’ committed
73 74
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
75 76 77 78
  std::vector<std::string> lines = utils::split(body, '\n', true);
  if (lines.empty())
    return ;
  const std::string first_line = lines[0];
louiz’'s avatar
louiz’ committed
79 80 81 82 83 84 85 86 87 88 89
  if (iid.chan.empty() || iid.server.empty())
    {
      std::cout << "Cannot send message to channel: [" << iid.chan << "] on server [" << iid.server << "]" << std::endl;
      return;
    }
  IrcClient* irc = this->get_irc_client(iid.server);
  if (!irc)
    {
      std::cout << "Cannot send message: no client exist for server " << iid.server << std::endl;
      return;
    }
90
  if (first_line.substr(0, 6) == "/mode ")
louiz’'s avatar
louiz’ committed
91
    {
92
      std::vector<std::string> args = utils::split(first_line.substr(6), ' ', false);
louiz’'s avatar
louiz’ committed
93 94 95
      irc->send_mode_command(iid.chan, args);
      return;
    }
96 97
  if (first_line.substr(0, 4) == "/me ")
    irc->send_channel_message(iid.chan, action_prefix + first_line.substr(4) + "\01");
98
  else
99 100 101 102
    irc->send_channel_message(iid.chan, first_line);
  // Send each of the other lines of the message as a separate IRC message
  for (std::vector<std::string>::const_iterator it = lines.begin() + 1; it != lines.end(); ++it)
    irc->send_channel_message(iid.chan, *it);
103 104
  // We do not need to convert body to utf-8: it comes from our XMPP server,
  // so it's ok to send it back
louiz’'s avatar
louiz’ committed
105 106 107
  this->xmpp->send_muc_message(iid.chan + "%" + iid.server, irc->get_own_nick(), body, this->user_jid);
}

louiz’'s avatar
louiz’ committed
108 109 110 111 112 113 114 115 116
void Bridge::send_private_message(const Iid& iid, const std::string& body)
{
  if (iid.chan.empty() || iid.server.empty())
    return ;
  IrcClient* irc = this->get_irc_client(iid.server);
  if (irc)
    irc->send_private_message(iid.chan, body);
}

louiz’'s avatar
louiz’ committed
117 118 119 120 121 122 123
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
124 125 126 127 128 129 130
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);
}

131 132 133 134 135 136 137
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);
}

louiz’'s avatar
louiz’ committed
138
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
139
{
louiz’'s avatar
louiz’ committed
140
  std::string utf8_body = this->sanitize_for_xmpp(body);
141 142 143
  if (utf8_body.substr(0, action_prefix_len) == action_prefix)
    { // Special case for ACTION (/me) messages:
      // "\01ACTION goes out\01" == "/me goes out"
louiz’'s avatar
louiz’ committed
144 145
      utf8_body = std::string("/me ") +
        utf8_body.substr(action_prefix_len, utf8_body.size() - action_prefix_len - 1);
146
    }
louiz’'s avatar
louiz’ committed
147
  if (muc)
148
    this->xmpp->send_muc_message(iid.chan + "%" + iid.server, nick, utf8_body, this->user_jid);
louiz’'s avatar
louiz’ committed
149 150
  else
    this->xmpp->send_message(iid.chan + "%" + iid.server, utf8_body, this->user_jid);
louiz’'s avatar
louiz’ committed
151 152
}

louiz’'s avatar
louiz’ committed
153
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
154 155 156 157
{
  this->xmpp->send_muc_leave(std::move(iid.chan) + "%" + std::move(iid.server), std::move(nick), this->sanitize_for_xmpp(message), this->user_jid, self);
}

louiz’'s avatar
louiz’ committed
158 159 160 161 162 163
void Bridge::send_nick_change(Iid&& iid, const std::string& old_nick, const std::string& new_nick, const bool self)
{
  this->xmpp->send_nick_change(std::move(iid.chan) + "%" + std::move(iid.server),
                               old_nick, new_nick, this->user_jid, self);
}

164 165
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
166 167 168 169 170
  std::string body;
  if (!author.empty())
    body = std::string("[") + author + std::string("] ") + msg;
  else
    body = msg;
171
  this->xmpp->send_message(from, this->sanitize_for_xmpp(body), this->user_jid);
172 173 174 175 176 177 178 179 180 181 182 183 184 185
}

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

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

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
186
  this->xmpp->send_topic(chan_name + "%" + hostname, this->sanitize_for_xmpp(topic), this->user_jid);
187
}
louiz’'s avatar
louiz’ committed
188 189 190 191 192 193 194 195

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 "";
}
196 197 198 199 200

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