56 lines
1.1 KiB
Ruby
56 lines
1.1 KiB
Ruby
class Budget
|
|
class Result
|
|
attr_accessor :budget, :heading, :current_investment
|
|
|
|
def initialize(budget, heading)
|
|
@budget = budget
|
|
@heading = heading
|
|
end
|
|
|
|
def calculate_winners
|
|
reset_winners
|
|
if @budget.hide_money?
|
|
investments.compatible.update_all(winner: true)
|
|
else
|
|
investments.compatible.each do |investment|
|
|
@current_investment = investment
|
|
set_winner if inside_budget?
|
|
end
|
|
end
|
|
end
|
|
|
|
def investments
|
|
heading.investments.selected.sort_by_ballots
|
|
end
|
|
|
|
def inside_budget?
|
|
available_budget >= @current_investment.price
|
|
end
|
|
|
|
def available_budget
|
|
total_budget - money_spent
|
|
end
|
|
|
|
def total_budget
|
|
heading.price
|
|
end
|
|
|
|
def money_spent
|
|
@money_spent ||= 0
|
|
end
|
|
|
|
def reset_winners
|
|
investments.update_all(winner: false)
|
|
end
|
|
|
|
def set_winner
|
|
@money_spent += @current_investment.price
|
|
@current_investment.update!(winner: true)
|
|
end
|
|
|
|
def winners
|
|
investments.where(winner: true)
|
|
end
|
|
end
|
|
end
|