Remove IRC colors from the body when forwarding it to XMPP
This commit is contained in:
+19
-4
@@ -1,6 +1,9 @@
|
|||||||
#include <bridge/bridge.hpp>
|
#include <bridge/bridge.hpp>
|
||||||
|
#include <bridge/colors.hpp>
|
||||||
#include <xmpp/xmpp_component.hpp>
|
#include <xmpp/xmpp_component.hpp>
|
||||||
#include <network/poller.hpp>
|
#include <network/poller.hpp>
|
||||||
|
#include <utils/encoding.hpp>
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
@@ -15,6 +18,17 @@ Bridge::~Bridge()
|
|||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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");
|
||||||
|
remove_irc_colors(res);
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
IrcClient* Bridge::get_irc_client(const std::string& hostname, const std::string& username)
|
IrcClient* Bridge::get_irc_client(const std::string& hostname, const std::string& username)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
@@ -43,7 +57,6 @@ IrcClient* Bridge::get_irc_client(const std::string& hostname)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void Bridge::join_irc_channel(const Iid& iid, const std::string& username)
|
void Bridge::join_irc_channel(const Iid& iid, const std::string& username)
|
||||||
{
|
{
|
||||||
IrcClient* irc = this->get_irc_client(iid.server, username);
|
IrcClient* irc = this->get_irc_client(iid.server, username);
|
||||||
@@ -64,12 +77,14 @@ void Bridge::send_channel_message(const Iid& iid, const std::string& body)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
irc->send_channel_message(iid.chan, body);
|
irc->send_channel_message(iid.chan, body);
|
||||||
|
// We do not need to convert body to utf-8: it comes from our XMPP server,
|
||||||
|
// so it's ok to send it back
|
||||||
this->xmpp->send_muc_message(iid.chan + "%" + iid.server, irc->get_own_nick(), body, this->user_jid);
|
this->xmpp->send_muc_message(iid.chan + "%" + iid.server, irc->get_own_nick(), body, this->user_jid);
|
||||||
}
|
}
|
||||||
|
|
||||||
void Bridge::send_muc_message(const Iid& iid, const std::string& nick, const std::string& body)
|
void Bridge::send_muc_message(const Iid& iid, const std::string& nick, const std::string& body)
|
||||||
{
|
{
|
||||||
this->xmpp->send_muc_message(iid.chan + "%" + iid.server, nick, body, this->user_jid);
|
this->xmpp->send_muc_message(iid.chan + "%" + iid.server, nick, this->sanitize_for_xmpp(body), this->user_jid);
|
||||||
}
|
}
|
||||||
|
|
||||||
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
|
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
|
||||||
@@ -79,7 +94,7 @@ void Bridge::send_xmpp_message(const std::string& from, const std::string& autho
|
|||||||
body = std::string("[") + author + std::string("] ") + msg;
|
body = std::string("[") + author + std::string("] ") + msg;
|
||||||
else
|
else
|
||||||
body = msg;
|
body = msg;
|
||||||
this->xmpp->send_message(from, body, this->user_jid);
|
this->xmpp->send_message(from, this->sanitize_for_xmpp(body), this->user_jid);
|
||||||
}
|
}
|
||||||
|
|
||||||
void Bridge::send_user_join(const std::string& hostname, const std::string& chan_name, const std::string nick)
|
void Bridge::send_user_join(const std::string& hostname, const std::string& chan_name, const std::string nick)
|
||||||
@@ -94,5 +109,5 @@ void Bridge::send_self_join(const std::string& hostname, const std::string& chan
|
|||||||
|
|
||||||
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
|
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
|
||||||
{
|
{
|
||||||
this->xmpp->send_topic(chan_name + "%" + hostname, topic, this->user_jid);
|
this->xmpp->send_topic(chan_name + "%" + hostname, this->sanitize_for_xmpp(topic), this->user_jid);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,6 +23,7 @@ public:
|
|||||||
explicit Bridge(const std::string& user_jid, XmppComponent* xmpp, Poller* poller);
|
explicit Bridge(const std::string& user_jid, XmppComponent* xmpp, Poller* poller);
|
||||||
~Bridge();
|
~Bridge();
|
||||||
|
|
||||||
|
static std::string sanitize_for_xmpp(const std::string& str);
|
||||||
/***
|
/***
|
||||||
**
|
**
|
||||||
** From XMPP to IRC.
|
** From XMPP to IRC.
|
||||||
|
|||||||
@@ -6,6 +6,7 @@
|
|||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
|
#include <bridge/colors.hpp>
|
||||||
#include <utils/encoding.hpp>
|
#include <utils/encoding.hpp>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
@@ -39,5 +40,9 @@ int main()
|
|||||||
// wrong charset)
|
// wrong charset)
|
||||||
std::string from_ascii = utils::convert_to_utf8(original_latin1, "US-ASCII");
|
std::string from_ascii = utils::convert_to_utf8(original_latin1, "US-ASCII");
|
||||||
assert(from_ascii == "couc�ou");
|
assert(from_ascii == "couc�ou");
|
||||||
|
|
||||||
|
std::string coucou("\u0002\u0002COUCOU\u0003");
|
||||||
|
remove_irc_colors(coucou);
|
||||||
|
assert(coucou == "COUCOU");
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user