a long variety of improvements and tweaks

[selectricity] / app / controllers / site_controller.rb

diff --git a/app/controllers/site_controller.rb b/app/controllers/site_controller.rb
index 8d2f8b2a63c46a719efac7da711c7ad646f7e458..47ae2cbd87af5dbde48f59c1fdbb6feb44c92f4d 100644 (file)
--- a/app/controllers/site_controller.rb
+++ b/app/controllers/site_controller.rb
@@ -1,12 +1,24 @@
 class SiteController < ApplicationController
-  layout 'vb'
+  layout 'hc'
   model :user, :election
 
   def index
-    @current_elections = session[:user].elections.sort do |a,b|
-      b.enddate <=> a.enddate
-    end
+    # if the person claims they are logged in
+    if session[:user]
 
-    
+      # check to see that we actually have record of them
+      if User.find_all(["id = ?", session[:user].id]).length == 1
+        # if we have record of them, grab the list of their elections
+        session[:user] = User.find(session[:user].id)
+        @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
+        session[:user] = nil
+      end
+    end
   end
+
 end