X-Git-Url: https://projects.mako.cc/source/selectricity-live/blobdiff_plain/34fe63727a8929d7e9c529feee1f0c1e827acdfa..08ca43a9196b53ae078f6f7737fa544dd2343923:/app/controllers/site_controller.rb

diff --git a/app/controllers/site_controller.rb b/app/controllers/site_controller.rb
index 47ae2cb..80fdfc4 100644
--- a/app/controllers/site_controller.rb
+++ b/app/controllers/site_controller.rb
@@ -1,22 +1,27 @@
 class SiteController < ApplicationController
-  layout 'hc'
-  model :user, :election
+  layout 'main'
+  require_dependency "user"
+  require_dependency "election"
+  require_dependency "account"
 
   def index
+    @quickvotes = QuickVote.find(:all).sort {|a,b| b.enddate <=> a.enddate}[0..1]
     # if the person claims they are logged in
-    if session[:user]
-
+   
+    if logged_in?
       # check to see that we actually have record of them
-      if User.find_all(["id = ?", session[:user].id]).length == 1
+      if User.exists?(session[:user])
+        username = User.find(session[:user]).login
         # if we have record of them, grab the list of their elections
-        session[:user] = User.find(session[:user].id)
+        session[:user] = User.find(session[:user])
         @current_elections = session[:user].elections.sort do |a,b|
           b.enddate <=> a.enddate
         end
       else
-        # if we have no record of them, set the session id back to
-	# nothing and start again
+        # if we have no record of them, set the user back to
+	      # nothing and start again
         session[:user] = nil
+        
       end
     end
   end