Gebruik option parsing in kat en tak

1 files changed, 54 insertions, 21 deletions

diff --git a/src/kat.c b/src/kat.c
index 7efa7e1..399e466 100644
--- a/src/kat.c
+++ b/src/kat.c
@@ -1,35 +1,68 @@
-#include <fcntl.h>
-#include <stdio.h>
 #include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
-#include <sys/mman.h>
-#include <sys/stat.h>
 #include <unistd.h>
 
 #include "util/map.h"
+#include "util/versie.h"
 
-int entry_kat(int argc, char **argv) {
-  if (argc == 2 && strcmp(argv[1], "-h") == 0) {
-    fprintf(stderr, "Gebruik: kat [BESTAND]...\n"
-                    "\n"
-                    "Schakel BESTAND(en) aaneen naar standaard uitvoer.\n");
-    return 0;
-  }
+static void usage(FILE *f) {
+  fprintf(f,
+      "Gebruik: kat [-hV] <bestand...>\n"
+      "\n"
+      "Schakel bestanden aaneen naar standaard uitvoer.\n"
+      "\n"
+      "  -h    Toon deze hulptekst\n"
+      "  -V    Toon versienummer\n");
+}
 
-  for (int i = 1; i < argc; i++) {
-    const char *fname = argv[i];
-    struct map *map = open_map(fname);
-    if (map == NULL) {
-      return 1;
+// Returns pointer to argument array containing the file names
+static char** parse_options(int argc, char **argv) {
+  int opt;
+  while ((opt = getopt(argc, argv, "hV")) != -1) {
+    switch (opt) {
+      case 'h':
+        usage(stdout);
+        exit(0);
+
+      case 'V':
+        drukkedoos_print_versie(stdout, "kat");
+        exit(0);
+
+      case '?':
+        usage(stderr);
+        exit(1);
     }
+  }
+
+  return argv + optind;
+}
+
+static void process(const char *fname) {
+  struct map *map = open_map(fname);
+  if (map == NULL) {
+    exit(1);
+  }
 
-    if (map->sb.st_size >= 0) {
-      for (size_t i = 0; i < (size_t)map->sb.st_size; i++) {
-        putchar(map->addr[i]);
-      }
+  if (map->sb.st_size >= 0) {
+    for (size_t i = 0; i < (size_t)map->sb.st_size; i++) {
+      putchar(map->addr[i]);
     }
+  }
 
-    close_map(map);
+  close_map(map);
+}
+
+int entry_kat(int argc, char **argv) {
+  char **args = parse_options(argc, argv);
+  if (*args == NULL) {
+    fprintf(stderr, "kat: Standaard invoer niet ondersteund in deze projectie-gebaseerde implementatie\n");
+    return 1;
+  } else {
+    while (*args != NULL) {
+      process(*args);
+      args++;
+    }
   }
 
   return 0;