From 395b9e4658ba4e702c014f2b0f91816debdb9a77 Mon Sep 17 00:00:00 2001
From: Adrian Cochrane <adrian@openwork.nz>
Date: Sun, 5 Apr 2020 15:15:33 +1200
Subject: [PATCH] Add commandline flags.

---
 src/main.c | 73 ++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 66 insertions(+), 7 deletions(-)

diff --git a/src/main.c b/src/main.c
index 0178bd3..afc5f5f 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,6 +1,9 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <unistd.h>
+#include <fcntl.h>
+
 #include "HsFFI.h"
 
 struct session;
@@ -17,19 +20,75 @@ char *c_renderDoc(struct session*, struct page*);
 char **c_extractLinks(struct page*);
 char **c_docLinksAndRendering(struct session*, struct page*); // FIXME segfaults.
 
+void write_links(FILE *dest, char **links) {
+    for (int i = 0; strcmp(links[i], " ") != 0; i++) {
+        fprintf(dest, "%s%c", links[i], (i % 3) == 2 ? '\n' : '\t');
+    }
+}
+
 int main(int argc, char **argv) {
     hs_init(&argc, &argv);
-    struct session *session = c_newSession();
 
+    char *mimes = "text/html text/xml application/xml application/xhtml+xml text/plain";
+    FILE *fd_ssml = NULL;
+    FILE *fd_links = NULL;
+
+    int c;
+    opterr = 0;
+    while ((c = getopt(argc, argv, "xs::l::h")) != -1) {
+        switch (c) {
+        case 'x':
+            mimes = "text/xml application/xml application/xhtml+xml text/html text/plain";
+            break;
+        case 's':
+            fd_ssml = optarg != NULL ? fopen(optarg, "w") : stdout;
+            if (fd_ssml == NULL) {
+                fprintf(stderr, "Failed to open file %s\n", optarg);
+                hs_exit();
+                return -1;
+            }
+            break;
+        case 'l':
+            fd_links = optarg != NULL ? fopen(optarg, "w") : stdout;
+            if (fd_links == NULL) {
+                fprintf(stderr, "Failed to open file %s\n", optarg);
+                hs_exit();
+                return -1;
+            }
+            break;
+        case '?':
+            fprintf(stderr, "Invalid flag %c\n\n", optopt);
+        case 'h':
+            fprintf(stderr, "USAGE: rhapsode [FLAGS] URL...\n");
+            fprintf(stderr, "\t-x\tX(HT)ML\tIndicates to expect an X(HT)ML file.\n");
+            fprintf(stderr, "\t-s\tsilent/SSML\tWrites SSML to the specified file or stdout.\n");
+            fprintf(stderr, "\t\t\thttps://xkcd.com/1692/\n");
+            fprintf(stderr, "\t-l\tlinks\tWrite extracted links to specifed file or stdout as TSV.\n");
+            fprintf(stderr, "\t-h\thelp\tOutputs this usage information to stderr.\n");
+            fprintf(stderr, "\t\t\tIf both -s & -l are enabled without an argument, writes to stderr instead.\n");
+
+            hs_exit();
+            return c == 'h' ? 0 : -1;
+        }
+    }
+    if (fd_ssml == stdout && fd_links == stdout) fd_links = stderr;
+    if (fd_ssml == NULL && fd_links == NULL) fd_ssml = stdout;
+
+    struct session *session = c_newSession();
     struct page *referer = c_initialReferer();
-    struct page *page = c_fetchURL(session, "text/xml text/html text/plain", referer, argv[1]);
-    char *ssml = c_renderDoc(session, page);
-    char **links = c_extractLinks(page);
 
-    printf("%s\n", ssml);
-    for (int i = 0; strcmp(links[i], " ") != 0; i++) fprintf(stderr, "%s\n", links[i]);
+    for (int i = optind; i < argc; i++) {
+        printf("%s\n", argv[i]);
+        struct page *page = c_fetchURL(session, mimes, referer, argv[i]);
+        char *ssml = c_renderDoc(session, page);
+        char **links = c_extractLinks(page);
+
+        if (fd_ssml != NULL) fprintf(fd_ssml, "%s\n", ssml);
+        if (fd_links != NULL) write_links(fd_links, links);
+
+        c_freePage(page);
+    }
 
-    c_freePage(page);
     c_freePage(referer);
     c_freeSession(session);
     hs_exit();
-- 
2.30.2