1 files changed, 12 insertions, 5 deletions

diff --git a/ui-shared.c b/ui-shared.c
index 4175bd8..cf06511 100644
--- a/ui-shared.c
+++ b/ui-shared.c
@@ -483,4 +483,7 @@ void cgit_print_docstart(struct cgit_context *ctx)
 {
-        if (ctx->cfg.embedded)
+        if (ctx->cfg.embedded) {
+                if (ctx->cfg.header)
+                        html_include(ctx->cfg.header);
                 return;
+        }
 
@@ -522,3 +525,9 @@ void cgit_print_docend()
 {
-        html("</div>");
+        html("</div> <!-- class=content -->\n");
+        if (ctx.cfg.embedded) {
+                html("</div> <!-- id=cgit -->\n");
+                if (ctx.cfg.footer)
+                        html_include(ctx.cfg.footer);
+                return;
+        }
         if (ctx.cfg.footer)
@@ -531,5 +540,3 @@ void cgit_print_docend()
         }
-        html("</div>");
-        if (ctx.cfg.embedded)
-                return;
+        html("</div> <!-- id=cgit -->\n");
         html("</body>\n</html>\n");