From 12e25237309a0e358418e17ed1a71e123b89a7be Mon Sep 17 00:00:00 2001
From: Andrew Kelley <andrew@ziglang.org>
Date: Fri, 2 Apr 2021 13:57:16 -0700
Subject: [PATCH] docgen: correct the progress bar

It wasn't showing progress for non-code nodes.
---
 doc/docgen.zig | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/doc/docgen.zig b/doc/docgen.zig
index c77f439f06..8dd299a6c5 100644
--- a/doc/docgen.zig
+++ b/doc/docgen.zig
@@ -1023,6 +1023,7 @@ fn genHtml(allocator: *mem.Allocator, tokenizer: *Tokenizer, toc: *Toc, out: any
     const builtin_code = try getBuiltinCode(allocator, &env_map, zig_exe);
 
     for (toc.nodes) |node| {
+        defer root_node.completeOne();
         switch (node) {
             .Content => |data| {
                 try out.writeAll(data);
@@ -1062,8 +1063,6 @@ fn genHtml(allocator: *mem.Allocator, tokenizer: *Tokenizer, toc: *Toc, out: any
                 try tokenizeAndPrint(tokenizer, out, content_tok);
             },
             .Code => |code| {
-                root_node.completeOne();
-
                 const raw_source = tokenizer.buffer[code.source_token.start..code.source_token.end];
                 const trimmed_raw_source = mem.trim(u8, raw_source, " \n");
                 if (!code.is_inline) {