newznab-tmux/nntmux/Contents.php

<?php
namespace nntmux;

use nntmux\db\DB;

class Contents
{
	const TYPEUSEFUL = 1;
	const TYPEARTICLE = 2;
	const TYPEINDEX = 3;

	/**
	 * @var DB
	 */
	public $pdo;

	/**
	 * @param array $options Class instances.
	 */
	public function __construct(array $options = [])
	{
		$defaults = [
			'Settings' => null,
		];
		$options += $defaults;

		$this->pdo = ($options['Settings'] instanceof DB ? $options['Settings'] : new DB());
	}

	/**
	 * @return array|bool
	 */
	public function get()
	{
		$arr = [];
		$rows = $this->data_get();
		if ($rows === false) {
			return false;
		}

		foreach ($rows as $row) {
			$arr[] = $this->row2Object($row);
		}

		return $arr;
	}

	/**
	 * @return array|bool
	 */
	public function getAll()
	{
		$arr = [];
		$rows = $this->data_getAll();
		if ($rows === false) {
			return false;
		}

		foreach ($rows as $row) {
			$arr[] = $this->row2Object($row);
		}

		return $arr;
	}

	/**
	 * Convert get all but from to object.
	 *
	 * @return array|bool
	 */
	public function getAllButFront()
	{
		$arr = [];
		$rows = $this->data_getAllButFront();
		if ($rows === false) {
			return false;
		}

		foreach ($rows as $row) {
			$arr[] = $this->row2Object($row);
		}

		return $arr;
	}

	/**
	 * @return array|bool
	 */
	public function getFrontPage()
	{
		$arr = [];
		$rows = $this->data_getFrontPage();
		if ($rows === false) {
			return false;
		}

		foreach ($rows as $row) {
			$arr[] = $this->row2Object($row);
		}

		return $arr;
	}

	/**
	 * @param $id
	 * @param $role
	 *
	 * @return array|bool
	 */
	public function getForMenuByTypeAndRole($id, $role)
	{

		$arr = [];
		$rows = $this->data_getForMenuByTypeAndRole($id, $role);
		if ($rows === false) {
			return false;
		}

		foreach ($rows as $row) {
			$arr[] = $this->row2Object($row);
		}

		return $arr;
	}

	/**
	 * @return bool|Content
	 */
	public function getIndex()
	{
		$row = $this->data_getIndex();
		if ($row === false) {
			return false;
		}

		return $this->row2Object($row);
	}

	/**
	 * @param $id
	 * @param $role
	 *
	 * @return bool|Content
	 */
	public function getByID($id, $role)
	{
		$row = $this->data_getByID($id, $role);
		if ($row === false) {
			return false;
		}

		return $this->row2Object($row);
	}

	/**
	 * @param $content
	 *
	 * @return mixed
	 */
	public function validate($content)
	{
		if ($content->url[0] !== '/') {
			$content->url = '/' . $content->url;
		}

		if (substr($content->url, strlen($content->url) - 1) !== '/') {
			$content->url .= '/';
		}

		return $content;
	}

	/**
	 * @param $form
	 *
	 * @return false|int|string
	 */
	public function add($form)
	{
		$content = $this->row2Object($form);
		$content = $this->validate($content);
		if ($content->ordinal === 1) {
			$this->pdo->queryDirect('UPDATE content SET ordinal = ordinal + 1 WHERE ordinal > 0');
		}
		return $this->data_add($content);
	}

	/**
	 * @param $id
	 *
	 * @return bool|\PDOStatement
	 */
	public function delete($id)
	{
		return $this->pdo->queryExec(sprintf('DELETE FROM content WHERE id = %d', $id));
	}

	/**
	 * @param $form
	 *
	 * @return mixed|Content
	 */
	public function update($form)
	{
		$content = $this->row2Object($form);
		$content = $this->validate($content);
		$this->data_update($content);

		return $content;
	}

