lib/ib/generator.rb in ib-0.2.3 vs lib/ib/generator.rb in ib-0.2.4

@@ -50,16 +50,21 @@
     end
     output
   end
 
   def generate_objc_impl files
-    src = files.map do |path, info|
-      <<-OBJC
+    output = ""
+    files.map do |path, infos|
+      infos.each do |info|
+        output << <<-OBJC
 @implementation #{info[:class][0][0]}
 
 @end
 OBJC
-    end.join("\n" * 2)
+        output << "\n\n"
+      end
+    end
+    output
   end
 
   def generate_type type
     type == "id" ? type : "#{type} *"
   end