silverstripe / silverstripe-environmentcheck

tests/Checks/HasFunctionCheckTest.php

Indentation +18 added lines, -18 removed lines

@@ -15,27 +15,27 @@
  */
 class HasFunctionCheckTest extends SapphireTest
 {
-    public function testCheckReportsMissingFunctions()
-    {
-        $check = new HasFunctionCheck('foo');
+	public function testCheckReportsMissingFunctions()
+	{
+		$check = new HasFunctionCheck('foo');
 
-        $expected = [
-            EnvironmentCheck::ERROR,
-            'foo() doesn\'t exist'
-        ];
+		$expected = [
+			EnvironmentCheck::ERROR,
+			'foo() doesn\'t exist'
+		];
 
-        $this->assertEquals($expected, $check->check());
-    }
+		$this->assertEquals($expected, $check->check());
+	}
 
-    public function testCheckReportsFoundFunctions()
-    {
-        $check = new HasFunctionCheck('class_exists');
+	public function testCheckReportsFoundFunctions()
+	{
+		$check = new HasFunctionCheck('class_exists');
 
-        $expected = [
-            EnvironmentCheck::OK,
-            'class_exists() exists'
-        ];
+		$expected = [
+			EnvironmentCheck::OK,
+			'class_exists() exists'
+		];
 
-        $this->assertEquals($expected, $check->check());
-    }
+		$this->assertEquals($expected, $check->check());
+	}
 }

tests/EnvironmentCheckerTest.php

Indentation +57 added lines, -57 removed lines

@@ -18,80 +18,80 @@
  */
 class EnvironmentCheckerTest extends SapphireTest
 {
-    protected $usesDatabase = true;
+	protected $usesDatabase = true;
 
-    protected function tearDown()
-    {
-        EnvironmentCheckSuite::reset();
-        parent::tearDown();
-    }
+	protected function tearDown()
+	{
+		EnvironmentCheckSuite::reset();
+		parent::tearDown();
+	}
 
-    public function testOnlyLogsWithErrors()
-    {
-        Config::modify()->set(EnvironmentChecker::class, 'log_results_warning', true);
-        Config::modify()->set(EnvironmentChecker::class, 'log_results_error', true);
+	public function testOnlyLogsWithErrors()
+	{
+		Config::modify()->set(EnvironmentChecker::class, 'log_results_warning', true);
+		Config::modify()->set(EnvironmentChecker::class, 'log_results_error', true);
 
-        EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckNoErrors());
+		EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckNoErrors());
 
-        $logger = $this->getMockBuilder(Logger::class)
-            ->disableOriginalConstructor()
-            ->setMethods(['log'])
-            ->getMock();
+		$logger = $this->getMockBuilder(Logger::class)
+			->disableOriginalConstructor()
+			->setMethods(['log'])
+			->getMock();
 
-        $logger->expects($this->never())->method('log');
+		$logger->expects($this->never())->method('log');
 
-        Injector::inst()->registerService($logger, LoggerInterface::class);
+		Injector::inst()->registerService($logger, LoggerInterface::class);
 
-        (new EnvironmentChecker('test suite', 'test'))->index();
-    }
+		(new EnvironmentChecker('test suite', 'test'))->index();
+	}
 
-    public function testLogsWithWarnings()
-    {
-        Config::modify()->set(EnvironmentChecker::class, 'log_results_warning', true);
-        Config::modify()->set(EnvironmentChecker::class, 'log_results_error', false);
+	public function testLogsWithWarnings()
+	{
+		Config::modify()->set(EnvironmentChecker::class, 'log_results_warning', true);
+		Config::modify()->set(EnvironmentChecker::class, 'log_results_error', false);
 
-        EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckWarnings());
-        EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckErrors());
+		EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckWarnings());
+		EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckErrors());
 
-        $logger = $this->getMockBuilder(Logger::class)
-            ->disableOriginalConstructor()
-            ->setMethods(['log'])
-            ->getMock();
+		$logger = $this->getMockBuilder(Logger::class)
+			->disableOriginalConstructor()
+			->setMethods(['log'])
+			->getMock();
 
-        $logger->expects($this->once())
-            ->method('log')
-            ->withConsecutive(
-                $this->equalTo(LogLevel::WARNING),
-                $this->anything()
-            );
+		$logger->expects($this->once())
+			->method('log')
+			->withConsecutive(
+				$this->equalTo(LogLevel::WARNING),
+				$this->anything()
+			);
 
-        Injector::inst()->registerService($logger, LoggerInterface::class);
+		Injector::inst()->registerService($logger, LoggerInterface::class);
 
-        (new EnvironmentChecker('test suite', 'test'))->index();
-    }
+		(new EnvironmentChecker('test suite', 'test'))->index();
+	}
 
