You are here

public function DrupalMessageSet::rewrite in Drupal 7 to 8/9 Module Upgrader 8

Tries to rewrite the original function call.

Parameters

\Pharborist\Functions\FunctionCallNode $call: The original function call.

\Drupal\drupalmoduleupgrader\TargetInterface $target: The target module.

Return value

\Pharborist\Node|null If the original function call is returned (determined by object identity), the function call is not replaced. If a different node is returned, it will replace the original call. And if nothing is returned, the original call is commented out with a FIXME.

Overrides FunctionCallModifier::rewrite

File

src/Plugin/DMU/Converter/Functions/DrupalMessageSet.php, line 32

Class

DrupalMessageSet
Plugin annotation @Converter( id = "drupal_set_message", description = @Translation("Rewrites calls to drupal_set_message()."), )

Namespace

Drupal\drupalmoduleupgrader\Plugin\DMU\Converter\Functions

Code

public function rewrite(FunctionCallNode $call, TargetInterface $target) {
  $arguments = $call
    ->getArguments();
  if (isset($arguments[0]) && !empty($arguments[0])) {
    if (isset($arguments[1]) && !empty($arguments[1])) {
      $arguments_1 = trim(trim((string) $arguments[1], "'"), '"');
      $object = ClassMethodCallNode::create('\\Drupal', 'messenger')
        ->appendMethodCall($this->messengers[$arguments_1])
        ->appendArgument(clone $arguments[0]);
      if ($arguments_1 == 'custom') {
        $object
          ->appendArgument(clone $arguments[1]);
      }
      return $object;
    }
    return ClassMethodCallNode::create('\\Drupal', 'messenger')
      ->appendMethodCall('addMessage')
      ->appendArgument(clone $arguments[0]);
  }
  return ClassMethodCallNode::create('\\Drupal', 'messenger')
    ->appendMethodCall('all');
}