cleaning up generator, now uses migration_template (instead of hand r…

…olled solution)
coneybeare · Mar 22, 2011 · 0b9e622 · 0b9e622
1 parent 5c01a53
commit 0b9e622
Show file tree

Hide file tree

Showing 16 changed files with 64 additions and 43 deletions.
diff --git a/apn_on_rails.gemspec b/apn_on_rails.gemspec
@@ -31,19 +31,20 @@ easily add Apple Push Notification (iPhone) support to your Rails application.
     "VERSION",
     "apn_on_rails.gemspec",
     "autotest/discover.rb",
-    "generators/apn_migrations_generator.rb",
-    "generators/templates/apn_migrations/001_create_apn_devices.rb",
-    "generators/templates/apn_migrations/002_create_apn_notifications.rb",
-    "generators/templates/apn_migrations/003_alter_apn_devices.rb",
-    "generators/templates/apn_migrations/004_create_apn_apps.rb",
-    "generators/templates/apn_migrations/005_create_groups.rb",
-    "generators/templates/apn_migrations/006_alter_apn_groups.rb",
-    "generators/templates/apn_migrations/007_create_device_groups.rb",
-    "generators/templates/apn_migrations/008_create_apn_group_notifications.rb",
-    "generators/templates/apn_migrations/009_create_pull_notifications.rb",
-    "generators/templates/apn_migrations/010_alter_apn_notifications.rb",
-    "generators/templates/apn_migrations/011_make_device_token_index_nonunique.rb",
-    "generators/templates/apn_migrations/012_add_launch_notification_to_apn_pull_notifications.rb",
+    "generators/apn_on_rails/migrations_generator.rb",
+    "generators/apn_on_rails/USAGE",    
+    "generators/apn_on_rails/templates/migrations/create_apn_devices.rb",
+    "generators/apn_on_rails/templates/migrations/create_apn_notifications.rb",
+    "generators/apn_on_rails/templates/migrations/alter_apn_devices.rb",
+    "generators/apn_on_rails/templates/migrations/create_apn_apps.rb",
+    "generators/apn_on_rails/templates/migrations/create_groups.rb",
+    "generators/apn_on_rails/templates/migrations/alter_apn_groups.rb",
+    "generators/apn_on_rails/templates/migrations/create_device_groups.rb",
+    "generators/apn_on_rails/templates/migrations/create_apn_group_notifications.rb",
+    "generators/apn_on_rails/templates/migrations/create_pull_notifications.rb",
+    "generators/apn_on_rails/templates/migrations/alter_apn_notifications.rb",
+    "generators/apn_on_rails/templates/migrations/make_device_token_index_nonunique.rb",
+    "generators/apn_on_rails/templates/migrations/add_launch_notification_to_apn_pull_notifications.rb",
     "lib/apn_on_rails.rb",
     "lib/apn_on_rails/apn_on_rails.rb",
     "lib/apn_on_rails/app/models/apn/app.rb",

