blob: 9d7a169512e5f7b37883f41a48d5846c152819c2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
{-# LANGUAGE OverloadedStrings #-}
module Plugin.Factorial
( factorial
) where
import Text.Regex.TDFA ((=~))
import TextShow (showt)
import Bot (Bot)
import qualified Message as M
import Plugin.Base
factorial :: Bot Plugin
factorial = do
return defaultPlugin
{ matchRegex = "^([0-9]+)!$"
, perform = factorialAction
, command = "<integer>!"
, description = "Calculate the factorial of <integer> for whole numbers \
\up to 35000."
}
factorialAction :: PluginAction
factorialAction message = do
plugin <- factorial
case M.textStr message =~ matchRegex plugin :: [[String]] of
[] -> return $ Left "I didn't understand"
(m:_) -> do
let number = last m
return $ Right $ result (read number :: Integer)
where
result n = case calculate n of
Nothing -> "Input is too large."
Just n -> showt n
calculate :: Integer -> Maybe Integer
calculate n
| n <= 0 = Just 0
| n <= 35000 = Just $ product [1..n]
| otherwise = Nothing
|
