make output-sql for ident-attributes do its best not to output invalid sql
authorRuss Tyndall <russ@acceleration.net>
Mon, 24 Aug 2009 21:14:09 +0000 (17:14 -0400)
committerRuss Tyndall <russ@acceleration.net>
Mon, 24 Aug 2009 21:14:09 +0000 (17:14 -0400)
sql/expressions.lisp

index 9157da04606a87929cd40057a2422e010ad060b2..aa9edcab98143431b00e3213906ea1fd01bbb605 100644 (file)
 ;;; KMR: The TYPE field is used by CommonSQL for type conversion -- it
 ;;; should not be output in SQL statements
   (let ((*print-pretty* nil))
-    (with-slots (qualifier name type) expr
-      (format *sql-stream* "~@[~a.~]~a"
-             (typecase qualifier
-               (string (format nil "~s" qualifier))
-               (symbol (safety-first (sql-escape qualifier))))
-             (typecase name
-               (string (format nil "~s" name))
-               (symbol (safety-first (sql-escape name)))))
-      t)))
+    (labels ((quoted-string-p (inp)
+              (and (char-equal #\" (elt inp 0))
+                   (char-equal #\" (elt inp (1- (length inp))))))
+            (safety-first (inp)
+              "do our best not to output sql that we can guarantee is invalid. 
+              if the ident has a space or quote in it, instead output a quoted
+             identifier containing those chars"
+              (when (and (not (quoted-string-p inp))
+                         (find-if
+                          (lambda (x) (member x '(#\space #\' #\")
+                                              :test #'char-equal)) inp))
+                (setf inp (format nil "~s" (substitute "\\\"" "\"" inp :test #'string-equal))))
+              inp))
+      (with-slots (qualifier name type) expr
+       (format *sql-stream* "~@[~a.~]~a"
+               (typecase qualifier
+                 (null nil)            ; nil is a symbol
+                 (string (format nil "~s" qualifier))
+                 (symbol (safety-first (sql-escape qualifier))))
+               (typecase name ;; could never get this to be nil without getting another error first
+                 (string (format nil "~s" name))
+                 (symbol (safety-first (sql-escape name)))))
+       t))))
 
 (defmethod output-sql-hash-key ((expr sql-ident-attribute) database)
   (with-slots (qualifier name type)