X-Git-Url: https://git.martlubbers.net/?p=cloogle-irc.git;a=blobdiff_plain;f=cloogle.icl;h=29cee8056d1a4ce513878379658490f1563132cc;hp=b6adaf32dcba8c17fc3091d70040c391775db031;hb=f828f3a79cfcfb21c6c7a7773905cc210a739f75;hpb=936f7a03dbe41c7c415ba63ecbf64556edb0993e

diff --git a/cloogle.icl b/cloogle.icl
index b6adaf3..29cee80 100644
--- a/cloogle.icl
+++ b/cloogle.icl
@@ -1,5 +1,6 @@
 module cloogle
 
+import Cloogle
 import GenPrint
 import IRC
 import StdEnv
@@ -9,6 +10,9 @@ import Data.Maybe
 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
@@ -73,8 +77,32 @@ where
 		= (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
@@ -91,12 +119,63 @@ shorten s w
 # 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
@@ -121,17 +200,21 @@ process io chan 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)