-    public function testLogsWithErrors()
-    {
-        Config::modify()->set(EnvironmentChecker::class, 'log_results_error', false);
-        Config::modify()->set(EnvironmentChecker::class, 'log_results_error', true);
+	public function testLogsWithErrors()
+	{
+		Config::modify()->set(EnvironmentChecker::class, 'log_results_error', false);
+		Config::modify()->set(EnvironmentChecker::class, 'log_results_error', true);
 
-        EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckWarnings());
-        EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckErrors());
+		EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckWarnings());
+		EnvironmentCheckSuite::register('test suite', new EnvironmentCheckerTest\CheckErrors());
 
-        $logger = $this->getMockBuilder(Logger::class)
-            ->disableOriginalConstructor()
-            ->setMethods(['log'])
-            ->getMock();
+		$logger = $this->getMockBuilder(Logger::class)
+			->disableOriginalConstructor()
+			->setMethods(['log'])
+			->getMock();
 
-        $logger->expects($this->once())
-            ->method('log')
-            ->withConsecutive(
-                [$this->equalTo(LogLevel::ALERT), $this->anything()],
-                [$this->equalTo(LogLevel::WARNING), $this->anything()]
-            );
+		$logger->expects($this->once())
+			->method('log')
+			->withConsecutive(
+				[$this->equalTo(LogLevel::ALERT), $this->anything()],
+				[$this->equalTo(LogLevel::WARNING), $this->anything()]
+			);
 
-        Injector::inst()->registerService($logger, LoggerInterface::class);
+		Injector::inst()->registerService($logger, LoggerInterface::class);
 
-        (new EnvironmentChecker('test suite', 'test'))->index();
-    }
+		(new EnvironmentChecker('test suite', 'test'))->index();
+	}
 }

tests/EnvironmentCheckerTest/CheckNoErrors.php

Indentation +4 added lines, -4 removed lines

@@ -7,8 +7,8 @@
 
 class CheckNoErrors implements EnvironmentCheck, TestOnly
 {
-    public function check()
-    {
-        return [EnvironmentCheck::OK, ''];
-    }
+	public function check()
+	{
+		return [EnvironmentCheck::OK, ''];
+	}
 }

tests/EnvironmentCheckerTest/CheckErrors.php

Indentation +4 added lines, -4 removed lines

@@ -7,8 +7,8 @@
 
 class CheckErrors implements EnvironmentCheck, TestOnly
 {
-    public function check()
-    {
-        return [EnvironmentCheck::ERROR, 'test error'];
-    }
+	public function check()
+	{
+		return [EnvironmentCheck::ERROR, 'test error'];
+	}
 }

tests/EnvironmentCheckerTest/CheckWarnings.php

Indentation +4 added lines, -4 removed lines

@@ -7,8 +7,8 @@
 
 class CheckWarnings implements EnvironmentCheck, TestOnly
 {
-    public function check()
-    {
-        return [EnvironmentCheck::WARNING, 'test warning'];
-    }
+	public function check()
+	{
+		return [EnvironmentCheck::WARNING, 'test warning'];
+	}
 }

src/EnvironmentCheckSuite.php

Indentation +180 added lines, -180 removed lines

