PollVerlaine/models/Poll.php

<?php

require __DIR__ . "/../config/app.php";

define("SAVE_PATH", __DIR__ . "/../db");

define("DEFAULT_SETTINGS", ["unique_ip" => true, "multiple_choices" => false]);

function settings_or_default($settings)
{
	return [
		"unique_ip" => isset($settings["unique_ip"]) ? $settings["unique_ip"] : DEFAULT_SETTINGS["unique_ip"],
		"multiple_choices" => isset($settings["multiple_choices"]) ? $settings["multiple_choices"] : DEFAULT_SETTINGS["multiple_choices"],
	];
}

class Poll
{
	/**
	 * Create a new poll and save it.
	 */
	public static function create_poll($request_data)
	{
		try
		{
			$poll = new Poll();
			$poll->title = $request_data->title;
			$poll->creation_date = (new DateTime())->getTimestamp();
			foreach ($request_data->options as $option)
			{
				$poll->options[] = [
					"label" => $option,
					"votes" => 0,
				];
			}
			$poll->settings = isset($request_data->settings) ? settings_or_default($request_data->settings) : DEFAULT_SETTINGS;
			$poll->gen_new_id();
			$poll->delete_token = bin2hex(openssl_random_pseudo_bytes(16));
			$poll->save();
			return $poll;
		}
		catch (Exception $e)
		{ return false; }
	}

	/**
	 * Try to load an existing poll.
	 * @param string id - Poll ID.
	 * @return boolean|Poll - Requested poll if found, false otherwise.
	 */
	public static function load_poll($id)
	{
		global $VERLAINE;

		$db = dba_open(SAVE_PATH . "/polls.db", "rd", $VERLAINE["db_handler"]);

		if (dba_exists($id, $db))
		{
			$poll = new Poll();
			$saved_poll_data = json_decode(dba_fetch($id, $db));
			$poll->id = $id;
			$poll->title = $saved_poll_data->title;
			$poll->creation_date = $saved_poll_data->creation_date;
			$poll->options = $saved_poll_data->options;
			$poll->delete_token = $saved_poll_data->delete_token;
			$poll->settings = $saved_poll_data->settings;
			$poll->ips = (array) $saved_poll_data->ips;

			dba_close($db);
			return $poll;
		}
		else
		{
			dba_close($db);
			return false;
		}
	}

	public $id = null;
	public $title;
	public $creation_date;
	public $options = [];
	public $settings;
	public $ips = [];
	public $delete_token;

	private function gen_new_id()
	{
		global $VERLAINE;

		$db = dba_open(SAVE_PATH . "/polls.db", "rd", $VERLAINE["db_handler"]);

		function gen_id()
		{ return bin2hex(openssl_random_pseudo_bytes(16)); }

		do
		{ $new_id = gen_id(); }
		while(dba_exists($new_id, $db));

		dba_close($db);
		$this->id = $new_id;
	}

	/**
	 * Vote for a list of options.
	 * @param array $options - Array of integers containing voted options.
	 * @return bool
	 */
	public function vote(array $options)
	{
		if (empty($options))
			return false; // Disallow void selection vote. TODO: Allow it in settings?

		if (!$this->settings->unique_ip)
		{ // If unique_ip option is enabled => Only allow unregistered IPs.
			if (isset($this->ips[Flight::request()->ip]))
				return false; // A vote is already registered with this IP: error.
			else
				$this->ips[Flight::request()->ip] = true; // We register the IP in the used IPs array.
		}

		if (!$this->settings->multiple_choices && count($options) > 1)
			return false; // If multiple_choices is not selected, disallow an options array with more than one selected option.

		// For each option in the list, add 1 to the vote number in the poll data.
		foreach ($options as $option)
			if (isset($this->options[intval($option)])) // Check invalid options id.
				$this->options[intval($option)]->votes++;
		return true;
	}

	public function save()
	{
		global $VERLAINE;

		$db = dba_open(SAVE_PATH . "/polls.db", "wd", $VERLAINE["db_handler"]);
		$func = (dba_exists($this->id, $db) ? "dba_replace" : "dba_insert");
		$func($this->id, json_encode([
			"title" => $this->title,
			"creation_date" => $this->creation_date,
			"options" => $this->options,
			"delete_token" => $this->delete_token,
			"ips" => $this->ips,
			"settings" => $this->settings,
		]), $db);
		dba_close($db);
	}

	public function delete()
	{
		global $VERLAINE;

		$db = dba_open(SAVE_PATH . "/polls.db", "wd", $VERLAINE["db_handler"]);
		dba_delete($this->id, $db);
		if ($VERLAINE["optimize_on_delete"])
			dba_optimize($db);
		dba_close($db);
	}
}