fix: adding code documentation with hasDoc rule

FossilOrigin-Name: 1dc867309571e17e96e5f3de12ef6952a5d3d545282275e3e351020f4d4236bb
thindil · Jun 2, 2024 · f5ab916 · f5ab916
1 parent 9a30ea9
commit f5ab916
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/rules/hasdoc.nim b/src/rules/hasdoc.nim
@@ -235,6 +235,10 @@ fixRule:
     return false
   if astNode.kind == nkObjectTy:
     astNode[2].comment = docTemplate
+  elif astNode.kind == nkStmtList:
+    let docNode: PNode = newNode(kind = nkCommentStmt)
+    docNode.comment = docTemplate
+    astNode.sons = docNode & astNode.sons
   else:
     astNode.comment = docTemplate
   return true