@@ -32,184 +32,184 @@
  */
 class EnvironmentCheckSuite
 {
-    use Configurable;
-    use Injectable;
-    use Extensible;
-    /**
-     * Name of this suite.
-     *
-     * @var string
-     */
-    protected $name;
-
-    /**
-     * @var array
-     */
-    protected $checks = [];
-
-    /**
-     * Associative array of named checks registered via the config system. Each check should specify:
-     * - definition (e.g. 'MyHealthCheck("my param")')
-     * - title (e.g. 'Is my feature working?')
-     * - state (setting this to 'disabled' will cause suites to skip this check entirely.
-     *
-     * @var array
-     */
-    private static $registered_checks = [];
-
-    /**
-     * Associative array of named suites registered via the config system. Each suite should enumerate
-     * named checks that have been configured in 'registered_checks'.
-     *
-     * @var array
-     */
-    private static $registered_suites = [];
-
-    /**
-     * Load checks for this suite from the configuration system. This is an alternative to the
-     * EnvironmentCheckSuite::register - both can be used, checks will be appended to the suite.
-     *
-     * @param string $suiteName The name of this suite.
-     */
-    public function __construct($suiteName)
-    {
-        if (empty($this->config()->registered_suites[$suiteName])) {
-            // Not registered via config system, but it still may be configured later via self::register.
-            return;
-        }
-
-        foreach ($this->config()->registered_suites[$suiteName] as $checkName) {
-            if (empty($this->config()->registered_checks[$checkName])) {
-                throw new InvalidArgumentException(
-                    "Bad EnvironmentCheck: '$checkName' - the named check has not been registered."
-                );
-            }
-
-            $check = $this->config()->registered_checks[$checkName];
-
-            // Existing named checks can be disabled by setting their 'state' to 'disabled'.
-            // This is handy for disabling checks mandated by modules.
-            if (!empty($check['state']) && $check['state'] === 'disabled') {
-                continue;
-            }
-
-            // Add the check to this suite.
-            $this->push($check['definition'], $check['title']);
-        }
-    }
-
-    /**
-     * Run this test suite and return the result code of the worst result.
-     *
-     * @return int
-     */
-    public function run()
-    {
-        $result = new EnvironmentCheckSuiteResult();
-
-        foreach ($this->checkInstances() as $check) {
-            list($checkClass, $checkTitle) = $check;
-            try {
-                list($status, $message) = $checkClass->check();
-            // If the check fails, register that as an error
-            } catch (Exception $e) {
-                $status = EnvironmentCheck::ERROR;
-                $message = $e->getMessage();
-            }
-            $result->addResult($status, $message, $checkTitle);
-        }
-
-        return $result;
-    }
-
-    /**
-     * Get instances of all the environment checks.
-     *
-     * @return array
-     * @throws InvalidArgumentException
-     */
-    protected function checkInstances()
-    {
-        $output = [];
-        foreach ($this->checks as $check) {
-            list($checkClass, $checkTitle) = $check;
-            if (is_string($checkClass)) {
-                $checkInst = Injector::inst()->create($checkClass);
-                if ($checkInst instanceof EnvironmentCheck) {
-                    $output[] = [$checkInst, $checkTitle];
-                } else {
-                    throw new InvalidArgumentException(
-                        "Bad EnvironmentCheck: '$checkClass' - the named class doesn't implement EnvironmentCheck"
-                    );
-                }
-            } elseif ($checkClass instanceof EnvironmentCheck) {
-                $output[] = [$checkClass, $checkTitle];
-            } else {
-                throw new InvalidArgumentException("Bad EnvironmentCheck: " . var_export($check, true));
-            }
-        }
-        return $output;
-    }
-
-    /**
-     * Add a check to this suite.
-     *
-     * @param mixed $check
-     * @param string $title
-     */
-    public function push($check, $title = null)
-    {
-        if (!$title) {
-            $title = is_string($check) ? $check : get_class($check);
-        }
-        $this->checks[] = [$check, $title];
-    }
-
-    /////////////////////////////////////////////////////////////////////////////////////////////
-
-    /**
-     * @var array
-     */
-    protected static $instances = [];
-
-    /**
-     * Return a named instance of EnvironmentCheckSuite.
-     *
-     * @param string $name
-     *
-     * @return EnvironmentCheckSuite
-     */
-    public static function inst($name)
-    {
-        if (!isset(self::$instances[$name])) {
-            self::$instances[$name] = new EnvironmentCheckSuite($name);
-        }
-        return self::$instances[$name];
-    }
-
-    /**
-     * Register a check against the named check suite.
-     *
-     * @param string|array $names
-     * @param EnvironmentCheck $check
-     * @param string|array
-     */
-    public static function register($names, $check, $title = null)
-    {
-        if (!is_array($names)) {
-            $names = [$names];
-        }
-
-        foreach ($names as $name) {
-            self::inst($name)->push($check, $title);
-        }
-    }
-
-    /**
-     * Unregisters all checks.
-     */
-    public static function reset()
-    {
-        self::$instances = [];
-    }
+	use Configurable;
+	use Injectable;
+	use Extensible;
+	/**
+	 * Name of this suite.
+	 *
+	 * @var string
+	 */
+	protected $name;
+
+	/**
+	 * @var array
+	 */
+	protected $checks = [];
+
+	/**
+	 * Associative array of named checks registered via the config system. Each check should specify:
+	 * - definition (e.g. 'MyHealthCheck("my param")')
+	 * - title (e.g. 'Is my feature working?')
+	 * - state (setting this to 'disabled' will cause suites to skip this check entirely.
+	 *
+	 * @var array
+	 */
+	private static $registered_checks = [];
+
+	/**
+	 * Associative array of named suites registered via the config system. Each suite should enumerate
+	 * named checks that have been configured in 'registered_checks'.
+	 *
+	 * @var array
+	 */
+	private static $registered_suites = [];
+
+	/**
+	 * Load checks for this suite from the configuration system. This is an alternative to the
+	 * EnvironmentCheckSuite::register - both can be used, checks will be appended to the suite.
+	 *
+	 * @param string $suiteName The name of this suite.
+	 */
+	public function __construct($suiteName)
+	{
+		if (empty($this->config()->registered_suites[$suiteName])) {
+			// Not registered via config system, but it still may be configured later via self::register.
+			return;
+		}
+
+		foreach ($this->config()->registered_suites[$suiteName] as $checkName) {
+			if (empty($this->config()->registered_checks[$checkName])) {
+				throw new InvalidArgumentException(
+					"Bad EnvironmentCheck: '$checkName' - the named check has not been registered."
+				);
+			}
+
+			$check = $this->config()->registered_checks[$checkName];
+
+			// Existing named checks can be disabled by setting their 'state' to 'disabled'.
+			// This is handy for disabling checks mandated by modules.
+			if (!empty($check['state']) && $check['state'] === 'disabled') {
+				continue;
+			}
+
+			// Add the check to this suite.
+			$this->push($check['definition'], $check['title']);
+		}
+	}
+
+	/**
+	 * Run this test suite and return the result code of the worst result.
+	 *
+	 * @return int
+	 */
+	public function run()
+	{
+		$result = new EnvironmentCheckSuiteResult();
+
+		foreach ($this->checkInstances() as $check) {
+			list($checkClass, $checkTitle) = $check;
+			try {
+				list($status, $message) = $checkClass->check();
+			// If the check fails, register that as an error
+			} catch (Exception $e) {
+				$status = EnvironmentCheck::ERROR;
+				$message = $e->getMessage();
+			}
+			$result->addResult($status, $message, $checkTitle);
+		}
+
+		return $result;
+	}
+
+	/**
+	 * Get instances of all the environment checks.
+	 *
+	 * @return array
+	 * @throws InvalidArgumentException
+	 */
+	protected function checkInstances()
+	{
+		$output = [];
+		foreach ($this->checks as $check) {
+			list($checkClass, $checkTitle) = $check;
+			if (is_string($checkClass)) {
+				$checkInst = Injector::inst()->create($checkClass);
+				if ($checkInst instanceof EnvironmentCheck) {
+					$output[] = [$checkInst, $checkTitle];
+				} else {
+					throw new InvalidArgumentException(
+						"Bad EnvironmentCheck: '$checkClass' - the named class doesn't implement EnvironmentCheck"
+					);
+				}
+			} elseif ($checkClass instanceof EnvironmentCheck) {
+				$output[] = [$checkClass, $checkTitle];
+			} else {
+				throw new InvalidArgumentException("Bad EnvironmentCheck: " . var_export($check, true));
+			}
+		}
+		return $output;
+	}
+
+	/**
+	 * Add a check to this suite.
+	 *
+	 * @param mixed $check
+	 * @param string $title
+	 */
+	public function push($check, $title = null)
+	{
+		if (!$title) {
+			$title = is_string($check) ? $check : get_class($check);
+		}
+		$this->checks[] = [$check, $title];
+	}
+
+	/////////////////////////////////////////////////////////////////////////////////////////////
+
+	/**
+	 * @var array
+	 */
+	protected static $instances = [];
+
+	/**
+	 * Return a named instance of EnvironmentCheckSuite.
+	 *
+	 * @param string $name
+	 *
+	 * @return EnvironmentCheckSuite
+	 */
+	public static function inst($name)
+	{
+		if (!isset(self::$instances[$name])) {
+			self::$instances[$name] = new EnvironmentCheckSuite($name);
+		}
+		return self::$instances[$name];
+	}
+
+	/**
+	 * Register a check against the named check suite.
+	 *
+	 * @param string|array $names
+	 * @param EnvironmentCheck $check
+	 * @param string|array
+	 */
+	public static function register($names, $check, $title = null)
+	{
+		if (!is_array($names)) {
+			$names = [$names];
+		}
+
+		foreach ($names as $name) {
+			self::inst($name)->push($check, $title);
+		}
+	}
+
+	/**
+	 * Unregisters all checks.
+	 */
+	public static function reset()
+	{
+		self::$instances = [];
+	}
 }

