scaffold_plus 1.3.0 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a08fcefd45e286fd8445802b84da15d435af04e8
-  data.tar.gz: 07646c0fcff9389f982d2b2b5c3972e43715bdae
+  metadata.gz: c5b490cd1736e29f658be2225acfb7acf13d583d
+  data.tar.gz: 65a025ea73ecce6f5a7ec16b745ed5a903d8e694
 SHA512:
-  metadata.gz: 2bc399c4a5be223a17ec05c73d750831c2648ed236539aefa70a10c5d878c794eeb9cd2be86e54e184979f878b19ad8ef443121b71a93406007fbb850bda7fb6
-  data.tar.gz: 82b65abe4d9d91a9c5890949608fd822833129a57bd586d784461f95e751c525c82281f4c89ae86115f7c3ae4939b2d9a90f02daef0ce20a279bc18fde14294d
+  metadata.gz: 1d45b528fc3807500c43485658ed1510bc97daef86d980fdf6d7af2d8604306d80867ccbd3dc80338afa70a2680b034e25f341e4f50be949a5e5fc1345fe0f0b
+  data.tar.gz: 73f83b76bacaf97b7248a5056735d928b8633ebae1649171ebc60cfe81a8a5924b7afef2e58d75a909288c2bd8a19f06d7e7d385c30db83f033fa26eb55d5e46

data/README.md

@@ -44,6 +44,12 @@ This helper creates a join table and updates the two parent resources.
 It can handle additional attributes in the join table incl. whitelisting
 and accepts_nested_attributes_for in one of the parents.
 
+### Add many-to-many association with has_and_belongs_to_many
+    rails generate scaffold_plus:habtm
+
+This helper scaffolds a has_and_belongs_to_many relationship with migration
+and updates to the models.
+
 ## Testing
 
 Since I have no experience with test driven development (yet), this is

data/lib/generators/scaffold_plus/habtm/habtm_generator.rb

@@ -0,0 +1,77 @@
+require 'rails/generators/active_record'
+
+module ScaffoldPlus
+  module Generators
+    class HabtmGenerator < ActiveRecord::Generators::Base
+      desc "Add many-to-many association with has_and_belongs_to_many"
+      argument :name, type: :string,
+               desc: "The first object that has_and_belongs_to_many OTHERs"
+      argument :other, type: :string,
+               desc: "The second object that has_and_belongs_to_many NAMEs"
+      class_option :migration, type: :boolean, default: true,
+               desc: 'Create a migration for the join table'
+      class_option :permit, type: :boolean, default: true,
+               desc: 'Allow mass assignment for added attributes in NAME'
+      class_option :before, type: :array,
+               desc: 'Add a line before generated text in models'
+      class_option :after, type: :array,
+               desc: 'Add a line after generated text in models'
+      source_root File.expand_path('../templates', __FILE__)
+      
+      def add_migration
+        return unless options.migration?
+        migration_template "habtm_migration.rb", "db/migrate/#{habtm_migration_name}.rb"
+      end
+      
+      def add_to_models
+        inject_into_class "app/models/#{name}.rb", class_name do
+          text = before_array.include?(name) ? "\n" : ""
+          text << "  has_and_belongs_to_many :#{other.pluralize}\n"
+          text << "\n" if after_array.include?(name)
+          text
+        end
+        inject_into_class "app/models/#{other}.rb", other.camelize do
+          text = before_array.include?(other) ? "\n" : ""
+          text << "  has_and_belongs_to_many :#{table_name}\n"
+          text << "\n" if after_array.include?(other)
+          text
+        end
+      end
+      
+      def add_to_permit
+        return unless options.permit?
+        text = "{ :#{other}_ids => [] }"
+        file = "app/controllers/#{table_name}_controller.rb"
+        gsub_file file, /(permit\(.*)\)/, "\\1, #{text})"
+        # Special case: no previous permit
+        gsub_file file, /^(\s*params)\[:#{name}\]$/, "\\1.require(:#{name}).permit(#{text})"
+      end
+      
+      protected
+      
+      def before_array
+        options['before'] || []
+      end
+      
+      def after_array
+        options['after'] || []
+      end
+      
+      def sorted_sg
+        [name, other].sort
+      end
+      
+      def sorted_pl
+        [table_name, other.pluralize].sort
+      end
+      
+      def habtm_migration_name
+        "create_#{sorted_pl.first}_and_#{sorted_pl.second}"
+      end
+      
+      def habtm_table_name
+        "#{sorted_pl.first}_#{sorted_pl.second}"
+      end
+    end
+  end
+end

data/lib/generators/scaffold_plus/habtm/templates/habtm_migration.rb

@@ -0,0 +1,10 @@
+class <%= habtm_migration_name.camelize %> < ActiveRecord::Migration
+  def change
+    create_table :<%= habtm_table_name %>, id: false do |t|
+      t.belongs_to :<%= sorted_sg[0] %>
+      t.belongs_to :<%= sorted_sg[1] %>
+    end
+    add_index :<%=  habtm_table_name %>, :<%= sorted_sg[0] %>_id
+    add_index :<%=  habtm_table_name %>, :<%= sorted_sg[1] %>_id
+  end
+end

data/lib/scaffold_plus/version.rb

@@ -1,3 +1,3 @@
 module ScaffoldPlus
-  VERSION = "1.3.0"
+  VERSION = "1.4.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: scaffold_plus
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: ruby
 authors:
 - Volker Wiegand
@@ -83,6 +83,8 @@ files:
 - lib/generators/scaffold_plus/ancestry/templates/ancestry_migration.rb
 - lib/generators/scaffold_plus/collection/collection_generator.rb
 - lib/generators/scaffold_plus/collection/templates/view.html.erb
+- lib/generators/scaffold_plus/habtm/habtm_generator.rb
+- lib/generators/scaffold_plus/habtm/templates/habtm_migration.rb
 - lib/generators/scaffold_plus/has_many/has_many_generator.rb
 - lib/generators/scaffold_plus/has_many/templates/child_migration.rb
 - lib/generators/scaffold_plus/has_many/templates/counter_migration.rb