From 3a031d74293b2adc3849f63cf436dd47f2aab48f Mon Sep 17 00:00:00 2001
From: Ala'a Mohammad Alawi <amalawi@gmail.com>
Date: Fri, 27 Apr 2012 19:26:57 +0400
Subject: declaration are supported per cl-who docs. This patch detect them and
 put them in proper place

---
 util.lisp | 10 ++++++++++
 who.lisp  |  8 ++++++--
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/util.lisp b/util.lisp
index aee8efe..ae34184 100644
--- a/util.lisp
+++ b/util.lisp
@@ -228,3 +228,13 @@ determine whether CHAR must be escaped."
 character set."
   (escape-string string :test #'non-7bit-ascii-escape-char-p))
 
+(defun extract-declarations (body)
+  "Given a FORM, the declarations - if any - will be exctracted
+   from the head of the FORM, and will return two values the declarations,
+   and the remaining of FORM"
+  (do ((sexp (first body) (first forms))
+       (forms (rest body) (rest forms))
+       (declarations nil))
+      ((not (eq (first sexp) 'cl:declare))
+       (values declarations (append (if (null sexp) sexp (list sexp)) forms)))
+    (push sexp declarations)))
diff --git a/who.lisp b/who.lisp
index 0c1cd19..a512f94 100644
--- a/who.lisp
+++ b/who.lisp
@@ -273,7 +273,9 @@ into Lisp code to write the corresponding HTML as strings to VAR -
 which should either hold a stream or which'll be bound to STREAM if
 supplied."
   (declare (ignore prologue))
+  (multiple-value-bind (declarations forms) (extract-declarations body)
   `(let ((,var ,(or stream var)))
+       ,@declarations
      (macrolet ((htm (&body body)
                   `(with-html-output (,',var nil :prologue nil :indent ,,indent)
                      ,@body))
@@ -287,7 +289,7 @@ supplied."
                   (with-unique-names (result)
                     `(let ((,result ,thing))
                        (when ,result (princ ,result ,',var))))))
-       ,@(apply 'tree-to-commands body var rest))))
+         ,@(apply 'tree-to-commands forms var rest)))))
 
 (defmacro with-html-output-to-string ((var &optional string-form
                                            &key (element-type #-:lispworks ''character
@@ -297,11 +299,13 @@ supplied."
                                       &body body)
   "Transform the enclosed BODY consisting of HTML as s-expressions
 into Lisp code which creates the corresponding HTML as a string."
+  (multiple-value-bind (declarations forms) (extract-declarations body)
   `(with-output-to-string (,var ,string-form
                                 #-(or :ecl :cmu :sbcl) :element-type
                                 #-(or :ecl :cmu :sbcl) ,element-type)
+     ,@declarations
     (with-html-output (,var nil :prologue ,prologue :indent ,indent)
-      ,@body)))
+      ,@forms))))
 
 ;; stuff for Nikodemus Siivola's HYPERDOC
 ;; see <http://common-lisp.net/project/hyperdoc/>
-- 
cgit v1.2.3