<?php
require_once dirname(__FILE__) . '/../../../../www/config.php';

use newznab\db\Settings;
use newznab\utility\Utility;


$pdo = new Settings();
$tRun = new \TmuxRun($pdo);
$tOut = new \TmuxOutput($pdo);

$runVar['paths']['misc'] = NN_MISC;
$runVar['paths']['lib'] = NN_LIB;
$db_name = DB_NAME;
$dbtype = DB_TYPE;
$tmux = $tRun->get('niceness');

$tmux_niceness = (isset($tmux->niceness) ? $tmux->niceness : 2);

$runVar['constants'] = $pdo->queryOneRow($tRun->getConstantSettings());

$PHP = ($tRun->command_exist("php5") ? 'php5' : 'php');
$PYTHON = ($tRun->command_exist("python3") ? 'python3 -OOu' : 'python -OOu');

//assign shell commands
$show_time = (NN_DEBUG ? "/usr/bin/time" : "");
$runVar['commands']['_php'] = $show_time . " nice -n{$tmux_niceness} $PHP";
$runVar['commands']['_phpn'] = "nice -n{$tmux_niceness} $PHP";
$runVar['commands']['_python'] = $show_time . " nice -n{$tmux_niceness} $PYTHON";
$runVar['commands']['_sleep'] = "{$runVar['commands']['_phpn']} {$runVar['paths']['misc']}update_scripts/nix_scripts/tmux/bin/showsleep.php";

//spawn IRCScraper as soon as possible
$tRun->runPane('scraper', $runVar);

//get list of panes by name
$runVar['panes'] = $tRun->getListOfPanes($runVar['constants']);

//totals per category in db, results by parentid
$catcntqry = "SELECT c.parentid AS parentid, COUNT(r.id) AS count FROM category c, releases r WHERE r.categoryid = c.id GROUP BY c.parentid";

//create timers and set to now
$runVar['timers']['timer1'] = $runVar['timers']['timer2'] = $runVar['timers']['timer3'] =
$runVar['timers']['timer4'] = $runVar['timers']['timer5'] = time();

$runVar['timers']['query']['tmux_time'] = $runVar['timers']['query']['split_time'] = $runVar['timers']['query']['init_time'] = $runVar['timers']['query']['proc1_time'] =
$runVar['timers']['query']['proc2_time'] = $runVar['timers']['query']['proc3_time'] = $runVar['timers']['query']['split1_time'] = $runVar['timers']['query']['init1_time'] =
$runVar['timers']['query']['proc11_time'] = $runVar['timers']['query']['proc21_time'] = $runVar['timers']['query']['proc31_time'] = $runVar['timers']['query']['tpg_time'] =
$runVar['timers']['query']['tpg1_time'] = 0;

// Analyze tables
printf($pdo->log->info("\nAnalyzing your tables to refresh your indexes."));
$pdo->optimise(false, 'analyze', false, ['releases']);
Utility::clearScreen();

$runVar['settings']['monitor'] = 0;
$runVar['counts']['iterations'] = 1;
$runVar['modsettings']['fc']['firstrun'] = true;
$runVar['modsettings']['fc']['num'] = 0;

$tblCount = "SELECT TABLE_ROWS AS count FROM information_schema.TABLES WHERE TABLE_NAME = :table AND TABLE_SCHEMA = " . $pdo->escapeString($db_name);
$psTableRowCount = $pdo->Prepare($tblCount);

