Add graph-build-depends subcommand

1 files changed, 67 insertions, 0 deletions

diff --git a/morphlib/plugins/graphing_plugin.py b/morphlib/plugins/graphing_plugin.py
new file mode 100644
index 00000000..5d679385
--- /dev/null
+++ b/morphlib/plugins/graphing_plugin.py
@@ -0,0 +1,67 @@
+# Copyright (C) 2012  Codethink Limited
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 of the License.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+
+import cliapp
+import os
+
+import morphlib
+
+
+class GraphingPlugin(cliapp.Plugin):
+
+    def enable(self):
+        self.app.add_subcommand('graph-build-depends', 
+                                self.graph_build_depends, 
+                                arg_synopsis='REPO REF MORPHOLOGY')
+
+    def disable(self):
+        pass
+    
+    def graph_build_depends(self, args):
+        for repo_name, ref, filename in self.app._itertriplets(args):
+            self.app.status(msg='Creating build order for '
+                                '%(repo_name)s %(ref)s %(filename)s',
+                            repo_name=repo_name, ref=ref, filename=filename)
+            builder = morphlib.app.BuildCommand(self.app)
+            order = builder.compute_build_order(repo_name, ref, filename)
+
+            basename, ext = os.path.splitext(filename)
+            dot_filename = basename + '.gv'
+            dep_fmt = '  "%s" -> "%s";\n'
+            shape_name = {
+                'system': 'octagon',
+                'stratum': 'box',
+                'chunk': 'ellipse',
+            }
+
+            self.app.status(msg='Writing DOT file to %(filename)s',
+                            filename=dot_filename)
+
+            with open(dot_filename, 'w') as f:
+                f.write('digraph "%s" {\n' % basename)
+                for i, group in enumerate(order.groups):
+                    for artifact in group:
+                        f.write('  "%s" [shape=%s];\n' % 
+                            (artifact.name, 
+                             shape_name[artifact.source.morphology['kind']]))
+                        for dep in artifact.dependencies:
+                            if artifact.source.morphology['kind'] == 'stratum':
+                                if dep.dependents == [artifact]:
+                                    f.write(dep_fmt % (artifact.name, dep.name))
+                            else:
+                                f.write(dep_fmt % (artifact.name, dep.name))
+                f.write('}\n')
+