More TAPification.

[open-adventure.git] / cheat.c

diff --git a/cheat.c b/cheat.c
index 0eee14bfb2d44b29be7d0d3aad4639cbe538cf2a..ef9b014087c1f1ca275e35e9a0ca84c047d35538 100644 (file)
--- a/cheat.c
+++ b/cheat.c
@@ -1,7 +1,7 @@
 /*
  * 'cheat' is a tool for generating save game files to test states that ought
  * not happen. It leverages chunks of advent, mostly initialize() and
- * savefile(), so we know we're always outputing save files that advent
+ * savefile(), so we know we're always outputting save files that advent
  * can import.
  *
  * Copyright (c) 1977, 2005 by Will Crowther and Don Woods
@@ -12,6 +12,7 @@
 #include <stdlib.h>
 #include <stdio.h>
 #include <stdbool.h>
+#include <editline/readline.h>
 #include "advent.h"
 
 int main(int argc, char *argv[])
@@ -20,6 +21,7 @@ int main(int argc, char *argv[])
     char *savefilename = NULL;
     int version = 0;
     FILE *fp = NULL;
+    char *tapmessage = NULL;
 
     // Initialize game variables
     initialise();
@@ -30,14 +32,15 @@ int main(int argc, char *argv[])
     game.saved = 1;
 
     /*  Options. */
-    const char* opts = "d:l:s:t:v:o:";
+    const char* opts = "d:l:s:t:v:o:m:";
     const char* usage = "Usage: %s [-d numdie] [-s numsaves] [-v version] -o savefilename \n"
                         "        -d number of deaths. Signed integer.\n"
                         "        -l lifetime of lamp in turns. Signed integer.\n"
                         "        -s number of saves. Signed integer.\n"
                         "        -t number of turns. Signed integer.\n"
                         "        -v version number of save format.\n"
-                        "        -o required. File name of save game to write.\n";
+                        "        -o required. File name of save game to write.\n"
+                       "        -m specify message for YAP output\n";
 
     while ((ch = getopt(argc, argv, opts)) != EOF) {
         switch (ch) {
@@ -64,6 +67,9 @@ int main(int argc, char *argv[])
         case 'o':
             savefilename = optarg;
             break;
+        case 'm':
+            tapmessage = optarg;
+            break;
         default:
             fprintf(stderr,
                     usage, argv[0]);
@@ -74,25 +80,48 @@ int main(int argc, char *argv[])
 
     // Save filename required; the point of cheat is to generate save file
     if (savefilename == NULL) {
-        fprintf(stderr,
-                usage, argv[0]);
-        fprintf(stderr,
-                "ERROR: filename required\n");
-        exit(EXIT_FAILURE);
+       if (tapmessage != NULL) {
+           printf("not ok - %s: filename required\n", tapmessage);
+       } else {
+           printf("not ok - filename required\n");
+           exit(EXIT_FAILURE);
+       }
     }
 
     fp = fopen(savefilename, WRITE_MODE);
     if (fp == NULL) {
-        fprintf(stderr,
-                "Can't open file %s. Exiting.\n", savefilename);
-        exit(EXIT_FAILURE);
+       if (tapmessage != NULL) {
+           printf("not ok - %s: can't open file %s. Exiting.\n", tapmessage, savefilename);
+       } else {
+           printf("not ok - can't open file %s.\n", savefilename);
+           exit(EXIT_FAILURE);
+       }
     }
 
     savefile(fp, version);
 
     fclose(fp);
 
-    printf("cheat: %s created.\n", savefilename);
-
+    if (tapmessage != NULL) {
+       printf("ok - %s\n", tapmessage);
+    } else {
+       printf("cheat: %s created.\n", savefilename);
+    }
     return EXIT_SUCCESS;
 }
+
+// LCOV_EXCL_START
+/*
+ * Ugh...unused, but required for linkage.
+ * See the actually useful version of this in main.c
+ */
+
+char *myreadline(const char *prompt)
+{
+    return readline(prompt);
+}
+// LCOV_EXCL_STOP
+
+/* end */
+
+