View source  
  <?php
namespace Drupal\drupalmoduleupgrader;
use Drupal\Component\Render\FormattableMarkup;
use Doctrine\Common\Collections\ArrayCollection;
use Pharborist\Node;
use Pharborist\Parser;
use Pharborist\RootNode;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Symfony\Component\Finder\Finder;
class Target implements TargetInterface {
  
  protected $id;
  
  protected $indexerManager;
  
  protected $basePath;
  
  protected $indexers = [];
  
  protected $services;
  
  protected $documents = [];
  
  public function __construct($path, ContainerInterface $container) {
    $this->indexerManager = $container
      ->get('plugin.manager.drupalmoduleupgrader.indexer');
    if (is_dir($path)) {
      $this->basePath = $path;
    }
    else {
      throw new \RuntimeException((new FormattableMarkup('Invalid base path: @path', [
        '@path' => $path,
      ]))
        ->__toString());
    }
  }
  
  public function id() {
    if (empty($this->id)) {
      $dir = $this
        ->getBasePath();
      $info = (new Finder())
        ->in($dir)
        ->depth('== 0')
        ->name('*.info')
        ->getIterator();
      $info
        ->rewind();
      if ($info_file = $info
        ->current()) {
        $this->id = substr($info_file
          ->getFilename(), 0, -5);
      }
      else {
        throw new \RuntimeException((new FormattableMarkup('Could not find info file in @dir', [
          '@dir' => $dir,
        ]))
          ->__toString());
      }
    }
    return $this->id;
  }
  
  public function getBasePath() {
    return $this->basePath;
  }
  
  public function getPath($file) {
    if ($file[0] == '.') {
      $file = $this
        ->id() . $file;
    }
    return $this
      ->getBasePath() . '/' . ltrim($file, '/');
  }
  
  public function getFinder() {
    
    $directories = (new Finder())
      ->directories()
      ->in($this
      ->getBasePath())
      ->filter(function (\SplFileInfo $dir) {
      return (new Finder())
        ->files()
        ->in($dir
        ->getPathname())
        ->depth('== 0')
        ->name('*.info')
        ->count() === 0;
    });
    $directories = array_keys(iterator_to_array($directories));
    $directories[] = $this
      ->getBasePath();
    return (new Finder())
      ->files()
      ->in($directories)
      ->depth('== 0')
      ->name('*.module')
      ->name('*.install')
      ->name('*.inc')
      ->name('*.php')
      ->name('*.test');
  }
  
  public function getIndexer($which) {
    if (empty($this->indexers[$which])) {
      
      $indexer = $this->indexerManager
        ->createInstance($which);
      $indexer
        ->bind($this);
      $this->indexers[$which] = $indexer;
    }
    return $this->indexers[$which];
  }
  
  public function getServices() {
    if (empty($this->services)) {
      $this->services = new ArrayCollection();
    }
    return $this->services;
  }
  
  public function buildIndex() {
    $indexers = array_keys($this->indexerManager
      ->getDefinitions());
    foreach ($indexers as $id) {
      $this
        ->getIndexer($id)
        ->build();
    }
    
    $this
      ->flush();
  }
  
  public function destroyIndex() {
    $indexers = array_keys($this->indexerManager
      ->getDefinitions());
    foreach ($indexers as $id) {
      $this
        ->getIndexer($id)
        ->destroy();
    }
  }
  
  public function implementsHook($hook) {
    return $this
      ->getIndexer('function')
      ->has('hook_' . $hook);
  }
  
  public function executeHook($hook, array $arguments = []) {
    if ($this
      ->implementsHook($hook)) {
      return $this
        ->getIndexer('function')
        ->execute('hook_' . $hook, $arguments);
    }
    else {
      $variables = [
        '@module' => $this
          ->id(),
        '@hook' => $hook,
      ];
      throw new \InvalidArgumentException((new FormattableMarkup('@module does not implement hook_@hook.', $variables))
        ->__toString());
    }
  }
  
  public function open($file) {
    if (empty($this->documents[$file])) {
      $this->documents[$file] = Parser::parseFile($file);
    }
    return $this->documents[$file];
  }
  
  public function save(Node $node = NULL) {
    if ($node) {
      $file = $this
        ->getFileOf($node);
      if ($file) {
        $doc = $node instanceof RootNode ? $node : $node
          ->parents()
          ->get(0);
        $victory = file_put_contents($file, $doc
          ->getText());
        if ($victory === FALSE) {
          throw new IOException((new FormattableMarkup('Failed to save @file.', [
            '@file' => $file,
          ]))
            ->__toString());
        }
      }
      else {
        throw new IOException('Cannot save a node that is not attached to an open document.');
      }
    }
    else {
      array_walk($this->documents, [
        $this,
        'save',
      ]);
    }
  }
  
  public function create($file, $ns = NULL) {
    $this->documents[$file] = RootNode::create($ns);
    return $this->documents[$file];
  }
  
  public function flush() {
    $this->documents = [];
  }
  
  public function getFileOf(Node $node) {
    if ($node instanceof RootNode) {
      $root = $node;
    }
    else {
      $parents = $node
        ->parents();
      if ($parents
        ->isEmpty()) {
        return NULL;
      }
      $root = $parents
        ->get(0);
    }
    foreach ($this->documents as $file => $doc) {
      if ($root === $doc) {
        return $file;
      }
    }
  }
}