diff --git a/app/controllers/budgets/results_controller.rb b/app/controllers/budgets/results_controller.rb index 8da1e9ef7..96f52b8ca 100644 --- a/app/controllers/budgets/results_controller.rb +++ b/app/controllers/budgets/results_controller.rb @@ -1,22 +1,31 @@ module Budgets class ResultsController < ApplicationController + before_action :load_budget + before_action :load_heading load_and_authorize_resource :budget def show authorize! :read_results, @budget @investments = load_result.investments - @heading = heading end private def load_result - Budget::Result.new(@budget, heading) + Budget::Result.new(@budget, @heading) end - def heading - @budget.headings.find(params[:heading_id]) + def load_budget + @budget = Budget.find_by(id: params[:budget_id]) + end + + def load_heading + if params[:heading_id].present? + @heading = @budget.headings.find(params[:heading_id]) + else + @heading = @budget.headings.first + end end end