From ffc8e99f73fc1e2870d133623688d0d0a53995e2 Mon Sep 17 00:00:00 2001
From: Stephen Seo <seo.disparate@gmail.com>
Date: Sun, 22 Sep 2024 14:42:41 +0900
Subject: [PATCH] Add new arg for program: --enable-cache-dir=<DIR>

---
 src/arg_parse.c | 39 +++++++++++++++++++++++++++++++++++++++
 src/arg_parse.h |  3 +++
 2 files changed, 42 insertions(+)

diff --git a/src/arg_parse.c b/src/arg_parse.c
index 7250e36..ce626e0 100644
--- a/src/arg_parse.c
+++ b/src/arg_parse.c
@@ -21,6 +21,14 @@
 #include <stdlib.h>
 #include <stdio.h>
 
+// libc includes.
+#include <sys/types.h>
+#include <dirent.h>
+#include <errno.h>
+
+// Posix includes.
+#include <sys/stat.h>
+
 void print_usage(void) {
   puts("Usage:");
   puts("  -p <port> | --port <port>");
@@ -30,6 +38,7 @@ void print_usage(void) {
   puts("    For example: --req-header-to-print=User-Agent");
   puts("    Note that this option is case-insensitive");
   puts("  --enable-reload-config-on-change");
+  puts("  --enable-cache-dir=<DIR>");
 }
 
 Args parse_args(int32_t argc, char **argv) {
@@ -69,6 +78,36 @@ Args parse_args(int32_t argc, char **argv) {
       }
     } else if (strcmp(argv[0], "--enable-reload-config-on-change") == 0) {
       args.flags |= 2;
+    } else if (strncmp(argv[0], "--enable-cache-dir=", 19) == 0) {
+      args.cache_dir = argv[0] + 19;
+      // Check if it actually is an existing directory.
+      DIR *d = opendir(args.cache_dir);
+      if (d == NULL) {
+        if (errno == ENOENT) {
+          printf(
+            "Directory \"%s\" doesn't exist, creating it...\n",
+            args.cache_dir);
+          int ret = mkdir(
+              args.cache_dir,
+              S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
+          if (ret == -1) {
+            fprintf(
+              stderr,
+              "ERROR Failed to create new directory (errno %d)\n",
+              errno);
+            exit(1);
+          }
+        } else {
+          fprintf(
+            stderr,
+            "ERROR Failed to open directory \"%s\" (errno %d)!\n",
+            args.cache_dir,
+            errno);
+          exit(1);
+        }
+      } else {
+        printf("Directory \"%s\" exists.\n", args.cache_dir);
+      }
     } else {
       puts("ERROR: Invalid args!\n");
       print_usage();
diff --git a/src/arg_parse.h b/src/arg_parse.h
index 8f56fcd..c14ee90 100644
--- a/src/arg_parse.h
+++ b/src/arg_parse.h
@@ -34,6 +34,9 @@ typedef struct Args {
   const char *config_file;
   // Needs to be free'd.
   SDArchiverLinkedList *list_of_headers_to_log;
+  // Non-NULL if cache-dir is specified and cache is to be used.
+  // Does not need to be free'd since it points to a string in argv.
+  const char *cache_dir;
 } Args;
 
 void print_usage(void);