phorge-phorge/src/applications/harbormaster/engine/HarbormasterBuildGraph.php

<?php

/**
 * Directed graph representing a build plan
 */
final class HarbormasterBuildGraph extends AbstractDirectedGraph {

  private $stepMap;

  public static function determineDependencyExecution(
    HarbormasterBuildPlan $plan) {

    $steps = id(new HarbormasterBuildStepQuery())
      ->setViewer(PhabricatorUser::getOmnipotentUser())
      ->withBuildPlanPHIDs(array($plan->getPHID()))
      ->execute();

    $steps_by_phid = mpull($steps, null, 'getPHID');
    $step_phids = mpull($steps, 'getPHID');

    if (count($steps) === 0) {
      return array();
    }

    $graph = id(new HarbormasterBuildGraph($steps_by_phid))
      ->addNodes($step_phids);

    $raw_results =
      $graph->getBestEffortTopographicallySortedNodes();

    $results = array();
    foreach ($raw_results as $node) {
      $results[] = array(
        'node' => $steps_by_phid[$node['node']],
        'depth' => $node['depth'],
        'cycle' => $node['cycle']);
    }

    return $results;
  }

  public function __construct($step_map) {
    $this->stepMap = $step_map;
  }

  protected function loadEdges(array $nodes) {
    $map = array();
    foreach ($nodes as $node) {
      $deps = $this->stepMap[$node]->getDetail('dependsOn', array());

      $map[$node] = array();
      foreach ($deps as $dep) {
        $map[$node][] = $dep;
      }
    }

    return $map;
  }

}