From 444817f145bc60f20701b8e92529bfb38aa0f50e Mon Sep 17 00:00:00 2001
From: Dafydd Francis <DaveOrdref@gmail.com>
Date: Sun, 1 Jan 2012 20:35:43 +0000
Subject: [PATCH] cleaned up the spelling module

---
 modules/spelling.js | 78 +++++++++++++++++++++------------------------
 1 file changed, 36 insertions(+), 42 deletions(-)

diff --git a/modules/spelling.js b/modules/spelling.js
index 5db8da4..cb5cc60 100644
--- a/modules/spelling.js
+++ b/modules/spelling.js
@@ -1,55 +1,49 @@
 var spelling = function(dbot) {
     var dbot = dbot;
     var last = {};
+
+    var correct = function (data, correction, candidate, output_callback) {
+        var candidates = last[data.channel][candidate].split(' ');
+        var winner = false;
+        var winnerDistance = Infinity; //urgh fix later
+
+        for(var i=0;i<candidates.length;i++) {
+            var distance = String.prototype.distance(correction, candidates[i]);
+            if(distance < winnerDistance) {
+                winner = candidates[i];
+                winnerDistance = distance;
+            }
+        }
+
+        if(winnerDistance < 3) {
+            if(winner !== correction) {
+                var fix = last[data.channel][candidate].replace(winner, correction);
+                if (/^.ACTION/.test(fix)) {
+                    fix = fix.replace(/^.ACTION/, '/me');
+                }
+                last[data.channel][candidate] = fix;
+                var output = {
+                    'fix': fix,
+                    'correcter': data.user,
+                    'candidate': candidate
+                };
+                output_callback(output);
+            }
+        }
+    }
     
     return {
         'listener': function(data, params) {
             var q = data.message.valMatch(/^\*\*?([\d\w\s]*)/, 2);
             var otherQ = data.message.valMatch(/^([\d\w\s]*): \*\*?([\d\w\s]*)/, 3);
             if(q) {
-                var correction = q[1];
-                var candidates = last[data.channel][data.user].split(' ');
-                var winner = false;
-                var winnerDistance = Infinity; //urgh fix later
-
-                for(var i=0;i<candidates.length;i++) {
-                    var distance = String.prototype.distance(correction, candidates[i]);
-                    if(distance < winnerDistance) {
-                        winner = candidates[i];
-                        winnerDistance = distance;
-                    }
-                }
-
-                if(winnerDistance < 3) {
-                    if(winner !== correction) {
-                        var fix = last[data.channel][data.user].replace(winner, correction); 
-                        last[data.channel][data.user] = fix;
-                        dbot.say(data.channel, data.user + ' meant: ' + fix);
-                    }
-                }
+                correct(data, q[1], data.user, function (e) {
+                    dbot.say(data.channel, e.correcter + ' meant: ' + e.fix);
+                });
             } else if(otherQ) {
-                if(last[data.channel].hasOwnProperty([otherQ[1]])) {
-                    var correction = otherQ[2];
-                    var candidates = last[data.channel][otherQ[1]].split(' ');
-                    var winner = false;
-                    var winnerDistance = Infinity; //urgh fix later
-
-                    for(var i=0;i<candidates.length;i++) {
-                        var distance = String.prototype.distance(correction, candidates[i]);
-                        if(distance < winnerDistance) {
-                            winner = candidates[i];
-                            winnerDistance = distance;
-                        }
-                    }
-
-                    if(winnerDistance < 3) {
-                        if(winner !== correction) {
-                            var fix = last[data.channel][otherQ[1]].replace(winner, correction); 
-                            last[data.channel][otherQ[1]] = fix;
-                            dbot.say(data.channel, data.user + ' thinks ' + otherQ[1] + ' meant: ' + fix);
-                        }
-                    }
-                }
+                correct(data, otherQ[2], otherQ[1], function (e) {
+                    dbot.say(data.channel, e.correcter + ' thinks ' + e.candidate + ' meant: ' + e.fix);
+                });
             } else {
                  if(last.hasOwnProperty(data.channel)) {
                    last[data.channel][data.user] = data.message;