irc.client: Uses fry instead of curry/compose, add server listeners.

db4
Bruno Deferrari 2008-06-09 15:36:44 -03:00
parent 7d1d574e40
commit 095506884b
1 changed files with 15 additions and 11 deletions

View File

@ -1,6 +1,6 @@
! Copyright (C) 2007 Doug Coleman, Slava Pestov. ! Copyright (C) 2007 Doug Coleman, Slava Pestov.
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: arrays combinators concurrency.mailboxes io USING: arrays combinators concurrency.mailboxes fry io
io.encodings.8-bit io.sockets kernel namespaces sequences io.encodings.8-bit io.sockets kernel namespaces sequences
sequences.lib splitting threads calendar classes.tuple sequences.lib splitting threads calendar classes.tuple
classes ascii assocs accessors destructors continuations ; classes ascii assocs accessors destructors continuations ;
@ -134,7 +134,7 @@ TUPLE: unhandled < irc-message ;
! ====================================== ! ======================================
: split-at-first ( seq separators -- before after ) : split-at-first ( seq separators -- before after )
dupd [ member? ] curry find dupd '[ , member? ] find
[ cut 1 tail ] [ cut 1 tail ]
[ swap ] [ swap ]
if ; if ;
@ -191,7 +191,8 @@ TUPLE: unhandled < irc-message ;
GENERIC: handle-incoming-irc ( irc-message -- ) GENERIC: handle-incoming-irc ( irc-message -- )
M: irc-message handle-incoming-irc ( irc-message -- ) M: irc-message handle-incoming-irc ( irc-message -- )
drop ; f irc> listeners>> at
[ in-messages>> mailbox-put ] [ drop ] if* ;
M: logged-in handle-incoming-irc ( logged-in -- ) M: logged-in handle-incoming-irc ( logged-in -- )
name>> irc> nick>> (>>name) ; name>> irc> nick>> (>>name) ;
@ -203,8 +204,8 @@ M: nick-in-use handle-incoming-irc ( nick-in-use -- )
name>> "_" append /NICK ; name>> "_" append /NICK ;
M: privmsg handle-incoming-irc ( privmsg -- ) M: privmsg handle-incoming-irc ( privmsg -- )
dup irc-message-origin irc> listeners>> at dup irc-message-origin irc> listeners>> [ at ] keep
[ in-messages>> mailbox-put ] [ drop ] if* ; '[ f , at ] unless* [ in-messages>> mailbox-put ] [ drop ] if* ;
M: join handle-incoming-irc ( join -- ) M: join handle-incoming-irc ( join -- )
irc> join-messages>> mailbox-put ; irc> join-messages>> mailbox-put ;
@ -226,7 +227,7 @@ M: privmsg handle-outgoing-irc ( privmsg -- )
! ====================================== ! ======================================
: irc-mailbox-get ( mailbox quot -- ) : irc-mailbox-get ( mailbox quot -- )
swap 5 seconds [ mailbox-get-timeout swap call ] 3curry [ drop ] recover ; swap 5 seconds '[ , , , mailbox-get-timeout swap call ] [ drop ] recover ;
: stream-readln-or-close ( stream -- str/f ) : stream-readln-or-close ( stream -- str/f )
dup stream-readln [ nip ] [ dispose f ] if* ; dup stream-readln [ nip ] [ dispose f ] if* ;
@ -272,7 +273,7 @@ DEFER: (connect-irc)
irc> out-messages>> mailbox-put ; irc> out-messages>> mailbox-put ;
: spawn-irc-loop ( quot name -- ) : spawn-irc-loop ( quot name -- )
[ [ irc> is-running>> ] compose ] dip [ '[ @ irc> is-running>> ] ] dip
spawn-server drop ; spawn-server drop ;
: spawn-irc ( -- ) : spawn-irc ( -- )
@ -285,7 +286,7 @@ DEFER: (connect-irc)
! ====================================== ! ======================================
: set+run-listener ( name irc-listener -- ) : set+run-listener ( name irc-listener -- )
[ [ listener-loop ] 2curry "listener" spawn-irc-loop ] [ '[ , , listener-loop ] "listener" spawn-irc-loop ]
[ swap irc> listeners>> set-at ] [ swap irc> listeners>> set-at ]
2bi ; 2bi ;
@ -294,13 +295,16 @@ M: irc-channel-listener (add-listener) ( irc-channel-listener -- )
[ [ name>> ] [ password>> ] bi /JOIN ] [ [ name>> ] [ password>> ] bi /JOIN ]
[ [ [ drop irc> join-messages>> ] [ [ [ drop irc> join-messages>> ]
[ timeout>> ] [ timeout>> ]
[ name>> [ swap trailing>> = ] curry ] [ name>> '[ trailing>> , = ] ]
tri mailbox-get-timeout? trailing>> ] keep set+run-listener tri mailbox-get-timeout? trailing>> ] keep set+run-listener
] bi ; ] bi ;
M: irc-nick-listener (add-listener) ( irc-nick-listener -- ) M: irc-nick-listener (add-listener) ( irc-nick-listener -- )
[ name>> ] keep set+run-listener ; [ name>> ] keep set+run-listener ;
M: irc-server-listener (add-listener) ( irc-server-listener -- )
f swap set+run-listener ;
: (connect-irc) ( irc-client -- ) : (connect-irc) ( irc-client -- )
[ profile>> [ server>> ] keep port>> /CONNECT ] keep [ profile>> [ server>> ] keep port>> /CONNECT ] keep
swap >>stream swap >>stream
@ -317,4 +321,4 @@ PRIVATE>
GENERIC: add-listener ( irc-client irc-listener -- ) GENERIC: add-listener ( irc-client irc-listener -- )
M: irc-listener add-listener ( irc-client irc-listener -- ) M: irc-listener add-listener ( irc-client irc-listener -- )
current-irc-client swap [ (add-listener) ] curry with-variable ; current-irc-client swap '[ , (add-listener) ] with-variable ;