<?php

require 'mods.php';
require 'election.php';
$mods = new Mods();
$mods->startConnection();
$action = $mods->escapeString($_POST['action']);
if ($action == $mods->getAction(1)) {
    $studentNumber = $mods->escapeString($_POST['studentNumber']);
    $mods->setQuery("SELECT * FROM voters WHERE Voter_ID='{$studentNumber}'");
    if ($mods->getCount() == 1) {
        $row = $mods->getResults('array');
        if ($row['Status'] == 1) {
            $mods->login($studentNumber, 'Voter');
            $mods->sendStatus($mods->getError(0), '../index.php');
        } else {
            if ($row['Status'] == 0) {
                $mods->sendStatus($mods->getError(3), '../index.php');
            } else {
                if ($row['Status'] == 2) {
                    $mods->sendStatus($mods->getError(4), '../index.php');
                }
            }
        }
    } else {
        $mods->sendStatus($mods->getError(2), '../index.php');
    }
} else {
    if ($action == $mods->getAction(2)) {
        $username = $mods->escapeString($_POST['username']);
        $password = $mods->escapeString($_POST['password']);
        <div id="content">
            <div id="sub-header">Voter Registration</div>
            <div id="side-bar" style="padding: 0 25px; width: 200px;">
                <form class="form" method="POST" action="" style="margin-top: 65%;">
                    <label>Student Number:</label>
                    <input class="input-box focused-input" type="text" name="studentNumber" maxlength="11" required autofocus>
                    <input class="input-button" type="submit" value="Register">
                </form>
            </div>
            <div id="container">
                <div id="msg">
                    <?php 
if (isset($_POST['studentNumber'])) {
    $mods->startConnection();
    $studentNumber = $mods->escapeString($_POST['studentNumber']);
    $mods->setQuery("SELECT * FROM voters WHERE Voter_ID='{$studentNumber}'");
    if ($mods->getCount() == 1) {
        $row = $mods->getResults('array');
        if ($row['Status'] == 0) {
            // update status here
            $mods->setQuery("UPDATE voters SET Status=1 WHERE Voter_ID='{$studentNumber}'");
            if ($mods->getCount() > 0) {
                echo '<img class="icon" src="" alt="icon">&nbsp;&nbsp;<span>Voter has been registered.</span>';
            } else {
                echo '<img class="icon" src="" alt="icon">&nbsp;&nbsp;<span>Failed to register voter.</span>';
            }
        } else {
            if ($row['Status'] == 1) {
                echo '<img class="icon" src="" alt="icon">&nbsp;&nbsp;<span>Voter is already registered.</span>';
            } else {
                if ($row['Status'] == 2) {
<?php

require 'mods.php';
require 'election.php';
$election = new Election();
$mods = new Mods();
$mods->startConnection();
$action = $mods->escapeString($_POST['action']);
if ($action == $mods->getAction(3)) {
    $position = $mods->escapeString($_POST['position']);
    $election->startReading('candidates');
    echo $election->getCandidates($position);
} else {
    if ($action == $mods->getAction(4)) {
        $sessionID = $mods->getSession('user_id');
        $mods->setQuery("SELECT * FROM voters WHERE Voter_ID='{$sessionID}'");
        $row = $mods->getResults('array');
        if ($row['Status'] == 0) {
            echo 'Voter not registered.';
        } else {
            if ($row['Status'] == 1) {
                $ctr = 0;
                foreach ($_POST['votes'] as $vote) {
                    if ($vote != '') {
                        $mods->setQuery("INSERT INTO votes (Voter_ID, Candidate_ID) VALUES ('{$sessionID}', '{$vote}')");
                        if ($mods->getCount() > 0) {
                            $ctr++;
                        }
                    }
                }
                if ($ctr > 0) {