Merge remote-tracking branch 'rafi/master' into rafi

1 files changed, 43 insertions, 18 deletions

diff --git a/application/migrations/003_add_referrers.php b/application/migrations/003_add_referrers.php
index 524e92ff0..9ca167eab 100644
--- a/application/migrations/003_add_referrers.php
+++ b/application/migrations/003_add_referrers.php
@@ -5,28 +5,53 @@ class Migration_Add_referrers extends CI_Migration {
 
 	public function up()
 	{
-		$this->db->query("
-			CREATE TABLE `invitations` (
-				`user` int(8) unsigned NOT NULL,
-				`key` varchar(16) CHARACTER SET ascii NOT NULL,
-				`date` int(11) unsigned NOT NULL,
-				PRIMARY KEY (`key`),
-		KEY `user` (`user`),
-		KEY `date` (`date`)
-			) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin
-		");
-		$this->db->query("
-			ALTER TABLE `users`
-			ADD `referrer` INT(8) UNSIGNED NOT NULL DEFAULT '0'
-		");
+		if ($this->db->dbdriver == 'postgre') {
+			$this->db->query('
+				CREATE TABLE "invitations" (
+					"user" integer NOT NULL,
+					"key" character varying(16) NOT NULL,
+					"date" integer NOT NULL,
+					PRIMARY KEY ("key")
+				);
+				CREATE INDEX "invitations_user_idx" ON "invitations" ("user");
+				CREATE INDEX "invitations_date_idx" ON "invitations" ("date");
+			');
+			$this->db->query('
+				ALTER TABLE "users"
+				ADD "referrer" integer NOT NULL DEFAULT 0
+			');
+
+		} else {
+
+			$this->db->query("
+				CREATE TABLE `invitations` (
+					`user` int(8) unsigned NOT NULL,
+					`key` varchar(16) CHARACTER SET ascii NOT NULL,
+					`date` int(11) unsigned NOT NULL,
+					PRIMARY KEY (`key`),
+			KEY `user` (`user`),
+			KEY `date` (`date`)
+				) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin
+			");
+			$this->db->query("
+				ALTER TABLE `users`
+				ADD `referrer` INT(8) UNSIGNED NOT NULL DEFAULT '0'
+			");
+		}
 	}
 
 	public function down()
 	{
-		$this->db->query("
-			ALTER TABLE `users`
-			DROP `referrer`
-		");
+		if ($this->db->dbdriver == 'postgre')
+		{
+			$this->db->query('
+				ALTER TABLE "users" DROP "referrer"
+			');
+		} else {
+			$this->db->query('
+				ALTER TABLE `users` DROP `referrer`
+			');
+		}
 		$this->dbforge->drop_table('invitations');
 
 	}