lib/generators/system_metrics/install/install_generator.rb in system-metrics-0.2.3 vs lib/generators/system_metrics/install/install_generator.rb in system-metrics-0.2.4

@@ -7,15 +7,19 @@
       desc "Install System Metrics public assets"
 
       source_root File.expand_path("../../../../../public", __FILE__)
 
       def copy_css_files
-        directory "stylesheets", "public/stylesheets/system_metrics", :recursive => true
+        directory "stylesheets", "#{assets_path_for_rails_versions}/stylesheets/system_metrics", :recursive => true
       end
 
       def copy_image_files
-        directory "images", "public/images/system_metrics", :recursive => true
+        directory "images", "#{assets_path_for_rails_versions}/images/system_metrics", :recursive => true
       end
 
+      private
+        def assets_path_for_rails_versions
+          ::Rails.version >= "3.1.0" ? "app/assets" : "public"
+        end
     end
   end
 end