module cloogle
+import Cloogle
import GenPrint
import IRC
import StdEnv
from Data.Func import $
from Text import class Text(..), instance Text String, instance + String
+import Text.JSON
+
+import Text.URI
import Control.Applicative
import qualified Control.Monad as CM
= (Error $ server_name + " hung up during transmission.", chan, w)
= receiveRest {resp & rsp_data=resp.rsp_data + toString (fromJust newresp)} chan w
+import StdMisc
+import StdDebug
+
+doRequestL :: HTTPRequest Int *World -> *(MaybeErrorString HTTPResponse, *World)
+doRequestL req 0 w = (Error "Maximal redirect numbe exceeded", w)
+doRequestL req maxRedirects w
+| not (trace_tn $ toString req) = undef
+# (er, w) = doRequest req w
+| isError er = (er, w)
+# resp = fromOk er
+| isMember resp.HTTPResponse.rsp_code [301, 302, 303, 307, 308]
+ = case lookup "Location" resp.HTTPResponse.rsp_headers of
+ Nothing = (Error $ "Redirect given but no Location header", w)
+ Just loc = case parseURI loc of
+ Nothing = (Error $ "Redirect URI couldn't be parsed", w)
+ Just uri = doRequestL {req
+ & server_name = maybe loc id uri.uriRegName
+ , server_port = maybe 80 id uri.uriPort
+ , req_path = uri.uriPath
+ , req_query = maybe "" ((+++) "?") uri.uriQuery
+ } (maxRedirects-1) w
+= (er, w)
+
shorten :: String *World -> (String, *World)
shorten s w
+# s = if (startsWith "http://" s) s (if (startsWith "https://" s) s ("http://" + s))
# data = "type=regular&url="+urlEncode s+"&token=a"
# (mer, w) = doRequest
{ newHTTPRequest
# resp = fromOk mer
= (resp.rsp_data, w)
+cloogle :: String *World -> (String, *World)
+cloogle data w
+# (mer, w) = doRequestL
+ { newHTTPRequest
+ & req_path = "/api.php"
+ , req_query = "?str=" + urlEncode data
+ , req_headers = 'DM'.fromList [("User-Agent", "cloogle-irc")]
+ , server_name = "cloogle.org"
+ , server_port = 80} 10 w
+| isError mer = ("request failed: " + fromError mer, w)
+# resp = fromOk mer
+= case fromJSON $ fromString resp.HTTPResponse.rsp_data of
+ Nothing = ("couldn't parse json", w)
+ Just clr = ("Results for " + data + " -- https://cloogle.org/#" + urlEncode data + "\n" +
+ processResults clr, w)
+ where
+ processResults :: Response -> String
+ processResults resp
+ | resp.return > 127 = "Failed: return code: " + toString resp.return + ", " + resp.msg
+ = join "\n" $ map processResult $ take 3 resp.data
+
+ processResult :: Result -> String
+ processResult (FunctionResult (br, {func}))
+ = "Function in " +++ br.library +++ ": " +++ br.modul +++ "\n" +++ func
+ processResult (TypeResult (br, {type}))
+ = "Type in " +++ br.library +++ ": " +++ br.modul +++ "\n" +++ limitResults type
+ processResult (ClassResult (br, {class_name,class_funs}))
+ = "Class in " +++ br.library +++ ": " +++ br.modul +++ "\n" +++ class_name +++ " with "
+ +++ toString (length class_funs) +++ " class functions"
+ processResult (MacroResult (br, {macro_name}))
+ = "Macro in " +++ br.library +++ ": " +++ br.modul +++ "\n" +++ macro_name
+ processResult (ModuleResult (br, _))
+ = "Module in " +++ br.library +++ ": " +++ br.modul
+
+ limitResults :: String -> String
+ limitResults s
+ # lines = split "\n" s
+ | length lines > 4 = limitResults (join "\n" (take 3 lines) + "\n...")
+ = join "\n" (map maxWidth lines)
+
+ maxWidth :: String -> String
+ maxWidth s
+ | size s > 80 = subString 0 77 s + "..."
+ = s
+
send :: [String] TCP_DuplexChannel *World -> (TCP_DuplexChannel, *World)
send [] chan w = (chan, w)
send [msg:msgs] {sChannel,rChannel} w
+# (_, w) = sleep 250000 w
# (rpt,i,sChannel,w) = send_MT TIMEOUT (toByteSeq msg) sChannel w
| rpt <> TR_Success = abort "Could not send request\n"
= send msgs {sChannel=sChannel,rChannel=rChannel} w
+ where
+ sleep :: !Int !*World -> (!Int, *World)
+ sleep i w = code {
+ ccall usleep "I:I:A"
+ }
recv :: TCP_DuplexChannel *World -> (Maybe String, TCP_DuplexChannel, *World)
recv {sChannel,rChannel} w
# (w, toSend) = case cmd of
["stop":_] = (w, Nothing)
["ping":xs] = (w, Just [msg $ "pong " +++ join " " xs])
+ ["query":xs]
+ # (s, w) = cloogle (join " " xs) w
+ = (w, Just $ map msg $ split "\n" s)
["short"] = (w, Just [msg $ "short requires an url argument"])
["short":xs]
# (s, w) = shorten (join " " xs) w
= (w, Just [msg s])
["help"] = (w, Just
[msg "type !help cmd for command specific help"
- ,msg "available commands: help, short, ping"])
+ ,msg "available commands: help, ping, query, short"])
["help":c:_] = (w, case c of
"help" = Just [msg "help [CMD] - I will print general help or the help of CMD"]
- "short" = Just [msg "short URL - I will give the url to https://cloo.gl shortening service and post back the result"]
"ping" = Just [msg "ping [TXT] - I will reply with pong and the optionar TXT"]
+ "query" = Just [msg "query QUERY - I will send QUERY to cloogle and post the results"]
+ "short" = Just [msg "short URL - I will give the url to https://cloo.gl shortening service and post back the result"]
_ = Just [msg "Unknown command"])
[c:_] = (w, Just [msg $ join " " ["unknown command: " , c, ", type !help to get help"]])
| isNothing toSend = (io, chan, w)