Add logging package. Handle flush prompts.
[eow] / empire.lisp
index fa088df853f29dc7613639fa036a139ce82ddcab..358d0010f5d8c5c1d1d715a60ebf89afd2c46ecb 100644 (file)
     :initarg :stream
     :accessor network-stream
     :documentation "Stream used to talk to the empire server.")
-   (logging-stream
-    :initarg :logging-stream
-    :accessor logging-stream
-    :initform t
-    :documentation "Messages coming back from the server is sent to
-this stream.")
    (connection-handler
     :accessor connection-handler
     :initform nil
-    :documentation "Handler function for incoming messages")))
+    :documentation "Handler function for incoming messages")
+   (mode
+    :accessor connection-mode
+    :initform nil
+    :documentation "TODO(gefla)")
+   (session
+    :accessor session
+    :initarg :session
+    :documentation "web session to which this connection belongs")))
 
 (defun make-connection (&key (user nil)
                        (password nil)
@@ -50,7 +52,7 @@ this stream.")
                        (server-port *default-empire-server-port*)
                        (socket nil)
                        (network-stream nil)
-                       (logging-stream nil))
+                       session)
   (make-instance 'connection
                 :user user
                 :password password
@@ -58,14 +60,14 @@ this stream.")
                 :server-port server-port
                 :socket socket
                 :stream network-stream
-                :logging-stream logging-stream))
+                :session session))
 
 ;;; interface
 (defun connect  (&key (user nil)
                 (password nil)
                 (server-name *default-empire-server*)
                 (server-port *default-empire-server-port*)
-                (logging-stream *standard-output*))
+                session)
   "Connect to server and return a connection object."
   (let* ((socket (usocket:socket-connect server-name server-port))
          (stream (usocket:socket-stream socket))
@@ -73,19 +75,31 @@ this stream.")
                                      :password password
                                      :socket socket
                                       :network-stream stream
-                                      :logging-stream logging-stream
                                       :server-name server-name
-                                     :server-port server-port))
+                                     :server-port server-port
+                                     :session session))
         (handler (spawn-with-name (format nil "empire-handler-~a" user)
                                   #'handle-connection connection)))
     (setf (connection-handler connection) handler)
+    (empire-log:info "empire:connect: ~a" connection)
     connection))
 
-
 (defgeneric quit (connection))
 (defgeneric handle-connection (connection))
 (defgeneric read-message (connection))
 (defgeneric send-message (connection message))