while ($runVar['counts']['iterations'] > 0) {

	//check the db connection
	if ($pdo->ping(true) == false) {
		unset($pdo);
		$pdo = new Settings();
	}

	$timer01 = time();
	// These queries are very fast, run every loop -- tmux and site settings
	$runVar['settings'] = $pdo->queryOneRow($tRun->getMonitorSettings(), false);
	$runVar['timers']['query']['tmux_time'] = (time() - $timer01);

	$runVar['settings']['book_reqids'] = (!empty($runVar['settings']['book_reqids'])
		? $runVar['settings']['book_reqids'] : \Category::CAT_PARENT_BOOK);

	//get usenet connection info
	$runVar['connections'] = $tOut->getConnectionsInfo($runVar['constants']);

	$runVar['modsettings']['clean'] = ($runVar['settings']['post_non'] == 2 ? ' clean ' : ' ');
	$runVar['constants']['pre_lim'] = ($runVar['counts']['iterations'] > 1 ? '7' : '');

	//assign scripts
	$runVar['scripts']['releases'] = ($runVar['constants']['tablepergroup'] == 0
		? "{$runVar['commands']['_php']} {$runVar['paths']['misc']}update_scripts/nix_scripts/tmux/bin/update_releases.php 1 false"
		: "{$runVar['commands']['_php']} {$runVar['paths']['misc']}update_scripts/nix_scripts/multiprocessing/releases.php"
	);

	switch((int) $runVar['settings']['binaries_run']) {
		case 1:
			$runVar['scripts']['binaries'] = "{$runVar['commands']['_php']} {$runVar['paths']['misc']}update_scripts/nix_scripts/multiprocessing/binaries.php 0";
			break;
		case 2:
			$runVar['scripts']['binaries'] = "{$runVar['commands']['_php']} {$runVar['paths']['misc']}update_scripts/nix_scripts/multiprocessing/safe.php binaries";
			break;
		default:
			$runVar['scripts']['binaries'] = 0;
	}

	switch ((int) $runVar['settings']['backfill']) {
		case 1:
			$runVar['scripts']['backfill'] = "{$runVar['commands']['_php']} {$runVar['paths']['misc']}update_scripts/nix_scripts/multiprocessing/backfill.php";
			break;
		case 2:
			$runVar['scripts']['backfill'] = "{$runVar['commands']['_python']} {$runVar['paths']['misc']}update_scripts/nix_scripts/tmux/python/backfill_threaded.py group";
			break;
		case 4:
			$runVar['scripts']['backfill'] = "{$runVar['commands']['_php']} {$runVar['paths']['misc']}update_scripts/nix_scripts/multiprocessing/safe.php backfill";
	}

	//get usenet connection counts
	unset ($runVar['conncounts']);
	$runVar['conncounts'] = $tOut->getUSPConnections('primary', $runVar['connections']);

	if ($runVar['constants']['alternate_nntp'] == 1) {
		$runVar['conncounts'] += $tOut->getUSPConnections('alternate', $runVar['connections']);
	}

	//run queries only after time exceeded, these queries can take awhile
	if ($runVar['counts']['iterations'] == 1 || (time() - $runVar['timers']['timer2'] >= $runVar['settings']['monitor'] && $runVar['settings']['is_running'] == 1)) {

		$runVar['counts']['proc1'] = $runVar['counts']['proc2'] = $runVar['counts']['proc3'] = $splitqry = $newOldqry = false;
		$runVar['counts']['now']['total_work'] = 0;
		$runVar['modsettings']['fix_crap'] = explode(', ', ($runVar['settings']['fix_crap']));

		echo $pdo->log->info("\nThe numbers(queries) above are currently being refreshed. \nNo pane(script) can be (re)started until these have completed.\n");
		$timer02 = time();

		$splitqry = $newOldqry = '';

		$splitqry = $tRun->proc_query(4, $runVar['settings']['book_reqids'], $runVar['settings']['request_hours'], $db_name);
		$newOldqry = $tRun->proc_query(6, $runVar['settings']['book_reqids'], $runVar['settings']['request_hours'], $db_name);

		$splitres = $pdo->queryOneRow($splitqry, false);
		$runVar['timers']['newOld'] = $pdo->queryOneRow($newOldqry, false);

		//assign split query results to main var
		if (is_array($splitres)) {
			foreach ($splitres AS $splitkey => $split) {
				$runVar['counts']['now'][$splitkey] = $split;
			}
		}

		$runVar['timers']['query']['split_time'] = (time() - $timer02);
		$runVar['timers']['query']['split1_time'] = (time() - $timer01);

		$timer03 = time();

		//This is subpartition compatible -- loops through all partitions and adds their total row counts instead of doing a slow query count
		$partitions = $pdo->queryDirect(
			sprintf("
				SELECT SUM(TABLE_ROWS) AS count, PARTITION_NAME AS category
				FROM INFORMATION_SCHEMA.PARTITIONS
				WHERE TABLE_NAME = 'releases'
				AND TABLE_SCHEMA = %s
				GROUP BY PARTITION_NAME",
				$pdo->escapeString($db_name)
			)
		);
		foreach ($partitions as $partition) {
			$runVar['counts']['now'][$partition['category']] = $partition['count'];
		}
		unset($partitions);

		$runVar['timers']['query']['init_time'] = (time() - $timer03);
		$runVar['timers']['query']['init1_time'] = (time() - $timer01);

		$timer04 = time();
		$proc1qry = $tRun->proc_query(1, $runVar['settings']['book_reqids'], $runVar['settings']['request_hours'], $db_name);
		$proc1res = $pdo->queryOneRow(($proc1qry !== false ? $proc1qry : ''), $tRun->rand_bool($runVar['counts']['iterations']));
		$runVar['timers']['query']['proc1_time'] = (time() - $timer04);
		$runVar['timers']['query']['proc11_time'] = (time() - $timer01);

		$timer05 = time();
		$proc2qry = $tRun->proc_query(2, $runVar['settings']['book_reqids'], $runVar['settings']['request_hours'], $db_name);
		$proc2res = $pdo->queryOneRow(($proc2qry !== false ? $proc2qry : ''), $tRun->rand_bool($runVar['counts']['iterations']));
		$runVar['timers']['query']['proc2_time'] = (time() - $timer05);
		$runVar['timers']['query']['proc21_time'] = (time() - $timer01);

		$timer06 = time();
		$proc3qry = $tRun->proc_query(3, $runVar['settings']['book_reqids'], $runVar['settings']['request_hours'], $db_name);
		$proc3res = $pdo->queryOneRow(($proc3qry !== false ? $proc3qry : ''), $tRun->rand_bool($runVar['counts']['iterations']));
		$runVar['timers']['query']['proc3_time'] = (time() - $timer06);
		$runVar['timers']['query']['proc31_time'] = (time() - $timer01);

		$timer07 = time();
		if ($runVar['constants']['tablepergroup'] == 1) {
			$sql = 'SHOW TABLE STATUS';

			$tables = $pdo->queryDirect($sql);
			$age = time();

			$runVar['counts']['now']['collections_table'] = $runVar['counts']['now']['binaries_table'] = 0;
			$runVar['counts']['now']['parts_table'] = $runVar['counts']['now']['parterpair_table'] = 0;

			if ($psTableRowCount === false) {
				echo "Unable to prepare statement, skipping monitor updates!";
			} else {
				if ($tables instanceof \Traversable) {
					foreach ($tables as $row) {
						$tbl   = $row['name'];
						$stamp = 'UNIX_TIMESTAMP(MIN(dateadded))';

						switch (true) {
							case strpos($tbl, 'collections_') !== false:
								$runVar['counts']['now']['collections_table'] += getTableRowCount($psTableRowCount,
									$tbl);
								$added = $pdo->queryOneRow(
									sprintf('SELECT %s AS dateadded FROM %s',
										$stamp,
										$tbl
									)
								);
								if (isset($added['dateadded']) && is_numeric($added['dateadded']) &&
									$added['dateadded'] < $age
								) {
									$age = $added['dateadded'];
								}
								break;
							case strpos($tbl, 'binaries_') !== false:
								$runVar['counts']['now']['binaries_table'] += getTableRowCount($psTableRowCount,
									$tbl);
								break;
							// This case must come before the 'parts_' one.
							case strpos($tbl, 'partrepair_') !== false:
								$runVar['counts']['now']['partrepair_table'] += getTableRowCount($psTableRowCount,
									$tbl);

								break;
							case strpos($tbl, 'parts_') !== false:
								$runVar['counts']['now']['parts_table'] += getTableRowCount($psTableRowCount,
									$tbl);
								break;
							default:
						}
					}
					$runVar['timers']['newOld']['oldestcollection'] = $age;

					//free up memory used by now stale data
					unset($age, $added, $tables);

					$runVar['timers']['query']['tpg_time']  = (time() - $timer07);
					$runVar['timers']['query']['tpg1_time'] = (time() - $timer01);
				}
			}
		}
		$runVar['timers']['timer2'] = time();

		//assign postprocess values from $proc
		if (is_array($proc1res)) {
			foreach ($proc1res AS $proc1key => $proc1) {
				$runVar['counts']['now'][$proc1key] = $proc1;
			}
		} else {
			errorOnSQL($pdo);
		}

		if (is_array($proc2res)) {
			foreach ($proc2res AS $proc2key => $proc2) {
				$runVar['counts']['now'][$proc2key] = $proc2;
			}
		} else {
			errorOnSQL($pdo);
		}
		if (is_array($proc3res)) {
			foreach ($proc3res AS $proc3key => $proc3) {
				$runVar['counts']['now'][$proc3key] = $proc3;
			}
		} else {
			errorOnSQL($pdo);
		}

		// now that we have merged our query data we can unset these to free up memory
		unset($proc1res, $proc2res, $proc3res, $splitres);

		// Zero out any post proc counts when that type of pp has been turned off
		foreach ($runVar['settings'] as $settingkey => $setting) {
			if (strpos($settingkey, 'process') == 0 && $setting == 0) {
				$runVar['counts']['now'][$settingkey] = $runVar['counts']['start'][$settingkey] = 0;
			}
		}

		//set initial start postproc values from work queries -- this is used to determine diff variables
		if ($runVar['counts']['iterations'] == 1) {
			$runVar['counts']['start'] = $runVar['counts']['now'];
		}

		foreach ($runVar['counts']['now'] as $key => $proc) {

			//if key is a process type, add it to total_work
			if (strpos($key, 'process') === 0) {
				$runVar['counts']['now']['total_work'] += $proc;
			}

			//calculate diffs
			$runVar['counts']['diff'][$key] = number_format($proc - $runVar['counts']['start'][$key]);

			//calculate percentages -- if user has no releases, set 0 for each key or this will fail on divide by zero
			$runVar['counts']['percent'][$key] = ($runVar['counts']['now']['releases'] > 0
				? sprintf("%02s", floor(($proc / $runVar['counts']['now']['releases']) * 100)) : 0);
		}

		$runVar['counts']['now']['total_work'] += $runVar['counts']['now']['work'];

		// Set initial total work count for diff
		if ($runVar['counts']['iterations'] == 1) {
			$runVar['counts']['start']['total_work'] = $runVar['counts']['now']['total_work'];
		}

		// Set diff total work count
		$runVar['counts']['diff']['total_work'] = number_format($runVar['counts']['now']['total_work'] - $runVar['counts']['start']['total_work']);
	}

	//set kill switches
	$runVar['killswitch']['pp'] = (($runVar['settings']['postprocess_kill'] < $runVar['counts']['now']['total_work']) && ($runVar['settings']['postprocess_kill'] != 0)
		? true
		: false
	);
	$runVar['killswitch']['coll'] = (($runVar['settings']['collections_kill'] < $runVar['counts']['now']['collections_table']) && ($runVar['settings']['collections_kill'] != 0)
		? true
		: false
	);

	$tOut->updateMonitorPane($runVar);

	//begin pane run execution
	if ($runVar['settings']['is_running'] === '1') {

		//run main updating function(s)
		$tRun->runPane('main', $runVar);

		//run nzb-import
		$tRun->runPane('import', $runVar);

		//run postprocess_releases amazon
		$tRun->runPane('amazon', $runVar);

		//respawn IRCScraper if it has been killed
		$tRun->runPane('scraper', $runVar);

		//run sharing regardless of sequential setting
		$tRun->runPane('sharing', $runVar);

		//update tv and theaters
		$tRun->runPane('updatetv', $runVar);

		//run these if complete sequential not set
		if ($runVar['constants']['sequential'] != 2) {

			//fix names
			$tRun->runPane('fixnames', $runVar);

			//dehash releases
			$tRun->runPane('dehash', $runVar);

			// Remove crap releases.
			$tRun->runPane('removecrap', $runVar);

			//run postprocess_releases additional
			$tRun->runPane('ppadditional', $runVar);

			//run postprocess_releases non amazon
			$tRun->runPane('nonamazon', $runVar);
		}

	} else  if ($runVar['settings']['is_running'] === '0') {
		$tRun->runPane('notrunning', $runVar);
	}

	$runVar['counts']['iterations']++;
	sleep(10);
}

function errorOnSQL($pdo)
{
	echo $pdo->log->error(PHP_EOL . "Monitor encountered severe errors retrieving process data from MySQL.  Please diagnose and try running again." . PHP_EOL);
	exit;
}

function getTableRowCount(PDOStatement &$ps, $table)
{
	$success = $ps->execute([':table' => $table]);
	if ($success) {
		$result = $ps->fetch();

		return is_numeric($result['count']) ? $result['count'] : 0;
	}

	return false;
}