better dot format handling

janalis · Jan 25, 2017 · 8044701 · 8044701
1 parent 11646b8
commit 8044701
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/Command/DoctrinevizCommand.php b/src/Command/DoctrinevizCommand.php
@@ -144,16 +144,16 @@ public function execute(InputInterface $input, OutputInterface $output)
         $binary = $input->getOption('binary', null);
         $path = $input->getOption('output-path', null);
         $graphviz = new Graphviz($format, $binary);
-        if (!$path) {
+        if (!$path && 'dot' === $format) {
+            $output->writeln((string) $graph);
+
+            return 0;
+        } elseif (!$path) {
             $graphviz->display($graph);
 
             return 0;
+        } else {
+            return !file_put_contents($path, $graphviz->createImageData($graph));
         }
-        $dotted = 'dot' === $format ? (string) $graph : $graphviz->createImageData($graph);
-        if ($path) {
-            return !file_put_contents($path, $dotted);
-        }
-
-        $output->write($dotted);
     }
 }