Add ClassCheckVisitor

vip-scanner/vip-scanner.php

@@ -45,6 +45,7 @@
 
 require_once( VIP_SCANNER_VISITORS_DIR . '/class-php-file-check-visitor.php' );
 require_once( VIP_SCANNER_VISITORS_DIR . '/class-function-call-check-visitor.php' );
+require_once( VIP_SCANNER_VISITORS_DIR . '/class-class-check-visitor.php' );
 
 
 if ( is_admin() ) {

vip-scanner/visitors/class-class-check-visitor.php

@@ -0,0 +1,28 @@
+<?php
+
+class ClassCheckVisitor extends PhpFileCheckVisitor {
+	protected $checks = array();
+
+	function __construct( $checks ) {
+		$this->checks = $checks;
+	}
+
+	function process( PhpParser\Node $node ) {
+
+		if ( $node instanceof PhpParser\Node\Expr\New_ ) {
+			foreach ( $this->checks as $check => $check_info ) {
+				$this->increment_check_count();
+
+				if ( $node->class->toString() === $check ) {
+					$this->add_error(
+						$check,
+						// If no slug is specified, use the function's name as slug.
+						empty( $check_info['slug'] ) ? $check : $check_info['slug'],
+						$check_info['note'],
+						$check_info['level']
+					);
+				}
+			}
+		}
+	}
+}