module Budgets
  module Ballot
    class LinesController < ApplicationController
      before_action :authenticate_user!
      #before_action :ensure_final_voting_allowed
      before_action :load_budget
      before_action :load_ballot

      before_action :load_investments

      load_and_authorize_resource :budget
      load_and_authorize_resource :ballot, class: "Budget::Ballot"
      load_and_authorize_resource :line, through: :ballot, find_by: :investment_id, class: "Budget::Ballot::Line"

      def create
        load_investment
        load_heading

        if @ballot.add_investment(@investment)
          #@ballot.set_geozone(@geozone)
          #@current_user.update(representative_id: nil)
          if request.get?
            redirect_to @investment, notice: t('budget_investments.notice.voted')
          end
        else
          if request.get?
            redirect_to @investment, notice: t('budget_investments.notice.could_not_vote')
          else
            render :new
          end
        end
      end

      def destroy
        @investment = @line.investment
        load_heading

        @line.destroy
        load_investments
        #@ballot.reset_geozone
      end

      private

        def ensure_final_voting_allowed
          return head(:forbidden) unless @budget.balloting?
        end

        def line_params
          params.permit(:investment_id)
        end

        def load_budget
          @budget = Budget.find(params[:budget_id])
        end

        def load_ballot
          @ballot = Budget::Ballot.where(user: current_user, budget: @budget).first_or_create
        end

        def load_investment
          @investment = Budget::Investment.find(params[:investment_id])
        end

        def load_investments
          if params[:investments_ids].present?
            @investments = Budget::Investment.where(id: params[:investments_ids])
          end
        end

        def load_heading
          @heading = @investment.heading
        end

    end
  end
end