hanneskod\classtools\Transformer\Action\NodeStripper

Complexity

Total Complexity

3

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	leaveNode()	0	8	2

<?php
/**
 * This program is free software. It comes without any warranty, to
 * the extent permitted by applicable law. You can redistribute it
 * and/or modify it under the terms of the Do What The Fuck You Want
 * To Public License, Version 2, as published by Sam Hocevar. See
 * http://www.wtfpl.net/ for more details.
 */

declare(strict_types = 1);

namespace hanneskod\classtools\Transformer\Action;

use PhpParser\NodeVisitorAbstract;
use PhpParser\Node;
use PhpParser\NodeTraverser;

/**
 * Strip nodes of registered type
 *
 * @author Hannes Forsgård <hannes.forsgard@fripost.org>
 */
class NodeStripper extends NodeVisitorAbstract
{
    /**
     * @var string Name of node to strip
     */
    private $nodeType;

    /**
     * Register what nodes to strip using a fully quilified PhpParser class name
     *
     * @param string $nodeType Node type (see Node::getType())
     */
    public function __construct(string $nodeType)
    {
        $this->nodeType = $nodeType;
    }

    public function leaveNode(Node $node)
    {
        if ($node->getType() == $this->nodeType) {
            return NodeTraverser::REMOVE_NODE;
        }

        return null;
    }
}