src/EnvironmentChecker.php

Indentation +289 added lines, -289 removed lines

@@ -24,293 +24,293 @@
  */
 class EnvironmentChecker extends RequestHandler
 {
-    /**
-     * @var array
-     */
-    private static $url_handlers = [
-        '' => 'index',
-    ];
-
-    /**
-     * @var string
-     */
-    protected $checkSuiteName;
-
-    /**
-     * @var string
-     */
-    protected $title;
-
-    /**
-     * @var int
-     */
-    protected $errorCode = 500;
-
-    /**
-     * @var null|string
-     */
-    private static $to_email_address = null;
-
-    /**
-     * @var null|string
-     */
-    private static $from_email_address = null;
-
-    /**
-     * @var bool
-     */
-    private static $email_results = false;
-
-    /**
-     * @var bool Log results via {@link \Psr\Log\LoggerInterface}
-     */
-    private static $log_results_warning = false;
-
-    /**
-     * @var string Maps to {@link \Psr\Log\LogLevel} levels. Defaults to LogLevel::WARNING
-     */
-    private static $log_results_warning_level = LogLevel::WARNING;
-
-    /**
-     * @var bool Log results via a {@link \Psr\Log\LoggerInterface}
-     */
-    private static $log_results_error = false;
-
-    /**
-     * @var int Maps to {@link \Psr\Log\LogLevel} levels. Defaults to LogLevel::ALERT
-     */
-    private static $log_results_error_level = LogLevel::ALERT;
-
-    /**
-     * @param string $checkSuiteName
-     * @param string $title
-     */
-    public function __construct($checkSuiteName, $title)
-    {
-        parent::__construct();
-
-        $this->checkSuiteName = $checkSuiteName;
-        $this->title = $title;
-    }
-
-    /**
-     * @param string $permission
-     *
-     * @throws HTTPResponse_Exception
-     */
-    public function init($permission = 'ADMIN')
-    {
-        // if the environment supports it, provide a basic auth challenge and see if it matches configured credentials
-        if (Environment::getEnv('ENVCHECK_BASICAUTH_USERNAME')
-            && Environment::getEnv('ENVCHECK_BASICAUTH_PASSWORD')
-        ) {
-            if (isset($_SERVER['PHP_AUTH_USER']) && isset($_SERVER['PHP_AUTH_PW'])) {
-                // authenticate the input user/pass with the configured credentials
-                if (!(
-                        $_SERVER['PHP_AUTH_USER'] == Environment::getEnv('ENVCHECK_BASICAUTH_USERNAME')
-                        && $_SERVER['PHP_AUTH_PW'] == Environment::getEnv('ENVCHECK_BASICAUTH_PASSWORD')
-                    )
-                ) {
-                    $response = new HTTPResponse(null, 401);
-                    $response->addHeader('WWW-Authenticate', "Basic realm=\"Environment check\"");
-                    // Exception is caught by RequestHandler->handleRequest() and will halt further execution
-                    $e = new HTTPResponse_Exception(null, 401);
-                    $e->setResponse($response);
-                    throw $e;
-                }
-            } else {
-                $response = new HTTPResponse(null, 401);
-                $response->addHeader('WWW-Authenticate', "Basic realm=\"Environment check\"");
-                // Exception is caught by RequestHandler->handleRequest() and will halt further execution
-                $e = new HTTPResponse_Exception(null, 401);
-                $e->setResponse($response);
-                throw $e;
-            }
-        } else {
-            if (!$this->canAccess(null, $permission)) {
-                return $this->httpError(403);
-            }
-        }
-    }
-
-    /**
-     * @param null|int|Member $member
-     * @param string $permission
-     *
-     * @return bool
-     *
-     * @throws HTTPResponse_Exception
-     */
-    public function canAccess($member = null, $permission = 'ADMIN')
-    {
-        if (!$member) {
-            $member = Security::getCurrentUser();
-        }
-
-        if (!$member) {
-            $member = BasicAuth::requireLogin($this->getRequest(), 'Environment Checker', $permission, false);
-        }
-
-        // We allow access to this controller regardless of live-status or ADMIN permission only
-        // if on CLI.  Access to this controller is always allowed in "dev-mode", or of the user is ADMIN.
-        if (Director::isDev()
-            || Director::is_cli()
-            || empty($permission)
-            || Permission::checkMember($member, $permission)
-        ) {
-            return true;
-        }
-
-        // Extended access checks.
-        // "Veto" style, return NULL to abstain vote.
-        $canExtended = null;
-        $results = $this->extend('canAccess', $member);
-        if ($results && is_array($results)) {
-            if (!min($results)) {
-                return false;
-            }
-            return true;
-        }
-
-        return false;
-    }
-
-    /**
-     * @return HTTPResponse
-     */
-    public function index()
-    {
-        $response = new HTTPResponse;
-        $result = EnvironmentCheckSuite::inst($this->checkSuiteName)->run();
-
-        if (!$result->ShouldPass()) {
-            $response->setStatusCode($this->errorCode);
-        }
-
-        $resultText = $result->customise([
-            'URL' => Director::absoluteBaseURL(),
-            'Title' => $this->title,
-            'Name' => $this->checkSuiteName,
-            'ErrorCode' => $this->errorCode,
-        ])->renderWith(__CLASS__);
-
-        if ($this->config()->get('email_results') && !$result->ShouldPass()) {
-            $email = new Email(
-                $this->config()->get('from_email_address'),
-                $this->config()->get('to_email_address'),
-                $this->title,
-                $resultText
-            );
-            $email->send();
-        }
-
-        // Optionally log errors and warnings individually
-        foreach ($result->Details() as $detail) {
-            if ($this->config()->get('log_results_warning') && $detail->StatusCode == EnvironmentCheck::WARNING) {
-                $this->log(
-                    sprintf('EnvironmentChecker warning at "%s" check. Message: %s', $detail->Check, $detail->Message),
-                    $this->config()->get('log_results_warning_level')
-                );
-            } elseif ($this->config()->get('log_results_error') && $detail->StatusCode == EnvironmentCheck::ERROR) {
-                $this->log(
-                    sprintf('EnvironmentChecker error at "%s" check. Message: %s', $detail->Check, $detail->Message),
-                    $this->config()->get('log_results_error_level')
-                );
-            }
-        }
-
-        // output the result as JSON if requested
-        if ($this->getRequest()->getExtension() == 'json'
-            || strpos($this->getRequest()->getHeader('Accept'), 'application/json') !== false
-        ) {
-            $response->setBody($result->toJSON());
-            $response->addHeader('Content-Type', 'application/json');
-            return $response;
-        }
-
-        $response->setBody($resultText);
-
-        return $response;
-    }
-
-    /**
-     * Sends a log entry to the configured PSR-3 LoggerInterface
-     *
-     * @param string $message
-     * @param int $level
-     */
-    public function log($message, $level)
-    {
-        Injector::inst()->get(LoggerInterface::class)->log($level, $message);
-    }
-
-    /**
-     * Set the HTTP status code that should be returned when there's an error.
-     *
-     * @param int $errorCode
-     */
-    public function setErrorCode($errorCode)
-    {
-        $this->errorCode = $errorCode;
-    }
-
-    /**
-     * @deprecated
-     * @param string $from
-     */
-    public static function set_from_email_address($from)
-    {
-        Deprecation::notice('2.0', 'Use config API instead');
-        static::config()->set('from_email_address', $from);
-    }
-
-    /**
-     * @deprecated
-     * @return null|string
-     */
-    public static function get_from_email_address()
-    {
-        Deprecation::notice('2.0', 'Use config API instead');
-        return static::config()->get('from_email_address');
-    }
-
-    /**
-     * @deprecated
-     * @param string $to
-     */
-    public static function set_to_email_address($to)
-    {
-        Deprecation::notice('2.0', 'Use config API instead');
-        static::config()->set('to_email_address', $to);
-    }
-
-    /**
-     * @deprecated
-     * @return null|string
-     */
-    public static function get_to_email_address()
-    {
-        Deprecation::notice('2.0', 'Use config API instead');
-        return static::config()->get('to_email_address');
-    }
-
-    /**
-     * @deprecated
-     * @param bool $results
-     */
-    public static function set_email_results($results)
-    {
-        Deprecation::notice('2.0', 'Use config API instead');
-        static::config()->set('email_results', $results);
-    }
-
-    /**
-     * @deprecated
-     * @return bool
-     */
-    public static function get_email_results()
-    {
-        Deprecation::notice('2.0', 'Use config API instead');
-        return static::config()->get('email_results');
-    }
+	/**
+	 * @var array
+	 */
+	private static $url_handlers = [
+		'' => 'index',
+	];
+
+	/**
+	 * @var string
+	 */
+	protected $checkSuiteName;
+
+	/**
+	 * @var string
+	 */
+	protected $title;
+
+	/**
+	 * @var int
+	 */
+	protected $errorCode = 500;
+
+	/**
+	 * @var null|string
+	 */
+	private static $to_email_address = null;
+
+	/**
+	 * @var null|string
+	 */
+	private static $from_email_address = null;
+
+	/**
+	 * @var bool
+	 */
+	private static $email_results = false;
+
+	/**
+	 * @var bool Log results via {@link \Psr\Log\LoggerInterface}
+	 */
+	private static $log_results_warning = false;
+
+	/**
+	 * @var string Maps to {@link \Psr\Log\LogLevel} levels. Defaults to LogLevel::WARNING
+	 */
+	private static $log_results_warning_level = LogLevel::WARNING;
+
+	/**
+	 * @var bool Log results via a {@link \Psr\Log\LoggerInterface}
+	 */
+	private static $log_results_error = false;
+
+	/**
+	 * @var int Maps to {@link \Psr\Log\LogLevel} levels. Defaults to LogLevel::ALERT
+	 */
+	private static $log_results_error_level = LogLevel::ALERT;
+
+	/**
+	 * @param string $checkSuiteName
+	 * @param string $title
+	 */
+	public function __construct($checkSuiteName, $title)
+	{
+		parent::__construct();
+
+		$this->checkSuiteName = $checkSuiteName;
+		$this->title = $title;
+	}
+
+	/**
+	 * @param string $permission
+	 *
+	 * @throws HTTPResponse_Exception
+	 */
+	public function init($permission = 'ADMIN')
+	{
+		// if the environment supports it, provide a basic auth challenge and see if it matches configured credentials
+		if (Environment::getEnv('ENVCHECK_BASICAUTH_USERNAME')
+			&& Environment::getEnv('ENVCHECK_BASICAUTH_PASSWORD')
+		) {
+			if (isset($_SERVER['PHP_AUTH_USER']) && isset($_SERVER['PHP_AUTH_PW'])) {
+				// authenticate the input user/pass with the configured credentials
+				if (!(
+						$_SERVER['PHP_AUTH_USER'] == Environment::getEnv('ENVCHECK_BASICAUTH_USERNAME')
+						&& $_SERVER['PHP_AUTH_PW'] == Environment::getEnv('ENVCHECK_BASICAUTH_PASSWORD')
+					)
+				) {
+					$response = new HTTPResponse(null, 401);
+					$response->addHeader('WWW-Authenticate', "Basic realm=\"Environment check\"");
+					// Exception is caught by RequestHandler->handleRequest() and will halt further execution
+					$e = new HTTPResponse_Exception(null, 401);
+					$e->setResponse($response);
+					throw $e;
+				}
+			} else {
+				$response = new HTTPResponse(null, 401);
+				$response->addHeader('WWW-Authenticate', "Basic realm=\"Environment check\"");
+				// Exception is caught by RequestHandler->handleRequest() and will halt further execution
+				$e = new HTTPResponse_Exception(null, 401);
+				$e->setResponse($response);
+				throw $e;
+			}
+		} else {
+			if (!$this->canAccess(null, $permission)) {
+				return $this->httpError(403);
+			}
+		}
+	}
+
+	/**
+	 * @param null|int|Member $member
+	 * @param string $permission
+	 *
+	 * @return bool
+	 *
+	 * @throws HTTPResponse_Exception
+	 */
+	public function canAccess($member = null, $permission = 'ADMIN')
+	{
+		if (!$member) {
+			$member = Security::getCurrentUser();
+		}
+
+		if (!$member) {
+			$member = BasicAuth::requireLogin($this->getRequest(), 'Environment Checker', $permission, false);
+		}
+
+		// We allow access to this controller regardless of live-status or ADMIN permission only
+		// if on CLI.  Access to this controller is always allowed in "dev-mode", or of the user is ADMIN.
+		if (Director::isDev()
+			|| Director::is_cli()
+			|| empty($permission)
+			|| Permission::checkMember($member, $permission)
+		) {
+			return true;
+		}
+
+		// Extended access checks.
+		// "Veto" style, return NULL to abstain vote.
+		$canExtended = null;
+		$results = $this->extend('canAccess', $member);
+		if ($results && is_array($results)) {
+			if (!min($results)) {
+				return false;
+			}
+			return true;
+		}
+
+		return false;
+	}
+
+	/**
+	 * @return HTTPResponse
+	 */
+	public function index()
+	{
+		$response = new HTTPResponse;
+		$result = EnvironmentCheckSuite::inst($this->checkSuiteName)->run();
+
+		if (!$result->ShouldPass()) {
+			$response->setStatusCode($this->errorCode);
+		}
+
+		$resultText = $result->customise([
+			'URL' => Director::absoluteBaseURL(),
+			'Title' => $this->title,
+			'Name' => $this->checkSuiteName,
+			'ErrorCode' => $this->errorCode,
+		])->renderWith(__CLASS__);
+
+		if ($this->config()->get('email_results') && !$result->ShouldPass()) {
+			$email = new Email(
+				$this->config()->get('from_email_address'),
+				$this->config()->get('to_email_address'),
+				$this->title,
+				$resultText
+			);
+			$email->send();
+		}
+
+		// Optionally log errors and warnings individually
+		foreach ($result->Details() as $detail) {
+			if ($this->config()->get('log_results_warning') && $detail->StatusCode == EnvironmentCheck::WARNING) {
+				$this->log(
+					sprintf('EnvironmentChecker warning at "%s" check. Message: %s', $detail->Check, $detail->Message),
+					$this->config()->get('log_results_warning_level')
+				);
+			} elseif ($this->config()->get('log_results_error') && $detail->StatusCode == EnvironmentCheck::ERROR) {
+				$this->log(
+					sprintf('EnvironmentChecker error at "%s" check. Message: %s', $detail->Check, $detail->Message),
+					$this->config()->get('log_results_error_level')
+				);
+			}
+		}
+
+		// output the result as JSON if requested
+		if ($this->getRequest()->getExtension() == 'json'
+			|| strpos($this->getRequest()->getHeader('Accept'), 'application/json') !== false
+		) {
+			$response->setBody($result->toJSON());
+			$response->addHeader('Content-Type', 'application/json');
+			return $response;
+		}
+
+		$response->setBody($resultText);
+
+		return $response;
+	}
+
+	/**
+	 * Sends a log entry to the configured PSR-3 LoggerInterface
+	 *
+	 * @param string $message
+	 * @param int $level
+	 */
+	public function log($message, $level)
+	{
+		Injector::inst()->get(LoggerInterface::class)->log($level, $message);
+	}
+
+	/**
+	 * Set the HTTP status code that should be returned when there's an error.
+	 *
+	 * @param int $errorCode
+	 */
+	public function setErrorCode($errorCode)
+	{
+		$this->errorCode = $errorCode;
+	}
+
+	/**
+	 * @deprecated
+	 * @param string $from
+	 */
+	public static function set_from_email_address($from)
+	{
+		Deprecation::notice('2.0', 'Use config API instead');
+		static::config()->set('from_email_address', $from);
+	}
+
+	/**
+	 * @deprecated
+	 * @return null|string
+	 */
+	public static function get_from_email_address()
+	{
+		Deprecation::notice('2.0', 'Use config API instead');
+		return static::config()->get('from_email_address');
+	}
+
+	/**
+	 * @deprecated
+	 * @param string $to
+	 */
+	public static function set_to_email_address($to)
+	{
+		Deprecation::notice('2.0', 'Use config API instead');
+		static::config()->set('to_email_address', $to);
+	}
+
+	/**
+	 * @deprecated
+	 * @return null|string
+	 */
+	public static function get_to_email_address()
+	{
+		Deprecation::notice('2.0', 'Use config API instead');
+		return static::config()->get('to_email_address');
+	}
+
+	/**
+	 * @deprecated
+	 * @param bool $results
+	 */
+	public static function set_email_results($results)
+	{
+		Deprecation::notice('2.0', 'Use config API instead');
+		static::config()->set('email_results', $results);
+	}
+
+	/**
+	 * @deprecated
+	 * @return bool
+	 */
+	public static function get_email_results()
+	{
+		Deprecation::notice('2.0', 'Use config API instead');
+		return static::config()->get('email_results');
+	}
 }

