Display score

app/controllers/habits_controller.rb

@@ -6,7 +6,8 @@ class HabitsController < ApplicationController
   # GET /habits.json
   def index
     @habits = current_user.habits.where(:habit_type => Habit::ALWAYS)
-    @todos = current_user.habits.where(:habit_type => Habit::DAILY)
+    @daily = current_user.habits.where(:habit_type => Habit::DAILY)
+    @score = current_user.habits.sum('score')
 
     respond_to do |format|
       format.html # index.html.erb
@@ -80,6 +81,7 @@ class HabitsController < ApplicationController
   def vote
     @habit = current_user.habits.find(params[:id])
     @habit.vote(params[:vote])
+    @score = current_user.habits.sum('score')
         
     respond_to do |format|
       if @habit.save

app/views/habits/index.html.erb

@@ -1,4 +1,4 @@
-
+<h1>Score: <span id="score"><%= @score %></span></h1>
 <%= link_to 'New Habit', new_habit_path %>
 
 <table>
@@ -17,7 +17,7 @@
   <div id="daily">
     <h2>Daily</h2>
     <ul>
-    <% @todos.each do |habit| %>
+    <% @daily.each do |habit| %>
       <li><%= render :partial => "habit", :locals => { :habit => habit } %></li>
     <% end %>
     </ul>

app/views/habits/vote.js.erb

@@ -1 +1,2 @@
-$('#<%= @habit.id %>').html("<%= escape_javascript(render :partial => "habit", :locals => { :habit => @habit }) %>")
+$('#<%= @habit.id %>').html("<%= escape_javascript(render :partial => "habit", :locals => { :habit => @habit }) %>");
+$('#score').html("<%= @score %>");