reader.php | searchcode

/nitrofuran/modules/reader/reader.php

https://github.com/axshavan/nitrofuran
PHP | 540 lines | 404 code | 27 blank | 109 comment | 31 complexity | 99fcf655714878e39e2bed56ca9405cb MD5 | raw file

<?php

/**
 * Описание классов моделей для модуля reader
 *
 * @author Dmitry Nikiforov <axshavan@yandex.ru>
 * @license http://sam.zoy.org/wtfpl WTFPL
 * This program is free software. It comes without any warranty, to
 * the extent permitted by applicable law. You can redistribute it
 * and/or modify it under the terms of the Do What The Fuck You Want
 * To Public License, Version 2, as published by Sam Hocevar. See
 * http://sam.zoy.org/wtfpl/COPYING for more details.
 */

require_once(DOCUMENT_ROOT.'/nitrofuran/crud.class.php');

class CReader
{
	protected $crud; // CRUD модель

	// конструктор
	public function __construct()
	{
		$this->crud = new CRUD();
		$this->someProcedures();
	}

	/**
	 * Добавление группы подписок
	 * @param  string $group_name      имя новой группы
	 * @param  int    $parent_group_id id группы-родителя
	 * @param  string &$error          тут возвращается код ошибки
	 * @return bool
	 */
	public function addGroup($group_name, $parent_group_id, &$error)
	{
		$error = '';
		if(!$group_name || !sizeof($group_name))
		{
			$error = 'EMPTY_NAME';
			return false;
		}
		if(!$this->crud->create(READER_SUBSCRIPTION_GROUP_TABLE, array('name' => $group_name, 'group_id' => (int)$parent_group_id)))
		{
			$error = 'DB_ERROR';
			return false;
		}
		return true;
	}

	/**
	 * Добавить непрочитанный элемент подписки.
	 * @param  array $item
	 * @param  array $subscription данные о подписке
	 * @return int   идентификатор добавленного элемента
	 */
	public function addItem($item, $subscription)
	{
		$res = $this->crud->read
		(
			READER_SUBSCRIPTION_ITEM_TABLE,
			array
			(
				'subscription_id' => $subscription['id'],
				'href'            => $item['href'],
			)
		);
		if(!$res[0] && !$res[0]['id'])
		{
			// за счёт разных часовых поясов и тупого php могут быть косяки со временными метками
			// накинем ещё сутки
			if((int)$item['date'] && $subscription['last_update'] > $item['date'] + 86400)
			{
				return -1;
			}
			$this->crud->create
			(
				READER_SUBSCRIPTION_ITEM_TABLE,
				array
				(
					'name'            => $item['title'],
					'href'            => $item['href'],
					'subscription_id' => $subscription['id'],
					'read_flag'       => 0,
					'date'            => (int)$item['date'] <= gmmktime() ? (int)$item['date'] : gmmktime(),
					'text'            => $item['description']
				)
			);
			return $this->crud->id();
		}
		else
		{
			if($res[0]['read_flag'])
			{
				return -1;
			}
			return $res[0]['id'];
		}
	}

	/**
	 * Добавить подписку
	 * @param  string $href     собственно ссылка на фид
	 * @param  int    $group_id id группы, куда добавляется подписка
	 * @param  string &$error   тут возвращается ошибка
	 * @return bool
	 */
	public function addSubscription($href, $group_id, &$error)
	{
		$error = '';
		if(!$href || !sizeof($href))
		{
			$error = 'EMPTY_HREF';
			return false;
		}
		global $AUTH;
		if
		(
			!$this->crud->create
			(
				READER_SUBSCRIPTION_TABLE,
				array
				(
					'name' => $href,
					'href' => $href,
					'group_id' => (int)$group_id,
					'user_id'  => $AUTH->sess_data['user_id']
				)
			)
		)
		{
			$error = 'DB_ERROR';
			return false;
		}
		return true;
	}