	/**
	 * @param        $row
	 * @param string $prefix
	 *
	 * @return Content
	 */
	public function row2Object($row, $prefix = '')
	{
		$obj = new Content();
		if (isset($row[$prefix . 'id'])) {
			$obj->id = $row[$prefix . 'id'];
		}
		$obj->title = $row[$prefix . 'title'];
		$obj->url = $row[$prefix . 'url'];
		$obj->body = $row[$prefix . 'body'];
		$obj->metadescription = $row[$prefix . 'metadescription'];
		$obj->metakeywords = $row[$prefix . 'metakeywords'];
		$obj->contenttype = $row[$prefix . 'contenttype'];
		$obj->showinmenu = $row[$prefix . 'showinmenu'];
		$obj->status = $row[$prefix . 'status'];
		$obj->ordinal = $row[$prefix . 'ordinal'];
		if (isset($row[$prefix . 'createddate'])) {
			$obj->createddate = $row[$prefix . 'createddate'];
		}
		$obj->role = $row[$prefix . 'role'];
		return $obj;
	}

	/**
	 * @param $content
	 *
	 * @return bool|\PDOStatement
	 */
	public function data_update($content)
	{
		return $this->pdo->queryExec(sprintf('UPDATE content SET role = %d, title = %s, url = %s, body = %s, metadescription = %s, metakeywords = %s, contenttype = %d, showinmenu = %d, status = %d, ordinal = %d WHERE id = %d', $content->role, $this->pdo->escapeString($content->title), $this->pdo->escapeString($content->url), $this->pdo->escapeString($content->body), $this->pdo->escapeString($content->metadescription), $this->pdo->escapeString($content->metakeywords), $content->contenttype, $content->showinmenu, $content->status, $content->ordinal, $content->id));
	}

	/**
	 * @param $content
	 *
	 * @return false|int|string
	 */
	public function data_add($content)
	{
		return $this->pdo->queryInsert(sprintf('INSERT INTO content (role, title, url, body, metadescription, metakeywords, contenttype, showinmenu, status, ordinal) values (%d, %s, %s, %s, %s, %s, %d, %d, %d, %d )', $content->role, $this->pdo->escapeString($content->title), $this->pdo->escapeString($content->url), $this->pdo->escapeString($content->body), $this->pdo->escapeString($content->metadescription), $this->pdo->escapeString($content->metakeywords), $content->contenttype, $content->showinmenu, $content->status, $content->ordinal));
	}

	/**
	 * @return array
	 */
	public function data_get(): array
	{
		return $this->pdo->query(sprintf('SELECT * FROM content WHERE status = 1 ORDER BY contenttype, COALESCE(ordinal, 1000000)'));
	}

	/**
	 * @return array
	 */
	public function data_getAll(): array
	{
		return $this->pdo->query(sprintf('SELECT * FROM content ORDER BY contenttype, COALESCE(ordinal, 1000000)'));
	}

	/**
	 * Get all but front page.
	 *
	 * @return array
	 */
	public function data_getAllButFront(): array
	{
		return $this->pdo->query(sprintf('SELECT * FROM content WHERE id != 1 ORDER BY contenttype, COALESCE(ordinal, 1000000)'));
	}

	/**
	 * @param $id
	 * @param $role
	 *
	 * @return array|bool
	 */
	public function data_getByID($id, $role)
	{
		if ($role === Users::ROLE_ADMIN) {
			$role = '';
		} else {
			$role = sprintf('AND (role = %d OR role = 0)', $role);
		}

		return $this->pdo->queryOneRow(sprintf('SELECT * FROM content WHERE id = %d %s', $id, $role));
	}

	/**
	 * @return array
	 */
	public function data_getFrontPage(): array
	{
		return $this->pdo->query(sprintf('SELECT * FROM content WHERE status = 1 AND contenttype = %d ORDER BY ordinal ASC, COALESCE(ordinal, 1000000), id', Contents::TYPEINDEX));
	}

	/**
	 * @return array|bool
	 */
	public function data_getIndex()
	{
		return $this->pdo->queryOneRow(sprintf('SELECT * FROM content WHERE status = 1 AND contenttype = %d', Contents::TYPEINDEX));
	}

	/**
	 * @param $id
	 * @param $role
	 *
	 * @return array
	 */
	public function data_getForMenuByTypeAndRole($id, $role): array
	{
		if ($role === Users::ROLE_ADMIN) {
			$role = '';
		} else {
			$role = sprintf('AND (role = %d OR role = 0)', $role);
		}
		return $this->pdo->query(sprintf('SELECT * FROM content WHERE showinmenu = 1 AND status = 1 AND contenttype = %d %s ', $id, $role));
	}
}