Support attr_accessible appears multi times.

Eric-Guo · Oct 3, 2014 · a1e688b · a1e688b
1 parent f06eadb
commit a1e688b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lib/attr_accessible2strong_params/converter.rb b/lib/attr_accessible2strong_params/converter.rb
@@ -5,9 +5,13 @@
 class AttrAccessible2StrongParams::Converter
   def read_attr_accessible(filename)
     root_node, comments = parse_file_with_comments(filename)
-    m = root_node.each_node(:send).select {|n| n.children[1] == :attr_accessible}.first
-    @model_class_name = m.parent.parent.children[0].children[1]
-    @aa_list = m.each_node(:sym).collect {|n| n.children[0]}
+    aa_nodes = root_node.each_node(:send).select {|n| n.children[1] == :attr_accessible}
+    aa_fields = []
+    aa_nodes.each do |m|
+      @model_class_name = m.parent.parent.children[0].children[1]
+      aa_fields <<= m.each_node(:sym).collect {|n| n.children[0]}
+    end
+    @model_fields = aa_fields.flatten
   end
 
   def write_controller_with_strong_params(filename)