	/**
	 * Получить список свежих элементов подписки с фида
	 * @param  array $subscription    данные о подписке
	 * @param  int   $mostEarlierDate таймстамп самого раннего поста
	 * @return array
	 */
	public function curlGetItems(&$subscription, &$mostEarlierDate)
	{
		$data            = array();
		$mostEarlierDate = $mostEarlierDate ? $mostEarlierDate : gmmktime();
		$curl            = curl_init($subscription['href']);
		curl_setopt($curl, CURLOPT_TIMEOUT, 30);
		curl_setopt($curl, CURLOPT_CONNECTTIMEOUT, 5);
		curl_setopt($curl, CURLOPT_FOLLOWLOCATION, true);
		curl_setopt
		(
			$curl,
			CURLOPT_HTTPHEADER,
			array
			(
				'Accept-Encoding: ',
			)
		);
		curl_setopt($curl, CURLOPT_USERAGENT, 'Nitrofuran Reader: PHP bot collecting RSS feeds');
		ob_start();
		curl_exec($curl);
		$raw_string = ob_get_clean();
		$xml = simplexml_load_string($raw_string);

		if(!$xml)
		{
			// может быть, кто-то игнорирует заголовок Accept-Encoding и отдаёт сжатый gzip текст?
			$tmp_file_name = DOCUMENT_ROOT.'/tmp/'.md5(time());
			file_put_contents($tmp_file_name, $raw_string);
			$cmd = 'cat '.$tmp_file_name.' | $(which gunzip) 2>/dev/null';
			ob_start();
			echo `$cmd`;
			$unpacked_string = ob_get_clean();
			if(strlen($unpacked_string) >= strlen($raw_string))
			{
				$raw_string = $unpacked_string;
			}
			unset($unpacked_string);
			unlink($tmp_file_name);
			$xml = simplexml_load_string($raw_string);
		}
		if($xml)
		{
			// может быть, это что-то похожее на RSS 2.0
			if((string)$xml->attributes()->version == '2.0' && $xml->channel)
			{
				$data = $this->parseRSS20($xml);
			}
			// может быть, это что-то похожее на Atom
			else
			{
				$data = $this->parseAtom($xml);
			}
		}
		foreach($data['items'] as $k => &$item)
		{
			if(!$item['date'])
			{
				$item['date'] = gmmktime();
			}
			elseif($item['date'] < $mostEarlierDate)
			{
				$mostEarlierDate = $item['date'];
			}
			$item['id'] = $this->addItem($item, $subscription);
			if($item['id'] == -1)
			{
				unset($item);
				unset($data['items'][$k]);
			}
		}
		$this->crud->update(READER_SUBSCRIPTION_TABLE, array('id' => $subscription['id']), array('last_update' => gmmktime()));
		$subscription['last_update'] = gmmktime();
		return $data;
	}

	/**
	 * Удаление подписки
	 * @param  int    $id идентификатор подписки
	 * @param  string $error
	 * @return bool
	 */
	public function deleteSubscription($id, &$error)
	{
		$error = '';
		if(!$id)
		{
			$error = 'NO_ID';
			return false;
		}
		$this->crud->delete(READER_SUBSCRIPTION_TABLE, array('id' => $id));
		$this->crud->delete(READER_SUBSCRIPTION_ITEM_TABLE, array('subscription_id' => $id));
		return true;
	}

	/**
	 * Получить список элементов подписки
	 * @param  array $subscription данные о подписке
	 * @param  bool  $forceRead    обновить элементы насильно
	 * @return array
	 */
	public function getItems(&$subscription, $forceRead = false)
	{
		$data = array();
		$mostEarlierDate = gmmktime();
		if(!get_param('reader', 'use_async_run') || $forceRead)
		{
			$data = $this->curlGetItems($subscription, $mostEarlierDate);
		}
		$res = $this->crud->read
		(
			READER_SUBSCRIPTION_ITEM_TABLE,
			array
			(
				'subscription_id' => $subscription['id'],
				'read_flag'       => 0,
				'<date'           => $mostEarlierDate
			)
		);
		foreach($res as $res_row)
		{
			if($res_row['date'] < $mostEarlierDate)
			{
				$mostEarlierDate = $res_row['date'];
			}
			foreach($data['items'] as &$item)
			{
				if($item['href'] == $res_row['href'])
				{
					continue 2;
				}
			}
			$res_row['title']       = $res_row['name'];
			$res_row['description'] = $res_row['text'];
			unset($res_row['text']);
			$data['items'][] = $res_row;
		}
		$res = $this->crud->read
		(
			READER_SUBSCRIPTION_ITEM_TABLE,
			array
			(
				'subscription_id' => $subscription['id'],
				'read_flag'       => 1,
				'>=date'          => $mostEarlierDate
			)
		);
		foreach($res as $res_row)
		{
			foreach($data['items'] as $k => &$item)
			{
				if($item['href'] == $res_row['href'])
				{
					unset($data['items'][$k]);
				}
			}
		}
		usort($data['items'], create_function('$a, $b', 'return $a["date"] < $b["date"] ? 1 : ($a["date"] > $b["date"] ? -1 : 0);'));
		return $data;
	}

	/**
	 * Получить данные об одной подписки
	 * @param  int $id идентификатор подписки
	 * @return array
	 */
	public function getSubscription($id)
	{
		$result = $this->crud->read(READER_SUBSCRIPTION_TABLE, array('id' => (int)$id));
		$result = $result[0];
		return $result;
	}

	/**
	 * Получить данные об одной группе подписок
	 * @param  int $id идентификатор группы подписок
	 * @return array
	 */
	public function getSubscriptionGroup($id)
	{
		$result = $this->crud->read(READER_SUBSCRIPTION_GROUP_TABLE, array('id' => (int)$id));
		return $result[0];
	}

