Add string escape for postgresql in custom query

.gitignore

@@ -4,4 +4,6 @@ test*.js
 .DS_STORE
 .DS_STORE
 node_modules
 node_modules
 npm-debug.log
 npm-debug.log
+spec/config/config.js
+spec-jasmine/config/config.js
 *~
 *~

lib/sequelize.js

@@ -171,7 +171,7 @@ module.exports = (function() {
 
 
   Sequelize.prototype.query = function(sql, callee, options, replacements) {
   Sequelize.prototype.query = function(sql, callee, options, replacements) {
     if (arguments.length === 4) {
     if (arguments.length === 4) {
-      sql = Utils.format([sql].concat(replacements))
+      sql = Utils.format([sql].concat(replacements), this.options.dialect)
     } else if (arguments.length === 3) {
     } else if (arguments.length === 3) {
       options = options
       options = options
     } else if (arguments.length === 2) {
     } else if (arguments.length === 2) {

lib/sql-string.js

@@ -7,7 +7,7 @@ SqlString.escapeId = function (val, forbidQualified) {
   return '`' + val.replace(/`/g, '``').replace(/\./g, '`.`') + '`';
   return '`' + val.replace(/`/g, '``').replace(/\./g, '`.`') + '`';
 };
 };
 
 
-SqlString.escape = function(val, stringifyObjects, timeZone) {
+SqlString.escape = function(val, stringifyObjects, timeZone, dialect) {
   if (val === undefined || val === null) {
   if (val === undefined || val === null) {
     return 'NULL';
     return 'NULL';
   }
   }
@@ -37,17 +37,22 @@ SqlString.escape = function(val, stringifyObjects, timeZone) {
     }
     }
   }
   }
 
 
-  val = val.replace(/[\0\n\r\b\t\\\'\"\x1a]/g, function(s) {
-    switch(s) {
-      case "\0": return "\\0";
-      case "\n": return "\\n";
-      case "\r": return "\\r";
-      case "\b": return "\\b";
-      case "\t": return "\\t";
-      case "\x1a": return "\\Z";
-      default: return "\\"+s;
-    }
-  });
+  if (dialect == "postgres") {
+    // http://www.postgresql.org/docs/8.2/static/sql-syntax-lexical.html#SQL-SYNTAX-STRINGS
+    val = val.replace(/'/g, "''");
+  } else {
+    val = val.replace(/[\0\n\r\b\t\\\'\"\x1a]/g, function(s) {
+      switch(s) {
+        case "\0": return "\\0";
+        case "\n": return "\\n";
+        case "\r": return "\\r";
+        case "\b": return "\\b";
+        case "\t": return "\\t";
+        case "\x1a": return "\\Z";
+        default: return "\\"+s;
+      }
+    });
+  }
   return "'"+val+"'";
   return "'"+val+"'";
 };
 };
 
 
@@ -58,7 +63,7 @@ SqlString.arrayToList = function(array, timeZone) {
   }).join(', ');
   }).join(', ');
 };
 };
 
 
-SqlString.format = function(sql, values, timeZone) {
+SqlString.format = function(sql, values, timeZone, dialect) {
   values = [].concat(values);
   values = [].concat(values);
 
 
   return sql.replace(/\?/g, function(match) {
   return sql.replace(/\?/g, function(match) {
@@ -66,7 +71,7 @@ SqlString.format = function(sql, values, timeZone) {
       return match;
       return match;
     }
     }
 
 
-    return SqlString.escape(values.shift(), false, timeZone);
+    return SqlString.escape(values.shift(), false, timeZone, dialect);
   });
   });
 };
 };
 
 

lib/utils.js

@@ -47,8 +47,9 @@ var Utils = module.exports = {
   escape: function(s) {
   escape: function(s) {
     return SqlString.escape(s, true, "local").replace(/\\"/g, '"')
     return SqlString.escape(s, true, "local").replace(/\\"/g, '"')
   },
   },
-  format: function(arr) {
-    return SqlString.format(arr.shift(), arr)
+  format: function(arr, dialect) {
+    var timeZone = null;
+    return SqlString.format(arr.shift(), arr, timeZone, dialect)
   },
   },
   isHash: function(obj) {
   isHash: function(obj) {
     return Utils._.isObject(obj) && !Array.isArray(obj);
     return Utils._.isObject(obj) && !Array.isArray(obj);

spec-jasmine/config/config.js

@@ -18,7 +18,8 @@ module.exports = {
 
 
   postgres: {
   postgres: {
     database: 'sequelize_test',
     database: 'sequelize_test',
-    username: "postgres",
+    username: "root",
+    password: "toor",
     port: 5432,
     port: 5432,
     pool: { maxConnections: 5, maxIdleTime: 30}
     pool: { maxConnections: 5, maxIdleTime: 30}
   }
   }

spec/config/config.js

@@ -24,7 +24,8 @@ module.exports = {
 
 
   postgres: {
   postgres: {
     database: 'sequelize_test',
     database: 'sequelize_test',
-    username: "postgres",
+    username: "root",
+    password: "toor",
     port: 5432,
     port: 5432,
     pool: { maxConnections: 5, maxIdleTime: 30}
     pool: { maxConnections: 5, maxIdleTime: 30}
   }
   }