1 files changed, 176 insertions, 0 deletions
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();
+}