projects.mako.cc - selectricity/blob - app/controllers/election_controller.rb

   1 # Selectricity: Voting Machinery for the Masses
   2 # Copyright (C) 2007, 2008 Benjamin Mako Hill <mako@atdot.cc>
   3 # Copyright (C) 2007 Massachusetts Institute of Technology
   4 #
   5 # This program is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU Affero General Public License as
   7 # published by the Free Software Foundation, either version 3 of the
   8 # License, or (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful, but
  11 # WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 # Affero General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU Affero General Public
  16 # License along with this program.  If not, see
  17 # <http://www.gnu.org/licenses/>.
  18
  19 class ElectionController < ApplicationController
  20   require_dependency "raw_voter_list"
  21   require_dependency "voter"
  22   require_dependency "vote"
  23   require_dependency "candidate"
  24   layout 'main'
  25
  26   ## methods for displaying, creating,
  27   ## and manipulating election overview data
  28   ####################################################################
  29
  30   def new
  31     redirect_to :action => 'general_information'
  32   end
  33
  34   def general_information
  35     @sidebar_content = render_to_string :partial => 'progress',
  36                                         :locals => { :page => 'overview' }
  37     @election = Election.new
  38     render :action => 'general_information'
  39   end
  40
  41   def create_election
  42     @election = Election.new(params[:election])
  43
  44     # default options
  45     @election.user = session[:user]
  46     @election.anonymous = 1
  47     @election.startdate = Time.now
  48     @election.type = 'Election'
  49
  50     holder = create_theme_hash
  51     unless holder.values.all? {|v| v.has_value?("")}
  52       token_generator = UniqueTokenGenerator.new( 16 )
  53       @election.embed_custom_string = token_generator.token
  54       add_theme(@election.embed_custom_string)
  55     end
  56
  57     if @election.save
  58       flash[:notice] = 'Election was successfully created.'
  59       redirect_to :action => 'edit_candidates', :id => @election.id
  60     else
  61       render :action => 'general_information'
  62     end
  63   end
  64
  65   def create_theme_hash
  66     target = Hash.new
  67     params.each do |k,v|
  68       target[k] = v if k=="top_bar" or k=="default_image" or k=="bg1" \
  69                     or k=="bg2" or k=="bottom_bar"
  70     end
  71     return target
  72   end
  73
  74   # TODO add filter to verify that the person working on or looking at
  75   # something is the owner
  76   def edit_general_information
  77     @election = Election.find(params[:id])
  78   end
  79
  80   def update_general_information
  81     @election = Election.find(params[:id])
  82
  83     holder = create_theme_hash
  84     unless holder.values.all? {|v| v.has_value?("")}
  85       unless @election.embed_custom_string
  86         token_generator = UniqueTokenGenerator.new( 16 )
  87         @election.embed_custom_string = token_generator.token
  88       end
  89
  90       add_theme(@election.embed_custom_string)
  91     end
  92
  93     if @election.update_attributes(params[:election])
  94       flash[:notice] = 'Election was successfully updated.'
  95       redirect_to :action => 'show', :id => @election
  96     else
  97       render :action => 'edit'
  98     end
  99   end
 100
 101   #Takes care of uploading custom images
 102   #unnecessarily long, how can I compress?
 103   def add_theme(prefix)
 104     holder = create_theme_hash
 105     unless params[:top_bar][:uploaded_data].to_s.empty?
 106       previous = SkinPicture.find(:first,
 107       :conditions => ["filename = ?", @election.embed_custom_string + "top_bar.png"])
 108       if previous
 109         previous.destroy
 110       end
 111       top_bar = SkinPicture.new(params[:top_bar])
 112       top_bar.filename = prefix + "top_bar." + params[:top_bar][:uploaded_data].content_type[6..-2]
 113       top_bar.save
 114     end
 115     unless params[:default_image][:uploaded_data].to_s.empty?
 116       previous = SkinPicture.find(:first,
 117       :conditions => ["filename = ?", @election.embed_custom_string + "default_image.png"])
 118       if previous
 119         previous.destroy
 120       end
 121       default_image = SkinPicture.new(params[:default_image])
 122       default_image.filename = prefix + "default_image." + params[:default_image][:uploaded_data].content_type[6..-2]
 123       default_image.save
 124     end
 125     unless params[:bg1][:uploaded_data].to_s.empty?
 126       previous = SkinPicture.find(:first,
 127       :conditions => ["filename = ?", @election.embed_custom_string + "bg1.png"])
 128       if previous
 129         previous.destroy
 130       end
 131       bg1 = SkinPicture.new(params[:bg1])
 132       bg1.filename = prefix + "bg1." + params[:bg1][:uploaded_data].content_type[6..-2]
 133       bg1.save
 134     end
 135     unless params[:bg2][:uploaded_data].to_s.empty?
 136       previous = SkinPicture.find(:first,
 137       :conditions => ["filename = ?", @election.embed_custom_string + "bg2.png"])
 138       if previous
 139         previous.destroy
 140       end
 141       bg2 = SkinPicture.new(params[:bg2])
 142       bg2.filename = prefix + "bg2." + params[:bg2][:uploaded_data].content_type[6..-2]
 143       bg2.save
 144     end
 145     unless params[:bottom_bar][:uploaded_data].to_s.empty?
 146       previous = SkinPicture.find(:first,
 147       :conditions => ["filename = ?", @election.embed_custom_string + "bottom_bar.png"])
 148       if previous
 149         previous.destroy
 150       end
 151       bottom_bar = SkinPicture.new(params[:bottom_bar])
 152       bottom_bar.filename = prefix + "bottom_bar." + params[:bottom_bar][:uploaded_data].content_type[6..-2]
 153       bottom_bar.save
 154     end
 155
 156   end
 157
 158   def show
 159     @sidebar_content = render_to_string :partial => 'progress',
 160                                         :locals => { :page => 'review' }
 161
 162     @election = Election.find(params[:id])
 163     if @election.class  == QuickVote
 164       redirect_to(:controller => 'quickvote', :action => 'index', :ident => @election.id)
 165     end
 166
 167   end
 168
 169   def start_election
 170     @election = Election.find(params[:id])
 171
 172     @election.voters.each do |voter|
 173       voter.vote = Vote.new
 174       email_voter voter unless voter.email.nil?
 175     end
 176
 177     @election.activate!
 178     redirect_to :action => 'show', :id => @election.id
 179   end
 180
 181   # methods fod display, adding, deleting, and manipulating candidate
 182   # information for elections
 183   ####################################################################
 184   def edit_candidates
 185     @sidebar_content = render_to_string :partial => 'progress',
 186                                         :locals => { :page => 'candidates' }
 187     @election = Election.find( params[:id] )
 188   end
 189
 190   def add_candidate
 191     @election = Election.find(params[:id])
 192     @candidate = Candidate.new(params[:candidate])
 193     @election.candidates << @candidate
 194
 195     if @candidate.save
 196       # check to see if they've uploaded a picture
 197       if params[:picture][:uploaded_data]
 198         picture = Picture.new(params[:picture])
 199         @candidate.picture = picture if picture.save
 200       end
 201
 202       @candidate = Candidate.new
 203       redirect_to :action => 'edit_candidates', :id => @election.id
 204     else
 205       render :action => 'edit_candidates', :id => @election.id
 206     end
 207   end
 208
 209   def delete_candidate
 210     candidate = Candidate.find( params[:id] )
 211     candidate.destroy
 212   end
 213
 214   def candidate_picture
 215     candidate = Candidate.find( params[:id] )
 216     send_data( candidate.picture.data,
 217                :filename => candidate.picture.filename,
 218                :type => candidate.picture.filetype,
 219                :disposition => 'inline' )
 220   end
 221
 222   ## methods for displaying, adding, deleting, and manipulating voters
 223   ## for a particular election
 224   ####################################################################
 225   def new_voters
 226     redirect_to :action => 'edit_voters', :id => params[:id]
 227   end
 228
 229   def edit_voters
 230     @sidebar_content = render_to_string :partial => 'progress',
 231                                         :locals => { :page => 'voters' }
 232
 233     @election = Election.find( params[:id] )
 234     if params.has_key?( :raw_voter_list )
 235       process_incoming_voters( params[:raw_voter_list] )
 236     end
 237     @edit = true
 238     @raw_voter_list = RawVoterList.new
 239   end
 240
 241   def delete_voter
 242     voter = FullVoter.find( params[:id] )
 243     voter.destroy
 244   end
 245
 246   def toggle_authenticated
 247     @election = Election.find(params[:id])
 248     if params[:authenticated] == "1"
 249       @election.authenticated = true
 250     else
 251       @election.authenticated = false
 252     end
 253     @election.save
 254   end
 255
 256   ## methods for computing and printing results
 257   ####################################################################
 258   def results
 259     @election = Election.find( params[:id] )
 260     votes = []
 261
 262     @election.voters.each do |voter|
 263       if voter.vote and voter.vote.confirmed?
 264         votes << voter.vote.rankings.sort.collect {|vote| vote.candidate_id}
 265       end
 266     end
 267
 268     @voteobj = CloneproofSSDVote.new(votes)
 269     @resultobj = @voteobj.result
 270     @winners = @resultobj.winners
 271
 272     @candidates_by_id = {}
 273     @election.candidates.each {|cand| @candidates_by_id[cand.id] = cand}
 274
 275   end
 276
 277   def detailed_results
 278
 279     self.results
 280
 281     @voter_list = []
 282     @vote_list = []
 283
 284     @election.voters.each do |voter|
 285       if voter.vote and voter.vote.confirmed?
 286         @voter_list << voter.email
 287               @vote_list << voter.vote
 288       end
 289     end
 290
 291     @vote_list.sort!
 292     @vote_list.sort! { |a,b| a.token <=> b.token }
 293   end
 294
 295   ## private methods
 296   ####################################################################
 297   private
 298
 299     def process_incoming_voters(raw_voter_list)
 300       incoming_voters = RawVoterList.new( raw_voter_list )
 301
 302       unless incoming_voters.entries.empty?
 303         incoming_voters.each do |new_voter|
 304           new_voter.email.strip! # There's a trailing \r on all but the last in
 305                                  # the list!
 306           if new_voter.valid?
 307             # the new voter should be in good shape. save add to the election
 308             @election.voters << new_voter
 309                   new_voter.save
 310           end
 311           # TODO: Can we do some kind of AJAX error message for the voter being invalid?
 312         end
 313         @election.save
 314       end
 315
 316       # reset the next time to have a the same default value for emailing
 317       @raw_voter_list = RawVoterList.new
 318       @raw_voter_list.email = incoming_voters.email
 319     end
 320
 321     def email_voter(voter=nil)
 322       if voter
 323         VoterNotify.deliver_votestart(voter)
 324         voter.contacted=1
 325         voter.save
 326       end
 327     end
 328
 329 end