support environments

Author: Karel Mareš

config/ip-filter.php

@@ -12,6 +12,9 @@
     |
     */
 
+	// Env - only use filter on listed environments
+	'env' => ['production'],
+
 	// White list - List of allowed IP addresses
 	'allowed' => [],
 

src/FilterIp.php

@@ -18,13 +18,16 @@ public function handle($request, Closure $next)
 	{
 		$whiteList = config('ip-filter.allowed', []);
 		$blackList = config('ip-filter.denied', []);
+		$environments = config('ip-filter.env', []);
+		$appEnvironment = config('app.env');
 		$ip = $request->ip();
 
 		if (
-			(!count($whiteList) && !count($blackList)) ||
-			count($whiteList) > 0 && in_array($ip, $whiteList) ||
-			count($blackList) > 0 && !in_array($ip, $blackList)
-		) {
+			!in_array($appEnvironment, $environments) || (
+				(!count($whiteList) && !count($blackList)) ||
+				count($whiteList) > 0 && in_array($ip, $whiteList) ||
+				count($blackList) > 0 && !in_array($ip, $blackList)
+			)) {
 			return $next($request);
 		}
 	}