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