Handle InlineSig construct (pragma), fixes #20

src-literatetests/tests.blt

@@ -253,6 +253,39 @@ func
   -> ()
 
 
+###############################################################################
+###############################################################################
+###############################################################################
+#group type signatures pragmas
+###############################################################################
+###############################################################################
+###############################################################################
+
+#test inline pragma 1
+func = f
+ where
+  {-# INLINE f #-}
+  f = id
+
+#test inline pragma 2
+func = ($)
+ where
+  {-# INLINE ($) #-}
+  ($) = id
+
+#test inline pragma 3
+func = f
+ where
+  {-# INLINE CONLIKE [1] f #-}
+  f = id
+
+#test inline pragma 4
+#pending this does not work with the compiler version we currently use yet (i think). should work with ghc-8.0.2.
+func = f
+ where
+  {-# INLINE [~] f #-}
+  f = id
+
 
 ###############################################################################
 ###############################################################################

src/Language/Haskell/Brittany/Layouters/Decl.hs

@@ -25,6 +25,11 @@ import           GHC ( runGhc, GenLocated(L), moduleNameString )
 import           SrcLoc ( SrcSpan )
 import           HsSyn
 import           Name
+import           BasicTypes ( InlinePragma(..)
+                            , Activation(..)
+                            , InlineSpec(..)
+                            , RuleMatchInfo(..)
+                            )
 import           Language.Haskell.GHC.ExactPrint.Types ( mkAnnKey )
 
 import           Language.Haskell.Brittany.Layouters.Type
@@ -60,6 +65,26 @@ layoutSig lsig@(L _loc sig) = case sig of
              ]
            )
          ]
+  InlineSig name (InlinePragma _ spec _arity phaseAct conlike) ->
+    docWrapNode lsig $ do
+      nameStr <- lrdrNameToTextAnn name
+      let specStr = case spec of
+            Inline          -> "INLINE "
+            Inlinable       -> "INLINABLE "
+            NoInline        -> "NOINLINE "
+            EmptyInlineSpec -> "" -- i have no idea if this is correct.
+      let phaseStr = case phaseAct of
+            NeverActive      -> "[] "
+            AlwaysActive     -> ""
+            ActiveBefore _ i -> "[~" ++ show i ++ "] "
+            ActiveAfter  _ i -> "[" ++ show i ++ "] "
+      let conlikeStr = case conlike of
+            FunLike -> ""
+            ConLike -> "CONLIKE "
+      docLit
+        $  Text.pack ("{-# " ++ specStr ++ conlikeStr ++ phaseStr)
+        <> nameStr
+        <> Text.pack " #-}"
   _ -> briDocByExactNoComment lsig -- TODO
 
 layoutGuardLStmt :: ToBriDoc' (Stmt RdrName (LHsExpr RdrName))