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
@@ -480,10 +480,13 @@ void cgit_print_http_headers(struct cgit_context *ctx)
 }
 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;
+        }
        char *host = cgit_hosturl();
        html(cgit_doctype);
        html("<html xmlns='http://www.w3.org/1999/xhtml' xml:lang='en' lang='en'>\n");
@@ -519,20 +522,24 @@ void cgit_print_docstart(struct cgit_context *ctx)
 }
 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)
                html_include(ctx.cfg.footer);
        else {
                htmlf("<div class='footer'>generated  by cgit %s at ",
                        cgit_version);
                cgit_print_date(time(NULL), FMT_LONGDATE, ctx.cfg.local_time);
                html("</div>\n");
        }
-        html("</div>");
+        html("</div> <!-- id=cgit -->\n");
-        if (ctx.cfg.embedded)
-                return;
        html("</body>\n</html>\n");
 }
 int print_branch_option(const char *refname, const unsigned char *sha1,