From ecb95519ec82cc6af5322aca57ae42719b5164e5 Mon Sep 17 00:00:00 2001
From: Tute Costa <tutecosta@gmail.com>
Date: Mon, 12 Mar 2012 18:19:29 -0300
Subject: [PATCH] Impressionist generator now accepts --orm option (defaults to
 active_record).

---
 .../active_record/impressionist_generator.rb  | 21 +++++++++++++++++++
 .../templates/create_impressions_table.rb     |  0
 .../impressionist/impressionist_generator.rb  | 20 ------------------
 lib/generators/impressionist_generator.rb     |  7 +++++++
 4 files changed, 28 insertions(+), 20 deletions(-)
 create mode 100644 lib/generators/active_record/impressionist_generator.rb
 rename lib/generators/{impressionist => active_record}/templates/create_impressions_table.rb (100%)
 delete mode 100644 lib/generators/impressionist/impressionist_generator.rb
 create mode 100644 lib/generators/impressionist_generator.rb

diff --git a/lib/generators/active_record/impressionist_generator.rb b/lib/generators/active_record/impressionist_generator.rb
new file mode 100644
index 0000000..31f4c1a
--- /dev/null
+++ b/lib/generators/active_record/impressionist_generator.rb
@@ -0,0 +1,21 @@
+module ActiveRecord
+  module Generators
+    class ImpressionistGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+      source_root File.join(File.dirname(__FILE__), 'templates')
+
+      def self.next_migration_number(dirname)
+        sleep 1
+        if ActiveRecord::Base.timestamped_migrations
+          Time.now.utc.strftime("%Y%m%d%H%M%S")
+        else
+          "%.3d" % (current_migration_number(dirname) + 1)
+        end
+      end
+
+      def create_migration_file
+        migration_template 'create_impressions_table.rb', 'db/migrate/create_impressions_table.rb'
+      end
+    end
+  end
+end
diff --git a/lib/generators/impressionist/templates/create_impressions_table.rb b/lib/generators/active_record/templates/create_impressions_table.rb
similarity index 100%
rename from lib/generators/impressionist/templates/create_impressions_table.rb
rename to lib/generators/active_record/templates/create_impressions_table.rb
diff --git a/lib/generators/impressionist/impressionist_generator.rb b/lib/generators/impressionist/impressionist_generator.rb
deleted file mode 100644
index 156f1a9..0000000
--- a/lib/generators/impressionist/impressionist_generator.rb
+++ /dev/null
@@ -1,20 +0,0 @@
-require 'rails/generators'
-require 'rails/generators/migration'
-
-class ImpressionistGenerator < Rails::Generators::Base
-  include Rails::Generators::Migration
-  source_root File.join(File.dirname(__FILE__), 'templates')
-
-  def self.next_migration_number(dirname)
-    sleep 1
-    if ActiveRecord::Base.timestamped_migrations
-      Time.now.utc.strftime("%Y%m%d%H%M%S")
-    else
-      "%.3d" % (current_migration_number(dirname) + 1)
-    end
-  end
-
-  def create_migration_file
-    migration_template 'create_impressions_table.rb', 'db/migrate/create_impressions_table.rb'
-  end
-end
diff --git a/lib/generators/impressionist_generator.rb b/lib/generators/impressionist_generator.rb
new file mode 100644
index 0000000..69763b7
--- /dev/null
+++ b/lib/generators/impressionist_generator.rb
@@ -0,0 +1,7 @@
+module Impressionist
+  module Generators
+    class ImpressionistGenerator < Rails::Generators::Base
+      hook_for :orm
+    end
+  end
+end