joshnabbott-desert 0.3.5.2 → 0.3.5.3

data/lib/desert/plugin_migrations/2.1/migrator.rb

@@ -0,0 +1,35 @@
+module Desert #:nodoc:
+  module PluginMigrations
+    class Migrator < ActiveRecord::Migrator
+      class << self
+        def current_version #:nodoc:
+          result = ActiveRecord::Base.connection.select_one("SELECT version FROM #{schema_migrations_table_name} WHERE plugin_name = '#{current_plugin.name}' order by version desc")
+          if result
+            result['version'].to_i
+          else
+            # There probably isn't an entry for this plugin in the migration info table.
+            0
+          end
+        end
+
+        def get_all_versions
+          ActiveRecord::Base.connection.select_values("SELECT version FROM #{schema_migrations_table_name} where plugin_name='#{current_plugin.name}'").map(&:to_i).sort
+        end
+      end
+
+      def record_version_state_after_migrating(version)
+        sm_table = self.class.schema_migrations_table_name
+
+        if down?
+          ActiveRecord::Base.connection.update("DELETE FROM #{sm_table} WHERE version = '#{version}' WHERE plugin_name = '#{current_plugin.name}'")
+        else
+          ActiveRecord::Base.connection.insert("INSERT INTO #{sm_table} (plugin_name, version) VALUES ('#{current_plugin.name}', '#{version}')")
+        end
+      end
+
+      def migrated
+        self.class.get_all_versions
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: joshnabbott-desert
 version: !ruby/object:Gem::Version 
-  version: 0.3.5.2
+  version: 0.3.5.3
 platform: ruby
 authors: 
 - Pivotal Labs
@@ -31,6 +31,7 @@ files:
 - lib/desert/manager.rb
 - lib/desert/plugin.rb
 - lib/desert/plugin_migrations/2.1/extensions/schema_statements.rb
+- lib/desert/plugin_migrations/2.1/migrator.rb
 - lib/desert/plugin_migrations/1.2/extensions/schema_statements.rb
 - lib/desert/plugin_migrations/migrator.rb
 - lib/desert/plugin_migrations.rb