<?php

require_once( "../../php/Snoopy.class.inc" );
require_once( "../../php/rtorrent.php" );
require_once( "../../php/util.php" );

require_once( "trackers/rutracker.php" );
require_once( "trackers/anidub.php" );
require_once( "trackers/kinozal.php" );
require_once( "trackers/nnmclub.php" );
require_once( "trackers/tapocheknet.php" );
require_once( "trackers/tfile.php" );
require_once( "trackers/toloka.php" );

eval(FileUtil::getPluginConf( "rutracker_check" ));

class ruTrackerChecker
{
	const STE_INPROGRESS		= 1;
	const STE_UPDATED		= 2;
	const STE_UPTODATE		= 3;
	const STE_DELETED		= 4;
	const STE_CANT_REACH_TRACKER	= 5;
	const STE_ERROR			= 6;
	const STE_NOT_NEED		= 7;
	const STE_IGNORED		= 8;
	
	const MAX_LOCK_TIME		= 900;	// 15 min
	
	private static $TRACKERS = array();
	private static $ANNOUNCES = array();	

	static public function registerTracker($commentFiler, $announceFilter, $handler)
	{
		if(!array_key_exists($commentFiler, self::$TRACKERS)) 
		{
			self::$TRACKERS[$commentFiler] = $handler;
			self::$ANNOUNCES[] = $announceFilter;
		}
	}

	static public function supportedTrackers()
	{
		return(self::$ANNOUNCES);
	}

	static protected function setState( $hash, $state )
	{
		$req = new rXMLRPCRequest( array(
			new rXMLRPCCommand( getCmd("d.set_custom"), array($hash, "chk-state", $state."")  ),
			new rXMLRPCCommand( getCmd("d.set_custom"), array($hash, "chk-time", time()."") )
			));
		if($state == self::STE_UPTODATE)
			$req->addCommand(new rXMLRPCCommand( getCmd("d.set_custom"), array($hash, "chk-stime", time()."") ));
		return($req->success());
	}

	static protected function getState( $hash, &$state, &$time, &$successful_time, &$label )
	{
		$req = new rXMLRPCRequest( array(
			new rXMLRPCCommand( getCmd("d.get_custom"), array($hash, "chk-state")  ),
			new rXMLRPCCommand( getCmd("d.get_custom"), array($hash, "chk-time") ),
			new rXMLRPCCommand( getCmd("d.get_custom"), array($hash, "chk-stime") ),
			new rXMLRPCCommand( getCmd("d.get_custom1"), $hash )
			));
		if($req->success())
		{
			$state = intval($req->val[0]);
			$time = intval($req->val[1]);
			$successful_time = intval($req->val[2]);
			$label = $req->val[3];
			return(true);
		}
		else
		{
			$state = self::STE_INPROGRESS;
			$time = time();
			$successful_time = 0;
			return(false);
		}
	}

	static public function createTorrent($torrent, $hash){
		global $saveUploadedTorrents;
		$torrent = new Torrent( $torrent );

		if( $torrent->errors() ) return self::STE_DELETED;

		if( $torrent->hash_info()==$hash ) return self::STE_UPTODATE;
		$req =  new rXMLRPCRequest( array(
			new rXMLRPCCommand("d.get_directory_base",$hash),
			new rXMLRPCCommand("d.get_custom1",$hash),
			new rXMLRPCCommand("d.get_throttle_name",$hash),
			new rXMLRPCCommand("d.get_connection_seed",$hash),
			new rXMLRPCCommand("d.is_open",$hash),
			new rXMLRPCCommand("d.is_active",$hash),
			new rXMLRPCCommand("d.get_state",$hash),
			new rXMLRPCCommand("d.stop",$hash),
			new rXMLRPCCommand("d.close",$hash),
		));

		if($req->success()){
			$addition = array(
				getCmd("d.set_connection_seed=").$req->val[3],
				getCmd("d.set_custom")."=chk-state,".self::STE_UPDATED,
				getCmd("d.set_custom")."=chk-time,".time(),
				getCmd("d.set_custom")."=chk-stime,".time()
			);
			$isStart = (($req->val[4]!=0) && ($req->val[5]!=0) && ($req->val[6]!=0));
			if(!empty($req->val[2]))
				$addition[] = getCmd("d.set_throttle_name=").$req->val[2];
			if(preg_match('/rat_(\d+)/',$req->val[3],$ratio))
				$addition[] = getCmd("view.set_visible=")."rat_".$ratio;
			$label = rawurldecode($req->val[1]);
			if(rTorrent::sendTorrent($torrent, $isStart, false, $req->val[0],
				$label, $saveUploadedTorrents, false, true, $addition))
			{
				$req = new rXMLRPCRequest( new rXMLRPCCommand("d.erase", $hash ) );
				if($req->success())
					return null;
			}
		}
		return self::STE_ERROR;
	}

	static public function run_ex($hash, $fname){
		$torrent = new Torrent( $fname );
		if(!$torrent->errors()){
			foreach (self::$TRACKERS as $key => $value)
			{
				if( preg_match($key, $torrent->comment()) ) 
				{
					return call_user_func($value, $torrent->comment(), $hash, $torrent);
				}
			}
		}
		return self::STE_NOT_NEED;
	}

	static public function makeClient( $url, $method="GET", $content_type="", $body="" )
	{
		$client = new Snoopy();
		$client->read_timeout = 5;
		$client->_fp_timeout = 5;
		@$client->fetchComplex($url,$method,$content_type,$body);
		return($client);
	}

	static public function run( $hash, $state = null, $time = null, $successful_time = null, $label = null )
	{
		global $ignoreLabels;

		if(is_null($state)) self::getState( $hash, $state, $time, $successful_time, $label );
		
		if (!is_null($label) && in_array($label, $ignoreLabels)) {
			$state = self::STE_IGNORED;
			self::setState($hash, $state);
			return(true);
		}

		if(($state==self::STE_INPROGRESS) && ((time()-$time)>self::MAX_LOCK_TIME)) $state = 0;

		if($state!==self::STE_INPROGRESS){
			$state = self::STE_INPROGRESS;
			if(!self::setState( $hash, $state )) return(false);

			$fname = rTorrentSettings::get()->session.$hash.".torrent";

			if(is_readable($fname))	$state = self::run_ex($hash, $fname);
			if($state==self::STE_INPROGRESS) $state=self::STE_ERROR;

			if(!is_null($state)) self::setState( $hash, $state );
		}
		return($state != self::STE_CANT_REACH_TRACKER);
	}

}
