From aecef373f910ecd33ee9ab92ac63dd949bcf90ab Mon Sep 17 00:00:00 2001
From: Lennart Spitzner <lsp@informatik.uni-kiel.de>
Date: Fri, 30 Dec 2016 22:15:23 +0100
Subject: [PATCH] Add ppUsageWithHelp

---
 src/UI/Butcher/Monadic/Pretty.hs | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/src/UI/Butcher/Monadic/Pretty.hs b/src/UI/Butcher/Monadic/Pretty.hs
index 6da7c16..a7c04a2 100644
--- a/src/UI/Butcher/Monadic/Pretty.hs
+++ b/src/UI/Butcher/Monadic/Pretty.hs
@@ -16,6 +16,7 @@ module UI.Butcher.Monadic.Pretty
   , ppHelpShallow
   , ppPartDescUsage
   , ppPartDescHeader
+  , ppUsageWithHelp
   )
 where
 
@@ -47,11 +48,31 @@ ppUsage (CommandDesc mParent _help _syn parts out children) =
     partDocs = parts <&> ppPartDescUsage
     subsDoc = case out of
       _ | null children -> PP.empty -- TODO: remove debug
-      Nothing           -> PP.parens   $ subDoc
-      Just{}            -> PP.brackets $ subDoc
+      Nothing | null parts ->               subDoc
+              | otherwise  -> PP.parens   $ subDoc
+      Just{}               -> PP.brackets $ subDoc
     subDoc = PP.fcat $ PP.punctuate (PP.text " | ") $ children <&> \(n, _) ->
       PP.text n
 
+ppUsageWithHelp :: CommandDesc a -> PP.Doc
+ppUsageWithHelp (CommandDesc mParent help _syn parts out children) =
+    pparents mParent <+> PP.fsep (partDocs ++ [subsDoc]) PP.<> helpDoc
+  where
+    pparents :: Maybe (String, CommandDesc out) -> PP.Doc
+    pparents Nothing = PP.empty
+    pparents (Just (n, cd)) = pparents (_cmd_mParent cd) <+> PP.text n
+    partDocs = parts <&> ppPartDescUsage
+    subsDoc = case out of
+      _ | null children -> PP.empty -- TODO: remove debug
+      Nothing | null parts ->               subDoc
+              | otherwise  -> PP.parens   $ subDoc
+      Just{}               -> PP.brackets $ subDoc
+    subDoc = PP.fcat $ PP.punctuate (PP.text " | ") $ children <&> \(n, _) ->
+      PP.text n
+    helpDoc = case help of
+      Nothing -> PP.empty
+      Just h  -> PP.text ":" PP.<+> h
+
 ppUsageAt :: [String] -- (sub)command sequence
           -> CommandDesc a
           -> Maybe PP.Doc