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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
{-# LANGUAGE TemplateHaskell, MultiParamTypeClasses, TypeSynonymInstances, FlexibleInstances #-}
-- |
-- Module : Where
-- Copyright : 2003 Shae Erisson
--
-- License: lGPL
--
-- Slightly specialised version of Where for associating projects with their urls.
-- Code almost all copied.
module Plugin.Where (theModule) where
import Plugin
import Lambdabot.Util (confirmation)
import qualified Data.ByteString.Char8 as P
import qualified Data.Map as M
$(plugin "Where")
type WhereState = M.Map P.ByteString P.ByteString
type WhereWriter = WhereState -> LB ()
-- type Where m a = ModuleT WhereState m a
instance Module WhereModule WhereState where
moduleCmds _ = ["where", "url", "what", "where+" ]
moduleHelp _ s = case s of
"where" -> "where <key>. Return element associated with key"
"what" -> "what <key>. Return element associated with key"
"url" -> "url <key>. Return element associated with key"
"where+" -> "where+ <key> <elem>. Define an association"
moduleDefState _ = return M.empty
moduleSerialize _ = Just mapPackedSerial
process_ _ cmd rest = list $ withMS $ \factFM writer ->
case words rest of
[] -> return "@where <key>, return element associated with key"
(fact:dat) -> processCommand factFM writer
(lowerCaseString fact) cmd (unwords dat)
------------------------------------------------------------------------
processCommand :: WhereState -> WhereWriter
-> String -> String -> String -> LB String
processCommand factFM writer fact cmd dat = case cmd of
"where" -> return $ getWhere factFM fact
"what" -> return $ getWhere factFM fact -- an alias
"url" -> return $ getWhere factFM fact -- an alias
"where+" -> updateWhere True factFM writer fact dat
_ -> return "Unknown command."
getWhere :: WhereState -> String -> String
getWhere fm fact =
case M.lookup (P.pack fact) fm of
Nothing -> "I know nothing about " ++ fact ++ "."
Just x -> P.unpack x
updateWhere :: Bool -> WhereState -> WhereWriter -> String -> String -> LB String
updateWhere _guard factFM writer fact dat = do
writer $ M.insert (P.pack fact) (P.pack dat) factFM
random confirmation
|