diff --git a/generators/apn_migrations_generator.rb b/generators/apn_migrations_generator.rb
diff --git a/generators/apn_on_rails/USAGE b/generators/apn_on_rails/USAGE
@@ -0,0 +1,12 @@
+Description:
+    Generate migrations needed to create APN tables
+
+Example:
+    rails generate apn_on_rails:migrations
+
+    This will create:
+        db/migrate/<timestamp>_create_apn_devices.rb
+        db/migrate/<timestamp>_create_apn_notifications.rb
+        db/migrate/<timestamp>_create_apn_group_notifications.rb
+        ...
+
diff --git a/generators/apn_on_rails/migrations_generator.rb b/generators/apn_on_rails/migrations_generator.rb
@@ -0,0 +1,38 @@
+require 'rails/generators/active_record'
+
+module ApnOnRails
+  module Generators
+    class MigrationsGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+      extend ActiveRecord::Generators::Migration
+
+      # Set the current directory as base for the inherited generators.
+      def self.base_root
+        File.dirname(__FILE__)
+      end
+
+      source_root File.expand_path('../templates/migrations', __FILE__)
+
+      def create_migrations
+        # Dir.glob(File.join(self.class.base_root, 'templates', 'migrations', '*.rb')).sort.each do |file|          
+        #   migration_template("#{File.basename(file)}", "db/migrate/#{File.basename(file)}");
+        # end
+
+        migration_template('create_apn_devices.rb', 'db/migrate/create_apn_devices.rb')
+        migration_template('create_apn_notifications.rb', 'db/migrate/create_apn_notifications.rb')
+        migration_template('create_apn_apps.rb', 'db/migrate/create_apn_apps.rb')
+        migration_template('alter_apn_devices.rb', 'db/migrate/alter_apn_devices.rb')
+        migration_template('create_groups.rb', 'db/migrate/create_groups.rb')
+        migration_template('alter_apn_groups.rb', 'db/migrate/alter_apn_groups.rb')
+        migration_template('create_device_groups.rb', 'db/migrate/create_device_groups.rb')
+        migration_template('create_apn_group_notifications.rb', 'db/migrate/create_apn_group_notifications.rb')
+        migration_template('create_pull_notifications.rb', 'db/migrate/create_pull_notifications.rb')
+        migration_template('alter_apn_notifications.rb', 'db/migrate/alter_apn_notifications.rb')
+        migration_template('make_device_token_index_nonunique.rb', 'db/migrate/make_device_token_index_nonunique.rb')
+        migration_template('add_launch_notification_to_apn_pull_notifications.rb', 'db/migrate/add_launch_notification_to_apn_pull_notifications.rb')
+
+      end
+    end
+  end
+end
+
diff --git a/...notification_to_apn_pull_notifications.rb → ...notification_to_apn_pull_notifications.rb b/...notification_to_apn_pull_notifications.rb → ...notification_to_apn_pull_notifications.rb
diff --git a/...s/apn_migrations/003_alter_apn_devices.rb → ...templates/migrations/alter_apn_devices.rb b/...s/apn_migrations/003_alter_apn_devices.rb → ...templates/migrations/alter_apn_devices.rb
diff --git a/...es/apn_migrations/006_alter_apn_groups.rb → .../templates/migrations/alter_apn_groups.rb b/...es/apn_migrations/006_alter_apn_groups.rb → .../templates/migrations/alter_apn_groups.rb
diff --git a/...migrations/010_alter_apn_notifications.rb → ...tes/migrations/alter_apn_notifications.rb b/...migrations/010_alter_apn_notifications.rb → ...tes/migrations/alter_apn_notifications.rb
diff --git a/...tes/apn_migrations/004_create_apn_apps.rb → ...s/templates/migrations/create_apn_apps.rb b/...tes/apn_migrations/004_create_apn_apps.rb → ...s/templates/migrations/create_apn_apps.rb
diff --git a/.../apn_migrations/001_create_apn_devices.rb → ...emplates/migrations/create_apn_devices.rb b/.../apn_migrations/001_create_apn_devices.rb → ...emplates/migrations/create_apn_devices.rb
diff --git a/...ons/008_create_apn_group_notifications.rb → ...rations/create_apn_group_notifications.rb b/...ons/008_create_apn_group_notifications.rb → ...rations/create_apn_group_notifications.rb
diff --git a/...igrations/002_create_apn_notifications.rb → ...es/migrations/create_apn_notifications.rb b/...igrations/002_create_apn_notifications.rb → ...es/migrations/create_apn_notifications.rb
diff --git a/...pn_migrations/007_create_device_groups.rb → ...plates/migrations/create_device_groups.rb b/...pn_migrations/007_create_device_groups.rb → ...plates/migrations/create_device_groups.rb
diff --git a/...lates/apn_migrations/005_create_groups.rb → ...ils/templates/migrations/create_groups.rb b/...lates/apn_migrations/005_create_groups.rb → ...ils/templates/migrations/create_groups.rb
diff --git a/...grations/009_create_pull_notifications.rb → ...s/migrations/create_pull_notifications.rb b/...grations/009_create_pull_notifications.rb → ...s/migrations/create_pull_notifications.rb
diff --git a/.../011_make_device_token_index_nonunique.rb → ...ions/make_device_token_index_nonunique.rb b/.../011_make_device_token_index_nonunique.rb → ...ions/make_device_token_index_nonunique.rb