From fff5e51426f586562b8438d84ced8330ef3c6975 Mon Sep 17 00:00:00 2001
From: Ben Firshman <ben@firshman.co.uk>
Date: Tue, 29 Apr 2014 09:31:57 +0100
Subject: [PATCH] Make log messages line up with each other

---
 fig/cli/log_printer.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/fig/cli/log_printer.py b/fig/cli/log_printer.py
index c528a126b..e302aecb7 100644
--- a/fig/cli/log_printer.py
+++ b/fig/cli/log_printer.py
@@ -13,6 +13,7 @@ class LogPrinter(object):
     def __init__(self, containers, attach_params=None):
         self.containers = containers
         self.attach_params = attach_params or {}
+        self.prefix_width = self._calculate_prefix_width(containers)
         self.generators = self._make_log_generators()
 
     def run(self):
@@ -20,6 +21,19 @@ class LogPrinter(object):
         for line in mux.loop():
             sys.stdout.write(line.encode(sys.__stdout__.encoding or 'utf-8'))
 
+    def _calculate_prefix_width(self, containers):
+        """
+        Calculate the maximum width of container names so we can make the log
+        prefixes line up like so:
+
+        db_1  | Listening
+        web_1 | Listening
+        """
+        prefix_width = 0
+        for container in containers:
+            prefix_width = max(prefix_width, len(container.name_without_project))
+        return prefix_width
+
     def _make_log_generators(self):
         color_fns = cycle(colors.rainbow())
         generators = []
@@ -31,7 +45,7 @@ class LogPrinter(object):
         return generators
 
     def _make_log_generator(self, container, color_fn):
-        prefix = color_fn(container.name_without_project + " | ")
+        prefix = color_fn(self._generate_prefix(container))
         # Attach to container before log printer starts running
         line_generator = split_buffer(self._attach(container), '\n')
 
@@ -42,6 +56,14 @@ class LogPrinter(object):
         yield color_fn("%s exited with code %s\n" % (container.name, exit_code))
         yield STOP
 
+    def _generate_prefix(self, container):
+        """
+        Generate the prefix for a log line without colour
+        """
+        name = container.name_without_project
+        padding = ' ' * (self.prefix_width - len(name))
+        return ''.join([name, padding, ' | '])
+
     def _attach(self, container):
         params = {
             'stdout': True,