can add investments to ballot

app/controllers/budgets/ballot/lines_controller.rb

@@ -0,0 +1,80 @@
+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 @line.save
+          #@ballot.set_geozone(@geozone)
+          #@current_user.update(representative_id: nil)
+          if request.get?
+            redirect_to @spending_proposal, notice: t('spending_proposals.notice.voted')
+          end
+        else
+          if request.get?
+            redirect_to @spending_proposal, notice: t('spending_proposals.notice.could_not_vote')
+          else
+            render :new
+          end
+        end
+      end
+
+      def destroy
+        load_spending_proposal
+        load_spending_proposals
+        load_geozone
+
+        @ballot_line.destroy
+        @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
+
+        def load_geozone
+          @geozone = @line.investment.geozone
+        end
+    end
+  end
+end

app/controllers/budgets/ballots_controller.rb

@@ -0,0 +1,17 @@
+module Budgets
+  class BallotsController < ApplicationController
+    before_action :authenticate_user!
+    before_action :load_ballot
+    load_and_authorize_resource
+
+    def show
+    end
+
+    private
+
+      def load_ballot
+        @ballot = Ballot.where(user: current_user).first_or_create
+      end
+
+  end
+end