Add rename clause for alter table

Chaoran Yang · Chaoran Yang · commit bcf08b79c7ff · 2013-04-20T23:56:45.000-05:00
diff --git a/lib/dialect/postgres.js b/lib/dialect/postgres.js
@@ -45,6 +45,7 @@ Postgres.prototype.visit = function(node) {
     case 'DEFAULT': return this.visitDefault(node);
     case 'IF EXISTS': return this.visitIfExists();
     case 'IF NOT EXISTS': return this.visitIfNotExists();
+    case 'RENAME': return this.visitRename(node);
     case 'ADD COLUMN': return this.visitAddColumn(node);
     case 'DROP COLUMN': return this.visitDropColumn(node);
     case 'RENAME COLUMN': return this.visitRenameColumn(node);
@@ -346,6 +347,10 @@ Postgres.prototype.visitRenameColumn = function(renameColumn) {
   return ['RENAME COLUMN ' + this.visit(renameColumn.nodes[0]) + ' TO ' + this.visit(renameColumn.nodes[1])];
 };
 
+Postgres.prototype.visitRename = function(rename) {
+  return ['RENAME TO ' + this.visit(rename.nodes[0])];
+};
+
 Postgres.prototype.visitIfExists = function() {
   return ['IF EXISTS'];
 };
diff --git a/lib/node/query.js b/lib/node/query.js
@@ -21,6 +21,7 @@ var Alter = require(__dirname + '/alter');
 var AddColumn = require(__dirname + '/addColumn');
 var DropColumn = require(__dirname + '/dropColumn');
 var RenameColumn = require(__dirname + '/renameColumn');
+var Rename = require(__dirname + '/rename');
 var Column = require(__dirname + '/../column');
 var ParameterNode = require(__dirname + '/parameter');
 var IfExists = require(__dirname + '/ifExists');
@@ -161,6 +162,14 @@ var Query = Node.define({
   alter: function() {
     return this.add(new Alter());
   },
+  rename: function(newName) {
+    var renameClause = new Rename();
+    if (!newName.toNode)
+      newName = new Column({ name: newName, table: this.table});
+    renameClause.add(newName.toNode());
+    this.nodes[0].add(renameClause);
+    return this;
+  },
   addColumn: function(column, dataType) {
     var addClause = new AddColumn();
     if (!column.toNode)
diff --git a/lib/node/rename.js b/lib/node/rename.js
@@ -0,0 +1,7 @@
+'use strict';
+
+var Node = require(__dirname);
+
+module.exports = Node.define({
+  type: 'RENAME'
+});
diff --git a/test/dialects/alter-table-tests.js b/test/dialects/alter-table-tests.js
@@ -37,6 +37,14 @@ Harness.test({
   params: []
 });
 
+Harness.test({
+  query : post.alter().rename('posts'),
+  pg    : 'ALTER TABLE "post" RENAME TO "posts"',
+  sqlite: 'ALTER TABLE "post" RENAME TO "posts"',
+  mysql : 'ALTER TABLE `post` RENAME TO `posts`',
+  params: []
+});
+
 var group = Table.define({
   name: 'group',
   columns: [{
