<?php

namespace Icinga\Less;

use Less_VisitorReplacing;

/**
 * Ensure that light mode calls have access to the environment in which the mode was defined
 */
class LightModeVisitor extends Less_VisitorReplacing
{
    use LightModeTrait;

    public $isPreVisitor = true;

    public function visitRulesetCall($c)
    {
        if ($this->getLightMode()->isModule($c->variable)) {
            return LightModeCall::fromRulesetCall($c)->setLightMode($this->getLightMode());
        }

        return $c;
    }

    public function run($node)
    {
        return $this->visitObj($node);
    }
}