add irc-line type, switch to (match) for handle-line

[8sync.git] / demos / irc.scm

diff --git a/demos/irc.scm b/demos/irc.scm
index 10436be6f91d886c0e682d0dffb2ee76e2803c64..b6a992cdf19e5be2148b7a8e92ec4f22e3728f0b 100755 (executable)
--- a/demos/irc.scm
+++ b/demos/irc.scm
@@ -21,7 +21,9 @@
 
 (use-modules (eightsync repl)
              (eightsync agenda)
+             (srfi srfi-9)
              (ice-9 getopt-long)
+             (ice-9 format)
              (ice-9 q)
              (ice-9 match))
 
@@ -43,11 +45,106 @@
   (display "Installing socket...\n")   ; debugging :)
   (make-port-request socket #:read handler))
 
-(define (handle-line socket line)
-  (display line)
-  (newline))
+(define irc-eol "\r\n")
 
-(define (make-simple-irc-handler handle-line)
+(define (irc-line line)
+  (string-concatenate (list line irc-eol)))
+
+(define-syntax-rule (irc-format dest format-string rest ...)
+  (let ((line (string-concatenate
+               (list (format #f format-string rest ...)
+                     irc-eol))))
+    (match dest
+      (#f line)
+      (#t (display line))
+      (else
+       (display line dest)))))
+
+(define* (irc-display line #:optional dest)
+  (if dest
+      (display (irc-line line) dest)
+      (display (irc-line dest))))
+
+(define* (handle-login socket username
+                       #:key
+                       (hostname "*")
+                       (servername "*")
+                       (realname username)
+                       (channels '()))
+  (irc-format socket "USER ~a ~a ~a :~a"
+              username hostname servername realname)
+  (irc-format socket "NICK ~a" username)
+  (for-each
+   (lambda (channel)
+     (irc-format socket "JOIN ~a" channel))
+   channels))
+
+(define (startswith-colon? str)
+  (and (> (string-length str) 0)
+       (eq? (string-ref str 0)
+            #\:)))
+
+(define-record-type <irc-line>
+  (make-irc-line prefix command params)
+  irc-line?
+  (prefix irc-line-prefix)
+  (command irc-line-command)
+  (params irc-line-params))
+
+
+(define (parse-line line)
+  (define (parse-params pre-params)
+    ;; This is stupid and imperative but I can't wrap my brain around
+    ;; the right way to do it in a functional way :\
+    (let ((param-list '())
+          (currently-building '()))
+      (for-each
+       (lambda (param-item)
+         (cond
+          ((startswith-colon? param-item)
+           (if (not (eq? currently-building '()))
+               (set! param-list
+                     (cons
+                      (reverse currently-building)
+                      param-list)))
+           (set! currently-building (list param-item)))
+          (else
+           (set! currently-building (cons param-item currently-building)))))
+       pre-params)
+      ;; We're still building something, so tack that on there
+      (if (not (eq? currently-building '()))
+          (set! param-list
+                (cons (reverse currently-building) param-list)))
+      ;; return the reverse of the param list
+      (reverse param-list)))
+
+  (match (string-split line #\space)
+    (((? startswith-colon? prefix)
+      command
+      pre-params ...)
+     (make-irc-line prefix command
+                    (parse-params pre-params)))
+    ((command pre-params ...)
+     (make-irc-line #f command
+                    (parse-params pre-params)))))
+
+
+(define (handle-line socket line my-username)
+  (let ((parsed-line (parse-line line)))
+    (match (irc-line-command parsed-line)
+      ("PING"
+       (irc-display "PONG" socket))
+      ("PRIVMSG"
+       (display "hey we got a PRIVMSG up in here!\n")
+       (display parsed-line)
+       (newline)
+       (display line)
+       (newline))
+      (_
+       (display line)
+       (newline)))))
+
+(define (make-simple-irc-handler handle-line username)
   (let ((buffer '()))
     (define (reset-buffer)
       (set! buffer '()))
@@ -60,19 +157,27 @@
           ((#\newline #\return (? char? line-chars) ...)
            (%sync (%run (handle-line
                          socket
-                         (list->string (reverse line-chars)))))
+                         (list->string (reverse line-chars))
+                         username)))
            ;; reset buffer
            (set! buffer '()))
           (_ #f))))
     irc-handler))
 
 (define* (queue-and-start-irc-agenda! agenda socket #:key
+                                      (username "syncbot")
                                       (inet-port default-irc-port)
-                                      (handler (make-simple-irc-handler handle-line)))
+                                      (handler (make-simple-irc-handler
+                                                (lambda args
+                                                  (apply handle-line args))
+                                                username))
+                                      (channels '()))
   (dynamic-wind
     (lambda () #f)
     (lambda ()
       (enq! (agenda-queue agenda) (wrap (install-socket socket handler)))
+      (enq! (agenda-queue agenda) (wrap (handle-login socket username
+                                                      #:channels channels)))
       (start-agenda agenda))
     (lambda ()
       (display "Cleaning up...\n")
@@ -91,6 +196,7 @@
            ,(lambda (s)
               (if (string->number s) #t #f))))
     (username (single-char #\u) (required? #t) (value #t))
+    (channels (value #t))
     (listen)))
 
 (define (main args)
@@ -99,12 +205,17 @@
          (port (or (option-ref options 'port #f)
                    default-irc-port))
          (username (option-ref options 'username #f))
-         (listen (option-ref options 'listen #f)))
+         (listen (option-ref options 'listen #f))
+         (channels (option-ref options 'channels ""))
+         (agenda (make-agenda)))
     (display `((server ,hostname) (port ,port) (username ,username)
-               (listen ,listen)))
+               (listen ,listen) (channels-split ,(string-split channels #\space))))
     (newline)
+    (if listen
+        (spawn-and-queue-repl-server! agenda))
     (queue-and-start-irc-agenda!
-     (make-agenda)
+     agenda
      (irc-socket-setup hostname port)
      #:inet-port port
-     #:handler (make-simple-irc-handler handle-line))))
+     #:username username
+     #:channels (string-split channels #\space))))