new line after comment

nammayatri · Nov 19, 2024 · f32e6c4 · f32e6c4
1 parent fb5bddf
commit f32e6c4
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/lib/namma-dsl/src/NammaDSL/Generator/Haskell/Common.hs b/lib/namma-dsl/src/NammaDSL/Generator/Haskell/Common.hs
@@ -344,13 +344,14 @@ mkFullUserActionType apiRead apiTT = do
 delimiterComment :: String -> Writer r CodeUnit
 delimiterComment handlerName = do
   let delimiterLength = 80
-  commentW $ " " <> replicate (delimiterLength - 3) '='
+  commentW $ " " <> replicate (delimiterLength - 4) '='
   commentW $
     replicate 11 '-'
       <> " "
       <> handlerName
       <> do
-        let remainder = delimiterLength - length handlerName - 15
+        let remainder = delimiterLength - length handlerName - 16
         if remainder > 0
           then " " <> replicate remainder '-'
           else ""
+  addNewLineW
diff --git a/lib/namma-dsl/src/NammaDSL/Lib/Interpreter.hs b/lib/namma-dsl/src/NammaDSL/Lib/Interpreter.hs
@@ -70,6 +70,7 @@ interpretSplice (Splice e) = "$(" <> pprint' e <> ")" <> "\n\n"
 
 interpretComment :: Comment -> String
 interpretComment (Comment str) = "--" <> str <> "\n"
+interpretComment AddNewLine = "\n"
 
 myStyle :: Pretty.Style
 myStyle = Pretty.style {Pretty.lineLength = 300}

diff --git a/lib/namma-dsl/src/NammaDSL/Lib/TH.hs b/lib/namma-dsl/src/NammaDSL/Lib/TH.hs
@@ -428,3 +428,6 @@ matchWOD patQ bodyQ = do
 
 commentW :: String -> Writer r CodeUnit
 commentW str = tell [CodeComment (Comment str)]
+
+addNewLineW :: Writer r CodeUnit
+addNewLineW = tell [CodeComment AddNewLine]
diff --git a/lib/namma-dsl/src/NammaDSL/Lib/Types.hs b/lib/namma-dsl/src/NammaDSL/Lib/Types.hs
@@ -25,7 +25,8 @@ data CodeUnit = CodeDec [TH.Dec] | CodeSplice Splice | CodeComment Comment
 
 newtype Splice = Splice TH.Exp deriving (Eq)
 
-newtype Comment = Comment String
+-- sometimes extra new line required after comment
+data Comment = Comment String | AddNewLine
 
 type Writer r w = WriterT [w] (Q r) ()