+(defgeneric reconnect (conncetion))
+
+(defmethod reconnect ((c connection))
+  (with-slots (server-name server-port user password socket stream connection-handler) c
+      (let* ((sock (usocket:socket-connect server-name server-port))
+            (s (usocket:socket-stream sock))
+            (handler (spawn-with-name (format nil "empire-handler-~a" user)
+                                      #'handle-connection c)))
+       (setf socket sock
+             stream s
+             connection-handler handler)))
+  (empire-log:info "empire:reconnect: ~a" c))
 
 (defparameter +C_CMDOK+ "0")
 (defparameter +C_DATA+ "1")
@@ -93,7 +107,7 @@ this stream.")
 (defparameter +C_EXIT+ "3")
 (defparameter +C_FLUSH+ "4")
 (defparameter +C_NOECHO+ "5")
-(defparameter +C_PROMPT+       "6")
+(defparameter +C_PROMPT+ "6")
 (defparameter +C_ABORT+ "7")
 (defparameter +C_REDIR+ "8")
 (defparameter +C_PIPE+ "9")
@@ -103,13 +117,74 @@ this stream.")
 (defparameter +C_FLASH+ "d")
 (defparameter +C_INFORM+ "e")
 
-(defgeneric init-mode (connection message))
-(defmethod init-mode ((c connection) message)
+(defclass base-mode ()
+  ((connection :initarg :connection :accessor connection)))
+(defclass init-mode (base-mode)
+  ((play-sent-p :initform nil
+               :documentation "Have we already sent the play command?")))
+(defclass play-mode (base-mode) nil)
+
+(defun make-mode (connection mode)
+  (make-instance mode :connection connection))
+
+(defgeneric set-new-mode (base-mode new-mode))
+(defmethod set-new-mode ((m base-mode) new-mode)
+  (let* ((c (connection m))
+        (mode (make-mode c new-mode)))
+    (setf (connection-mode c) mode)))
+
+(defgeneric handle-data (base-mode message))
+(defgeneric handle-init (base-mode message))
+(defgeneric handle-exit (base-mode message))
+(defgeneric handle-prompt (base-mode message))
+(defgeneric handle-flush (base-mode message))
+(defgeneric handle-simple-message (base-mode message))
+
+(defmethod handle-init ((m init-mode) message)
+  (declare (ignorable message))
+  (with-slots (play-sent-p) m
+    (if (not play-sent-p)
+       (progn 
+         (play (connection m))
+         (setf play-sent-p t))
+       (set-new-mode m 'play-mode))))
+
+(defmethod handle-exit ((m init-mode) message)
+  (declare (ignorable message))
+  (quit (connection m)))
+
+(defmethod handle-exit ((m play-mode) message)
   (declare (ignorable message))
-  (play c))
+  (quit (connection m)))
+
+(defmethod handle-data ((m play-mode) message)
+  (let* ((c (connection m)))
+    (empire-web:data (session c) message)))
+
+(defmethod handle-simple-message ((m play-mode) message)
+  (let* ((c (connection m)))
+    (empire-web:data (session c) message)))
+
+(defmethod handle-flush ((m play-mode) message)
+  (let* ((c (connection m)))
+    (empire-web:prompt (session c) message)))
+
+(defmethod handle-prompt ((m play-mode) message)
+  (let* ((c (connection m)))
+    (with-input-from-string (s message)
+      (let* ((minutes (read s)) ;;FIXME DANGEROUS
+            (btus (read s)))
+       (empire-web:prompt (session c)
+                          (format nil "[~a,~a]: " minutes btus))))))
 
-(defvar *line-type-dispatch*
-  (list `(,+C_INIT+ . init-mode)))
+(defparameter *line-type-dispatch*
+  (list `(,+C_DATA+ . handle-data)
+       `(,+C_INIT+ . handle-init)
+       `(,+C_EXIT+ . handle-exit)
+       `(,+C_FLUSH+ . handle-flush)
+       `(,+C_BADCMD+ . handle-simple-message)
+       `(,+C_FLASH+ . handle-simple-message)
+       `(,+C_PROMPT+ . handle-prompt)))
 
 (defun parse-server-line (line)
   (let* ((first-space-index
@@ -117,32 +192,62 @@ this stream.")
             if (eq (char line i) #\Space)
             do (return i)))
         (message-type (subseq line 0 first-space-index))
-        (message (subseq line (+ first-space-index 1))))
+        (message (if first-space-index
+                     (subseq line (+ first-space-index 1))
+                     "")))
     (values message message-type)))
 
 (defmethod handle-connection ((c connection))
+  (setf (connection-mode c) (make-mode c 'init-mode))
   (loop while (read-message c)))
 
 (defun lookup-handler (message-type)
   (cdr (assoc message-type *line-type-dispatch* :test 'string=)))
 
+(define-condition no-handler (error)
+  ((mode :initarg :mode)
+   (type :initarg :type))
+  (:report (lambda (condition stream)
+            (with-slots (mode type) condition
+              (format stream "No handler for ~a in mode ~a."
+                      type mode)))))
 (defmethod read-message ((c connection))
-  (let* ((s (network-stream c))
-        (log (logging-stream c))
-        (line (read-line s)))
-    (format log "< ~a~%" line)
-    (multiple-value-bind (message type) (parse-server-line line)
-      (let ((handler (lookup-handler type)))
-       (funcall handler c message)))
-    line))
-  
-(defmethod send-message ((c connection) message)
-  (let ((s (network-stream c))
-       (log (logging-stream c)))
-    (format log "< ~a~%" message)
+  (handler-case
+      (let* ((s (network-stream c))
+            (line (read-line s)))
+       (empire-log:info "< ~a~%" line)
+       (multiple-value-bind (message type) (parse-server-line line)
+         (let ((handler (lookup-handler type))
+               (mode (connection-mode c)))
+           (if handler
+               (funcall handler mode message)
+               (error 'no-handler :mode mode :type type))))
+       line)
+    (sb-int:closed-stream-error () nil)
+    (end-of-file () nil)))
+
+(defun raw-send-message (s message)
     (write-sequence message s)
     (terpri s)
-    (force-output s)
+    (force-output s))
+  
+(defmethod send-message ((c connection) message)
+  (let ((s (network-stream c)))
+    (empire-log:info "> ~a~%" message)
+    (let ((sent-p nil)
+         (tries 3))
+      (loop
+        while (and (not sent-p)
+                   (> tries 0))
+        do (handler-case
+               (progn
+                 (raw-send-message s message)
+                 (setf sent-p t))
+             (sb-int:closed-stream-error ()
+               (progn
+                 (decf tries)
+                 (empire-log:info "Connection close - retrying (~a tries left)" tries)
+                 (reconnect c))))))
     message))
 
 (defgeneric play (connection))