Small extra function.

[gedcom-parse.git] / t / src / gom_write.c

diff --git a/t/src/gom_write.c b/t/src/gom_write.c
index 388432887ed9b449e05dcb47f1ced029d771e0c7..cb28844aa81f08b5d96dda6f885c992a27015f6d 100644 (file)
--- a/t/src/gom_write.c
+++ b/t/src/gom_write.c
@@ -51,6 +51,7 @@ void show_help ()
   printf("  -h    Show this help text\n");
   printf("  -q    No output to standard output\n");
   printf("  -o <outfile>  File to generate errors to (def. testgedcom.out)\n");
+  printf("  -i <gedfile>  File to read gedcom from (default: new file)\n");
   printf("  -w <gedfile>  File to write gedcom to (def. %s)\n", WRITE_GEDCOM);
   printf("  -e <encoding> Encoding (UNICODE, ASCII, ANSEL, ...: see gedcom.enc)\n");
   printf("  -u <unicode_enc> Encoding details for Unicode\n");
@@ -59,20 +60,30 @@ void show_help ()
   printf("        <terminator> can be CR, LF, CR_LF, LF_CR\n");
 }
 
-int update_charset(char* encoding)
+int update_header(char* encoding)
 {
   struct header* head = NULL;
   char* value;
+  char* long_note = "This note is for testing the continuation stuff\n"
+    "Some Specials: This line is very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very very long but too long (255 caharcters is the limit), so this is going over the border\n"
+    "And now we have an at character: @, which should be doubled";
 
   head = gom_get_header();
   if (head == NULL)
     return 1;
   else {
+    /*
     value = gom_set_string(&head->charset.name, encoding);
     if (value == NULL || strcmp(value, encoding))
       return 1;
     else
       return 0;
+    */
+    value = gom_set_string(&head->note, long_note);
+    if (value == NULL || strcmp(value, long_note))
+      return 1;
+    else
+      return 0;
   }
 }
 
@@ -81,6 +92,7 @@ int main(int argc, char* argv[])
   int result;
   int total_conv_fails = 0;
   char* outfilename = NULL;
+  char* infilename  = NULL;
   char* gedfilename = WRITE_GEDCOM;
   char* encoding    = "ASCII";
   Encoding enc      = ONE_BYTE;
@@ -119,6 +131,17 @@ int main(int argc, char* argv[])
          exit(1);
        }
       }
+      else if (!strncmp(argv[i], "-i", 3)) {
+       i++;
+       if (i < argc) {
+         infilename = argv[i];
+       }
+       else {
+         printf ("Missing input file name\n");
+         show_help();
+         exit(1);
+       }
+      }
       else if (!strncmp(argv[i], "-e", 3)) {
        i++;
        if (i < argc) {
@@ -191,12 +214,15 @@ int main(int argc, char* argv[])
   gedcom_write_set_line_terminator(end);
 
   output_open(outfilename);
-  
-  result = gom_new_model();
-  /*
-  if (result == 0)
-    result |= update_charset(encoding);
-  */
+
+  if (infilename) {
+    result = gom_parse_file(infilename);
+  }
+  else {
+    result = gom_new_model();
+    if (result == 0)
+      result |= update_header(encoding);
+  }
   if (result == 0)
     result |= gom_write_file(gedfilename, &total_conv_fails);
   if (result == 0 && total_conv_fails == 0) {