build

Author: porsager

cjs/src/connection.js

@@ -3,7 +3,7 @@ const tls = require('tls')
 const crypto = require('crypto')
 const Stream = require('stream')
 
-const { stringify, handleValue, arrayParser, arraySerializer } = require('./types.js')
+const { serialize, stringify, handleValue, arrayParser, arraySerializer } = require('./types.js')
 const { Errors } = require('./errors.js')
 const Result = require('./result.js')
 const Queue = require('./queue.js')
@@ -180,7 +180,7 @@ function Connection(options, queues = {}, { onopen = noop, onend = noop, onclose
       throw Errors.generic('MAX_PARAMETERS_EXCEEDED', 'Max number of parameters (65534) exceeded')
 
     return q.options.simple
-      ? b().Q().str(q.strings[0] + b.N).end()
+      ? b().Q().str(q.statement.string + b.N).end()
       : q.describeFirst
         ? Buffer.concat([describe(q), Flush])
         : q.prepare
@@ -912,7 +912,7 @@ function Connection(options, queues = {}, { onopen = noop, onend = noop, onclose
       type = types[i]
       parameters[i] = x = type in options.serializers
         ? options.serializers[type](x)
-        : '' + x
+        : serialize(x)
 
       prev = b.i
       b.inc(4).str(x).i32(b.i - prev - 4, prev)

cjs/src/types.js

@@ -1,11 +1,19 @@
 const { Query } = require('./query.js')
 const { Errors } = require('./errors.js')
 
+const serialize = module.exports.serialize = function serialize(x) {
+  return typeof x === 'string' ? x :
+    x instanceof Date ? types.date.serialize(x) :
+    x instanceof Uint8Array ? types.bytea.serialize(x) :
+    (x === true || x === false) ? types.boolean.serialize(x) :
+    '' + x
+}
+
 const types = module.exports.types = {
   string: {
     to: 25,
     from: null,             // defaults to string
-    serialize: x => '' + x
+    serialize
   },
   number: {
     to: 0,
@@ -66,10 +74,9 @@ const Builder = module.exports.Builder = class Builder extends NotTagged {
 
   build(before, parameters, types, options) {
     const keyword = builders.map(([x, fn]) => ({ fn, i: before.search(x) })).sort((a, b) => a.i - b.i).pop()
-    if (keyword.i === -1)
-      throw new Error('Could not infer helper mode')
-
-    return keyword.fn(this.first, this.rest, parameters, types, options)
+    return keyword.i === -1
+      ? escapeIdentifiers(this.first, options)
+      : keyword.fn(this.first, this.rest, parameters, types, options)
   }
 }
 
@@ -137,7 +144,7 @@ function values(first, rest, parameters, types, options) {
 function select(first, rest, parameters, types, options) {
   typeof first === 'string' && (first = [first].concat(rest))
   if (Array.isArray(first))
-    return first.map(x => escapeIdentifier(options.transform.column.to ? options.transform.column.to(x) : x)).join(',')
+    return escapeIdentifiers(first, options)
 
   let value
   const columns = rest.length ? rest.flat() : Object.keys(first)
@@ -170,9 +177,7 @@ const builders = Object.entries({
 
   insert(first, rest, parameters, types, options) {
     const columns = rest.length ? rest.flat() : Object.keys(Array.isArray(first) ? first[0] : first)
-    return '(' + columns.map(x =>
-      escapeIdentifier(options.transform.column.to ? options.transform.column.to(x) : x)
-    ).join(',') + ')values' +
+    return '(' + escapeIdentifiers(columns, options) + ')values' +
     valuesBuilder(Array.isArray(first) ? first : [first], parameters, types, columns, options)
   }
 }).map(([x, fn]) => ([new RegExp('((?:^|[\\s(])' + x + '(?:$|[\\s(]))(?![\\s\\S]*\\1)', 'i'), fn]))
@@ -209,16 +214,18 @@ function typeHandlers(types) {
   }, { parsers: {}, serializers: {} })
 }
 
+function escapeIdentifiers(xs, { transform: { column } }) {
+  return xs.map(x => escapeIdentifier(column.to ? column.to(x) : x)).join(',')
+}
+
 const escapeIdentifier = module.exports.escapeIdentifier = function escape(str) {
   return '"' + str.replace(/"/g, '""').replace(/\./g, '"."') + '"'
 }
 
 const inferType = module.exports.inferType = function inferType(x) {
   return x instanceof Parameter
     ? x.type
-    : Array.isArray(x)
-      ? inferType(x[0])
-      : 0
+    : 0
 }
 
 const escapeBackslash = /\\/g

cjs/tests/index.js

@@ -483,7 +483,7 @@ t('Point type array', async() => {
   })
 
   await sql`create table test (x point[])`
-  await sql`insert into test (x) values (${ sql.array([sql.types.point([10, 20]), sql.types.point([20, 30])]) })`
+  await sql`insert into test (x) values (${ [sql.types.point([10, 20]), sql.types.point([20, 30])] })`
   return [30, (await sql`select x from test`)[0].x[1][1], await sql`drop table test`]
 })
 

deno/src/connection.js

@@ -7,7 +7,7 @@ import { tls } from '../polyfills.js'
 import crypto from 'https://deno.land/[email protected]/node/crypto.ts'
 import Stream from 'https://deno.land/[email protected]/node/stream.ts'
 
-import { stringify, handleValue, arrayParser, arraySerializer } from './types.js'
+import { serialize, stringify, handleValue, arrayParser, arraySerializer } from './types.js'
 import { Errors } from './errors.js'
 import Result from './result.js'
 import Queue from './queue.js'
@@ -184,7 +184,7 @@ function Connection(options, queues = {}, { onopen = noop, onend = noop, onclose
       throw Errors.generic('MAX_PARAMETERS_EXCEEDED', 'Max number of parameters (65534) exceeded')
 
     return q.options.simple
-      ? b().Q().str(q.strings[0] + b.N).end()
+      ? b().Q().str(q.statement.string + b.N).end()
       : q.describeFirst
         ? Buffer.concat([describe(q), Flush])
         : q.prepare
@@ -916,7 +916,7 @@ function Connection(options, queues = {}, { onopen = noop, onend = noop, onclose
       type = types[i]
       parameters[i] = x = type in options.serializers
         ? options.serializers[type](x)
-        : '' + x
+        : serialize(x)
 
       prev = b.i
       b.inc(4).str(x).i32(b.i - prev - 4, prev)

deno/src/types.js

@@ -2,11 +2,19 @@ import { Buffer } from 'https://deno.land/[email protected]/node/buffer.ts'
 import { Query } from './query.js'
 import { Errors } from './errors.js'
 
+export const serialize = function serialize(x) {
+  return typeof x === 'string' ? x :
+    x instanceof Date ? types.date.serialize(x) :
+    x instanceof Uint8Array ? types.bytea.serialize(x) :
+    (x === true || x === false) ? types.boolean.serialize(x) :
+    '' + x
+}
+
 export const types = {
   string: {
     to: 25,
     from: null,             // defaults to string
-    serialize: x => '' + x
+    serialize
   },
   number: {
     to: 0,
@@ -67,10 +75,9 @@ export class Builder extends NotTagged {
 
   build(before, parameters, types, options) {
     const keyword = builders.map(([x, fn]) => ({ fn, i: before.search(x) })).sort((a, b) => a.i - b.i).pop()
-    if (keyword.i === -1)
-      throw new Error('Could not infer helper mode')
-
-    return keyword.fn(this.first, this.rest, parameters, types, options)
+    return keyword.i === -1
+      ? escapeIdentifiers(this.first, options)
+      : keyword.fn(this.first, this.rest, parameters, types, options)
   }
 }
 
@@ -138,7 +145,7 @@ function values(first, rest, parameters, types, options) {
 function select(first, rest, parameters, types, options) {
   typeof first === 'string' && (first = [first].concat(rest))
   if (Array.isArray(first))
-    return first.map(x => escapeIdentifier(options.transform.column.to ? options.transform.column.to(x) : x)).join(',')
+    return escapeIdentifiers(first, options)
 
   let value
   const columns = rest.length ? rest.flat() : Object.keys(first)
@@ -171,9 +178,7 @@ const builders = Object.entries({
 
   insert(first, rest, parameters, types, options) {
     const columns = rest.length ? rest.flat() : Object.keys(Array.isArray(first) ? first[0] : first)
-    return '(' + columns.map(x =>
-      escapeIdentifier(options.transform.column.to ? options.transform.column.to(x) : x)
-    ).join(',') + ')values' +
+    return '(' + escapeIdentifiers(columns, options) + ')values' +
     valuesBuilder(Array.isArray(first) ? first : [first], parameters, types, columns, options)
   }
 }).map(([x, fn]) => ([new RegExp('((?:^|[\\s(])' + x + '(?:$|[\\s(]))(?![\\s\\S]*\\1)', 'i'), fn]))
@@ -210,16 +215,18 @@ function typeHandlers(types) {
   }, { parsers: {}, serializers: {} })
 }
 
+function escapeIdentifiers(xs, { transform: { column } }) {
+  return xs.map(x => escapeIdentifier(column.to ? column.to(x) : x)).join(',')
+}
+
 export const escapeIdentifier = function escape(str) {
   return '"' + str.replace(/"/g, '""').replace(/\./g, '"."') + '"'
 }
 
 export const inferType = function inferType(x) {
   return x instanceof Parameter
     ? x.type
-    : Array.isArray(x)
-      ? inferType(x[0])
-      : 0
+    : 0
 }
 
 const escapeBackslash = /\\/g

deno/tests/index.js

@@ -485,7 +485,7 @@ t('Point type array', async() => {
   })
 
   await sql`create table test (x point[])`
-  await sql`insert into test (x) values (${ sql.array([sql.types.point([10, 20]), sql.types.point([20, 30])]) })`
+  await sql`insert into test (x) values (${ [sql.types.point([10, 20]), sql.types.point([20, 30])] })`
   return [30, (await sql`select x from test`)[0].x[1][1], await sql`drop table test`]
 })