Add --custom-function parameter to check

src/Settings.php

@@ -74,6 +74,7 @@ public static function parseArguments(array $arguments)
     {
         $arguments = new ArrayIterator(array_slice($arguments, 1));
         $setting = new self;
+        $customFunctions = array();
 
         foreach ($arguments as $argument) {
             if ($argument[0] !== '-') {
@@ -124,6 +125,10 @@ public static function parseArguments(array $arguments)
                         $setting->functionsToCheck[] = self::LARAVEL_DUMP;
                         break;
 
+                    case '--custom-function':
+                        $customFunctions = array_map('trim', explode(',', $arguments->getNext()));
+                        break;
+
                     case '--doctrine':
                         $setting->functionsToCheck[] = self::DOCTRINE_DUMP;
                         $setting->functionsToCheck[] = self::DOCTRINE_DUMP_2;
@@ -134,7 +139,11 @@ public static function parseArguments(array $arguments)
                 }
             }
         }
+
+        // Merge if any custom function is given
+        $setting->functionsToCheck = array_merge($setting->functionsToCheck, $customFunctions);
         $setting->functionsToCheck = array_unique($setting->functionsToCheck);
+
         return $setting;
     }
 

tests/JakubOnderka/PhpVarDumpCheck/CustomFunctionTest.php

@@ -0,0 +1,79 @@
+<?php
+
+use JakubOnderka\PhpVarDumpCheck;
+
+class CustomFunctionTest extends PHPUnit_Framework_TestCase
+{
+
+    public function testCheck_noDebugFunction()
+    {
+        $settings = new PhpVarDumpCheck\Settings();
+
+        $settings->functionsToCheck = array_merge($settings->functionsToCheck, array(
+            'functionName1'
+        ));
+
+        $uut     = new PhpVarDumpCheck\Checker($settings);
+        $content = <<<PHP
+<?php
+nonDebugFunction1(\$var);
+\$i++;
+nonDebugFunction2(\$i);
+PHP;
+        $result = $uut->check($content);
+        $this->assertCount(0, $result);
+    }
+
+    public function testCheck_singleFunction()
+    {
+        $settings = new PhpVarDumpCheck\Settings();
+
+        $settings->functionsToCheck = array_merge($settings->functionsToCheck, array(
+            'functionName1'
+        ));
+
+        $uut     = new PhpVarDumpCheck\Checker($settings);
+        $content = <<<PHP
+<?php
+functionName1(\$var);
+PHP;
+        $result = $uut->check($content);
+        $this->assertCount(1, $result);
+    }
+
+
+    public function testCheck_multipleFunction()
+    {
+        $settings = new PhpVarDumpCheck\Settings();
+
+        $settings->functionsToCheck = array_merge($settings->functionsToCheck, array(
+            'functionName1',
+            'functionName2'
+        ));
+
+        $uut     = new PhpVarDumpCheck\Checker($settings);
+
+        $content1 = <<<PHP
+<?php
+functionName1(\$var);
+PHP;
+        $content2 = <<<PHP
+<?php
+functionName2(\$var);
+PHP;
+        $content3 = <<<PHP
+<?php
+functionName1(\$var);
+sleep(4);
+functionName2(\$var);
+PHP;
+        $result = $uut->check($content1);
+        $this->assertCount(1, $result);
+
+        $result = $uut->check($content2);
+        $this->assertCount(1, $result);
+
+        $result = $uut->check($content3);
+        $this->assertCount(2, $result);
+    }
+}

var-dump-check.php

@@ -21,6 +21,7 @@ function showOptions() {
     --symfony     Enable support for Symfony2 (dump, VarDumper::dump, VarDumper::setHandler, Vardumper::dd())
     --doctrine    Enable support for Doctrine (Doctrine::dump, \Doctrine\Common\Util\Debug::dump)
     --laravel     Enable support for Laravel (dd, dump)
+    --custom-function Comma separated custom function name(s) to check like "pre_echo". 
     --extensions  Check only files with selected extensions separated by comma
                   (default: php, php3, php4, php5, phtml)
     --exclude     Exclude directory. If you want exclude multiple directory, use