0
@@ -16,10 +16,42 @@ class ControllerGenerator < Genosaurus
0
   end
0
   
0
   def after_generate # :nodoc:
0
+    add_actions
0
+    update_routes_file
0
+    ControllerHelperGenerator.run(@options)
0
+  end
0
+  
0
+  private
0
+  def update_routes_file # :nodoc:
0
+    unless @actions.empty?
0
+      routes = File.join(Mack.root, "config", "routes.rb")
0
+      rf = File.open(routes).read
0
+      unless rf.match(".resource :#{@name_plural}")
0
+        puts "Updating routes.rb"
0
+        nrf = ""
0
+        rf.each do |line|
0
+          if line.match("Mack::Routes.build")
0
+            x = line.match(/\|(.+)\|/).captures
0
+            line << "\n  # Added by rake generate:controller name=#{param(:name)} actions=#{param(:actions)}\n"
0
+            line << "\n  r.with_options(:controller => :#{@name_plural}) do |map|\n"
0
+            @actions.each do |action|
0
+              line << "\n    map.#{@name_plural}_#{action}_url \"/#{@name_plural}#{action == "index" ? "" : "/#{action}"}\", :action => :#{action}"
0
+            end
0
+            line << "\n  end # #{@name_plural}\n"
0
+          end
0
+          nrf << line
0
+        end
0
+        File.open(routes, "w") do |f|
0
+          f.puts nrf
0
+        end
0
+      end
0
+    end
0
+  end
0
+  
0
+  def add_actions
0
     @actions.each do |action|
0
       template(action_template(action), File.join("app", "views", @name_plural, "#{action}.html.erb"))
0
-    end
0
-    ControllerHelperGenerator.run(@options)
0
+    end    
0
   end
0
   
0
   def action_template(action) # :nodoc:

0
@@ -12,10 +12,15 @@ describe ControllerGenerator do
0
   
0
   before(:each) do
0
     common_cleanup
0
+    @route_file_loc = File.join(Mack.root, "config", "routes.rb")
0
+    @original_route_file = File.read(@route_file_loc)
0
   end
0
   
0
   after(:each) do
0
     common_cleanup
0
+    File.open(@route_file_loc, "w") do |f|
0
+      f.puts @original_route_file
0
+    end
0
   end
0
   
0
   it "should require a name parameter" do
0
@@ -36,6 +41,15 @@ describe ControllerGenerator do
0
     File.read(show).should match(/<p>You can find me in app\/views\/zoos\/show.html.erb<\/p>/)
0
   end
0
   
0
+  it "should update the routes file if there are actions" do
0
+    ControllerGenerator.run("name" => "zoo", "actions" => "index,show")
0
+    File.read(@route_file_loc).should match(/  # Added by rake generate:controller name=zoo actions=index,show/)
0
+    File.read(@route_file_loc).should match(/  r.with_options\(:controller => :zoos\) do \|map\|/)
0
+    File.read(@route_file_loc).should match(/    map.zoos_index_url "\/zoos", :action => :index/)
0
+    File.read(@route_file_loc).should match(/    map.zoos_show_url "\/zoos\/show", :action => :show/)
0
+    File.read(@route_file_loc).should match(/  end # zoos/)
0
+  end
0
+  
0
   it "should generate a controller" do
0
     file = File.join(Mack.root, "app", "controllers", "zoos_controller.rb")
0
     File.should_not be_exists(file)