automatic reconnect. handle more message types.
[eow] / empire.lisp
1 (in-package :empire)
2
3 (defparameter *version* "0.0.1")
4
5 ;;
6 ;; Connection
7 ;;
8
9 (defvar *default-empire-server* "localhost")
10 (defvar *default-empire-server-port* 6665)
11
12 (defclass connection ()
13   ((user
14     :initarg :user
15     :accessor user)
16    (password
17     :initarg :password
18     :accessor password
19     :initform nil)
20    (server-name
21     :initarg :server-name
22     :accessor server-name
23     :initform "Unknown server")
24    (server-port
25     :initarg :server-port
26     :accessor server-port
27     :initform *default-empire-server-port*)
28    (socket
29     :initarg :socket
30     :accessor socket
31     :documentation "Slot to store socket (for internal use only).")
32    (stream
33     :initarg :stream
34     :accessor network-stream
35     :documentation "Stream used to talk to the empire server.")
36    (logging-stream
37     :initarg :logging-stream
38     :accessor logging-stream
39     :initform t
40     :documentation "Messages coming back from the server is sent to
41 this stream.")
42    (connection-handler
43     :accessor connection-handler
44     :initform nil
45     :documentation "Handler function for incoming messages")
46    (mode
47     :accessor connection-mode
48     :initform nil
49     :documentation "TODO(gefla)")
50    (session
51     :accessor session
52     :initarg :session
53     :documentation "web session to which this connection belongs")))
54
55 (defun make-connection (&key (user nil)
56                         (password nil)
57                         (server-name *default-empire-server*)
58                         (server-port *default-empire-server-port*)
59                         (socket nil)
60                         (network-stream nil)
61                         (logging-stream nil)
62                         session)
63   (make-instance 'connection
64                  :user user
65                  :password password
66                  :server-name server-name
67                  :server-port server-port
68                  :socket socket
69                  :stream network-stream
70                  :logging-stream logging-stream
71                  :session session))
72
73 ;;; interface
74 (defun connect  (&key (user nil)
75                  (password nil)
76                  (server-name *default-empire-server*)
77                  (server-port *default-empire-server-port*)
78                  (logging-stream *standard-output*)
79                  session)
80   "Connect to server and return a connection object."
81   (let* ((socket (usocket:socket-connect server-name server-port))
82          (stream (usocket:socket-stream socket))
83          (connection (make-connection :user user
84                                       :password password
85                                       :socket socket
86                                       :network-stream stream
87                                       :logging-stream logging-stream
88                                       :server-name server-name
89                                       :server-port server-port
90                                       :session session))
91          (handler (spawn-with-name (format nil "empire-handler-~a" user)
92                                    #'handle-connection connection)))
93     (setf (connection-handler connection) handler)
94     connection))
95
96 (defgeneric quit (connection))
97 (defgeneric handle-connection (connection))
98 (defgeneric read-message (connection))
99 (defgeneric send-message (connection message))
100 (defgeneric reconnect (conncetion))
101
102 (defmethod reconnect ((c connection))
103   (with-slots (server-name server-port user password socket stream connection-handler) c
104       (let* ((sock (usocket:socket-connect server-name server-port))
105              (s (usocket:socket-stream sock))
106              (handler (spawn-with-name (format nil "empire-handler-~a" user)
107                                        #'handle-connection c)))
108         (setf socket sock
109               stream s
110               connection-handler handler))))
111
112 (defparameter +C_CMDOK+ "0")
113 (defparameter +C_DATA+ "1")
114 (defparameter +C_INIT+ "2")
115 (defparameter +C_EXIT+ "3")
116 (defparameter +C_FLUSH+ "4")
117 (defparameter +C_NOECHO+ "5")
118 (defparameter +C_PROMPT+ "6")
119 (defparameter +C_ABORT+ "7")
120 (defparameter +C_REDIR+ "8")
121 (defparameter +C_PIPE+ "9")
122 (defparameter +C_CMDERR+ "a")
123 (defparameter +C_BADCMD+ "b")
124 (defparameter +C_EXECUTE+ "c")
125 (defparameter +C_FLASH+ "d")
126 (defparameter +C_INFORM+ "e")
127
128 (defclass base-mode ()
129   ((connection :initarg :connection :accessor connection)))
130 (defclass init-mode (base-mode)
131   ((play-sent-p :initform nil
132                 :documentation "Have we already sent the play command?")))
133 (defclass play-mode (base-mode) nil)
134
135 (defun make-mode (connection mode)
136   (make-instance mode :connection connection))
137
138 (defgeneric set-new-mode (base-mode new-mode))
139 (defmethod set-new-mode ((m base-mode) new-mode)
140   (let* ((c (connection m))
141          (mode (make-mode c new-mode)))
142     (setf (connection-mode c) mode)))
143
144 (defgeneric handle-data (base-mode message))
145 (defgeneric handle-init (base-mode message))
146 (defgeneric handle-exit (base-mode message))
147 (defgeneric handle-prompt (base-mode message))
148 (defgeneric handle-simple-message (base-mode message))
149
150 (defmethod handle-init ((m init-mode) message)
151   (declare (ignorable message))
152   (with-slots (play-sent-p) m
153     (if (not play-sent-p)
154         (progn 
155           (play (connection m))
156           (setf play-sent-p t))
157         (set-new-mode m 'play-mode))))
158
159 (defmethod handle-exit ((m init-mode) message)
160   (declare (ignorable message))
161   (quit (connection m)))
162
163 (defmethod handle-exit ((m play-mode) message)
164   (declare (ignorable message))
165   (quit (connection m)))
166
167 (defmethod handle-data ((m play-mode) message)
168   (let* ((c (connection m))
169          (log (logging-stream c)))
170     (format log "handle-data ~a~%" message)
171     (empire-web:data (session c) message)))
172
173 (defmethod handle-simple-message ((m play-mode) message)
174   (let* ((c (connection m))
175          (log (logging-stream c)))
176     (format log "handle-simple-message ~a~%" message)
177     (empire-web:data (session c) message)))
178
179 (defmethod handle-prompt ((m play-mode) message)
180   (let* ((c (connection m))
181          (log (logging-stream c)))
182     (with-input-from-string (s message)
183       (let* ((minutes (read s)) ;;FIXME DANGEROUS
184              (btus (read s)))
185         (format log "[~a:~a]: ~%" minutes btus)
186         (empire-web:prompt (session c) minutes btus)))))
187
188 (defparameter *line-type-dispatch*
189   (list `(,+C_DATA+ . handle-data)
190         `(,+C_INIT+ . handle-init)
191         `(,+C_EXIT+ . handle-exit)
192         `(,+C_FLUSH+ . handle-simple-message)
193         `(,+C_BADCMD+ . handle-simple-message)
194         `(,+C_FLASH+ . handle-simple-message)
195         `(,+C_PROMPT+ . handle-prompt)))
196
197 (defun parse-server-line (line)
198   (let* ((first-space-index
199           (loop for i from 0 below (length line)
200              if (eq (char line i) #\Space)
201              do (return i)))
202          (message-type (subseq line 0 first-space-index))
203          (message (if first-space-index
204                       (subseq line (+ first-space-index 1))
205                       "")))
206     (values message message-type)))
207
208 (defmethod handle-connection ((c connection))
209   (setf (connection-mode c) (make-mode c 'init-mode))
210   (loop while (read-message c)))
211
212 (defun lookup-handler (message-type)
213   (cdr (assoc message-type *line-type-dispatch* :test 'string=)))
214
215 (define-condition no-handler (error)
216   ((mode :initarg :mode)
217    (type :initarg :type))
218   (:report (lambda (condition stream)
219              (with-slots (mode type) condition
220                (format stream "No handler for ~a in mode ~a."
221                        type mode)))))
222 (defmethod read-message ((c connection))
223   (handler-case
224       (let* ((s (network-stream c))
225              (log (logging-stream c))
226              (line (read-line s)))
227         (format log "< ~a~%" line)
228         (multiple-value-bind (message type) (parse-server-line line)
229           (let ((handler (lookup-handler type))
230                 (mode (connection-mode c)))
231             (if handler
232                 (funcall handler mode message)
233                 (error 'no-handler :mode mode :type type))))
234         line)
235     (sb-int:closed-stream-error () nil)
236     (end-of-file () nil)))
237
238 (defun raw-send-message (s message)
239     (write-sequence message s)
240     (terpri s)
241     (force-output s))
242   
243 (defmethod send-message ((c connection) message)
244   (let ((s (network-stream c))
245         (log (logging-stream c)))
246     (format log "< ~a~%" message)
247     (let ((sent-p nil)
248           (tries 3))
249       (loop
250          while (and (not sent-p)
251                     (> tries 0))
252          do (handler-case
253                 (progn
254                   (raw-send-message s message)
255                   (setf sent-p t))
256               (sb-int:closed-stream-error ()
257                 (progn
258                   (decf tries)
259                   (reconnect c))))))
260     message))
261
262 (defgeneric play (connection))
263 (defmethod play ((c connection))
264   (with-slots (user password) c
265     (send-message c (format nil "play FIXME ~a ~a" user password))))
266
267 (defmethod quit ((c connection))
268   (usocket:socket-close (socket c)))
269
270
271 ;;; Tests
272 (defvar *l1* "2 Empire server ready")
273 ;; (parse-server-line *l1*)