Merge pull request #6 from kaspth/infer-model

Infer model from form name.

Author: kaspth

lib/active_form/base.rb

@@ -5,16 +5,16 @@ class Base
 
     define_model_callbacks :save, only: [:after]
     after_save :update_form_models
-    
+
     delegate :persisted?, :to_model, :to_key, :to_param, :to_partial_path, to: :model
     attr_reader :model, :forms
-    
+
     def initialize(model)
       @model = model
       @forms = []
       populate_forms
     end
-    
+
     def submit(params)
       params.each do |key, value|
         if nested_params?(value)
@@ -48,12 +48,13 @@ def valid?
 
       collect_errors_from(model)
       aggregate_form_errors
-      
+
       errors.empty?
     end
 
     class << self
-      attr_accessor :main_class, :main_model
+      attr_accessor :main_class
+      attr_writer :main_model
       delegate :reflect_on_association, to: :main_class
 
       def attributes(*names)
@@ -69,7 +70,11 @@ def attributes(*names)
       end
 
       def main_class
-        @main_class = main_model.to_s.camelize.constantize
+        @main_class ||= main_model.to_s.camelize.constantize
+      end
+
+      def main_model
+        @main_model ||= name.sub(/Form$/, '').singularize
       end
 
       alias_method :attribute, :attributes

test/dummy/app/forms/conference_form.rb

@@ -1,5 +1,4 @@
 class ConferenceForm < ActiveForm::Base
-  self.main_model = :conference
   attributes :name, :city, required: true
 
   association :speaker do
@@ -9,4 +8,4 @@ class ConferenceForm < ActiveForm::Base
       attribute :topic, :duration, required: true
     end
   end
-end
+end

test/dummy/app/forms/project_form.rb

@@ -1,5 +1,4 @@
 class ProjectForm < ActiveForm::Base
-  self.main_model = :project
   attributes :name, :description, :owner_id
 
   association :tasks do
@@ -16,7 +15,7 @@ class ProjectForm < ActiveForm::Base
 
   association :project_tags do
     attribute :tag_id
-    
+
     association :tag do
       attribute :name
     end
@@ -25,4 +24,4 @@ class ProjectForm < ActiveForm::Base
   association :owner do
     attributes :name, :description, :role
   end
-end
+end

test/dummy/app/forms/song_form.rb

@@ -1,5 +1,4 @@
 class SongForm < ActiveForm::Base
-  self.main_model = :song
   attributes :title, :length, required: true
 
   association :artist do
@@ -9,4 +8,4 @@ class SongForm < ActiveForm::Base
       attributes :name, :studio, required: true
     end
   end
-end
+end

test/dummy/app/forms/survey_form.rb

@@ -1,5 +1,4 @@
 class SurveyForm < ActiveForm::Base
-  self.main_model = :survey
   attribute :name, required: true
 
   association :questions do
@@ -9,4 +8,4 @@ class SurveyForm < ActiveForm::Base
       attribute :content, required: true
     end
   end
-end
+end

test/fixtures/user_form_fixture.rb

@@ -1,5 +1,6 @@
 class UserFormFixture < ActiveForm::Base
   self.main_model = :user
+
   attributes :name, :age, :gender, required: true
 
   validates :name, length: { in: 6..20 }