From 0b655daaff3cdd41f48b96fe34f786f10deed56a Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@gmail.com>
Date: Tue, 29 Apr 2014 13:19:02 -0400
Subject: [PATCH] Finish migration to v2.

---
 bitshift/database/__init__.py  |  2 ++
 bitshift/database/migration.py | 21 +++++++++++++++------
 bitshift/database/schema.sql   |  3 ++-
 3 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/bitshift/database/__init__.py b/bitshift/database/__init__.py
index 50486b6..14f7575 100644
--- a/bitshift/database/__init__.py
+++ b/bitshift/database/__init__.py
@@ -29,8 +29,10 @@ class Database(object):
     def _migrate(self, cursor, current):
         """Migrate the database to the latest schema version."""
         for version in xrange(current, VERSION):
+            print "Migrating to %d..." % version + 1
             for query in MIGRATIONS[version - 1]:
                 cursor.execute(query)
+            cursor.execute("UPDATE version SET version = ?", (version + 1,))
 
     def _check_version(self, migrate):
         """Check the database schema version and respond accordingly.
diff --git a/bitshift/database/migration.py b/bitshift/database/migration.py
index c9fdd39..2ea9666 100644
--- a/bitshift/database/migration.py
+++ b/bitshift/database/migration.py
@@ -8,12 +8,21 @@ VERSION = 2
 MIGRATIONS = [
     # 1 -> 2
     [
-        # drop index on code_hash
-        "ALTER TABLE code DROP COLUMN code_hash",
-        # change code_id to BIGINT NOT NULL,
-        # add key on codelets to codelet_lang
-        # add symbol_end_row INT UNSIGNED NOT NULL
-        # add symbol_end_col INT UNSIGNED NOT NULL
+        """ALTER TABLE `codelets`
+           DROP FOREIGN KEY `codelets_ibfk_1`""",
+        """ALTER TABLE `code`
+           DROP KEY `code_hash`,
+           DROP COLUMN `code_hash`,
+           MODIFY COLUMN `code_id` BIGINT NOT NULL""",
+        """ALTER TABLE `codelets`
+           MODIFY COLUMN `codelet_code_id` BIGINT NOT NULL,
+           ADD KEY (`codelet_lang`),
+           ADD FOREIGN KEY (`codelet_code_id`)
+               REFERENCES `code` (`code_id`)
+               ON DELETE RESTRICT ON UPDATE CASCADE""",
+        """ALTER TABLE `symbols`
+           ADD COLUMN `symbol_end_row` INT UNSIGNED NOT NULL,
+           ADD COLUMN `symbol_end_col` INT UNSIGNED NOT NULL"""
     ]
 ]
 
diff --git a/bitshift/database/schema.sql b/bitshift/database/schema.sql
index 159f85a..56a2d85 100644
--- a/bitshift/database/schema.sql
+++ b/bitshift/database/schema.sql
@@ -6,6 +6,7 @@ USE `bitshift`;
 CREATE TABLE `version` (
     `version` INT UNSIGNED NOT NULL
 ) ENGINE=InnoDB;
+INSERT INTO `version` VALUES (2);
 
 CREATE TABLE `origins` (
     `origin_id` TINYINT UNSIGNED NOT NULL AUTO_INCREMENT,
@@ -26,7 +27,7 @@ CREATE TABLE `code` (
 CREATE TABLE `codelets` (
     `codelet_id` BIGINT UNSIGNED NOT NULL AUTO_INCREMENT,
     `codelet_name` VARCHAR(300) NOT NULL,
-    `codelet_code_id` BIGINT UNSIGNED NOT NULL,
+    `codelet_code_id` BIGINT NOT NULL,
     `codelet_lang` SMALLINT UNSIGNED DEFAULT NULL,
     `codelet_origin` TINYINT UNSIGNED NOT NULL,
     `codelet_url` VARCHAR(512) NOT NULL,