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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
{-# LANGUAGE TemplateHaskell #-}
--
-- Copyright (c) 2006 Don Stewart - http://www.cse.unsw.edu.au/~dons
-- GPL version 2 or later (see http://www.gnu.org/copyleft/gpl.html)
--
-- Syntactic sugar for developing plugins.
-- Simplifies import lists, and abstracts over common patterns
--
module Plugin (
ios, box, list, ios80, plugin, modules,
module Lambdabot,
module LBState,
module Config,
module Lambdabot.Util,
module Lambdabot.Serial,
module Lambdabot.Process,
module Lambdabot.MiniHTTP,
module Lambdabot.Url,
module Lambdabot.Regex,
module Data.List,
module Data.Char,
module Data.Maybe,
module Data.Either,
module System.IO,
module Control.Monad.Error
) where
import Lambdabot
import LBState
import Config
import Lambdabot.Util
import Lambdabot.Serial
import Lambdabot.Process
import Lambdabot.MiniHTTP
import Lambdabot.Url
import Lambdabot.Regex
import Message
import Data.List
import Data.Char
import Data.Maybe
import Data.Either
import System.IO
import Control.Monad.Error
import Control.Monad.Trans
import Codec.Binary.UTF8.String
import Language.Haskell.TH
import Codec.Binary.UTF8.String
import Language.Haskell.TH
import Codec.Binary.UTF8.String
import Language.Haskell.TH
-- | convenience, we often want to perform some io, get a string, and box it.
ios :: (Functor m, MonadIO m) => IO a -> m [a]
ios = list . io
list :: (Functor m, Monad m) => m a -> m [a]
list = (return `fmap`)
box :: (Functor m, Monad m) => a -> m [a]
box = return . return
-- | convenience, similar to ios but also cut output to channel to 80 characters
-- usage: @process _ _ to _ s = ios80 to (plugs s)@
ios80 :: (Functor m, MonadIO m) => Nick -> IO String -> m [String]
ios80 to what = list . io $ what >>= return . lim . encodeString . spaceOut . removeControl . decodeString
where lim = case nName to of
('#':_) -> abbr 80 -- message to channel: be nice
_ -> id -- private message: get everything
spaceOut = unlines . map (' ':) . lines
removeControl = filter (\x -> isSpace x || not (isControl x))
abbr n s = let (b, t) = splitAt n s in
if null t then b else take (n-3) b ++ "..."
plugin :: String -> Q [Dec]
plugin n = sequence [typedec, fundec]
where
fundec = funD themod [clause [] (normalB ([| MODULE $(conE mod) |]) ) []]
-- typedec = newtypeD (cxt []) mod [] (normalC mod [strictType notStrict [t|()|]]) []
typedec = dataD (cxt []) mod [] [normalC mod []] []
themod = mkName "theModule"
mod = mkName $ n ++ "Module"
modules :: [String] -> Q Exp
modules xs = [| ($install, $names) |]
where
names = listE $ map (stringE . map toLower) xs
install = [| sequence_ $(listE $ map instalify xs) |]
instalify x = let mod = varE $ mkName $ concat $ ["Plugin.", x, ".theModule"]
low = stringE $ map toLower x
in [| ircInstallModule $mod $low |]
|
