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    (logging-stream
  37     :initarg :logging-stream
  38     :accessor logging-stream
  39     :initform t
  40     :documentation "Messages coming back from the server are 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*)