(no commit message)

[ppastats.git] / src / main.c

diff --git a/src/main.c b/src/main.c
index b590c67..cc7f625 100644 (file)
--- 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);