diff --git a/src/Analysis/Abstract/ImportGraph.hs b/src/Analysis/Abstract/ImportGraph.hs index 0f0011366..caa3f9d41 100644 --- a/src/Analysis/Abstract/ImportGraph.hs +++ b/src/Analysis/Abstract/ImportGraph.hs @@ -25,6 +25,12 @@ import Prologue hiding (empty, packageName) newtype ImportGraph = ImportGraph { unImportGraph :: G.Graph Name } deriving (Eq, Graph, Show) +data Vertex + = Package PackageName + | Module ModuleName + | Variable Name + deriving (Eq, Ord, Show) + -- | Render a 'ImportGraph' to a 'ByteString' in DOT notation. renderImportGraph :: ImportGraph -> ByteString renderImportGraph = export (defaultStyle friendlyName) . unImportGraph