Fix #44. Allow pattern in environment detection.

themosis · Nov 13, 2016 · cc9d9b8 · cc9d9b8
1 parent c21a665
commit cc9d9b8
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 4 deletions.
diff --git a/config/environment.php b/config/environment.php
@@ -3,4 +3,7 @@
 /*----------------------------------------------------*/
 // Define environment type
 /*----------------------------------------------------*/
-return '';
+return [
+    'local' => 'INSERT-HOSTNAME',
+    'production' => 'INSERT-HOSTNAME'
+];
diff --git a/library/Thms/Config/Environment.php b/library/Thms/Config/Environment.php
@@ -44,16 +44,32 @@ public function which($hostname = '')
         // by comparing the given hostnames from an array.
         if (is_array($this->locations) && !empty($this->locations)) {
             foreach ($this->locations as $location => $host) {
-                $host = is_array($host) ? $host : [$host];
-                if (in_array($hostname, $host)) {
-                    return $location;
+
+                /*
+                 * Test against an array of hosts.
+                 */
+                if (is_array($host)) {
+                    if (in_array($hostname, $host)) {
+                        return $location;
+                    }
+                } else {
+                    /*
+                     * Test against a string/regular expression.
+                     */
+                    $pattern = ($host !== '/') ? str_replace('*', '(.*)', $host).'\z' : '^/$';
+
+                    if ((bool) preg_match('/'.$pattern.'/', $hostname)) {
+                        return $location;
+                    }
                 }
             }
         }
 
         // If not using array, a single string is provided to define the
         // environment.
         if (is_string($this->locations)) {
+
+            // Default string.
             return $this->locations;
         }
 

diff --git a/tests/config/EnvironmentTest.php b/tests/config/EnvironmentTest.php
@@ -61,4 +61,26 @@ public function testWhichEnvironmentWithString()
         $env = new \Thms\Config\Environment($locations);
         $this->assertEquals('local', $env->which());
     }
+
+    public function testWhichEnvironmentWithRegEx()
+    {
+        $locations = [
+            'local' => '*.themosis.dev',
+            'production' => '*.aws.elastic456278s9d.com',
+            'staging' => 'xyz\..+\.net',
+            'custom' => 'my-hostname',
+        ];
+
+        $env = new Thms\Config\Environment($locations);
+
+        $this->assertEquals('local', $env->which('server001.themosis.dev'));
+        $this->assertEquals('local', $env->which('abcserver3578sdd67.themosis.dev'));
+
+        $this->assertEquals('production', $env->which('us2-467.aws.elastic456278s9d.com'));
+
+        $this->assertEquals('staging', $env->which('xyz.some-name.net'));
+        $this->assertEquals('staging', $env->which('xyz.another_name215658.net'));
+
+        $this->assertEquals('custom', $env->which('my-hostname'));
+    }
 }