An @edit=true declaration was missing from edit_voters that prevented the voter delet...
[selectricity-live] / 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     
49     holder = create_theme_hash
50     unless holder.values.all? {|v| v.has_value?("")}
51       token_generator = UniqueTokenGenerator.new( 16 )
52       @election.embed_custom_string = token_generator.token
53       add_theme(@election.embed_custom_string)
54     end
55     
56     if @election.save
57       flash[:notice] = 'Election was successfully created.'
58       redirect_to :action => 'edit_candidates', :id => @election.id
59     else
60       render :action => 'general_information'
61     end
62   end
63   
64   def create_theme_hash
65     target = Hash.new
66     params.each do |k,v|
67       target[k] = v if k=="top_bar" or k=="default_image" or k=="bg1" \
68                     or k=="bg2" or k=="bottom_bar"
69     end
70     return target
71   end
72   
73   # TODO add filter to verify that the person working on or looking at
74   # something is the owner
75   def edit_general_information
76     @election = Election.find(params[:id])
77   end
78   
79   def update_general_information
80     @election = Election.find(params[:id])
81     
82     holder = create_theme_hash
83     unless holder.values.all? {|v| v.has_value?("")}
84       unless @election.embed_custom_string
85         token_generator = UniqueTokenGenerator.new( 16 )
86         @election.embed_custom_string = token_generator.token
87       end
88       
89       add_theme(@election.embed_custom_string)
90     end
91     
92     if @election.update_attributes(params[:election])
93       flash[:notice] = 'Election was successfully updated.'
94       redirect_to :action => 'show', :id => @election
95     else
96       render :action => 'edit'
97     end
98   end
99   
100   #Takes care of uploading custom images 
101   #unnecessarily long, how can I compress?
102   def add_theme(prefix)
103     holder = create_theme_hash
104     unless params[:top_bar][:uploaded_data].to_s.empty?
105       previous = SkinPicture.find(:first,
106       :conditions => ["filename = ?", @election.embed_custom_string + "top_bar.png"])
107       if previous
108         previous.destroy
109       end
110       top_bar = SkinPicture.new(params[:top_bar])
111       top_bar.filename = prefix + "top_bar." + params[:top_bar][:uploaded_data].content_type[6..-2]
112       top_bar.save
113     end
114     unless params[:default_image][:uploaded_data].to_s.empty?
115       previous = SkinPicture.find(:first,
116       :conditions => ["filename = ?", @election.embed_custom_string + "default_image.png"])
117       if previous
118         previous.destroy
119       end
120       default_image = SkinPicture.new(params[:default_image])
121       default_image.filename = prefix + "default_image." + params[:default_image][:uploaded_data].content_type[6..-2]
122       default_image.save
123     end
124     unless params[:bg1][:uploaded_data].to_s.empty?
125       previous = SkinPicture.find(:first,
126       :conditions => ["filename = ?", @election.embed_custom_string + "bg1.png"])
127       if previous
128         previous.destroy
129       end
130       bg1 = SkinPicture.new(params[:bg1])  
131       bg1.filename = prefix + "bg1." + params[:bg1][:uploaded_data].content_type[6..-2]
132       bg1.save
133     end
134     unless params[:bg2][:uploaded_data].to_s.empty?
135       previous = SkinPicture.find(:first,
136       :conditions => ["filename = ?", @election.embed_custom_string + "bg2.png"])
137       if previous
138         previous.destroy
139       end
140       bg2 = SkinPicture.new(params[:bg2]) 
141       bg2.filename = prefix + "bg2." + params[:bg2][:uploaded_data].content_type[6..-2]
142       bg2.save
143     end
144     unless params[:bottom_bar][:uploaded_data].to_s.empty?
145       previous = SkinPicture.find(:first,
146       :conditions => ["filename = ?", @election.embed_custom_string + "bottom_bar.png"])
147       if previous
148         previous.destroy
149       end
150       bottom_bar = SkinPicture.new(params[:bottom_bar])
151       bottom_bar.filename = prefix + "bottom_bar." + params[:bottom_bar][:uploaded_data].content_type[6..-2]
152       bottom_bar.save
153     end
154         
155   end
156   
157   def show
158     @sidebar_content = render_to_string :partial => 'progress',
159                                         :locals => { :page => 'review' }
160
161     @election = Election.find(params[:id])
162     if @election.type == QuickVote
163       redirect_to(:controller => 'quickvote', :action => 'index', :ident => @election.id)
164     end
165       
166   end
167
168   def start_election
169     @election = Election.find(params[:id])
170     
171     @election.voters.each do |voter|
172       voter.vote = Vote.new
173       email_voter voter unless voter.email.nil?
174     end
175
176     @election.activate!
177     redirect_to :action => 'show', :id => @election.id
178   end
179
180   # methods fod display, adding, deleting, and manipulating candidate
181   # information for elections
182   ####################################################################
183   def edit_candidates
184     @sidebar_content = render_to_string :partial => 'progress',
185                                         :locals => { :page => 'candidates' }
186     @election = Election.find( params[:id] )
187   end
188
189   def add_candidate
190     @election = Election.find(params[:id])
191     @candidate = Candidate.new(params[:candidate])
192     @election.candidates << @candidate
193
194     if @candidate.save
195       # check to see if they've uploaded a picture
196       if params[:picture][:uploaded_data]
197         picture = Picture.new(params[:picture])
198         @candidate.picture = picture if picture.save
199       end
200
201       @candidate = Candidate.new
202       redirect_to :action => 'edit_candidates', :id => @election.id
203     else
204       render :action => 'edit_candidates', :id => @election.id
205     end
206   end
207   
208   def delete_candidate
209     candidate = Candidate.find( params[:id] )
210     candidate.destroy
211   end
212
213   def candidate_picture
214     candidate = Candidate.find( params[:id] )
215     send_data( candidate.picture.data,
216                :filename => candidate.picture.filename,
217                :type => candidate.picture.filetype,
218                :disposition => 'inline' )
219   end
220
221   ## methods for displaying, adding, deleting, and manipulating voters
222   ## for a particular election
223   ####################################################################
224   def new_voters
225     redirect_to :action => 'edit_voters', :id => params[:id]
226   end
227   
228   def edit_voters
229     @sidebar_content = render_to_string :partial => 'progress',
230                                         :locals => { :page => 'voters' }
231
232     @election = Election.find( params[:id] )
233     if params.has_key?( :raw_voter_list )
234       process_incoming_voters( params[:raw_voter_list] )
235     end
236     @edit = true
237     @raw_voter_list = RawVoterList.new
238   end
239   
240   def delete_voter
241     voter = Voter.find( params[:id] )
242     voter.destroy
243   end
244
245   def toggle_authenticated
246     @election = Election.find(params[:id])
247     if params[:authenticated] == "1"
248       @election.authenticated = true
249     else
250       @election.authenticated = false
251     end
252     @election.save
253   end
254   
255   ## methods for computing and printing results
256   ####################################################################
257   def results
258     @election = Election.find( params[:id] )
259     votes = []
260     
261     @election.voters.each do |voter|
262       if voter.vote and voter.vote.confirmed?
263         votes << voter.vote.rankings.sort.collect {|vote| vote.candidate_id}
264       end
265     end
266     
267     @voteobj = CloneproofSSDVote.new(votes)
268     @resultobj = @voteobj.result
269     @winners = @resultobj.winners
270     
271     @candidates_by_id = {}
272     @election.candidates.each {|cand| @candidates_by_id[cand.id] = cand}
273     
274   end
275   
276   def detailed_results
277    
278     self.results
279
280     @voter_list = []
281     @vote_list = []
282     
283     @election.voters.each do |voter|
284       if voter.vote and voter.vote.confirmed?
285         @voter_list << voter.email
286               @vote_list << voter.vote
287       end
288     end
289
290     @vote_list.sort!
291     @vote_list.sort! { |a,b| a.token <=> b.token }
292   end
293
294   ## private methods
295   ####################################################################
296   private
297
298     def process_incoming_voters(raw_voter_list)
299       incoming_voters = RawVoterList.new( raw_voter_list )
300
301       unless incoming_voters.entries.empty?
302         incoming_voters.each do |new_voter|
303           new_voter.email.strip! # There's a trailing \r on all but the last in
304                                  # the list!
305           if new_voter.valid?
306             # the new voter should be in good shape. save add to the election
307             @election.voters << new_voter
308                   new_voter.save
309           end
310           # TODO: Can we do some kind of AJAX error message for the voter being invalid?
311         end
312         @election.save
313       end
314  
315       # reset the next time to have a the same default value for emailing
316       @raw_voter_list = RawVoterList.new
317       @raw_voter_list.email = incoming_voters.email
318     end
319
320     def email_voter(voter=nil)
321       if voter
322         VoterNotify.deliver_votestart(voter)
323         voter.contacted=1
324         voter.save
325       end
326     end
327
328 end

Benjamin Mako Hill || Want to submit a patch?