mod_compression: Moving compression setup in a dedicated function.

1 files changed, 17 insertions, 14 deletions

diff --git a/plugins/mod_compression.lua b/plugins/mod_compression.lua
index 72573b93..2a6c0871 100644
--- a/plugins/mod_compression.lua
+++ b/plugins/mod_compression.lua
@@ -108,21 +108,24 @@ module:add_handler({"c2s_unauthed", "c2s", "s2sin_unauthed", "s2sin"}, "compress
 				end
 				
 				-- setup compression for session.w
-				local old_send = session.send;
+				local function setup_compression(session)
+					local old_send = session.send;
 				
-				session.send = function(t)
-						local status, compressed, eof = pcall(deflate_stream, tostring(t), 'sync');
-						if status == false then
-							session:close({
-								condition = "undefined-condition";
-								text = compressed;
-								extra = st.stanza("failure", {xmlns="http://jabber.org/protocol/compress"}):tag("processing-failed");
-							});
-							module:log("warn", compressed);
-							return;
-						end
-						old_send(compressed);
-					end;
+					session.send = function(t)
+							local status, compressed, eof = pcall(deflate_stream, tostring(t), 'sync');
+							if status == false then
+								session:close({
+									condition = "undefined-condition";
+									text = compressed;
+									extra = st.stanza("failure", {xmlns="http://jabber.org/protocol/compress"}):tag("processing-failed");
+								});
+								module:log("warn", compressed);
+								return;
+							end
+							old_send(compressed);
+						end;
+				end
+				setup_compression(session);
 					
 				-- setup decompression for session.data
 				local function setup_decompression(session)