newznab-tmux/tests/Install/InstallTest.php

<?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;
			// 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'),
					]
				);
			} catch (\PDOException $e) {
				ColorCLI::doEcho(ColorCLI::error('Unable to connect to MySQL server.'));
				$error = true;
			} 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.
			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;');

            $DbSetup->processSQLFile(); // Setup default schema
            //Insert admin user into database
            if (env('ADMIN_USER') === '' || env('ADMIN_PASS') === '' || env('ADMIN_EMAIL') === '') {
                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();
            }
            $pdo->queryExec(sprintf('INSERT INTO users (username, email, password, user_roles_id, created_at) VALUES (%s, %s, %s, 2, NOW())', $pdo->escapeString(env('ADMIN_USER')), $pdo->escapeString(env('ADMIN_EMAIL')), $pdo->escapeString(User::hashPassword(env('ADMIN_PASS')))));
            ColorCLI::doEcho(ColorCLI::header('Migrating tables and populating them'));
            passthru('php '.NN_ROOT.'artisan migrate');
            passthru('php '.NN_ROOT.'artisan db:seed');

			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) {
					$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->queryExec(
							"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 {
					$error = true;
					ColorCLI::doEcho(ColorCLI::warning('Could not select data from your database.'));
				}
			}
		}

		if (!$error) {
			$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');
	}
}