#!/usr/bin/perl
#
# survey.cgi
#
# User program to submit survey answers for presidential politics survey.
#
# Brian Casey
# Root Level Services, LLC
# brian@rootlevelservices.com
# May 2007
######################################################################

BEGIN { unshift (@INC,"/home/jgroom/www/president/lib"); }
use CAM::Template;
use CGI;
use mydb::DB;
require "scores.pl";
require "ads.pl";


$VERBOSE = 1;

######################################################################


&initialize;
my $cmd = $CGI->param('cmd');

if ($cmd eq "save")         { &save; }
elsif ($cmd eq "thanks")         { &thanks; }
else { 
   &survey; 
}

######################################################################
######################################################################
# survey
#    Display the input form.
#
sub survey { 

   my $tmpl = new CAM::Template("admin/templates/usersurvey.html");

   #####
   # Get list of all questions 
   #####
   $style = "";
   foreach my $q_ref ( @questions ) { 
      $tmpl->addLoop("question", 
                     $q_ref,
                    );
      $style .= qq!#pro_$$q_ref{question_id} { position: absolute; display: none; }\n!;
      $style .= qq!#con_$$q_ref{question_id} { position: absolute; display: none; }\n!;
   }

   $tmpl->addParams("style" => $style);

   ####  
   # Fill in ads.
   ####  
   &fillRandomAds( $tmpl, 2 );

   $tmpl->print();
   exit;
}
######################################################################
######################################################################
# save
#	Save a survey entry
#
sub save { 

   ############
   # Generate the answer string for this user's answers
   ############
   my $answer = "";
   my @q_query = ();
   foreach my $qref ( @questions ) { 
      my $qid = $$qref{question_id};
      my $a = $CGI->param("answer-$qid") || "_";
      if ($CGI->param("important-$qid")) { 
         $a = uc $a; 
      }

      if (lc $a eq "y") { 
         push(@q_query,"UPDATE $questionsTableName set numYes=numYes+1 WHERE question_id=$qid");
      }
      elsif (lc $a eq "n") { 
         push(@q_query,"UPDATE $questionsTableName set numNo=numNo+1 WHERE question_id=$qid");
      }
      else {
         push(@q_query,"UPDATE $questionsTableName set numUnanswered=numUnanswered+1 WHERE question_id=$qid");
      }
      if ($CGI->param("important-$qid")) { 
         push(@q_query,"UPDATE $questionsTableName set numImportant=numImportant+1 WHERE question_id=$qid");
      }
      $answer .= $a;
   }

   ############
   # Update counts for each question
   ############
   foreach my $query ( @q_query) { 
      &verbose($query);
      $sth = $MYSQLDB_DBH->query($query);
   }


   ############
   # Get/generate an answer key
   # If one needs to be generated, entries in $presidentKeyScores will
   # be created scoring this new answer against all the others.
   ############
   $answer_id = &getAnswerId( $answer );

   ############
   # Update the stats for this answer...
   # Increment the count of users that answered this, the total
   # scores of this answer against others and the average.
   ############

   # Get current stats
   &verbose("Get current stats.");
   my @stats = DBLIB_loadTable($keyStatsTableName,"key_id=$answer_id");
   my %stats = %{$stats[0]};
   $stats{key_id} = $answer_id;
   $stats{count}++;

   # Write new stats
   &verbose("Write new stats for $answer_id");
   DBLIB_writeTable($keyStatsTableName, %stats);


   &thanks;
   
}
######################################################################
######################################################################
# thanks
#	Display a "thanks" page with output data.
#
sub thanks { 

   my $tmpl = new CAM::Template("admin/templates/userthanks.html");

   if (!$answer_id) { $answer_id = $CGI->param('answer_id'); }
   $answer = &getAnswer($answer_id);

   ###############
   # Rank the presidential candidates for this user.
   ###############
   my $query = "SELECT president_id,answer_id,answer,score FROM $presidentTableName LEFT JOIN $keyScoresTableName ON (answer_id=key_id) LEFT JOIN $answerKeyTableName ON (answer_id=id) WHERE survey_id=$answer_id";
   my $sth = $MYSQLDB_DBH->query($query);
   %candidateAnswers = ();
   foreach my $i ( 1 .. $sth->numrows ) { 
      my ($pid,$aid,$a,$score) = $sth->fetchrow;
      $candidateAnswer{$pid} = { answer_id => $aid, answer => $a, score => $score };
      &verbose("candidate score ($pid) = $candidateAnswer{$pid}{score}");
   }
   foreach my $cref ( sort({ $candidateAnswer{$$b{president_id}}{score} <=> $candidateAnswer{$$a{president_id}}{score} } @candidates) ) { 
      $tmpl->addLoop("user_candidates", 
                     $cref, 
                     "score"  => $candidateAnswer{$$cref{president_id}}{score},
                     "answer" => $candidateAnswer{$$cref{president_id}}{answer}
                    );
   }

   ###############
   # Get the rank of this user against other users.
   ###############
#   $query  = "
#SELECT SUM(score*count),SUM(count) FROM $keyScoresTableName 
#JOIN $keyStatsTableName ON ($keyStatsTableName.key_id=$keyScoresTableName.survey_id)
#WHERE $keyScoresTableName.key_id=$answer_id 
#AND $keyScoresTableName.survey_id != $answer_id
#";
#   &verbose($query);
#   $sth = $MYSQLDB_DBH->query($query);
#   my ($user_total,$user_count) = $sth->fetchrow;
#   my $user_average = 0;
#   if ($user_count > 0) { $user_average = sprintf("%.2f", $user_total / $user_count ); }


   $user_average = sprintf("%.2f", &scoreVsAverage( $answer_id ));

   $tmpl->addParams("average_user_score", $user_average); 


   ###############
   # Rank the presidential candidates against each other
   # Stick in user when he/she should appear
   ###############
   $query  = "
SELECT president_id,SUM(score*count),SUM(count) FROM $presidentTableName
LEFT JOIN $keyScoresTableName ON (answer_id=$keyScoresTableName.key_id)
LEFT JOIN $keyStatsTableName  ON ($keyStatsTableName.key_id=$keyScoresTableName.survey_id)
GROUP BY president_id
";
   &verbose($query);
   $sth = $MYSQLDB_DBH->query($query);
   foreach my $i ( 1 .. $sth->numrows ) { 
      my ($pid,$total,$count) = $sth->fetchrow;
      my $average = 0;
      if ($count > 0) { 
         $average = sprintf("%.2f", $total/$count);
      }
      $candidateAverage{$pid} = $average;
      $total_count += $count;
   }

   my $user_in = 0;
   foreach my $cref ( sort({ $candidateAverage{$$b{president_id}} <=> $candidateAverage{$$a{president_id}} } @candidates) ) { 

      if ($user_average >= $candidateAverage{$$cref{president_id}} & !$user_in) { 
         $user_in = 1;
         $tmpl->addLoop("candidates", 
                        "name" => "YOU",
                        "average"  => $user_average,
                       );
      }

      $tmpl->addLoop("candidates", 
                     $cref, 
                     "average"  => $candidateAverage{$$cref{president_id}}
                    );
   }
   if (!$user_in) { 
         $tmpl->addLoop("candidates", 
                        "name" => "YOU",
                        "average"  => $user_average,
                       );
   }


   ###############
   # Total count of surveys taken
   ###############
   $query = "SELECT SUM(count) FROM $keyStatsTableName";
   &verbose($query);
   $sth = $MYSQLDB_DBH->query($query);
   my ($total_survey_count) = $sth->fetchrow;
   $tmpl->addParams("total_survey_count", $total_survey_count); 


   ###############
   # Answer distribution
   ###############
   my @questions = DBLIB_loadTable($questionsTableName,"question_id > 0 ORDER BY survey_number");
   foreach $qref ( @questions ) { 
      $tmpl->addLoop("question_hdr", $qref);
      $tmpl->addLoop("yes_count", $qref);
      $tmpl->addLoop("no_count", $qref);
      $tmpl->addLoop("unanswered_count", $qref);
      $tmpl->addLoop("important_count", $qref);
   }
   my $nq = $#questions + 1;
   $tmpl->addParams("num_questions", $nq);


   ###############
   # Some other data
   ###############
   $tmpl->addParams(
             "answer_id" => $answer_id,
             "answer" => $answer,
          );
   
   ####  
   # Fill in ads.
   ####  
   &fillRandomAds( $tmpl, 2 );

   $tmpl->print();
   exit;

}
######################################################################



######################################################################
# initialize
# 	Get everything ready to go.
#
sub initialize { 

   $CGI = new CGI;
   print $CGI->header();

   DBLIB_openDatabase;

   @questions = DBLIB_loadTable($questionsTableName,"question_id > 0 ORDER BY survey_number");
   @candidates = DBLIB_loadTable($presidentTableName,"president_id > 0 ORDER BY name");
   foreach my $c_ref ( @candidates ) { 
      $candidateName{$$c_ref{president_id}} = $$c_ref{name};
   }


}
######################################################################
######################################################################
# comment
#
sub comment { 

   print STDERR "$_[0]\n";

}
######################################################################