Add support for compound UNIQUE and PRIMARY KEY constraints

Author: brianc

helpers/query-types.js

@@ -30,7 +30,7 @@ queryTypes.add(
 
 queryTypes.add(
   'create-table'
-, '{with} create table {ifNotExists} {table} ({definition})'
+, '{with} create table {ifNotExists} {table} ({definition}{constraints})'
 );
 
 queryTypes.add(

helpers/query/constraints.js

@@ -0,0 +1,26 @@
+var helpers = require('../../lib/query-helpers');
+var utils   = require('../../lib/utils');
+
+var quoteList = function(vals){
+  return vals.map(function(val){
+    return utils.quoteObject(val)
+  }).join(', ');
+};
+
+helpers.register('constraints', function(constraints, values, query){
+  var output = [];
+
+  for (var k in constraints) {
+    switch(k) {
+      case 'primaryKey':
+        output.push('primary key (' + quoteList(constraints[k]) +')');
+        break;
+      case 'unique':
+        output.push('unique (' + quoteList(constraints[k]) + ')');
+        break;
+    }
+  }
+
+  if (!output.length) return '';
+  return ', ' + output.join(', ');
+});

index.js

@@ -22,6 +22,7 @@ require('./helpers/query/boolean-helpers');
 require('./helpers/query/cascade');
 require('./helpers/query/column-constraint');
 require('./helpers/query/columns');
+require('./helpers/query/constraints');
 require('./helpers/query/definition');
 require('./helpers/query/distinct');
 require('./helpers/query/expression');

test/create-table.js

@@ -258,7 +258,6 @@ describe('Built-In Query Types', function(){
           }
         }
       });
-      console.log(query.toString());
       assert.equal(
         query.toString()
       , [ 'create table "posts" ('
@@ -270,5 +269,30 @@ describe('Built-In Query Types', function(){
       );
     });
 
+    it('Should support table constraints', function(){
+      var query = builder.sql({
+        type: 'create-table'
+      , table: 'foo'
+      , definition: {
+          name: { type: 'text' }
+        , email: { type: 'text' }
+        }
+
+      , constraints: {
+          primaryKey: ['name', 'email']
+        , unique: ['email', 'name']
+        }
+      });
+      assert.equal(
+        query.toString()
+      , [ 'create table "foo" ('
+        , '"name" text, "email" text, '
+        , 'primary key ("name", "email"), '
+        , 'unique ("email", "name")'
+        , ')'
+        ].join('')
+      );
+    });
+
   });
 });