bug fix change session check to database check

solve #14

Author: manh21

application/controllers/Home.php

@@ -22,13 +22,19 @@ class Home extends CI_Controller
      * @see https://codeigniter.com/user_guide/general/urls.html
      */
 
+    /**
+     * Constructor
+     * 
+     * @return  void
+     */
     public function __construct()
     {
         parent::__construct();
         $this->load->database();
         $this->load->library('ion_auth', 'form_validation', 'session');
         $this->load->helper('url', 'language', 'my_helper');
         $this->load->model('Home_model');
+        $this->load->model('Setting_model');
     }
 
     public function index()
@@ -54,24 +60,28 @@ public function vote()
         // Get All Kandidat
         $kandidat_data = $this->Home_model->get_all('nourut', 'kandidat', 'ASC');
 
-        $data = array(
+        // Get Settings
+        $setting_q = $this->Setting_model->get_all('id', 'settings', 'ASC');
+        $settings = $setting_q[0];
+        $selesai = $selesai = DateTime::createFromFormat('Y-m-d H:i:s', $settings->selesai)->format('Y/m/d H:i:s');
 
+        $data = [
             // Data kandidat diambil dari database
             'kandidat_data' => $kandidat_data,
+            'waktu_selesai' => $selesai,
+        ];
 
-        );
+        $idpemilih = $this->session->userdata('userid');
 
         // Check status sudah memilih atau belum
-        $status = $this->session->userdata('status');
-        if ($status === 'Belum Memilih') {
-            $this->load->view('front/vote', $data);
-        } elseif ($status === 'Sudah Memilih') {
-
+        if ($this->home_model->is_voted($idpemilih)) {
             $data = array(
                 'nama' => $this->session->userdata('nama'),
             );
 
-            $this->load->view('front/terimakasih', $data);
+            return $this->load->view('front/terimakasih', $data);
+        } else {
+            return $this->load->view('front/vote', $data);
         }
     }
 
@@ -86,55 +96,55 @@ public function doVote($idkandidat)
         $idpemilih = $this->session->userdata('userid');
         // Tipe pemilih apakah guru atau siswa
         $tipe = $this->session->userdata('level');
-
+       
         // Check status sudah memilih atau belum
-        $status = $this->session->userdata('status');
-        if ($status === 'Belum Memilih') {
-
-            // insertData
-            $insertData = array(
-                'tipe' => $tipe,
-                'idpemilih' => $idpemilih,
-                'idkandidat' => $idkandidat,
-            );
-
-            // Insert data
-            $this->Home_model->insert('data_pemilihan', $insertData);
-
-            // Update Session data
-            $userData = array(
-                'status' => 'Sudah Memilih'
-            );
-            $this->session->set_userdata($userData);
-
-            // Update Database data
-            $updateData = array(
-                'status' => 'Sudah Memilih'
-            );
-            $this->Home_model->update('id', $idpemilih, 'data_pemilih', $updateData);
-
-            // Menghitung jumlah perolehan suara
-            $kandidatData = $this->Home_model->get_all('nourut', 'kandidat', 'DESC');
-            foreach ($kandidatData as $row) {
-                // Berdasarkan idkandidat yang ada
-                $jumlahSuara = $this->Home_model->tampil_data('idkandidat', $row->idkandidat, 'data_pemilihan');
-                $suaraData = array(
-                    'jumlahsuara' => $jumlahSuara,
-                );
-                // Update jumlah suara counter ke database
-                $this->Home_model->update('idkandidat', $row->idkandidat, 'kandidat', $suaraData);
-            };
-
-            redirect('vote', 'refresh');
-        } else {
-            redirect('home', 'refresh');
+        if ($this->home_model->is_voted($idpemilih)) {
             $this->session->set_flashdata(
                 'message',
                 '<div class="alert alert-warning alert-dismissible">
                 <button type="button" class="close" data-dismiss="alert" aria-hidden="true">&times;</button>
                 Anda sudah memilih </div>'
             );
+            return redirect('home', 'refresh');
         }
+
+        // insertData
+        $insertData = array(
+            'tipe' => $tipe,
+            'idpemilih' => $idpemilih,
+            'idkandidat' => $idkandidat,
+        );
+
+        // Insert data
+        $this->Home_model->insert('data_pemilihan', $insertData);
+
+        // Update Session data
+        $userData = array(
+            'status' => 'Sudah Memilih'
+        );
+
+        $this->session->set_userdata($userData);
+
+        // Update Database data
+        $updateData = array(
+            'status' => 'Sudah Memilih'
+        );
+        $this->Home_model->update('id', $idpemilih, 'data_pemilih', $updateData);
+
+        // Menghitung jumlah perolehan suara
+        $kandidatData = $this->Home_model->get_all('nourut', 'kandidat', 'DESC');
+        foreach ($kandidatData as $row) {
+            // Berdasarkan idkandidat yang ada
+            $jumlahSuara = $this->Home_model->tampil_data('idkandidat', $row->idkandidat, 'data_pemilihan');
+            $suaraData = array(
+                'jumlahsuara' => $jumlahSuara,
+            );
+            // Update jumlah suara counter ke database
+            $this->Home_model->update('idkandidat', $row->idkandidat, 'kandidat', $suaraData);
+        };
+
+        return redirect('vote', 'refresh');
+
     }
 }
 

application/models/Home_model.php

@@ -46,6 +46,16 @@ public function tampil_data($q, $idkandidat, $table)
         $this->db->where($q, $idkandidat);
         return $this->db->count_all_results($table);
     }
+
+    // Check sudah memilih atau belum
+    public function is_voted($idpemilih)
+    {
+        $result = $this->get_by_id('id', $idpemilih, 'data_pemilih');
+        if(!empty($result) && $result->status === 'Belum Memilih'){
+            return false;
+        }
+        return true;
+    }
 }
 
 /* End of file Home_modal.php */