From 238765e28bae6516c65b47602cd42007c19366da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Frederik=20Hangh=C3=B8j=20Iversen?= <fhi.1990@gmail.com>
Date: Wed, 16 Oct 2019 21:51:47 +0200
Subject: [PATCH] Output dot graph

---
 src/Data/Graphviz.hs | 14 ++++++++++++++
 src/Rubyhs.hs        | 29 +++++++++++++++++++++++++----
 2 files changed, 39 insertions(+), 4 deletions(-)
 create mode 100644 src/Data/Graphviz.hs

diff --git a/src/Data/Graphviz.hs b/src/Data/Graphviz.hs
new file mode 100644
index 0000000..06e60e4
--- /dev/null
+++ b/src/Data/Graphviz.hs
@@ -0,0 +1,14 @@
+module Data.Graphviz (digraph) where
+
+import Frelude
+import qualified Data.Text as Text
+
+digraph :: [NonEmpty Text] -> Text
+digraph es = Text.unlines
+  [ "digraph _ {"
+  , edges es
+  , "}"
+  ]
+
+edges :: [NonEmpty Text] -> Text
+edges = Text.unlines . fmap ((\(x :| xs) -> Text.unlines $ (\y -> x <> " -> " <> y <> ";") <$> xs))
diff --git a/src/Rubyhs.hs b/src/Rubyhs.hs
index fbeb7cc..46a38a5 100644
--- a/src/Rubyhs.hs
+++ b/src/Rubyhs.hs
@@ -4,12 +4,16 @@ module Rubyhs (main) where
 
 import Data.Foldable (traverse_)
 import Data.Language.Ruby (Block)
-import qualified Data.Language.Ruby as Ruby
+import Data.Map (Map)
+import Data.Set (Set)
 import Data.Tree (Tree(Node), Forest)
 import Frelude
 import Options.Applicative (Parser)
 import qualified Data.Aeson as Aeson
 import qualified Data.ByteString.Lazy.Char8 as ByteString
+import qualified Data.Graphviz as Graphviz
+import qualified Data.Language.Ruby as Ruby
+import qualified Data.Text.IO as Text
 import qualified Options.Applicative as Options
 import qualified Rubyhs.References
 
@@ -19,9 +23,26 @@ main = do
   blocks <- parseInput @Block targets
   if printAST
   then traverse_ @[] putEncoded blocks
-  else do
-    traverse_ (putEncoded . Rubyhs.References.references) blocks
-    traverse_ (ByteString.putStrLn . Aeson.encode . toJSONForest . Rubyhs.References.graph) blocks
+  else traverse_ act blocks
+  where
+  act block = do
+    let res@(Rubyhs.References.Result x) = Rubyhs.References.references $ block
+    putEncoded res
+    ByteString.putStrLn . Aeson.encode . toJSONForest . Rubyhs.References.graph $ block
+    drawDot x
+
+graphOut :: FilePath
+graphOut = "out.dot"
+
+drawDot :: Map Rubyhs.References.Node (Set Rubyhs.References.Node)  -> IO ()
+drawDot
+  = Text.writeFile graphOut
+  . Graphviz.digraph
+  . fmap (fmap (qoutes . Rubyhs.References.prettyContext) . \(x, xs) -> x :| toList xs)
+  . toList
+
+qoutes :: Text -> Text
+qoutes x = "\"" <> x <> "\""
 
 toJSONForest :: Forest Rubyhs.References.Node -> Aeson.Value
 toJSONForest = Aeson.Object . fromList . fmap go