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