X-Git-Url: https://git.martlubbers.net/?p=cloogle-irc.git;a=blobdiff_plain;f=cloogleirc.icl;h=9edcb80690dd5c939eedb963cce15dbecdc293f3;hp=596f8c47e06c1e7b67707aa2b8a20b006004e4d3;hb=1c65613d8732d28f6c6c11864ccbc761918ac594;hpb=cd75fc36319812f7f36d29719596950abdb3895a diff --git a/cloogleirc.icl b/cloogleirc.icl index 596f8c4..9edcb80 100644 --- a/cloogleirc.icl +++ b/cloogleirc.icl @@ -1,40 +1,29 @@ module cloogleirc -import Cloogle -import GenPrint import StdEnv -import Data.Functor -import Data.Maybe -import Data.Either -from Data.Func import $, mapSt -from Text import class Text(..), instance Text String, instance + String - -import Internet.HTTP - -import Text.JSON - -import Text.URI - import Control.Applicative -import qualified Control.Monad as CM -import qualified Data.Map as DM -from Control.Monad import class Monad, instance Monad Maybe, >>= -from Text.Encodings.UrlEncoding import urlEncode -import System.CommandLine -import Internet.HTTP +import Control.Monad => qualified join +import Data.Either import Data.Error -import Data.List +import Data.Func import Data.Functor +import Data.List +import qualified Data.Map as DM +import Data.Maybe import Data.Tuple +import Internet.HTTP +import System.CommandLine +import System.Time +import Text +import Text.Encodings.UrlEncoding +import Text.GenJSON + +import Cloogle.API -import TCPIP import IRC import IRCBot -TIMEOUT :== Just 10000 -SERVER :== "irc.freenode.net" - shorten :: String *World -> (String, *World) shorten s w # s = if (startsWith "http://" s) s (if (startsWith "https://" s) s ("http://" + s)) @@ -56,7 +45,7 @@ shorten s w cloogle :: String *World -> (String, *World) cloogle data w -# (mer, w) = doHTTPRequestL +# (mer, w) = doHTTPRequestFollowRedirects { newHTTPRequest & req_path = "/api.php" , req_query = "?str=" + urlEncode data @@ -67,6 +56,7 @@ cloogle data w # resp = fromOk mer = case fromJSON $ fromString resp.HTTPResponse.rsp_data of Nothing = ("couldn't parse json", w) + Just {return=127} = ("No results for " + data, w) Just clr = ("Results for " + data + " -- https://cloogle.org/#" + replaceSubString "+" "%20" (urlEncode data) + "\n" + processResults clr, w) @@ -86,6 +76,14 @@ cloogle data w +++ toString (length class_funs) +++ " class functions" processResult (ModuleResult (br, _)) = "Module in " +++ br.library +++ ": " +++ br.modul + processResult (SyntaxResult (br, re)) + = "Clean syntax: " +++ re.syntax_title +++ "\n" + +++ concat (intersperse "; " re.syntax_code) + processResult (ABCInstructionResult (br, re)) + = "ABC instruction: " +++ re.abc_instruction + processResult (ProblemResult pr) + = "Common problem: " +++ pr.problem_title + +++ "; see https://gitlab.science.ru.nl/cloogle/common-problems/blob/master/" +++ pr.problem_key +++ ".md" limitResults :: String -> String limitResults s @@ -94,9 +92,7 @@ cloogle data w = join "\n" (map maxWidth lines) maxWidth :: String -> String - maxWidth s - | size s > 80 = subString 0 77 s + "..." - = s + maxWidth s = if (size s > 80) (subString 0 77 s + "...") s :: BotSettings = { bs_nick :: String @@ -104,25 +100,34 @@ cloogle data w , bs_autojoin :: [String] , bs_port :: Int , bs_server :: String + , bs_strftime :: String } -Start :: *World -> (MaybeErrorString (), *World) +Start :: *World -> (Maybe String, *World) Start w -# ([arg0:args], w) = getCommandLine w -# bs = parseCLI args -| isError bs = (Error $ "\n" +++ fromError bs +++ "\n", w) +# ([cmd:args], w) = getCommandLine w +# (io, w) = stdio w +# bs = parseCLI cmd args +| isError bs + # io = io <<< fromError bs <<< "\n" + = (Nothing, snd $ fclose io w) # (Ok bs) = bs -= bot (bs.bs_server, bs.bs_port) (startup bs) shutdown () process w +# (_, w) = fclose io w +# (merr, _, w) = bot (bs.bs_server, bs.bs_port) (startup bs) shutdown () (process bs.bs_strftime) w += (merr, w) where - parseCLI :: [String] -> MaybeErrorString BotSettings - parseCLI [] = Ok + parseCLI :: String [String] -> MaybeErrorString BotSettings + parseCLI _ [] = Ok { bs_nick = "clooglebot" , bs_nickserv = Nothing , bs_autojoin = [] , bs_port = 6667 , bs_server = "irc.freenode.net" + , bs_strftime = "%s" } - parseCLI [a:as] + parseCLI cmd [a:as] + | a == "-f" || a == "--strftime" + = arg1 "--strftime" as \a c->{c & bs_strftime=a} | a == "-n" || a == "--nick" = arg1 "--nick" as \a c->{c & bs_nick=a} | a == "-ns" || a == "--nickserv" @@ -132,10 +137,11 @@ Start w | a == "-p" || a == "--port" = arg1 "--port" as \a c->{c & bs_port=toInt a} | a == "-s" || a == "--server" - = arg1 "--port" as \a c->{c & bs_server=a} + = arg1 "--server" as \a c->{c & bs_server=a} | a == "-h" || a == "--help" = Error $ join "\n" $ - [ "Usage: cloogle [OPTS]" + [ "Usage: " + cmd + " [OPTS]" , "Options:" + , "\t--strftime/-f FORMAT strftime format used in the output. default: %s\n" , "\t--nick/-n NICKNAME Use the given nickname instead of clooglebot" , "\t--nickserv/-ns PW Identify via the given password with NickServ" , "\t--port/-p PORT Use the given port instead of port 6667" @@ -145,9 +151,9 @@ Start w , "\t has to be escaped in most shells" ] = Error $ "Unknown option: " +++ a - - arg1 name [] _ = Error $ name +++ " requires an argument" - arg1 name [a:as] f = parseCLI as >>= Ok o f a + where + arg1 name [] _ = Error $ name +++ " requires an argument" + arg1 name [a:as] f = parseCLI cmd as >>= Ok o f a nickserv pw = PRIVMSG (CSepList ["NickServ"]) $ "IDENTIFY " +++ pw @@ -161,24 +167,36 @@ Start w [JOIN (CSepList bs.bs_autojoin) Nothing] shutdown = map toPrefix [QUIT $ Just "Bye"] - process :: IRCMessage () *World -> (Maybe [IRCMessage], (), *World) - process im s w = case im.irc_command of + process :: String !IRCMessage () !*World -> (Maybe [IRCMessage], (), !*World) + process strf im _ w + # (io ,w) = stdio w + # (io, w) = log strf " (r): " im (io, w) + # (_, w) = fclose io w + = case im.irc_command of Left numr = (Just [], (), w) Right cmd = case process` im.irc_prefix cmd w of (Nothing, w) = (Nothing, (), w) - (Just cs, w) = (Just $ map toPrefix cs, (), w) + (Just cs, w) + # msgs = map toPrefix cs + = (Just msgs, (), w) + + log :: String String IRCMessage (!*File, !*World) -> (!*File, !*World) + log strf pref m (io, w) + #! (t, w) = localTime w + = (io <<< strfTime strf t <<< pref <<< toString m <<< "\n", w) process` :: (Maybe (Either IRCUser String)) IRCCommand *World -> (Maybe [IRCCommand], *World) process` (Just (Left user)) (PRIVMSG t m) w | m == "!restart" = (Nothing, w) | m.[0] == '!' # (msgs, w) = realProcess (split " " $ m % (1, size m)) w - = (Just $ map (NOTICE recipient) msgs, w) + = (Just $ map reply msgs, w) + | m % (0,4) == "\001PING" = (Just [reply m], w) = (Just [], w) where - recipient = case (\(CSepList [t:_]) -> t.[0]) t of - '#' -> t - _ -> CSepList [user.irc_nick] + reply = case (\(CSepList [t:_]) -> t.[0]) t of + '#' -> PRIVMSG t + _ -> NOTICE user.irc_nick process` _ (PING t mt) w = (Just [PONG t mt], w) process` _ _ w = (Just [], w)