From 265ff3dc70c11d97be1bc0fd298b8dd397662b66 Mon Sep 17 00:00:00 2001
From: Tom Quackenbush <tom@madglory.com>
Date: Tue, 2 Aug 2016 14:42:37 +0000
Subject: [PATCH] add pattern to match group chat messages for filtering

---
 mod_pottymouth/src/mod_pottymouth.erl | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/mod_pottymouth/src/mod_pottymouth.erl b/mod_pottymouth/src/mod_pottymouth.erl
index 7ae8f77..0260344 100644
--- a/mod_pottymouth/src/mod_pottymouth.erl
+++ b/mod_pottymouth/src/mod_pottymouth.erl
@@ -2,6 +2,8 @@
 
 -behaviour(gen_mod).
 
+-include("logger.hrl").
+
 -export([
   start/2,
   stop/1,
@@ -37,6 +39,12 @@ censorWord({Lang, Word} = MessageTerm) ->
 filterWords(L) ->
   lists:map(fun censorWord/1, L).
 
+filterMessageText(MessageAttrs, MessageText) ->
+  Lang = getMessageLang(MessageAttrs),
+  MessageWords = string:tokens(unicode:characters_to_list(MessageText, utf8), " "),
+  MessageTerms = [{Lang, Word} || Word <- MessageWords],
+  unicode:characters_to_binary(string:join(filterWords(MessageTerms), " ")).
+
 start(_Host, Opts) ->
   Blacklists = gen_mod:get_opt(blacklists, Opts, fun(A) -> A end, []),
   lists:map(fun bloom_gen_server:start/1, Blacklists),
@@ -55,12 +63,13 @@ on_filter_packet(drop) ->
   drop;
 
 on_filter_packet({_From, _To, {xmlel, <<"message">>, Attrs, [_chatState, {xmlel, <<"body">>, _BodyAttr, [{xmlcdata, MessageText}] = _BodyCData} = _MessageBody] = _Els} = _Packet} = _Msg) ->
-  Lang = getMessageLang(Attrs),
-  MessageWords = string:tokens(unicode:characters_to_list(MessageText, utf8), " "),
-  MessageTerms = [{Lang, Word} || Word <- MessageWords],
-  FilteredMessageWords = unicode:characters_to_binary(string:join(filterWords(MessageTerms), " ")),
+  FilteredMessageWords = filterMessageText(Attrs, MessageText),
   {_From, _To, {xmlel, <<"message">>, Attrs, [_chatState, {xmlel, <<"body">>, _BodyAttr, [{xmlcdata, FilteredMessageWords}]}]}};
 
+on_filter_packet({_From, _To, {xmlel, <<"message">>, Attrs, [{xmlel, <<"body">>, _BodyAttr, [{xmlcdata, MessageText}] = _BodyCData} = _MessageBody] = _Els} = _Packet} = _Msg) ->
+  FilteredMessageWords = filterMessageText(Attrs, MessageText),
+  {_From, _To, {xmlel, <<"message">>, Attrs, [{xmlel, <<"body">>, _BodyAttr, [{xmlcdata, FilteredMessageWords}]}]}};
+
 on_filter_packet(Msg) ->
   % Handle the generic case (any packet that isn't a message with a body).
   Msg.