public function EntityLoad::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/EntityLoad.php, line 21

Class

EntityLoad

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

Namespace

Drupal\drupalmoduleupgrader\Plugin\DMU\Converter\Functions

Code

public function rewrite(FunctionCallNode $call, TargetInterface $target) {
  $arguments = $call
    ->getArguments();

  // If there were three arguments, the call is affecting the internal
  // entity cache. Unfortunately, it's pretty much impossible to reliably
  // determine whether or not they wanted to reset the cache, so let's just
  // leave a FIXME.
  if (sizeof($arguments) == 3) {
    $this
      ->buildFixMe('To reset the entity cache, use EntityStorageInterface::resetCache().')
      ->insertBefore($call);
  }
  $rewritten = ClassMethodCallNode::create('\\Drupal', 'entityTypeManager')
    ->appendMethodCall('getStorage')
    ->appendArgument(clone $arguments[0]);

  // If there's a third argument, conditions were passed. Not a recommended
  // practice, but certain modules might have done it anyway. If we detect
  // conditions, use loadByProperties().
  if (sizeof($arguments) > 2) {
    return $rewritten
      ->appendMethodCall('loadByProperties')
      ->appendArgument(clone $arguments[2]);
  }
  else {
    $rewritten
      ->appendMethodCall('load');
    if (sizeof($arguments) > 1 && $arguments[1] instanceof ArrayNode) {
      $rewritten
        ->appendArgument(clone $arguments[1]);
    }
    return $rewritten;
  }
}