lib/sugarcube-nsdata.rb in sugarcube-1.7.0 vs lib/sugarcube-nsdata.rb in sugarcube-2.0.0

@@ -1,15 +1,20 @@
 unless defined?(Motion::Project::Config)
   raise "The sugarcube gem must be required within a RubyMotion project Rakefile."
 end
 
 
+require 'sugarcube'
+
 Motion::Project::App.setup do |app|
   # scans app.files until it finds app/ (the default)
   # if found, it inserts just before those files, otherwise it will insert to
   # the end of the list
   insert_point = app.files.find_index { |file| file =~ /^(?:\.\/)?app\// } || 0
 
-  Dir.glob(File.join(File.dirname(__FILE__), 'sugarcube-nsdata/**/*.rb')).reverse.each do |file|
+  Dir.glob(File.join(File.dirname(__FILE__), App.template.to_s, 'sugarcube-nsdata/**/*.rb')).reverse.each do |file|
+    app.files.insert(insert_point, file)
+  end
+  Dir.glob(File.join(File.dirname(__FILE__), 'cocoa/sugarcube-nsdata/**/*.rb')).reverse.each do |file|
     app.files.insert(insert_point, file)
   end
 end