lib/ib/generator.rb in ib-0.1.0 vs lib/ib/generator.rb in ib-0.1.1

@@ -33,11 +33,11 @@
   def generate_objc files
     src = files.map do |path, info|
 <<-OBJC
 @interface #{info[:class][0][0]} : #{info[:class][0][1]}
 
-#{info[:outlets].map {|name, type| "@property IBOutlet #{type} * #{name};" }.join("\n")}
+#{info[:outlets].map {|name, type| "@property IBOutlet #{generate_type(type)} #{name};" }.join("\n")}
 
 #{info[:actions].map {|action| "-(IBAction) #{action[0]}:(id) sender;" }.join("\n")}
 
 @end
 OBJC
@@ -50,7 +50,11 @@
 @implementation #{info[:class][0][0]}
 
 @end
 OBJC
     end.join("\n" * 2)
+  end
+
+  def generate_type type
+    type == "id" ? type : "#{type} *"
   end
 end
\ No newline at end of file