More stuff for the new surveys.

[platal.git] / modules / survey / answer.inc.php

diff --git a/modules/survey/answer.inc.php b/modules/survey/answer.inc.php
new file mode 100644 (file)
index 0000000..8781a6b
--- /dev/null
+++ b/modules/survey/answer.inc.php
@@ -0,0 +1,163 @@
+<?php
+/***************************************************************************
+ *  Copyright (C) 2003-2010 Polytechnique.org                              *
+ *  http://opensource.polytechnique.org/                                   *
+ *                                                                         *
+ *  This program is free software; you can redistribute it and/or modify   *
+ *  it under the terms of the GNU General Public License as published by   *
+ *  the Free Software Foundation; either version 2 of the License, or      *
+ *  (at your option) any later version.                                    *
+ *                                                                         *
+ *  This program is distributed in the hope that it will be useful,        *
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of         *
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the          *
+ *  GNU General Public License for more details.                           *
+ *                                                                         *
+ *  You should have received a copy of the GNU General Public License      *
+ *  along with this program; if not, write to the Free Software            *
+ *  Foundation, Inc.,                                                      *
+ *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA                *
+ ***************************************************************************/
+
+class SurveyVote extends PlDBTableEntry
+{
+    protected $survey;
+    protected $user;
+
+    private $answers = array();
+    private $fetchAnswers;
+
+    public function __construct(Survey $survey, User $user)
+    {
+        parent::__construct('survey_votes');
+        $this->survey = $survey;
+        $this->user = $user;
+        $this->sid = $survey->id;
+    }
+
+    protected function postSave()
+    {
+        Platal::assert(!is_null($this->vid), "Cannot process a vote without its identifier");
+        XDB::execute("REPLACE INTO  survey_voters (sid, uid, vid)
+                            VALUES  ({?}, {?}, {?})",
+                     $this->survey->id, $this->user->id(),
+                     $this->survey->flags->hasFlag('anonymous') ? null : $this->vid);
+
+        /* Save answers */
+        $selector = new SurveyAnswer($this);
+        $selector->delete();
+
+        $answers = array();
+        foreach ($this->answers as $key=>$answer) {
+            if (!is_null($answer)) {
+                $answer->vid = $this->vid;
+                $answers[] = $answer;
+            }
+        }
+        PlDBTableEntry::insertBatch($answers);
+        return true;
+    }
+
+    protected function postFetch()
+    {
+        $selector = new SurveyAnswer($this);
+        foreach ($selector as $answer) {
+            $question = $this->survey->questionForId($answer->qid);
+            $this->answers[$answer->qid] = $answer;
+        }
+        return true;
+    }
+
+    public function inError()
+    {
+        foreach ($this->answers as $answer) {
+            if ($answer->inError !== false) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public function getAnswer(SurveyQuestion $question)
+    {
+        if (!isset($this->answers[$question->qid])) {
+            $val = new SurveyAnswer($this);
+            $val->qid = $question->qid;
+            $this->answers[$question->qid] = $val;
+        }
+        return $this->answers[$question->qid];
+    }
+
+    public function export()
+    {
+        $export = array();
+        foreach ($this->answers as $qid=>$answer) {
+            $export[$qid] = $answer->export();
+        }
+        return $export;
+    }
+
+    public static function getVote(Survey $survey, User $user, $fetchAnswers = true)
+    {
+        $vid = XDB::query('SELECT  vid
+                             FROM  survey_voters
+                            WHERE  sid = {?} AND uid = {?}',
+                          $survey->id, $user->id());
+        if ($vid->numRows() == 0) {
+            $vote = new SurveyVote($survey, $user);
+            $vote->fetchAnswers = $fetchAnswers;
+            return $vote;
+        }
+        $vid = $vid->fetchOneCell();
+        if (is_null($vid)) {
+            /* User already vote, but survey is anonymous and the vote
+             * cannot be changed
+             */
+            return null;
+        }
+        $vote = new SurveyVote($survey, $user);
+        $vote->vid = $vid;
+        $vote->fetchAnswers = $fetchAnswers;
+        $vote->fetch();
+        return $vote;
+    }
+}
+
+class SurveyAnswer extends PlDBTableEntry
+{
+    public $inError = false;
+    public $vote;
+
+    public function __construct(SurveyVote $vote)
+    {
+        parent::__construct('survey_vote_answers');
+        $this->registerFieldFormatter('answer', 'JSonFieldFormatter');
+        $this->vote = $vote;
+        if (!is_null($vote->vid)) {
+            $this->vid = $vote->vid;
+        }
+    }
+
+    protected function preSave()
+    {
+        Platal::assert(!$this->inError, "Cannot save an invalid answer");
+        $this->sid = $this->vote->sid;
+        $this->vid = $this->vote->vid;
+        return true;
+    }
+
+    public function export()
+    {
+        $export = array();
+        if (!is_null($this->answer)) {
+            $export['value'] = $this->answer->export();
+        }
+        if ($this->inError !== false) {
+            $export['error'] = $this->inError;
+        }
+        return $export;
+    }
+}
+
+// vim:set et sw=4 sts=4 ts=4 foldmethod=marker enc=utf-8:
+?>