Make some changes to avoid unnacessaty lines of code

2 files changed, 21 insertions, 16 deletions

diff --git a/app/controllers/calendars_controller.rb b/app/controllers/calendars_controller.rb
index 27c944f5f..193680342 100644
--- a/app/controllers/calendars_controller.rb
+++ b/app/controllers/calendars_controller.rb
@@ -1,10 +1,11 @@
 class CalendarsController < ChouetteController
-  include WorkgroupSupport
   include PolicyChecker
   defaults resource_class: Calendar
   before_action :ransack_contains_date, only: [:index]
   respond_to :html
   respond_to :js, only: :index
+  
+  belongs_to :workgroup
 
   def index
     index! do
@@ -15,7 +16,7 @@ class CalendarsController < ChouetteController
   def show
     show! do
       @calendar = @calendar.decorate(context: {
-        workgroup: current_workgroup
+        workgroup: workgroup
       })
     end
   end
@@ -27,7 +28,7 @@ class CalendarsController < ChouetteController
       calendars,
       with: CalendarDecorator,
       context: {
-        workgroup: current_workgroup
+        workgroup: workgroup
       }
     )
   end
@@ -47,26 +48,30 @@ class CalendarsController < ChouetteController
   end
 
   protected
+
+  alias_method :workgroup, :parent
+  helper_method :workgroup
+
   def resource
-    @calendar = Calendar.where('(organisation_id = ? OR shared = ?) AND workgroup_id = ?', current_organisation.id).find_by_id(params[:id], true, @workgroup.id)
+    @calendar ||= workgroup.calendars.where('(organisation_id = ? OR shared = ?)', current_organisation.id, true).find_by_id(params[:id])
   end
 
   def build_resource
     super.tap do |calendar|
-      calendar.workgroup = current_workgroup
+      calendar.workgroup = workgroup
       calendar.organisation = current_organisation
     end
   end
 
   def collection
-    return @calendars if @calendars
-    scope = Calendar.where('(organisation_id = ? OR shared = ?) AND workgroup_id = ?', current_organisation.id, true, @workgroup.id)
-    scope = shared_scope(scope)
-    @q = scope.ransack(params[:q])
-
-    calendars = @q.result
-    calendars = calendars.order(sort_column + ' ' + sort_direction) if sort_column && sort_direction
-    @calendars = calendars.paginate(page: params[:page])
+    @calendars ||= begin
+		    scope = workgroup.calendars.where('(organisation_id = ? OR shared = ?)', current_organisation.id, true)
+		    scope = shared_scope(scope)
+		    @q = scope.ransack(params[:q])
+		    calendars = @q.result
+		    calendars = calendars.order(sort_column + ' ' + sort_direction) if sort_column && sort_direction
+		    calendars = calendars.paginate(page: params[:page])
+                  end 
   end
 
   def ransack_contains_date
@@ -91,4 +96,4 @@ class CalendarsController < ChouetteController
     scope
   end
 
-end
+end
\ No newline at end of file
diff --git a/app/views/calendars/index.html.slim b/app/views/calendars/index.html.slim
index 309f0e6f9..0b58c0c72 100644
--- a/app/views/calendars/index.html.slim
+++ b/app/views/calendars/index.html.slim
@@ -1,4 +1,4 @@
-- breadcrumb :calendars, current_workgroup
+- breadcrumb :calendars, workgroup
 
 .page_content
   .container-fluid
@@ -16,7 +16,7 @@
                 key: :name, \
                 attribute: 'name', \
                 link_to: lambda do |calendar| \
-                  workgroup_calendar_path(current_workgroup, calendar) \
+                  workgroup_calendar_path(workgroup, calendar) \
                 end \
               ), \
               TableBuilderHelper::Column.new( \