--
-- (c) Josef Svenningsson, 2005
-- Licence: No licence, public domain
--
-- Inspired by the following page:
-- http://www.microsoft.com/athome/security/children/kidtalk.mspx
--
module Plugin.Elite (theModule) where

import Plugin
import qualified Text.Regex as R

import Control.Arrow
import Control.Monad.State

PLUGIN Elite

instance Module EliteModule () where
    moduleCmds _   = ["elite"]
    moduleHelp _ _ = "elite <phrase>. Translate English to elitespeak"
    process_ _ _ args = ios $
        case words args of
             [] -> return "Say again?"
             wds -> do let instr = map toLower (unwords wds)
                       transWords <- translate instr
                       return transWords

translate :: MonadIO m => String -> m String
translate []  = return []
translate str = case alts of
                  [] -> do rest <- translate (tail str)
                           return (head str : rest)
                  _ -> do (subst,rest) <- liftIO $ stdGetRandItem alts
                          translatedRest <- translate rest
                          return (subst ++ translatedRest)
  where alts = (map (\ (Just (_,_,rest,_),subst) -> (subst,rest)) $
                filter isJustEmpty $
                map (\ (re, subst) -> (R.matchRegexAll re str,subst)) $
                ruleList)
               ++ [([toUpper $ head str],tail str)
                  ,([head str],tail str)
                  ]

isJustEmpty :: (Maybe([a],b,c,d),e) -> Bool
isJustEmpty (Just ([],_,_,_),_) = True
isJustEmpty (_,_)                = False

ruleList :: [(Regex,String)]
ruleList = map (first regex')
           [("a","4")
           ,("b","8")
           ,("c","(")
           ,("ck","xx")
           ,("cks ","x ")
           ,(" cool "," kewl ")
           ,("e","3")
           ,("elite","1337")
           ,("f","ph")
           ,(" for "," 4 ")
           ,("g","9")
           ,("h","|-|")
           ,("k","x")
           ,("l","|")
           ,("l","1")
           ,("m","/\\/\\")
           ,("o","0")
           ,("ph","f")
           ,("s","z")
           ,("s","$")
           ,("s","5")
           ,("s ","z0rz ")
           ,("t","7")
           ,("t","+")
           ,("v","\\/")
           ,("w","\\/\\/")
           ,(" you "," u ")
           ,(" you "," joo ")
           ,("z","s")
           ]