From 1771a6ef114c7c085fc66301df438045953371b2 Mon Sep 17 00:00:00 2001
From: Frank Mullenger <frank@silverstripe.com>
Date: Fri, 15 Feb 2019 10:07:05 +1300
Subject: [PATCH] NEW: Environment type check.

---
 src/Checks/EnvTypeCheck.php | 40 +++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 src/Checks/EnvTypeCheck.php

diff --git a/src/Checks/EnvTypeCheck.php b/src/Checks/EnvTypeCheck.php
new file mode 100644
index 0000000..1efbfb1
--- /dev/null
+++ b/src/Checks/EnvTypeCheck.php
@@ -0,0 +1,40 @@
+<?php
+
+namespace SilverStripe\EnvironmentCheck\Checks;
+
+use SilverStripe\Control\Director;
+use SilverStripe\EnvironmentCheck\EnvironmentCheck;
+
+/**
+ * Check whether the environment setting is safe. Useful for live sites where a
+ * non "Live" setting might disclose sensitive information.
+ *
+ * @package environmentcheck
+ */
+class EnvTypeCheck implements EnvironmentCheck
+{
+    /**
+     * Check the environment setting.
+     *
+     * @return array
+     */
+    public function check() : array
+    {
+        $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",
+                ];
+        }
+    }
+}