<?php

class Paginate
{
	private $query;
	private $currentPage;
	private $perPage;
	private $url;
	private $totalRows = false;

	/**
	 * @query - The query that will return lines to show
	 * @currentPage - 0based index of currently viewed page
	 * @perPage - Number of items to show per page
	 * @url - URL of current wegpage
	 */
	public function __construct($query, $perPage, $url = false)
	{
		$this->currentPage = (isset($_GET['page']) ? (int)$_GET['page'] : 0);
		$this->perPage = (int)$perPage;
		if ($this->currentPage < 0) {
			Util::traceError('Current page < 0');
		}
		if ($this->perPage < 1) {
			Util::traceError('Per page < 1');
		}
		// Query
		if (!preg_match('/\s*SELECT\s/is', $query)) {
			Util::traceError('Query has to start with SELECT!');
		}
		// XXX: MySQL only
		if (preg_match('/^mysql/i', CONFIG_SQL_DSN)) {
			// Sanity: Check for LIMIT specification at the end
			if (preg_match('/LIMIT\s+(\d+|\:\w+|\?)\s*,\s*(\d+|\:\w+|\?)(\s|;)*(\-\-.*)?$/is', $query)) {
				Util::traceError("You cannot pass a query containing a LIMIT to the Paginator class!");
			}
			// Sanity: no comment or semi-colon at end (sloppy, might lead to false negatives)
			if (preg_match('/(\-\-|;)(\s|[^\'"`])*$/is', $query)) {
				Util::traceError("Your query must not end in a comment or semi-colon!");
			}
			// Don't use SQL_CALC_FOUND_ROWS as it leads to filesort frequently thus being slower than two queries
			// See https://www.percona.com/blog/2007/08/28/to-sql_calc_found_rows-or-not-to-sql_calc_found_rows/

		} else {
			Util::traceError('Unsupported database engine');
		}
		// Mangle URL
		if ($url === false) $url = $_SERVER['REQUEST_URI'];
		if (strpos($url, '?') === false) {
			$url .= '?';
		} else {
			$url = preg_replace('/(\?|&)&*page=[^&]*(&+|$)/i', '$1', $url);
			if (substr($url, -1) !== '&') $url .= '&';
		}
		//
		$this->query = $query;
		$this->url = $url;
	}

	/**
	 * Execute the query, returning the PDO query object
	 */
	public function exec($args = array())
	{
		$countQuery = preg_replace('/ORDER\s+BY\s.*?(\sASC|\sDESC|$)/is', '', $this->query);
		$countQuery = preg_replace('/SELECT\s.*?\sFROM\s/is', 'SELECT Count(*) AS rowcount FROM ', $countQuery);
		$countRes = Database::queryFirst($countQuery, $args);
		$query = $this->query . ' LIMIT ' . ($this->currentPage * $this->perPage) . ', ' . $this->perPage;
		$retval = Database::simpleQuery($query, $args);
		$this->totalRows = (int)$countRes['rowcount'];
		return $retval;
	}

	public function render($template, $data)
	{
		if ($this->totalRows == 0) {
			// Shortcut for no content
			Render::addTemplate($template, $data);
			return;
		}
		// The real thing
		$pages = array();
		$pageCount = floor(($this->totalRows - 1) / $this->perPage) + 1;
		$skip = false;
		for ($i = 0; $i < $pageCount; ++$i) {
			if (($i > 0 && $i < $this->currentPage - 3) || ($i > $this->currentPage + 3 && $i < $pageCount - 1)) {
				if (!$skip) {
					$skip = true;
					$pages[] = array(
						'text'     => false,
						'current'  => false
					);
				}
				continue;
			}
			$skip = false;
			$pages[] = array(
				'text'     => $i + 1,
				'page'     => $i,
				'current'  => $i == $this->currentPage,
			);
		}
		$pages = Render::parse('pagenav', array(
			'url'       => $this->url,
			'pages'     => $pages,
		),'main');
		$data['page'] = $this->currentPage;
		$data['pagenav'] = $pages;
		Render::addTemplate($template, $data);
	}

}