Move directory parameter to constructor

Author: clue

bin/graph-composer

@@ -6,5 +6,5 @@ require_once __DIR__ . '/../vendor/autoload.php';
 // directory to scan
 $dir = isset($argv[1]) ? $argv[1] : '.';
 
-$graph = new Clue\GraphComposer();
-$graph->displayGraph($dir);
+$graph = new Clue\GraphComposer($dir);
+$graph->displayGraph();

src/Clue/GraphComposer.php

@@ -28,19 +28,28 @@ class GraphComposer
         'style' => 'dashed'
     );
 
+    private $dependencyGraph;
+    
     /**
      * 
      * @param string $dir
-     * @return \Fhaculty\Graph\Graph
      */
-    public function createGraph($dir)
+    public function __construct($dir)
     {
         $analyzer = new \JMS\Composer\DependencyAnalyzer();
-        $dependencyGraph = $analyzer->analyze($dir);
-        
+        $this->dependencyGraph = $analyzer->analyze($dir);
+    }
+    
+    /**
+     * 
+     * @param string $dir
+     * @return \Fhaculty\Graph\Graph
+     */
+    public function createGraph()
+    {
         $graph = new Graph();
 
-        foreach ($dependencyGraph->getPackages() as $package) {
+        foreach ($this->dependencyGraph->getPackages() as $package) {
             $name = $package->getName();
             $start = $graph->createVertex($name, true);
 
@@ -65,14 +74,14 @@ public function createGraph($dir)
             }
         }
 
-        $graph->getVertex($dependencyGraph->getRootPackage()->getName())->setLayout($this->layoutVertexRoot);
+        $graph->getVertex($this->dependencyGraph->getRootPackage()->getName())->setLayout($this->layoutVertexRoot);
 
         return $graph;
     }
 
-    public function displayGraph($dir)
+    public function displayGraph()
     {
-        $graph = $this->createGraph($dir);
+        $graph = $this->createGraph();
 
         $graphviz = new GraphViz($graph);
         $graphviz->setFormat('svg');

tests/GraphComposerTest.php

@@ -8,8 +8,8 @@ public function testCreateGraph()
     {
         $dir = __DIR__ . '/../';
 
-        $graphComposer = new Clue\GraphComposer();
-        $graph = $graphComposer->createGraph($dir);
+        $graphComposer = new Clue\GraphComposer($dir);
+        $graph = $graphComposer->createGraph();
 
         $this->assertInstanceOf('Fhaculty\Graph\Graph', $graph);
         $this->assertTrue($graph->getNumberOfVertices() > 0);