From 8b84b9a4a26d7f2fb07a7e5dcd49b678d1c9be91 Mon Sep 17 00:00:00 2001
From: Karl Zylinski <karl@zylinski.se>
Date: Sat, 14 Sep 2024 14:32:46 +0200
Subject: [PATCH] Docs are generated as expected again.

---
 src/parser.cpp | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/parser.cpp b/src/parser.cpp
index 51da21e9d..523dad8b8 100644
--- a/src/parser.cpp
+++ b/src/parser.cpp
@@ -6388,9 +6388,13 @@ gb_internal bool parse_file(Parser *p, AstFile *f) {
 
 	String filepath = f->tokenizer.fullpath;
 	String base_dir = dir_from_path(filepath);
+	if (f->curr_token.kind == Token_Comment) {
+		consume_comment_groups(f, f->prev_token);
+	}
 
-	Array<Token> tags = array_make<Token>(ast_allocator(f));
+	CommentGroup *docs = f->lead_comment;
 
+	Array<Token> tags = array_make<Token>(temporary_allocator());
 	bool first_invalid_token_set = false;
 	Token first_invalid_token = {};
 
@@ -6410,8 +6414,6 @@ gb_internal bool parse_file(Parser *p, AstFile *f) {
 		}
 	}
 
-	CommentGroup *docs = f->lead_comment;
-
 	if (f->curr_token.kind != Token_package) {
 		ERROR_BLOCK();
 
@@ -6451,6 +6453,8 @@ gb_internal bool parse_file(Parser *p, AstFile *f) {
 	}
 	f->package_name = package_name.string;
 
+	// TODO: Shouldn't single file only matter for build tags? no-instrumentation for example
+	// should be respected even when in single file mode.
 	if (!f->pkg->is_single_file) {
 		if (docs != nullptr && docs->list.count > 0) {
 			for (Token const &tok : docs->list) {
@@ -6485,8 +6489,6 @@ gb_internal bool parse_file(Parser *p, AstFile *f) {
 		}
 	}
 
-	array_free(&tags);
-
 	Ast *pd = ast_package_decl(f, f->package_token, package_name, docs, f->line_comment);
 	expect_semicolon(f);
 	f->pkg_decl = pd;