X-Git-Url: http://git.wpitchoune.net/gitweb/?p=ppastats.git;a=blobdiff_plain;f=src%2Fmain.c;h=cc7f62503c343b26e28c234da8e431c12cd88f52;hp=b590c67169ac52511ddbd110d829b84a01565079;hb=677d2bfc3beae998b73f48a4be3036f09d6848ff;hpb=87f9e3fc5960cb9abfd918d7214c848bb988c91d

diff --git a/src/main.c b/src/main.c
index b590c67..cc7f625 100644
--- a/src/main.c
+++ b/src/main.c
@@ -97,6 +97,7 @@ static struct option long_options[] = {
 	{"status", required_argument, 0, 's'},
 	{"skip-js-css", no_argument, 0, 'S'},
 	{"get-bpph-size", required_argument, 0, 0},
+	{"theme-dir", required_argument, 0, 't'},
 	{0, 0, 0, 0}
 };
 
@@ -129,6 +130,7 @@ static void print_help()
 	puts("");
 
 	puts(_("  -o, --output-dir=[PATH]  generates HTML pages into 'PATH'"));
+	puts(_("  -t, --theme-dir=[PATH]  set theme dir to 'PATH'"));
 
 	puts(_(
 "  -s, --status=[STATUS]    retrieves only package of the given status\n"
@@ -149,7 +151,7 @@ static void print_help()
 
 int main(int argc, char **argv)
 {
-	char *owner, *ppa, *package_status, *output_dir;
+	char *owner, *ppa, *package_status, *output_dir, *theme_dir;
 	int optc, output_html, cmdok, install_static_files, ws_size, opti;
 
 	program_name = argv[0];
@@ -164,8 +166,9 @@ int main(int argc, char **argv)
 	package_status = NULL;
 	output_html = 0;
 	ws_size = -1;
+	theme_dir = NULL;
 
-	while ((optc = getopt_long(argc, argv, "vho:ds:S", long_options,
+	while ((optc = getopt_long(argc, argv, "vho:t:ds:S", long_options,
 				   &opti)) != -1) {
 		switch (optc) {
 		case 0:
@@ -189,6 +192,10 @@ int main(int argc, char **argv)
 			if (optarg)
 				package_status = strdup(optarg);
 			break;
+		case 't':
+			if (optarg)
+				theme_dir = strdup(optarg);
+			break;
 		case 'S':
 			install_static_files = 0;
 			break;
@@ -210,15 +217,20 @@ int main(int argc, char **argv)
 	owner = argv[optind];
 	ppa = argv[optind+1];
 
-	if (output_html)
+	if (output_html) {
+		if (theme_dir)
+			html_set_theme_dir(theme_dir);
+		else
+			html_set_theme_dir(DEFAULT_THEME_DIR);
 		ppa_to_html(owner,
 			    ppa,
 			    package_status,
 			    output_dir,
 			    install_static_files,
 			    ws_size);
-	else
+	} else {
 		display_published_binaries(owner, ppa, package_status, ws_size);
+	}
 
 	/* for valgrind.... */
 	free(package_status);