src/Checks/SolrIndexCheck.php

Indentation +30 added lines, -30 removed lines

@@ -15,38 +15,38 @@
  */
 class SolrIndexCheck implements EnvironmentCheck
 {
-    /**
-     * {@inheritDoc}
-     *
-     * @return array
-     */
-    public function check()
-    {
-        $brokenCores = [];
+	/**
+	 * {@inheritDoc}
+	 *
+	 * @return array
+	 */
+	public function check()
+	{
+		$brokenCores = [];
 
-        if (!class_exists(Solr::class)) {
-            return [
-                EnvironmentCheck::ERROR,
-                'Class `' . Solr::class . '` not found. Is the fulltextsearch module installed?'
-            ];
-        }
+		if (!class_exists(Solr::class)) {
+			return [
+				EnvironmentCheck::ERROR,
+				'Class `' . Solr::class . '` not found. Is the fulltextsearch module installed?'
+			];
+		}
 
-        $service = Solr::service();
-        foreach (Solr::get_indexes() as $index) {
-            /** @var SolrIndex $core */
-            $core = $index->getIndexName();
-            if (!$service->coreIsActive($core)) {
-                $brokenCores[] = $core;
-            }
-        }
+		$service = Solr::service();
+		foreach (Solr::get_indexes() as $index) {
+			/** @var SolrIndex $core */
+			$core = $index->getIndexName();
+			if (!$service->coreIsActive($core)) {
+				$brokenCores[] = $core;
+			}
+		}
 
-        if (!empty($brokenCores)) {
-            return [
-                EnvironmentCheck::ERROR,
-                'The following indexes are unavailable: ' . implode($brokenCores, ', ')
-            ];
-        }
+		if (!empty($brokenCores)) {
+			return [
+				EnvironmentCheck::ERROR,
+				'The following indexes are unavailable: ' . implode($brokenCores, ', ')
+			];
+		}
 
-        return [EnvironmentCheck::OK, 'Expected indexes are available.'];
-    }
+		return [EnvironmentCheck::OK, 'Expected indexes are available.'];
+	}
 }

