<?php
/**
 * This file is part of NNTmux.
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 *
 * @package       NNTmux
 * @author        DariusIII
 * @copyright (c) 2017, NNTmux
 * @version       0.0.1
 */

namespace tests;

require_once \dirname(__DIR__, 2) . DIRECTORY_SEPARATOR . 'bootstrap/autoload.php';

use App\Extensions\util\Versions;
use App\Models\User;
use nntmux\config\Configure;
use nntmux\db\DB;
use nntmux\db\DbUpdate;
use nntmux\ColorCLI;
use Illuminate\Database\Capsule\Manager as Capsule;

/**
 * Class InstallTest
 *
 * @package tests
 */
class InstallTest extends \PHPUnit\Framework\TestCase
{
	/**
	 * @var Configure
	 */
	public $config;


	public function testFullInstall()
	{
		if (! \defined('NN_INSTALLER')) {
			\define('NN_INSTALLER', true);
		}

		$this->config = new Configure('install');

		$pdo = new DB();
		$error = false;

		if (file_exists(NN_ROOT . '_install/install.lock')) {
			ColorCLI::doEcho(ColorCLI::notice('Installation is locked. If you want to reinstall NNTmux, please remove install.lock file from _install folder. ' . PHP_EOL . ColorCLI::warning('This will wipe your database!')));
			exit();
		}

		// Check if user selected right DB type.
		if (env('DB_SYSTEM') !== 'mysql') {
			ColorCLI::doEcho(ColorCLI::error('Invalid database system. Must be: mysql ; Not: ' . env('DB_SYSTEM')));
			$error = true;
		} else {
			// Connect to the SQL server.
			try {
				// HAS to be DB because settings table does not exist yet.
				$pdo = new DB(
					[
						'checkVersion' => true,
						'createDb'     => true,
						'dbhost'       => env('DB_HOST'),
						'dbname'       => env('DB_NAME'),
						'dbpass'       => env('DB_PASSWORD'),
						'dbport'       => env('DB_PORT'),
						'dbsock'       => env('DB_SOCKET'),
						'dbtype'       => env('DB_SYSTEM'),
						'dbuser'       => env('DB_USER'),
					]
				);
				$dbConnCheck = true;
			} catch (\PDOException $e) {
				ColorCLI::doEcho(ColorCLI::error('Unable to connect to MySQL server.'));
				$error = true;
				$dbConnCheck = false;
			} catch (\RuntimeException $e) {
				switch ($e->getCode()) {
					case 1:
					case 2:
					case 3:
						$error = true;
						ColorCLI::doEcho(ColorCLI::alternate($e->getMessage()));
						break;
					default:
						throw new \RuntimeException($e->getMessage(), $e->getCode(), $e);
				}
			}

			// Check if the MySQL version is correct.
			$goodVersion = false;
			if (!$error) {
				try {
					$goodVersion = $pdo->isDbVersionAtLeast(NN_MINIMUM_MYSQL_VERSION);
				} catch (\PDOException $e) {
					$goodVersion = false;
					$error = true;
					ColorCLI::doEcho(ColorCLI::error('Could not get version from MySQL server.'));
				}

				if ($goodVersion === false) {
					$error = true;
					ColorCLI::doEcho(ColorCLI::error(
						'You are using an unsupported version of ' .
						env('DB_SYSTEM') .
						' the minimum allowed version is ' .
						NN_MINIMUM_MYSQL_VERSION
					)
					);
				}
			}
		}

// Start inserting data into the DB.
		if (!$error) {
			$DbSetup = new DbUpdate(
				[
					'backup' => false,
					'db'     => $pdo,
				]
			);
            $pdo->exec('SET FOREIGN_KEY_CHECKS=0;');

			try {
				$DbSetup->processSQLFile(); // Setup default schema
				$DbSetup->processSQLFile( // Process any custom stuff.
					[
						'filepath' => NN_RES . 'db' . DS . 'schema' . DS . 'mysql-data.sql'
					]
				);
				$DbSetup->loadTables(); // Load default data files
			} catch (\PDOException $err) {
				$error = true;
				ColorCLI::doEcho(ColorCLI::error('Error inserting: (' . $err->getMessage() . ')'));
			}

			if (!$error) {
				// Check one of the standard tables was created and has data.
				$dbInstallWorked = false;
				$reschk = $pdo->query('SELECT COUNT(id) AS num FROM tmux');
				if ($reschk === false) {
					$dbCreateCheck = false;
					$error = true;
					ColorCLI::doEcho(ColorCLI::warningOver('Could not select data from your database, check that tables and data are properly created/inserted.'));
				} else {
					foreach ($reschk as $row) {
						if ($row['num'] > 0) {
							$dbInstallWorked = true;
							break;
						}
					}
				}
				$ver = new Versions();
				$patch = $ver->getSQLPatchFromFile();
				if ($dbInstallWorked) {
					$updateSettings = false;
					if ($patch > 0) {
						$updateSettings = $pdo->exec(
							"UPDATE settings SET value = '$patch' WHERE section = '' AND subsection = '' AND name = 'sqlpatch'"
						);
					}
					// If it all worked, continue the install process.
					if ($updateSettings) {
						$message = 'Database updated successfully';
						echo $message . PHP_EOL;
					} else {
						$error = true;
						$message = 'Could not update sqlpatch to ' . $patch . ' for your database.';
						echo $message . PHP_EOL;
					}
				} else {
					$dbCreateCheck = false;
					$error = true;
					ColorCLI::doEcho(ColorCLI::warning('Could not select data from your database.'));
				}
			}
		}
//Insert admin user into database
		if (env('ADMIN_USER') === '' || env('ADMIN_PASS') === '' || env('ADMIN_EMAIL') === '') {
			$error = true;
			ColorCLI::doEcho(ColorCLI::error('Admin user data cannot be empty! Please edit .env file and fill in admin user details and run this script again!'));
			exit();
		}

		$capsule = new Capsule;
		// Same as database configuration file of Laravel.
		$capsule->addConnection([
			'driver'      => env('DB_SYSTEM'),
			'host'        => env('DB_HOST', '127.0.0.1'),
			'port'        => env('DB_PORT', '3306'),
			'database'    => env('DB_NAME', 'nntmux'),
			'username'    => env('DB_USER', 'root'),
			'password'    => env('DB_PASSWORD', ''),
			'unix_socket' => env('DB_SOCKET', ''),
			'charset'     => 'utf8',
			'collation'   => 'utf8_unicode_ci',
			'strict'      => false
		], 'default'
		);
		$capsule->bootEloquent();

		if (!User::isValidUsername(env('ADMIN_USER'))) {
			$error = true;
		} else {
			$usrCheck = User::getByUsername(env('ADMIN_USER'));
			if ($usrCheck) {
				$error = true;
			}
		}
		if (!User::isValidEmail(env('ADMIN_EMAIL'))) {
			$error = true;
		}

		if (!$error) {
			$adminCheck = User::add(env('ADMIN_USER'), env('ADMIN_PASS'), env('ADMIN_EMAIL'), 2, '', '');
			if (!is_numeric($adminCheck)) {
				$error = true;
			}
		}

		if (!$error) {
			$doCheck = true;

			$covers_path = NN_RES . 'covers' . DS;
			$nzb_path = NN_RES . 'nzb' . DS;
			$tmp_path = NN_RES . 'tmp' . DS;
			$unrar_path = $tmp_path . 'unrar' . DS;


			$nzbPathCheck = is_writable($nzb_path);
			if ($nzbPathCheck === false) {
				$error = true;
				$message = $nzb_path . ' is not writable. Please fix folder permissions';
				echo $message . PHP_EOL;
			}

			$lastchar = substr($nzb_path, strlen($nzb_path) - 1);
			if ($lastchar !== '/') {
				$nzb_path .= '/';
			}

			if (!file_exists($unrar_path)) {
				ColorCLI::doEcho(ColorCLI::primary('Creating missing ' . $unrar_path . ' folder'));
				if (!@mkdir($unrar_path) && !is_dir($unrar_path)) {
					throw new \RuntimeException('Unable to create ' . $unrar_path . ' folder');
				}
				$message = 'Folder ' . $unrar_path . ' successfully created';
				echo $message;
			}
			$unrarPathCheck = is_writable($unrar_path);
			if ($unrarPathCheck === false) {
				$error = true;
				$message = $unrar_path . ' is not writable. Please fix folder permissions';
				echo $message . PHP_EOL;
			}

			$lastchar = substr($unrar_path, strlen($unrar_path) - 1);
			if ($lastchar !== '/') {
				$unrar_path .= '/';
			}

			$coversPathCheck = is_writable($covers_path);
			if ($coversPathCheck === false) {
				$error = true;
				$message = $covers_path . ' is not writable. Please fix folder permissions';
				echo $message . PHP_EOL;
			}

			$lastchar = substr($covers_path, strlen($covers_path) - 1);
			if ($lastchar !== '/') {
				$covers_path .= '/';
			}

			if (!$error) {

				$sql1 = sprintf("UPDATE settings SET value = %s WHERE setting = 'nzbpath'", $pdo->escapeString($nzb_path));
				$sql2 = sprintf("UPDATE settings SET value = %s WHERE setting = 'tmpunrarpath'", $pdo->escapeString($unrar_path));
				$sql3 = sprintf("UPDATE settings SET value = %s WHERE setting = 'coverspath'", $pdo->escapeString($covers_path));
				if ($pdo->queryExec($sql1) === false || $pdo->queryExec($sql2) === false || $pdo->queryExec($sql3) === false) {
					$error = true;
				} else {
					$message = 'Settings table updated successfully';
					echo $message . PHP_EOL;
				}
			}
		}

		if (!$error) {
			@file_put_contents(NN_ROOT . '_install/install.lock', '');
			$message = 'NNTmux installation completed successfully';
            $pdo->exec('SET FOREIGN_KEY_CHECKS=1;');
			echo $message . PHP_EOL;
		} else {
			$message = 'NNTmux installation failed. Please fix reported problems and run this script again';
			echo $message . PHP_EOL;
		}
		$this->assertEquals('NNTmux installation completed successfully', $message, 'Test Failed');
	}
}