X-Git-Url: https://git.armaanb.net/?a=blobdiff_plain;f=src%2Fstagit.c;h=217021ee896c6a2124693b829175f65c69315ea1;hb=bf62b511ebf4be03e07a0f2aa4d01bf0289ca8b0;hp=d1bf33c3eacb72bccab8391a3b10516fcc1c3aa8;hpb=338741f03a073f0963d737d05c15e9d472fa467d;p=stagit.git

diff --git a/src/stagit.c b/src/stagit.c
index d1bf33c..217021e 100644
--- a/src/stagit.c
+++ b/src/stagit.c
@@ -8,11 +8,13 @@
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <stdbool.h>
 #include <string.h>
 #include <time.h>
 #include <unistd.h>
 
 #include <git2.h>
+#include <cmark-gfm.h>
 
 #include "cp.h"
 #include "compat.h"
@@ -65,6 +67,8 @@ static char *readmefiles[] = { "HEAD:README", "HEAD:README.md" };
 static char *readme;
 static long long nlogcommits = -1; /* < 0 indicates not used */
 
+bool htmlized; /* true if markdoown converted to HTML */
+
 /* cache */
 static git_oid lastoid;
 static char lastoidstr[GIT_OID_HEXSZ + 2]; /* id + newline + NUL byte */
@@ -387,6 +391,8 @@ writeheader(FILE *fp, const char *title)
 	if (license)
 		fprintf(fp, " | <a href=\"%sfile/%s.html\">LICENSE</a>",
 		        relpath, license);
+	fprintf(fp, " | <a href=\"%s%s.tar.gz\">Download</a>",
+					relpath, strippedname);
 	fputs("</td></tr></table>\n<hr/>\n<div id=\"content\">\n", fp);
 }
 
@@ -396,38 +402,57 @@ writefooter(FILE *fp)
 	fputs("</div>\n</body>\n</html>\n", fp);
 }
 
+const char *
+get_ext(const char *filename)
+{
+	const char *dot = strrchr(filename, '.');
+	if(!dot || dot == filename) return "";
+	return dot + 1;
+}
+
 int
 call_chroma(const char *filename, FILE *fp, const char *s, size_t len)
 {
+	htmlized = false;
 	// Flush HTML-file
 	fflush(fp);
-	// Copy STDOUT
-	int stdout_copy = dup(1);
-	// Redirect STDOUT
-	dup2(fileno(fp), 1);
-
-	char cmd[] = "chroma --html --html-only --html-lines --html-lines-table --filename ";
-	strcat(cmd, filename);
-
-	FILE *child = popen(cmd, "w");
-	if (child == NULL) {
-		printf("child is null: %s", strerror(errno));
-		exit(1);
-	}
-	// Give filename through STDIN:
-	fprintf(child, "%s\n", filename);
-	// Give code to highlight through STDIN:
+
+	char *html = cmark_markdown_to_html(s, len, CMARK_OPT_DEFAULT);
+	if (strcmp(get_ext(filename), "md") == 0) htmlized = true;
+
 	int lc;
-	size_t i;
-	for (i = 0; *s && i < len; s++, i++) {
-		if (*s == '\n') lc++;
-		fprintf(child, "%c", *s);
+	if (!htmlized) {
+		// Copy STDOUT
+		int stdout_copy = dup(1);
+
+		// Redirect STDOUT
+		dup2(fileno(fp), 1);
+
+		char cmd[255] = "chroma --html --html-only --html-lines --html-lines-table --filename ";
+		strncat(cmd, filename, strlen(filename) + 1);
+		FILE *child = popen(cmd, "w");
+		if (child == NULL) {
+			printf("child is null: %s", strerror(errno));
+			exit(1);
+		}
+
+		// Give code to highlight through STDIN:
+		size_t i;
+		for (i = 0; *s && i < len; s++, i++) {
+			if (*s == '\n') lc++;
+			fprintf(child, "%c", *s);
+		}
+
+		pclose(child);
+		fflush(stdout);
+
+		// Give back STDOUT.
+		dup2(stdout_copy, 1);
+
+	} else {
+		fprintf(fp, "%s", html);
 	}
 
-	pclose(child);
-	fflush(stdout);
-	// Give back STDOUT.
-	dup2(stdout_copy, 1);
 	return lc;
 }
 
@@ -803,6 +828,7 @@ writeblob(git_object *obj, const char *fpath, const char *filename, git_off_t fi
 	fputs("<p> ", fp);
 	xmlencode(fp, filename, strlen(filename));
 	fprintf(fp, " (%juB)", (uintmax_t)filesize);
+
 	fputs("</p><hr/>", fp);
 
 	if (git_blob_is_binary((git_blob *)obj)) {
@@ -812,6 +838,7 @@ writeblob(git_object *obj, const char *fpath, const char *filename, git_off_t fi
 		if (ferror(fp))
 			err(1, "fwrite");
 	}
+
 	writefooter(fp);
 	fclose(fp);
 
@@ -1209,6 +1236,12 @@ main(int argc, char *argv[])
 		submodules = ".gitmodules";
 	git_object_free(obj);
 
+	/* Generate tarball */
+	char tarball[255];
+	sprintf(tarball, "tar -zcf %s.tar.gz --ignore-failed-read --exclude='.git' %s",
+			    strippedname, repodir);
+	system(tarball);
+
 	/* log for HEAD */
 	fp = efopen("log.html", "w");
 	relpath = "";
@@ -1270,6 +1303,8 @@ main(int argc, char *argv[])
 	writefooter(fp);
 	fclose(fp);
 
+	cp("files.html", "index.html");
+
 	/* summary page with branches and tags */
 	fp = efopen("refs.html", "w");
 	writeheader(fp, "Refs");