Close votes automatically after 7 days

Closes #1

vedia.rb

@@ -5,7 +5,12 @@ require 'gettext'
 require 'securerandom'
 require 'chartkick'
 require 'mail'
+require 'tzinfo'
+
+include GetText
+
 require_relative 'mj'
+require_relative 'config/environments/defaults.rb'
 require_relative "config/environments/#{settings.environment}"
 
 class Vote < ActiveRecord::Base
@@ -55,19 +60,11 @@ def verify_password(password, hash)
   BCrypt::Password.new(hash) == password
 end
 
-include GetText
 set_output_charset('UTF-8')
 bindtextdomain('vedia', 'locale')
 set_locale('ca')
 
 enable :sessions
-set :values, [ { :id => 1, :label => _("Awful"), :color => '#ff4500' },
-               { :id => 2, :label => _("Very bad"), :color => '#ffa500' },
-               { :id => 3, :label => _("Bad"), :color => '#ffff00' },
-               { :id => 4, :label => _("Mediocre"), :color => '#9acd32' },
-               { :id => 5, :label => _("Good"), :color => '#228b22' },
-               { :id => 6, :label => _("Very good"), :color => '#006400' } ]
-set :admin_email, 'vedia@potager.org'
 MajorityJudgment.values = settings.values
 
 get '/' do
@@ -123,6 +120,10 @@ post '/login' do
   end
 end
 
+post '/timezone' do
+  session[:timezone] = JSON.parse(request.body.read)['timezone']
+end
+
 get '/reset' do
   erb :reset
 end
@@ -220,10 +221,6 @@ get '/votes/:id' do
     erb :votes_show_open
   when 'closed'
     erb :votes_show_closed
-  else
-    @vote.state = 'draft'
-    @vote.save
-    erb :votes_edit
   end
 end
 
@@ -260,6 +257,15 @@ post '/votes/:id/candidates/:cid/delete' do
   redirect '/votes/' + @vote.secure_id
 end
 
+get '/votes/:id/open' do
+  require_login
+  find_vote
+  require_organizer
+  require_draft_vote
+  @expire_on = Time.now.utc + settings.expire_after
+  erb :votes_open
+end
+
 post '/votes/:id/open' do
   require_login
   find_vote
@@ -267,6 +273,7 @@ post '/votes/:id/open' do
   require_draft_vote
   if not @vote.candidates.length < 2
     @vote.state = 'open'
+    @vote.expire_on = Time.now.utc + settings.expire_after
     @vote.save
   end
   redirect '/votes/' + @vote.secure_id
@@ -277,6 +284,7 @@ post '/votes/:id/draft' do
   find_vote
   require_organizer
   require_open_vote
+  require_no_expire_on
   @vote.ratings.each {|r| r.destroy}
   @vote.state = 'draft'
   @vote.save
@@ -288,6 +296,7 @@ post '/votes/:id/close' do
   find_vote
   require_organizer
   require_open_vote
+  require_no_expire_on
   @vote.state = 'closed'
   @vote.save
   redirect '/votes/' + @vote.secure_id
@@ -298,6 +307,7 @@ post '/votes/:id/reopen' do
   find_vote
   require_organizer
   require_closed_vote
+  require_no_expire_on
   @vote.state = 'open'
   @vote.save
   redirect '/votes/' + @vote.secure_id
@@ -328,10 +338,29 @@ post '/votes/:id/delete' do
   require_login
   find_vote
   require_organizer
+  require_draft_vote
   @vote.destroy
   redirect '/'
 end
 
+def close_expired_votes
+  Vote.where(state: 'open').where("expire_on < :now", { now: Time.now.utc }).each do |vote|
+    puts "#{Time.now.utc} Closing vote \"#{vote.title}\" because it expired on #{vote.expire_on}..."
+    vote.state = 'closed'
+    vote.save
+    User.all.each do |user|
+      puts "#{Time.now.utc} Sending results by email to #{user.email}..."
+      mail = Mail.new
+      mail.from = settings.admin_email
+      mail.to = user.email
+      mail.subject = _("Results of the vote: #{vote.title}")
+      template = ERB.new(File.read("views/votes_close_email.erb"))
+      mail.body = template.result(binding)
+      mail.deliver
+    end
+  end
+end
+
 helpers do
   def current_user
     if session[:user_id]
@@ -364,4 +393,12 @@ helpers do
   def require_closed_vote
     redirect '/votes/' + @vote.secure_id unless @vote.state == 'closed'
   end
+
+  def require_no_expire_on
+    redirect '/votes/' + @vote.secure_id unless @vote.expire_on.nil?
+  end
+
+  def format_date(timestamp)
+    "#{TZInfo::Timezone.get(session[:timezone]).to_local(timestamp).strftime('%F %R')} (#{session[:timezone].gsub('_', ' ')})"
+  end
 end