resource_generator.rb in adminpanel-1.1.0

- old
+ new

@@ -16,20 +16,29 @@
 			def create_model
         		template 'resource.rb', "app/models/adminpanel/#{lower_name}.rb"
 			end
 
 			def create_controller
-				template "controller.rb", "app/controllers/adminpanel/#{pluralized_name}_controller.rb"
+				if is_a_resource?
+					template "controller.rb", "app/controllers/adminpanel/#{pluralized_name}_controller.rb"
+				end
 			end
 
 			def create_migrations
 				migration_template "migration.rb", "db/migrate/create_#{pluralized_name}_table"
-				# generate "migration.rb", "db/migrate/create_#{pluralized_name}_table"
 			end
 
 
-			private
+		private
+			def is_a_resource?
+				fields.each do |field, type|
+					if type != "belongs_to"
+						return true
+					end
+				end
+				false
+			end
 
 			def lower_name
 				resource_name.singularize.downcase
 			end
 
@@ -39,15 +48,44 @@
 
 			def pluralized_name
 				"#{lower_name.pluralize}"
 			end
 
+			def belongs_to_field(resource)
+				"#{resource.singularize.downcase}_id"
+			end
+
+			def has_many_field(resource)
+				"#{resource.singularize.downcase}_ids"
+			end
+
+			def resource_class_name(resource)
+				"#{resource.singularize.capitalize}"
+			end
+
+			def models_in_parameter(field)
+				models = []
+				field.split(",").each do |member|
+					models << member.downcase.pluralize
+				end
+				models
+			end
+
 			def symbolized_attributes
 		        attributes = ""
 		        fields.each do |field, type|
 		        	if type == "images"
 		            	attributes = attributes + ":images_attributes, "
+		        	elsif type == "belongs_to"
+		        		attributes = "#{attributes}:#{belongs_to_field(field)}, "
+		        	elsif type == "has_many" || type == "has_many_through"
+		        		if field.split(",").second == nil
+		        			attributes = "#{attributes}:#{has_many_field(field)}, "
+		        		else
+		        			model_name = models_in_parameter(field).first
+		        			attributes = "#{attributes}:#{has_many_field(model_name)}, "
+		        		end
 		        	else
 		            	attributes = attributes + ":#{field}, "
 		        	end
 		        end
 		        2.times do
@@ -61,45 +99,91 @@
 				fields.each do |field, type|
 					if type == "string" || type == "float"
 						form_hash = form_hash + "\n\t\t\t\t{\"#{field}\" => {\"type\" => \"text_field\", \"name\" => \"#{field}\", \"label\" => \"#{field}\", \"placeholder\" => \"#{field}\"}},"
 					elsif type == "text" || type == "wysiwyg"
 						form_hash = form_hash + "\n\t\t\t\t{\"#{field}\" => {\"type\" => \"wysiwyg_field\", \"name\" => \"#{field}\", \"label\" => \"#{field}\", \"placeholder\" => \"#{field}\"}},"
-					elsif type == "integer"
+					elsif type == "integer" 
 						form_hash = form_hash + "\n\t\t\t\t{\"#{field}\" => {\"type\" => \"number_field\", \"name\" => \"#{field}\", \"label\" => \"#{field}\", \"placeholder\" => \"#{field}\"}},"
 					elsif type == "datepicker"
 						form_hash = form_hash + "\n\t\t\t\t{\"#{field}\" => {\"type\" => \"datepicker\", \"name\" => \"#{field}\", \"label\" => \"#{field}\", \"placeholder\" => \"#{field}\"}},"
 					elsif type == "images"
 						form_hash = form_hash + "\n\t\t\t\t{\"#{field}\" => {\"type\" => \"adminpanel_file_field\", \"name\" => \"#{field}\"}},"
+					elsif type == "belongs_to"
+						form_hash = form_hash + "\n\t\t\t\t{\"#{belongs_to_field(field)}\" => {\"type\" => \"belongs_to\", \"model\" => \"Adminpanel\:\:#{resource_class_name(field)}\", \"name\" => \"#{belongs_to_field(field)}\"}},"
+					elsif type == "has_many" || type == "has_many_through"
+						if models_in_parameter(field).second.nil? 
+							through_model = field
+						else 
+							through_model = models_in_parameter(field).first
+						end 
+						through_model = resource_class_name(through_model)
+						form_hash = form_hash + "\n\t\t\t\t{\"#{has_many_field(through_model)}\" => {\"type\" => \"has_many\", \"model\" => \"Adminpanel\:\:#{through_model}\", \"name\" => \"#{has_many_field(through_model)}\"}},"
 					end
 				end
 				form_hash
 			end
 
 			def migration_string(field, type)
 				if type == "datepicker"
 					"t.string :#{field}"
-				elsif type == "images"
-					""
 				elsif type == "wysiwyg"
 					"t.text :#{field}"
+				elsif type == "belongs_to"
+					"t.integer :#{belongs_to_field(field)}"
+				elsif type == "images" || type == "has_many" || type == "has_many_through"
+					""# no need for an association here.
 				else
 					"t.#{type} :#{field}"
 				end
 			end
 
-			def has_images?
+			def has_associations?
 				fields.each do |field, type|
-					if type == "images"
+					if type == "images" || type == "belongs_to" || type == "has_many" || type == "has_many_through"
 						return true
 					end
 				end
 				return false
 			end
 
-			def image_relationship
-				return "has_many :images, :foreign_key => \"foreign_key\", :conditions => { :model => \"#{lower_name}\" }
-		accepts_nested_attributes_for :images, :allow_destroy => true
-		#remember to change the relationship if you change this model display_name"
+			def associations
+				association = ""
+				fields.each do |field, type|
+					if type == "belongs_to"
+						association = "#{association}#{belongs_to_association(field)}"
+					elsif type == "images"
+						association = "#{association}#{image_association}"
+					elsif type == "has_many" || type == "has_many_through"
+						association = "#{association}#{has_many_association(field)}"
+					end
+						
+				end
+				association
+			end
+
+			def belongs_to_association(field)
+				"belongs_to :#{field.singularize.downcase}\n\t\t"
+			end
+
+			def has_many_association(field)
+				if models_in_parameter(field).second.nil?
+					return "has_many :#{models_in_parameter(field).first}\n\t\t" + 
+					"has_many :#{models_in_parameter(field).first}, " + 
+					":through => :#{models_in_parameter(field).first}ation, " + 
+					":dependent => :destroy\n\t\t"
+				else
+					return "has_many :#{models_in_parameter(field).second}\n\t\t" +
+					"has_many :#{models_in_parameter(field).first}, " + 
+					":through => :#{models_in_parameter(field).second}, " + 
+					":dependent => :destroy\n\t\t"
+				end
+			end
+
+			def image_association
+				return "has_many :images, :foreign_key => \"foreign_key\", " + 
+				":conditions => { :model => \"#{lower_name}\" } \n\t\t " +
+				"accepts_nested_attributes_for :images, :allow_destroy => true\n\t\t" +
+				"#remember to change the association if you change this model display_name\n\t\t"
 			end
 		end
 	end
 end
\ No newline at end of file