src/Checks/EnvTypeCheck.php

Indentation +24 added lines, -24 removed lines

@@ -13,28 +13,28 @@
  */
 class EnvTypeCheck implements EnvironmentCheck
 {
-    /**
-     * Check the environment setting.
-     *
-     * @return array
-     */
-    public function check()
-    {
-        $envSetting = Director::get_environment_type();
-        switch ($envSetting) {
-            case 'live':
-                return [
-                    EnvironmentCheck::OK,
-                    "Env setting is 'live'",
-                ];
-            // Fallthrough
-            default:
-            case 'dev':
-            case 'test':
-                return [
-                    EnvironmentCheck::ERROR,
-                    "Env setting is '{$envSetting}' and may disclose information",
-                ];
-        }
-    }
+	/**
+	 * Check the environment setting.
+	 *
+	 * @return array
+	 */
+	public function check()
+	{
+		$envSetting = Director::get_environment_type();
+		switch ($envSetting) {
+			case 'live':
+				return [
+					EnvironmentCheck::OK,
+					"Env setting is 'live'",
+				];
+			// Fallthrough
+			default:
+			case 'dev':
+			case 'test':
+				return [
+					EnvironmentCheck::ERROR,
+					"Env setting is '{$envSetting}' and may disclose information",
+				];
+		}
+	}
 }