From 7392603b7df67013d81f8d156b0ccdd8b9de34fb Mon Sep 17 00:00:00 2001
From: Aaron Traas <aaron@traas.org>
Date: Tue, 27 Jun 2017 10:33:10 -0400
Subject: [PATCH] "cheat" now has command-line arguments for generating cheat
 files

-d number of deaths. Signed integer value
-s number of saves. Signed integer value
-o file name of save game to write
---
 cheat.c        | 57 ++++++++++++++++++++++++++++++++++++++++++++------
 tests/Makefile |  2 +-
 2 files changed, 52 insertions(+), 7 deletions(-)

diff --git a/cheat.c b/cheat.c
index 844b66f..cb20eb7 100644
--- a/cheat.c
+++ b/cheat.c
@@ -1,4 +1,5 @@
 #define DEFINE_GLOBALS_FROM_INCLUDES
+#include <getopt.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <stdbool.h>
@@ -20,6 +21,48 @@ bool prompt = true;
 
 int main(int argc, char *argv[])
 {
+    int ch;
+    char *savefilename = NULL;
+    long numdie = 0;
+    long saved = 1;
+
+    /*  Options. */
+    const char* opts = "d:s:o:";
+    const char* usage = "Usage: %s [-d numdie] [-s numsaves] -o savefilename \n";
+    while ((ch = getopt(argc, argv, opts)) != EOF) {
+        switch (ch) {
+        case 'd':
+            numdie = (long)atoi(optarg);
+            break;
+        case 's':
+            saved = (long)atoi(optarg);
+            break;
+        case 'o':
+            savefilename = optarg;
+            break;
+        default:
+            fprintf(stderr,
+                    usage, argv[0]);
+            fprintf(stderr,
+                    "        -d number of deaths. Signed integer value.'\n");
+            fprintf(stderr,
+                    "        -s number of saves. Signed integer value.\n");
+            fprintf(stderr,
+                    "        -o file name of save game to write.\n");
+            exit(-1);
+            break;
+        }
+    }
+    
+    if (savefilename == NULL)
+    {
+        fprintf(stderr,
+                usage, argv[0]);
+        fprintf(stderr,
+                "ERROR: filename required\n");
+        exit(-1);
+    }
+
     FILE *fp = NULL;
 
     game.lcg_a = 1093;
@@ -36,17 +79,19 @@ int main(int argc, char *argv[])
     game.newloc = LOC_START;
     game.loc = LOC_START;
     game.limit = GAMELIMIT;
-    game.numdie = -1000;
-    game.saved = 1;
     
-    fp = fopen("cheat_numdie.adv", WRITE_MODE);
+    // apply cheats
+    game.numdie = numdie;
+    game.saved = saved;
+    
+    fp = fopen(savefilename, WRITE_MODE);
     if (fp == NULL)
     {
-        printf("Can't open file. Exiting.\n");
-        exit(0);
+        printf("Can't open file %s. Exiting.\n", savefilename);
+        exit(-1);
     }        
 
     savefile(fp);
-    printf("cheat: tests/cheat_numdie.adv created.\n");
+    printf("cheat: %s created.\n", savefilename);
     return 0;
 }
diff --git a/tests/Makefile b/tests/Makefile
index 412ec6a..cbef142 100644
--- a/tests/Makefile
+++ b/tests/Makefile
@@ -38,7 +38,7 @@ buildregress:
 	rm -f scratch.tmp
 regress:
 	$(ECHO) -n "Generate cheat file: "
-	../cheat
+	../cheat -d -1000 -o cheat_numdie.adv
 	@for file in $(TESTLOADS); do \
 	    $(ECHO) -n "  $${file} "; grep '##' $${file}.log  || echo ' ## (no description)'; \
 	    OPTS=`sed -n /#options:/s///p <$${file}.log`; \
-- 
2.31.1