Add new voting system

include/vote.php

@@ -3,6 +3,189 @@ require_once(BASE."include/util.php");
 /* max votes per user */
 define('MAX_VOTES',3);
 
+class vote
+{
+    public $iUserId;
+    public $iVoteId;
+    public $iSlotIndex;
+    public $iVersionId;
+
+    public function vote($iVoteId = null, $oRow = null)
+    {
+        if(!$iVoteId && !$oRow) /* Nothing to do */
+            return;
+
+        if(!$oRow)
+        {
+            $hResult = query_parameters("SELECT * FROM appVotes WHERE id = '?'", $iVoteId);
+
+            $oRow = mysql_fetch_object($hResult);
+        }
+
+        if($oRow)
+        {
+            $this->iUserId = $oRow->userId;
+            $this->iVoteId = $oRow->id;
+            $this->iSlotIndex = $oRow->slot;
+            $this->iVersionId = $oRow->versionId;
+        }
+    }
+
+    public function update()
+    {
+        /* Check for valid vote slot index */
+        if($this->iSlotIndex < 1 || $this->iSlotIndex > MAX_VOTES)
+            return;
+
+        /* Avoid pointless votes */
+        if(!$this->iVersionId)
+            return;
+
+        if(!$this->iVoteId)
+        {
+            $hResult = query_parameters("INSERT INTO appVotes (versionId,userId,slot) VALUES('?','?','?')",
+                                        $this->iVersionId, $_SESSION['current']->iUserId, $this->iSlotIndex);
+        } else
+        {
+            $hResult = query_parameters("UPDATE appVotes SET versionId = '?' WHERE id = '?'", $this->iVersionId, $this->iVoteId);
+        }
+
+        if(!$hResult)
+            return FALSE;
+
+        return TRUE;
+    }
+}
+
+class voteManager
+{
+    private $iUserId;
+    private $aVotes;
+
+    public function voteManager($iUserId = null, $oRow = null)
+    {
+        $this->iUserId = $iUserId;
+    }
+
+    public function objectGetCustomVars($sAction)
+    {
+        switch($sAction)
+        {
+            case "edit":
+                return array("iVersionId");
+                break;
+
+            default:
+                return null;
+        }
+    }
+
+    public function outputEditor($aClean = null)
+    {
+        echo "The following shows your current votes.  Check the boxes next to the apps you wish to replace with a vote for ".version::fullNameLink($aClean['iVersionId']).".";
+
+        $oTable = new table();
+        $this->aVotes = $this->getVotes();
+
+        for($i = 0; $i < MAX_VOTES; $i++)
+        {
+            $sVersionText = $this->aVotes[$i]->iVersionId ? version::fullNameLink($this->aVotes[$i]->iVersionId) : "No app selected";
+            $oTableRow = new tableRow();
+            $oTableRow->addTextCell('<input type="checkbox" name="iSlot'.$i.'" value="'.$aClean['iVersionId'].'" />');
+            $oTableRow->addTextCell($sVersionText);
+            $oTable->addRow($oTableRow);
+        }
+
+        echo $oTable->getString();
+    }
+
+    public function canEdit()
+    {
+        if($_SESSION['current']->iUserId == $this->iUserId)
+            return TRUE;
+
+        return FALSE;
+    }
+
+    public function mustBeQueued()
+    {
+        return FALSE;
+    }
+
+    public function objectGetEntries($bQueued, $bRejected)
+    {
+        return query_parameters("SELECT * FROM appVotes");
+    }
+
+    public function objectGetId()
+    {
+        return $this->iUserId;
+    }
+
+    public function create()
+    {
+        return TRUE;
+    }
+
+    public function update()
+    {
+        foreach($this->aVotes as $oVote)
+            $oVote->update();
+    }
+
+    public function getOutputEditorValues($aClean)
+    {
+        $this->aVotes = $this->getVotes();
+
+        for($i = 0; $i < MAX_VOTES; $i++)
+            $this->aVotes[$i]->iVersionId = $aClean["iSlot$i"];
+    }
+
+    public function objectGetEntriesCount()
+    {
+        $hResult = query_parameters("SELECT COUNT(id) as count FROM appVotes");
+
+        if(!$hResult)
+            return FALSE;
+
+        if(!($oRow = mysql_fetch_object($hResult)))
+            return FALSE;
+
+        return $oRow->count;
+    }
+
+    public function objectGetSubmitterId()
+    {
+        return $this->iUserId;
+    }
+
+    public function getVotes()
+    {
+        $aVotes = array();
+        $hResult = query_parameters("SELECT * FROM appVotes WHERE userId = '?' ORDER BY slot", $this->iUserId);
+
+        if(!$hResult)
+            return $aVotes;
+
+        for($i = 0; $i < MAX_VOTES; $i++)
+            $aVotes[$i] = null;
+
+        while($oRow = mysql_fetch_object($hResult))
+            $aVotes[$oRow->slot-1] = new vote(null, $oRow);
+
+        for($i = 0; $i < MAX_VOTES; $i++)
+        {
+            if(!$aVotes[$i])
+            {
+                $aVotes[$i] = new vote();
+                $aVotes[$i]->iSlotIndex = $i+1;
+            }
+        }
+
+        return $aVotes;
+    }
+}
+
 /**
  * count the number of votes for appId by userId
  */