1 files changed, 55 insertions, 24 deletions

diff --git a/ui-shared.c b/ui-shared.c
index 40060ba..07d5dd4 100644
--- a/ui-shared.c
+++ b/ui-shared.c
@@ -36,20 +36,19 @@ void cgit_print_error(char *msg)
 
-char *cgit_hosturl()
+char *cgit_httpscheme()
 {
-        char *host, *port;
-
-        host = getenv("HTTP_HOST");
-        if (host) {
-                host = xstrdup(host);
-        } else {
-                host = getenv("SERVER_NAME");
-                if (!host)
-                        return NULL;
-                port = getenv("SERVER_PORT");
-                if (port && atoi(port) != 80)
-                        host = xstrdup(fmt("%s:%d", host, atoi(port)));
+        if (ctx.env.https && !strcmp(ctx.env.https, "on"))
+                return "https://";
                 else
-                        host = xstrdup(host);
+                return "http://";
         }
-        return host;
+
+char *cgit_hosturl()
+{
+        if (ctx.env.http_host)
+                return ctx.env.http_host;
+        if (!ctx.env.server_name)
+                return NULL;
+        if (!ctx.env.server_port || atoi(ctx.env.server_port) == 80)
+                return ctx.env.server_name;
+        return xstrdup(fmt("%s:%s", ctx.env.server_name, ctx.env.server_port));
 }
@@ -458,2 +457,7 @@ void cgit_print_http_headers(struct cgit_context *ctx)
 {
+        if (ctx->env.no_http && !strcmp(ctx->env.no_http, "1"))
+                return;
+
+        if (ctx->page.status)
+                htmlf("Status: %d %s\n", ctx->page.status, ctx->page.statusmsg);
         if (ctx->page.mimetype && ctx->page.charset)
@@ -470,3 +474,7 @@ void cgit_print_http_headers(struct cgit_context *ctx)
         htmlf("Expires: %s\n", http_date(ctx->page.expires));
+        if (ctx->page.etag)
+                htmlf("ETag: \"%s\"\n", ctx->page.etag);
         html("\n");
+        if (ctx->env.request_method && !strcmp(ctx->env.request_method, "HEAD"))
+                exit(0);
 }
@@ -475,2 +483,8 @@ void cgit_print_docstart(struct cgit_context *ctx)
 {
+        if (ctx->cfg.embedded) {
+                if (ctx->cfg.header)
+                        html_include(ctx->cfg.header);
+                return;
+        }
+
         char *host = cgit_hosturl();
@@ -494,3 +508,4 @@ void cgit_print_docstart(struct cgit_context *ctx)
         if (host && ctx->repo) {
-                html("<link rel='alternate' title='Atom feed' href='http://");
+                html("<link rel='alternate' title='Atom feed' href='");
+                html(cgit_httpscheme());
                 html_attr(cgit_hosturl());
@@ -498,4 +513,6 @@ void cgit_print_docstart(struct cgit_context *ctx)
                                        fmt("h=%s", ctx->qry.head)));
-                html("' type='application/atom+xml'/>");
+                html("' type='application/atom+xml'/>\n");
         }
+        if (ctx->cfg.head_include)
+                html_include(ctx->cfg.head_include);
         html("</head>\n");
@@ -508,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)
@@ -517,2 +540,3 @@ void cgit_print_docend()
         }
+        html("</div> <!-- id=cgit -->\n");
         html("</body>\n</html>\n");
@@ -604,9 +628,4 @@ char *hc(struct cgit_cmd *cmd, const char *page)
 
-void cgit_print_pageheader(struct cgit_context *ctx)
+static void print_header(struct cgit_context *ctx)
 {
-        struct cgit_cmd *cmd = cgit_get_cmd(ctx);
-
-        if (!cmd && ctx->repo)
-                fallback_cmd = "summary";
-
         html("<table id='header'>\n");
@@ -654,2 +673,14 @@ void cgit_print_pageheader(struct cgit_context *ctx)
         html("</td></tr></table>\n");
+}
+
+void cgit_print_pageheader(struct cgit_context *ctx)
+{
+        struct cgit_cmd *cmd = cgit_get_cmd(ctx);
+
+        if (!cmd && ctx->repo)
+                fallback_cmd = "summary";
+
+        html("<div id='cgit'>");
+        if (!ctx->cfg.noheader)
+                print_header(ctx);