casey_jones 0.0.113 → 0.0.114

data/lib/generators/extra_user_attributes/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Explain the generator
+
+Example:
+    rails generate extra_user_attributes Thing
+
+    This will create:
+        what/will/it/create

data/lib/generators/extra_user_attributes/extra_user_attributes_generator.rb

@@ -0,0 +1,38 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+require 'rails/generators/generated_attribute'
+require 'rails/generators/active_record'
+
+class ExtraUserAttributesGenerator < Rails::Generators::Base
+  include Rails::Generators::Migration
+  source_root File.expand_path('../templates', __FILE__)
+  require File.expand_path('../migration_emitter', __FILE__)
+
+  def manifest
+    self.class_name = "User"
+    self.emitter = MigrationEmitter.entity(:users)
+    self.emitter.column :confirmation_token, :string
+    self.emitter.column :confirmed_at, :datetime
+    self.emitter.column :confirmation_sent_at, :datetime
+    self.emitter.column :failed_attempts, :integer, :default=>0
+    self.emitter.column :unlock_token, :string
+    self.emitter.column :locked_at, :datetime
+    self.emitter.column :authentication_token, :string
+    self.emitter.column :login, :string
+    self.emitter.column :name, :string
+
+    migration_template 'add_fields_migration.rb', "db/migrate/add_fields_to_users"
+
+  end
+  attr_accessor :emitter, :class_name
+
+  def self.next_migration_number(dirname)
+    if ActiveRecord::Base.timestamped_migrations
+      Time.now.utc.strftime("%Y%m%d%H%M%S")
+    else
+      "%.3d" % (current_migration_number(dirname) + 1)
+    end
+  end
+
+end
+

data/lib/generators/extra_user_attributes/migration_emitter.rb

@@ -0,0 +1,79 @@
+class MigrationEmitter
+  attr_accessor :table_name
+  attr_accessor :columns
+  attr_accessor :options
+
+  def initialize(table_name, initial = [], options={})
+    self.table_name = table_name
+    self.columns = initial
+    self.options = options
+  end
+
+  def self.join_table(*classes)
+    a = self.new classes.map_to(:to_s).map_to(:tableize).sort.join('_').to_sym
+    (0..1).each {|i|
+      a.column classes[i].foreign_key, :integer, :index=>:true
+    }
+    a.options[:id] = false
+    a
+  end
+
+  def self.entity(class_name, *initial)
+    return self.new(class_name.to_s.tableize.to_sym, initial)
+  end
+
+  def column(name, datatype, options={})
+    self.columns << {:name=>name, :datatype=>datatype, :options=>options}
+  end
+
+  def create_table
+    return "
+    create_table #{create_table_def} do |t|
+      #{columns.map{|c| coldef(c)}.join("\n    ")}
+      t.timestamps
+    end
+    " + create_indexes
+  end
+
+  def drop_table
+    "drop_table #{table_name.inspect}"
+  end
+
+  def add_columns
+    columns.map{|c|
+      "add_column " +
+        [table_name.inspect, c[:name].inspect, c[:datatype].inspect,
+          map_inspect(c[:options].except(:index))
+        ].flatten.join(', ')
+    }.join("\n    ") + "\n    " + create_indexes
+  end
+
+  def remove_columns
+    columns.map{|c| "remove_column #{table_name.inspect}, #{c[:name].inspect}"}.join("\n    ")
+  end
+
+
+  private
+
+  def create_indexes
+    columns.select{|t| t[:options][:index] }.map{|t|
+      "add_index :#{table_name}, :#{t[:name]}" + (t[:options][:index]==:unique ? ", :unique=>true" : "")
+    }.join('\n    ')
+  end
+
+  def coldef(col)
+    ["t.#{col[:datatype]} :#{col[:name]}",
+      map_inspect(col[:options].except(:index))].
+      flatten.join(', ')
+  end
+
+  def create_table_def
+    [table_name.inspect, map_inspect(options)].
+      flatten.join(", ")
+  end
+
+  def map_inspect(map)
+    map.map{|a,b| "#{a.inspect} => #{b.inspect}"}
+  end
+end
+

data/lib/generators/extra_user_attributes/templates/add_fields_migration.rb

@@ -0,0 +1,10 @@
+class AddFieldsTo<%= class_name.pluralize.camelize %> < ActiveRecord::Migration
+  def self.up
+    <%= emitter.add_columns %>
+  end
+
+  def self.down
+    <%= emitter.remove_columns %>
+  end
+end
+

data/lib/generators/extra_user_attributes/templates/create_table_migration.rb

@@ -0,0 +1,10 @@
+class CreateTable<%= class_name.pluralize.camelize %> < ActiveRecord::Migration
+  def self.up
+    <%= emitter.create_table %>
+  end
+
+  def self.down
+    <%= emitter.drop_table %>
+  end
+end
+

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: casey_jones
 version: !ruby/object:Gem::Version 
-  hash: 253
+  hash: 251
   prerelease: false
   segments: 
   - 0
   - 0
-  - 113
-  version: 0.0.113
+  - 114
+  version: 0.0.114
 platform: ruby
 authors: 
 - Tyler Gannon
@@ -45,6 +45,11 @@ files:
 - lib/generators/anaf_habtm/anaf_habtm_generator.rb
 - lib/generators/anaf_habtm/assets/nested_attributes.css
 - lib/generators/anaf_habtm/assets/nested_attributes.js
+- lib/generators/extra_user_attributes/USAGE
+- lib/generators/extra_user_attributes/extra_user_attributes_generator.rb
+- lib/generators/extra_user_attributes/migration_emitter.rb
+- lib/generators/extra_user_attributes/templates/add_fields_migration.rb
+- lib/generators/extra_user_attributes/templates/create_table_migration.rb
 - lib/generators/loadbehind/USAGE
 - lib/generators/loadbehind/loadbehind_generator.rb
 - lib/generators/loadbehind/templates/destroy.js.haml