From f07f7dd153d6c605e98143c594d2a578457a724e Mon Sep 17 00:00:00 2001 From: Treeki Date: Fri, 14 Feb 2014 01:26:01 +0100 Subject: add richtext support and message priorities --- bouncer/build.sh | 2 +- bouncer/build_static.sh | 2 +- bouncer/core.h | 4 +- bouncer/richtext.cpp | 176 ++++++++++++++++++++++++++++++++++++++++++++++++ bouncer/richtext.h | 92 +++++++++++++++++++++++++ bouncer/window.cpp | 114 ++++++++++++++----------------- 6 files changed, 323 insertions(+), 67 deletions(-) create mode 100644 bouncer/richtext.cpp create mode 100644 bouncer/richtext.h (limited to 'bouncer') diff --git a/bouncer/build.sh b/bouncer/build.sh index 092cca1..be4826d 100755 --- a/bouncer/build.sh +++ b/bouncer/build.sh @@ -2,7 +2,7 @@ mkdir -p binary NETCODE="socketcommon.cpp client.cpp mobileclient.cpp server.cpp ircserver.cpp netcore.cpp" -SOURCES="$NETCODE main.cpp window.cpp dns.cpp ini.cpp" +SOURCES="$NETCODE main.cpp window.cpp dns.cpp ini.cpp richtext.cpp" FLAGS="-std=c++11 -DUSE_GNUTLS -lgnutls -pthread -g" g++ -o binary/nb4 $FLAGS $SOURCES diff --git a/bouncer/build_static.sh b/bouncer/build_static.sh index e75d469..5807c1f 100755 --- a/bouncer/build_static.sh +++ b/bouncer/build_static.sh @@ -2,7 +2,7 @@ mkdir -p binary NETCODE="socketcommon.cpp client.cpp mobileclient.cpp server.cpp ircserver.cpp netcore.cpp" -SOURCES="$NETCODE main.cpp window.cpp dns.cpp ini.cpp" +SOURCES="$NETCODE main.cpp window.cpp dns.cpp ini.cpp richtext.cpp" FLAGS="-static -static-libgcc -static-libstdc++ -std=c++11 -pthread" g++ -o binary/nb4_static $FLAGS $SOURCES diff --git a/bouncer/core.h b/bouncer/core.h index 1a13a1d..28b99f2 100644 --- a/bouncer/core.h +++ b/bouncer/core.h @@ -66,7 +66,7 @@ public: virtual void handleUserInput(const char *str) { } virtual void handleUserClosed(); - void pushMessage(const char *str); + void pushMessage(const char *str, int priority = 0); void notifyWindowRename(); }; @@ -109,6 +109,8 @@ public: void handleTopic(const UserRef &user, const char *message); void handleTopicInfo(const char *user, int timestamp); + void outputUserMessage(const char *nick, const char *message, bool isAction); + char getEffectivePrefixChar(const char *nick) const; void disconnected(); diff --git a/bouncer/richtext.cpp b/bouncer/richtext.cpp new file mode 100644 index 0000000..22c6330 --- /dev/null +++ b/bouncer/richtext.cpp @@ -0,0 +1,176 @@ +#include "richtext.h" + +static int matchIRCColourString(const char *str, int pos, int *result) { + // First, try to match a single digit. + if (str[pos] >= '0' && str[pos] <= '9') { + int first = str[pos] - '0'; + if (str[pos + 1] >= '0' && str[pos + 1] <= '9') { + int second = str[pos + 1] - '0'; + + int attempt = (first * 10) + second; + if (attempt < 0 || attempt > 15) { + // Invalid number, give up here + return pos; + } else { + // Valid number! + *result = attempt; + return pos + 2; + } + } else { + // Second character was not a digit, so + // let's take the colour we *did* parse as is + *result = first; + return pos + 1; + } + } else { + // First character was not a digit, so pass entirely. + return pos; + } +} + +static int matchIRCColourPair(const char *str, int pos, int *whichFG, int *whichBG) { + // str[pos] is the first character following the + // \x03 colour code. + + *whichFG = -1; + *whichBG = -1; + + pos = matchIRCColourString(str, pos, whichFG); + if (*whichFG != -1) { + if (str[pos] == ',') { + pos = matchIRCColourString(str, pos + 1, whichBG); + } + } + + return pos; +} + + +void RichTextBuilder::appendIRC(const char *str) { + // We're up for fun here! + const int colLayer = COL_LEVEL_IRC; + + bool b = false, i = false, u = false; + int activeFG = COL_DEFAULT_FG, activeBG = COL_DEFAULT_BG; + + int pos = 0; + int len = strlen(str); + + while (pos < len) { + char ch = str[pos]; + + if (ch < 0 || ch == 7 || ch == 10 || ch >= 0x20) { + // Pass this character through unfiltered. + writeU8(ch); + pos++; + continue; + } + + // IRC control code, but what? + if (ch == 2) { + b = !b; + if (b) + bold(); + else + endBold(); + } else if (ch == 0x1D) { + i = !i; + if (i) + italic(); + else + endItalic(); + } else if (ch == 0x1F) { + u = !u; + if (u) + underline(); + else + endUnderline(); + } else if (ch == 0xF) { + // Reset all formatting + if (b) { + endBold(); + b = false; + } + if (i) { + endItalic(); + i = false; + } + if (u) { + endUnderline(); + u = false; + } + if (activeFG != COL_DEFAULT_FG) { + endForeground(colLayer); + activeFG = COL_DEFAULT_FG; + } + if (activeBG != COL_DEFAULT_BG) { + endBackground(colLayer); + activeBG = COL_DEFAULT_BG; + } + } else if (ch == 0x16) { + // Reverse + int swap = activeBG; + activeBG = activeFG; + activeFG = swap; + + if (activeFG == COL_DEFAULT_FG) + endForeground(colLayer); + else + foreground(colLayer, activeFG); + + if (activeBG == COL_DEFAULT_BG) + endBackground(colLayer); + else + background(colLayer, activeBG); + + } else if (ch == 3) { + // Colours! + + int whichFG = -1, whichBG = -1; + + pos = matchIRCColourPair(str, pos + 1, &whichFG, &whichBG); + + if (whichFG == -1) { + if (activeFG != COL_DEFAULT_FG) { + activeFG = COL_DEFAULT_FG; + endForeground(colLayer); + } + } else { + activeFG = whichFG; + foreground(colLayer, whichFG); + } + + if (whichBG == -1) { + if (whichFG == -1 && activeBG != COL_DEFAULT_BG) { + activeBG = COL_DEFAULT_BG; + endBackground(colLayer); + } + } else { + activeBG = whichBG; + background(colLayer, whichBG); + } + + continue; + } + pos++; + } + + // Clean up any leftover state. + if (b) + endBold(); + if (i) + endItalic(); + if (u) + endUnderline(); + if (activeFG != COL_DEFAULT_FG) + endForeground(colLayer); + if (activeBG != COL_DEFAULT_BG) + endBackground(colLayer); +} + +const char *RichTextBuilder::c_str() { + if (size() == 0 || data()[size() - 1] != 0) + writeU8(0); + + return data(); +} diff --git a/bouncer/richtext.h b/bouncer/richtext.h new file mode 100644 index 0000000..608a1d6 --- /dev/null +++ b/bouncer/richtext.h @@ -0,0 +1,92 @@ +#include "buffer.h" + +enum ColourPresets { + COL_IRC_WHITE = 0, + COL_IRC_BLACK = 1, + COL_IRC_BLUE = 2, + COL_IRC_GREEN = 3, + COL_IRC_RED = 4, + COL_IRC_BROWN = 5, + COL_IRC_PURPLE = 6, + COL_IRC_ORANGE = 7, + COL_IRC_YELLOW = 8, + COL_IRC_LIME = 9, + COL_IRC_TEAL = 10, + COL_IRC_CYAN = 11, + COL_IRC_LIGHT_BLUE = 12, + COL_IRC_PINK = 13, + COL_IRC_GREY = 14, + COL_IRC_LIGHT_GREY = 15, + + COL_DEFAULT_FG = 16, + COL_DEFAULT_BG = 17, + + COL_ACTION = 18, + COL_JOIN = 19, + COL_PART = 20, + COL_QUIT = 21, + COL_KICK = 22, + COL_CHANNEL_NOTICE = 23, +}; + +enum ColourLevels { + COL_LEVEL_BASE = 0, + COL_LEVEL_IRC = 1, + COL_LEVEL_NICK = 2, +}; + +class RichTextBuilder : public Buffer { +public: + void bold() { writeU8(1); } + void endBold() { writeU8(2); } + + void italic() { writeU8(3); } + void endItalic() { writeU8(4); } + + void underline() { writeU8(5); } + void endUnderline() { writeU8(6); } + + void colour(bool background, int layer, int r, int g, int b) { + writeU8(0x10 + (background ? 4 : 0) + layer); + writeU8((r==0)?2:(r&254)); + writeU8((g==0)?1:g); + writeU8((b==0)?1:b); + } + void colour(bool background, int layer, int col) { + writeU8(0x10 + (background ? 4 : 0) + layer); + writeU8((col << 1) | 1); + } + + void endColour(bool background, int layer) { + writeU8(0x18 + (background ? 4 : 0) + layer); + } + + void foreground(int layer, int r, int g, int b) { + colour(false, layer, r, g, b); + } + void foreground(int layer, int col) { + colour(false, layer, col); + } + void endForeground(int layer) { + endColour(false, layer); + } + + void background(int layer, int r, int g, int b) { + colour(true, layer, r, g, b); + } + void background(int layer, int col) { + colour(true, layer, col); + } + void endBackground(int layer) { + endColour(true, layer); + } + + void append(const char *str) { + Buffer::append(str, strlen(str)); + } + + void appendIRC(const char *str); + + const char *c_str(); +}; + diff --git a/bouncer/window.cpp b/bouncer/window.cpp index 88e8247..6085304 100644 --- a/bouncer/window.cpp +++ b/bouncer/window.cpp @@ -1,4 +1,5 @@ #include "core.h" +#include "richtext.h" Window::Window(NetCore *_core) { core = _core; @@ -29,7 +30,7 @@ void Window::notifyWindowRename() { Packet::B2C_WINDOW_RENAME, packet); } -void Window::pushMessage(const char *str) { +void Window::pushMessage(const char *str, int priority) { messages.push_back(str); bool createdPacket = false; @@ -39,6 +40,7 @@ void Window::pushMessage(const char *str) { if (core->clients[i]->isAuthed()) { if (!createdPacket) { packet.writeU32(id); + packet.writeU8(priority); packet.writeStr(str); createdPacket = true; } @@ -198,18 +200,7 @@ void Channel::handleUserInput(const char *str) { if (str[0] == '/') { if (strncmp(str, "/me ", 4) == 0) { - // The duplication of code between here and - // handlePrivmsg is ugly. TODO: fixme. - char prefix[2]; - prefix[0] = getEffectivePrefixChar(server->currentNick); - prefix[1] = 0; - - snprintf(msgBuf, sizeof(msgBuf), - "* %s%s %s", - prefix, - server->currentNick, - &str[4]); - pushMessage(msgBuf); + outputUserMessage(server->currentNick, &str[4], /*isAction=*/true); snprintf(msgBuf, sizeof(msgBuf), "PRIVMSG %s :\x01" "ACTION %s\x01", @@ -218,18 +209,7 @@ void Channel::handleUserInput(const char *str) { server->sendLine(msgBuf); } } else { - // Aaaand this is also pretty ugly ><;; - // TODO: fixme. - char prefix[2]; - prefix[0] = getEffectivePrefixChar(server->currentNick); - prefix[1] = 0; - - snprintf(msgBuf, sizeof(msgBuf), - "<%s%s> %s", - prefix, - server->currentNick, - str); - pushMessage(msgBuf); + outputUserMessage(server->currentNick, str, /*isAction=*/false); snprintf(msgBuf, sizeof(msgBuf), "PRIVMSG %s :%s", @@ -573,61 +553,61 @@ void Channel::handleMode(const UserRef &user, const char *str) { } void Channel::handlePrivmsg(const UserRef &user, const char *str) { - char prefix[2]; - prefix[0] = getEffectivePrefixChar(user.nick.c_str()); - prefix[1] = 0; + outputUserMessage(user.nick.c_str(), str, /*isAction=*/false); +} - char buf[15000]; - snprintf(buf, 15000, - "<%s%s> %s", - prefix, - user.nick.c_str(), - str); +void Channel::outputUserMessage(const char *nick, const char *str, bool isAction) { + RichTextBuilder rt; - pushMessage(buf); + if (isAction) { + rt.foreground(COL_LEVEL_BASE, COL_ACTION); + rt.append("* "); + } else { + rt.writeS8('<'); + } + + char prefix = getEffectivePrefixChar(nick); + if (prefix != 0) + rt.writeS8(prefix); + + rt.append(nick); + rt.append(isAction ? " " : "> "); + + rt.appendIRC(str); + + pushMessage(rt.c_str(), 2); } void Channel::handleCtcp(const UserRef &user, const char *type, const char *params) { char buf[15000]; if (strcmp(type, "ACTION") == 0) { - char prefix[2]; - prefix[0] = getEffectivePrefixChar(user.nick.c_str()); - prefix[1] = 0; - - snprintf(buf, sizeof(buf), - "* %s%s %s", - prefix, - user.nick.c_str(), - params); - + outputUserMessage(user.nick.c_str(), params, /*isAction=*/true); } else { snprintf(buf, sizeof(buf), "CTCP from %s : %s %s", user.nick.c_str(), type, params); + pushMessage(buf, 2); } - - pushMessage(buf); } void Channel::handleTopic(const UserRef &user, const char *message) { - char buf[1024]; + RichTextBuilder rt; + rt.foreground(COL_LEVEL_BASE, COL_CHANNEL_NOTICE); if (user.isValid) { - snprintf(buf, sizeof(buf), - "%s changed the topic to: %s", - user.nick.c_str(), - message); + rt.append(user.nick.c_str()); + rt.append(" changed the topic to: "); + rt.appendIRC(message); } else { - snprintf(buf, sizeof(buf), - "Topic: %s", - message); + rt.append("Topic: "); + rt.appendIRC(message); } - pushMessage(buf); + pushMessage(rt.c_str()); topic = message; @@ -639,12 +619,18 @@ void Channel::handleTopic(const UserRef &user, const char *message) { } void Channel::handleTopicInfo(const char *user, int timestamp) { - char buf[1024]; - snprintf(buf, sizeof(buf), - "Topic set by %s at %d", - user, - timestamp); - pushMessage(buf); + char intConv[50]; + snprintf(intConv, sizeof(intConv), "%d", timestamp); + + RichTextBuilder rt; + rt.foreground(COL_LEVEL_BASE, COL_CHANNEL_NOTICE); + + rt.append("Topic set by "); + rt.append(user); + rt.append(" at "); + rt.append(intConv); + + pushMessage(rt.c_str()); } @@ -764,7 +750,7 @@ void Query::handlePrivmsg(const char *str) { partner.c_str(), str); - pushMessage(buf); + pushMessage(buf, 2); } void Query::handleCtcp(const char *type, const char *params) { @@ -784,7 +770,7 @@ void Query::handleCtcp(const char *type, const char *params) { params); } - pushMessage(buf); + pushMessage(buf, 2); } void Query::renamePartner(const char *_partner) { -- cgit v1.2.3