1 files changed, 129 insertions, 0 deletions
diff --git a/src/args.c b/src/args.c
new file mode 100644
index 0000000..9544e39
--- /dev/null
+++ b/src/args.c
@@ -0,0 +1,129 @@
+#include "../inc/args.h"
+#include "../inc/commands.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+
+void parsecmdstr(cmdinput *cmd, char *cmdstr)
+{
+    switch (cmd->settype)
+    {
+    case 'i':
+        if (sscanf(cmdstr, "%i:%23[^\n]", &cmd->id, cmd->to) != 2)
+        {
+            fprintf(stderr, "ERROR: Failed to parse argument '%s' to type 'ID:RESULT'\n", cmdstr);
+            exit(EXIT_FAILURE);
+        }
+        break;
+    case 'n':
+        if (sscanf(cmdstr, "%23[^:]:%23[^\n]", cmd->from, cmd->to) != 2)
+        {
+            fprintf(stderr, "ERROR: Failed to parse argument '%s' to type 'NAME:RESULT'\n", cmdstr);
+            exit(EXIT_FAILURE);
+        }
+        break;
+    case '\0':
+        fprintf(stderr, "ERROR: Found no flag modifier\n");
+        exit(EXIT_FAILURE);
+        break;
+    default:
+        fprintf(stderr, "ERROR: Failed to parse flag with '%c' modifier\n", cmd->settype);
+        exit(EXIT_FAILURE);
+        break;
+    }
+}
+
+void parseargs(input *in, int argc, char *argv[])
+{
+    in->demopath = NULL;
+    in->mappath = NULL;
+    in->outpath = NULL;
+    in->print = 0;
+    in->extractmap = 0;
+    
+    in->numcmds = 0;
+
+    int argi;
+    for (argi = 1; argi < argc; argi++)
+    {
+        if (argv[argi][0] == '-')
+        {
+            cmdinput *cmd = &in->commands[in->numcmds];
+            switch (argv[argi][1])
+            {
+            case 'n':
+                if (argi + 1 >= argc)
+                {
+                    fprintf(stderr, "ERROR: Flag 'n' requires additional argument\n");
+                    exit(EXIT_FAILURE);
+                }
+                cmd->cmdtype = argv[argi][1];
+                cmd->settype = argv[argi][2];
+                parsecmdstr(cmd, argv[argi + 1]);
+                in->numcmds++;
+                argi++;
+                break;
+            case 's':
+                if (argi + 1 >= argc)
+                {
+                    fprintf(stderr, "ERROR: Flag 's' requires additional argument\n");
+                    exit(EXIT_FAILURE);
+                }
+                cmd->cmdtype = argv[argi][1];
+                cmd->settype = argv[argi][2];
+                parsecmdstr(cmd, argv[argi + 1]);
+                in->numcmds++;
+                argi++;
+                break;
+            case 'o':
+                if (argi + 1 >= argc)
+                {    
+                    fprintf(stderr, "ERROR: Flag 'o' requires additional argument\n");
+                    exit(EXIT_FAILURE);
+                }
+                in->outpath = argv[argi + 1];
+                argi++;
+                break;
+            case 'p':
+                in->print = 1;
+                break;
+            case 'e':
+                if (argi + 1 >= argc)
+                {    
+                    fprintf(stderr, "ERROR: Flag 'e' requires additional argument\n");
+                    exit(EXIT_FAILURE);
+                }
+                in->extractmap = argv[argi + 1];
+                argi++;
+                break;
+            case 'm':
+                if (argi + 1 >= argc)
+                {    
+                    fprintf(stderr, "ERROR: Flag 'm' requires additional argument\n");
+                    exit(EXIT_FAILURE);
+                }
+                in->mappath = argv[argi + 1];
+                argi++;
+                break;
+            default:
+                fprintf(stderr, "ERROR: Unknown flag '%c'\n", argv[argi][1]);
+                exit(EXIT_FAILURE);
+                break;
+            }    
+        }
+        else 
+        {
+            if (in->demopath)
+            {
+                fprintf(stderr, "ERROR: Found two non-flag arguments\n");
+                exit(EXIT_FAILURE);
+            }
+            in->demopath = argv[argi];
+        }
+    }
+    if (!in->demopath)
+    {
+        fprintf(stderr, "ERROR: Found no demo input\n");
+        exit(EXIT_FAILURE);
+    }
+}