-o for output dir option

[ppastats.git] / src / main.c

diff --git a/src/main.c b/src/main.c
index 662a87e..f7b615a 100644 (file)
--- a/src/main.c
+++ b/src/main.c
@@ -31,7 +31,9 @@
 int debug;
 static const char *program_name;
 
-static void display_published_binaries(const char *owner, const char *ppa)
+static void display_published_binaries(const char *owner,
+                                      const char *ppa,
+                                      const char *package_status)
 {
        struct ppa_stats *ppastats;
        struct package_stats **packages;
@@ -39,7 +41,7 @@ static void display_published_binaries(const char *owner, const char *ppa)
        struct distro_stats **distros;
        struct arch_stats **archs;
 
-       ppastats = create_ppa_stats(owner, ppa);
+       ppastats = create_ppa_stats(owner, ppa, package_status);
        packages = ppastats->packages;
        while (packages && *packages) {
                struct package_stats *p = *packages;
@@ -84,8 +86,9 @@ static void display_published_binaries(const char *owner, const char *ppa)
 static struct option long_options[] = {
        {"version", no_argument, 0, 'v'},
        {"help", no_argument, 0, 'h'},
-       {"html", no_argument, 0, 't'},
+       {"output_dir", required_argument, 0, 'o'},
        {"debug", no_argument, 0, 'd'},
+       {"status", required_argument, 0, 's'},
        {0, 0, 0, 0}
 };
 
@@ -119,8 +122,8 @@ static void print_help()
        puts("");
 
        puts("\
-  -t, --html       \
-generates an HTML output.");
+  -o, --output_dir=[PATH]       \
+generates HTML pages into 'PATH'.");
 
        puts("");
 
@@ -135,15 +138,18 @@ int main(int argc, char **argv)
        char *owner, *ppa;
        int optc;
        int output_html = 0;
+       char *package_status = NULL;
        int cmdok = 1;
+       char *output_dir = NULL;
 
        program_name = argv[0];
 
-       while ((optc = getopt_long(argc, argv, "vhtd", long_options,
+       while ((optc = getopt_long(argc, argv, "vho:ds:", long_options,
                                   NULL)) != -1) {
                switch (optc) {
-               case 't':
+               case 'o':
                        output_html = 1;
+                       output_dir = strdup(optarg);
                        break;
                case 'd':
                        debug = 1;
@@ -154,6 +160,10 @@ int main(int argc, char **argv)
                case 'v':
                        print_version();
                        exit(EXIT_SUCCESS);
+               case 's':
+                       if (optarg)
+                               package_status = strdup(optarg);
+                       break;
                default:
                        cmdok = 0;
                        break;
@@ -170,11 +180,13 @@ int main(int argc, char **argv)
        ppa = argv[optind+1];
 
        if (output_html)
-               ppa_to_html(owner, ppa);
+               ppa_to_html(owner, ppa, package_status, output_dir);
        else
-               display_published_binaries(owner, ppa);
+               display_published_binaries(owner, ppa, package_status);
 
        /* for valgrind.... */
+       free(package_status);
+       free(output_dir);
        lp_ws_cleanup();
        cache_cleanup();