manage stop areas for connection links

1 files changed, 9 insertions, 2 deletions

diff --git a/app/controllers/connection_links_controller.rb b/app/controllers/connection_links_controller.rb
index 327fb7989..8790426f7 100644
--- a/app/controllers/connection_links_controller.rb
+++ b/app/controllers/connection_links_controller.rb
@@ -2,14 +2,21 @@ class ConnectionLinksController < ChouetteController
   defaults :resource_class => Chouette::ConnectionLink
 
   belongs_to :referential do
-    #belongs_to :departure, :parent_class => Chouette::StopArea, :optional => false
-    #belongs_to :arrival, :parent_class => Chouette::StopArea, :optional => false
+    belongs_to :departure, :parent_class => Chouette::StopArea, :optional => true
+    belongs_to :arrival, :parent_class => Chouette::StopArea, :optional => true
   end
 
   respond_to :html, :xml, :json
 
+  def select_areas
+    @connection_link = connection_link
+    @departure = connection_link.departure
+    @arrival = connection_link.arrival
+  end
 
   protected
+  
+  alias_method :connection_link, :resource
 
   def collection    
     @q = referential.connection_links.search(params[:q])