	/**
	 * Получить список папок и подписок
	 * @return array
	 */
	public function getSubscriptions()
	{
		require_once(DOCUMENT_ROOT.'/nitrofuran/graph.class.php');
		global $AUTH;
		global $DB;

		// надо собрать массив с ключами, соответствующими id групп
		$res_g = array();
		$tmp   = $this->crud->read(READER_SUBSCRIPTION_GROUP_TABLE);
		foreach($tmp as $v)
		{
			$res_g[$v['id']] = $v;
		}
		unset($tmp);

		// выборка подписок и раскладывание их по папкам
		//$res_s = $this->crud->read(READER_SUBSCRIPTION_TABLE, array('user_id' => $AUTH->sess_data['user_id']));
		$res_s = $DB->QueryFetched("select s.*, count(i.id) as unread_count from
			`".READER_SUBSCRIPTION_TABLE."` s
			left join `".READER_SUBSCRIPTION_ITEM_TABLE."` i on (i.`subscription_id` = s.id and !i.`read_flag`)
			where s.`user_id` = '".$AUTH->sess_data['user_id']."'
			group by s.`id`");
		foreach($res_s as $v)
		{
			$res_g[$v['group_id']]['subscriptions'][$v['id']] = $v;
		}
		unset($res_s);

		// а теперь завернём всё в граф
		$graph = new CGraph();
		$graph->CreateFromArray($res_g, 'group_id');
		$result = $graph->GetAsArray(true);

		// вынос подписок без группы в корень
		foreach($result['children'][0]['data']['subscriptions'] as $v)
		{
			$result['data']['subscriptions'][] = $v;
		}
		unset($result['children'][0]);
		return $result;
	}

	/**
	 * Пометить элемент прочитанным
	 * @param  int $id
     * @return bool
	 */
	public function readItem($id)
	{
        $item = $this->crud->read(READER_SUBSCRIPTION_ITEM_TABLE, array('id' => $id), array('id' => 'desc'));
        if($item[0]['read_flag'])
        {
            return false;
        }
		$this->crud->update(READER_SUBSCRIPTION_ITEM_TABLE, array('id' => $id), array('read_flag' => 1));
        return true;
	}

	/**
	 * Обновить группу подписок
	 * @param  $id       идентификатор группы
	 * @param  $name     новое название группы
	 * @param  $group_id новая родительяская группа для этой группы
	 * @return bool
	 */
	public function updateGroup($id, $name, $group_id)
	{
		return $this->crud->update
		(
			READER_SUBSCRIPTION_GROUP_TABLE,
			array('id' => (int)$id),
			array
			(
				'name'     => $name,
				'group_id' => (int)$group_id
			)
		);
	}

	/**
	 * Обновить подписку
	 * @param  int    $id       идентификатор обновляемой подписки
	 * @param  string $name     новое название для подписки
	 * @param  int    $group_id новая группа для подписки
	 * @return bool
	 */
	public function updateSubscription($id, $name, $group_id)
	{
		return $this->crud->update
		(
			READER_SUBSCRIPTION_TABLE,
			array('id' => (int)$id),
			array
			(
				'name'     => $name,
				'group_id' => (int)$group_id
			)
		);
	}

	// PROTECTED AREA

	/**
	 * Достать элементы из Atom
	 * @param  SimpleXMLElement $xml
	 * @return array
	 */
	protected function parseAtom(&$xml)
	{
		$_result = array
		(
			'meta'  => array
			(
				'title' => (string)$xml->title
			),
			'items' => array()
		);
		foreach($xml->entry as $entry)
		{
			$href = '';
			foreach($entry->link as $link)
			{
				if
				(
					(string)$link->attributes()->type == 'text/html'
					&& (string)$link->attributes()->rel == 'alternate'
				)
				{
					$href = (string)$link->attributes()->href;
				}
			}
			$_result['items'][] = array
			(
				'title'       => (string)$entry->title,
				'href'        => $href,
				'description' => (string)$entry->content,
				'date'        => strtotime((string)$entry->updated)
			);
		}
		return $_result;
	}

	/**
	 * Достать элементы из  RSS 2.0
	 * @param  SimpleXMLElement $xml
	 * @return array
	 */
	protected function parseRSS20(&$xml)
	{
		$_result = array
		(
			'meta' => array
			(
				'title'         => (string)$xml->channel->title,
				'link'          => (string)$xml->channel->link,
				'description'   => (string)$xml->channel->description,
				'lastBuildDate' => (string)$xml->channel->lastBuildDate,
				'image'         => array
				(
					'url'   => (string)$xml->channel->image->url,
					'link'  => (string)$xml->channel->image->link,
					'title' => (string)$xml->channel->image->title
				)
			),
			'items' => array()
		);
		foreach($xml->channel->item as $item)
		{
			$_result['items'][] = array
			(
				'title'       => (string)$item->title,
                'href'        => (string)$item->link,
                'description' => (string)$item->description,
                'date'        => (string)$item->pubDateUT ? (string)$item->pubDateUT : strtotime((string)$item->pubDate)
			);
		}
		return $_result;
	}

	/**
	 * Некоторые процедуры
	 */
	protected function someProcedures()
	{
		// у прочитанных постов старше двух недель удалим текст
		$this->crud->update
		(
			READER_SUBSCRIPTION_ITEM_TABLE,
			array
			(
				'read_flag' => 1,
				'<date'     => time() - 86400 * 14
			),
			array('text' => '')
		);

		// прочитанные посты старше месяца просто удалим
		$this->crud->delete
		(
			READER_SUBSCRIPTION_ITEM_TABLE,
			array
			(
				'read_flag' => 1,
				'<date'     => time() - 86400 * 31
			)
		);
	}
}

?>