From 982e6f9b6b15400f989b439f8b3e1672d09f5c3a Mon Sep 17 00:00:00 2001
From: Pistos <github.pistos@purepistos.net>
Date: Mon, 17 Oct 2011 10:52:03 -0400
Subject: [PATCH] Added missing rollback for photo migration.

---
 ...12215141_move_photos_to_their_own_table.rb | 33 +++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/db/migrate/20111012215141_move_photos_to_their_own_table.rb b/db/migrate/20111012215141_move_photos_to_their_own_table.rb
index 0bec98f25..fba0c3fca 100644
--- a/db/migrate/20111012215141_move_photos_to_their_own_table.rb
+++ b/db/migrate/20111012215141_move_photos_to_their_own_table.rb
@@ -48,7 +48,35 @@ INSERT INTO posts
   FROM photos
 SQL
 
-    execute <<SQL
+    if postgres?
+      execute %{
+        UPDATE
+          aspect_visibilities
+        SET
+            shareable_id=posts.id
+          , shareable_type='Post'
+        FROM
+            posts
+          , photos
+        WHERE
+          posts.guid=photos.guid
+          AND photos.id=aspect_visibilities.shareable_id
+        }
+      execute %{
+        UPDATE
+          share_visibilities
+        SET
+            shareable_id=posts.id
+          , shareable_type='Post'
+        FROM
+            posts
+          , photos
+        WHERE
+          posts.guid=photos.guid
+          AND photos.id=share_visibilities.shareable_id
+        }
+    else
+      execute <<SQL
 UPDATE aspect_visibilities, posts, photos
 SET
 aspect_visibilities.shareable_id=posts.id,
@@ -58,7 +86,7 @@ posts.guid=photos.guid AND
 photos.id=aspect_visibilities.shareable_id
 SQL
 
-    execute <<SQL
+      execute <<SQL
 UPDATE share_visibilities, posts, photos
 SET
 share_visibilities.shareable_id=posts.id,
@@ -67,6 +95,7 @@ WHERE
 posts.guid=photos.guid AND
 photos.id=share_visibilities.shareable_id
 SQL
+      end
 
     execute "DROP TABLE photos"
   end