Add cleaning for reverse, underline, and italic formatting.

[xudocci.git] / src / main / java / net / pterodactylus / irc / util / MessageCleaner.java

diff --git a/src/main/java/net/pterodactylus/irc/util/MessageCleaner.java b/src/main/java/net/pterodactylus/irc/util/MessageCleaner.java
index 6d0e3a2..9c5aab7 100644 (file)
--- a/src/main/java/net/pterodactylus/irc/util/MessageCleaner.java
+++ b/src/main/java/net/pterodactylus/irc/util/MessageCleaner.java
@@ -17,6 +17,10 @@
 
 package net.pterodactylus.irc.util;
 
+import static net.pterodactylus.irc.util.MessageCleaner.Attributes.italics;
+import static net.pterodactylus.irc.util.MessageCleaner.Attributes.reverse;
+import static net.pterodactylus.irc.util.MessageCleaner.Attributes.underline;
+
 import java.io.IOException;
 import java.io.StringReader;
 import java.util.EnumSet;
@@ -34,7 +38,10 @@ public class MessageCleaner {
 
                bold,
                color,
-               clear
+               clear,
+               reverse,
+               underline,
+               italics
 
        }
 
@@ -84,6 +91,15 @@ public class MessageCleaner {
                                if ((c == 15) && (attributes.contains(Attributes.clear))) {
                                        continue;
                                }
+                               if ((c == 22) && attributes.contains(reverse)) {
+                                       continue;
+                               }
+                               if ((c == 29) && attributes.contains(italics)) {
+                                       continue;
+                               }
+                               if ((c == 31) && attributes.contains(underline)) {
+                                       continue;
+                               }
                                if (inColorCode > 0) {
                                        if (inColorCode == 1) {
                                                if ((c < '0') || (c > '9')) {