Implemented GZIP functionality for the data file
[wordblah.git] / wordblox.c
index 6ad88e1..fae8b83 100644 (file)
@@ -4,11 +4,96 @@
 #include <stdbool.h>
 #include <string.h>
 #include <ctype.h>
+#include <unistd.h>
 
 #include "wordblox.h"
 #include "constantstrings.h"
 
-void set_clue_across_word (Puzzle *p)
+/* export the clues to a text file */
+void do_export_clues (Puzzle *p)
+{
+       if (p->grid_frozen == false)
+       {
+               printf (UNFROZEN_GRID);
+               char ch = getchar ();
+               return;
+       }
+       char fname[256];
+       printf (INPUT_FILE);
+       fgets (fname, 256, stdin);
+       if (strlen(fname) == 1)
+               return;
+       char *filename = strtok (fname, "\n");
+       
+       export_clues (p, filename);
+       printf (FILE_SAVED);
+       char ch = getchar ();
+}
+
+/* export the puzzle to a png file */
+void do_export_puzzle (Puzzle *p)
+{
+       if (p->grid_frozen == false)
+       {
+               printf (UNFROZEN_GRID);
+               char ch = getchar ();
+               return;
+       }
+       char fname[256];
+       printf (INPUT_FILE);
+       fgets (fname, 256, stdin);
+       if (strlen (fname) == 1)
+               return;
+       char* filename = strtok (fname, "\n");
+
+       printf (INPUT_EXPORT_ANSWERS);
+       char ans[3];
+       fgets (ans, 3, stdin);
+       bool solution;
+       solution = (toupper (ans[0]) == 'Y') ? true : false;
+               
+       export_grid_image (p, filename, solution);
+       printf (FILE_SAVED);
+       char ch = getchar ();
+}
+
+/* reset the grid */
+void do_reset_puzzle (Puzzle *p)
+{
+       int grid_size = p->grid_size;
+       printf (INPUT_CONFIRM_RESET);
+       char conf[3];
+       fgets (conf, 3, stdin);
+       if (toupper (conf[0]) == 'Y')
+               init_puzzle (p, grid_size);
+
+        print_puzzle (p);
+        char ch = getchar ();
+}
+
+/* set the password for the puzzle */
+void do_set_password (Puzzle *p)
+{
+       char* password;
+       password = getpass (INPUT_PASSWORD);
+       /* if empty reset the password to nothing */
+       if (strlen(password) == 0)
+       {
+               set_puzzle_password (p, "\0");
+               printf (PASSWORD_RESET);
+               char ch = getchar ();
+       }
+       /* set the password */
+       else 
+       {
+               set_puzzle_password (p, (const char* )password);
+               printf (PASSWORD_SET);
+               char ch = getchar ();
+       }
+}
+
+/* set clue for a word - only for frozen grid */
+void do_set_clue_word (Puzzle *p, enum ORIENTATION orient)
 {
        print_puzzle (p);
        if (p->grid_frozen == false)
@@ -25,13 +110,18 @@ void set_clue_across_word (Puzzle *p)
        fgets (clue, MAX_CLUE_LENGTH, stdin);
        char* cl = strtok (clue, "\n");
        
-       bool res = set_across_clue (p, cl, index);
+       bool res;
+       res = set_clue (p, cl, index, orient);
+
        if (res == false)
+       {
                printf (NO_WORD_INDEX);
+               char ch = getchar ();
+       }
 }
 
 /* clear a cell in the grid */
-void clear_cell (Puzzle *p)
+void do_clear_cell (Puzzle *p)
 {
        print_puzzle (p);
        if (p->grid_frozen == true) 
@@ -58,7 +148,7 @@ void clear_cell (Puzzle *p)
 }
 
 /* add a down word to the grid */
-void add_down_word (Puzzle *p)
+void do_add_down_word (Puzzle *p)
 {
        print_puzzle (p);
        if (p->grid_frozen == true) 
@@ -71,8 +161,8 @@ void add_down_word (Puzzle *p)
        int row, col;
        printf (INPUT_WORD);
        fgets (wd, MAX_PUZZLE_SIZE, stdin);
-       char *word = strtok (wd, "\n");
-       if (! is_valid_word (word))
+       char *word = is_valid_word (wd);
+       if (word == NULL)
        {
                printf (INVALID_WORD);
                char ch = getchar ();
@@ -103,7 +193,7 @@ void add_down_word (Puzzle *p)
 }
 
 /* add an across word to the grid */
-void add_across_word (Puzzle *p)
+void do_add_across_word (Puzzle *p)
 {
        print_puzzle (p);
        if (p->grid_frozen == true) 
@@ -115,14 +205,14 @@ void add_across_word (Puzzle *p)
        char wd[MAX_PUZZLE_SIZE];
        int row, col;   
        printf (INPUT_WORD);
-       fgets (wd, p->grid_size, stdin);
-       char *word = strtok (wd, "\n");
-       if (! is_valid_word (word))
+       fgets (wd, MAX_PUZZLE_SIZE, stdin);
+       char *word = is_valid_word (wd);
+       if (word == NULL)
        {
                printf (INVALID_WORD);
                char ch = getchar ();
                return;
-       }       
+       }
        printf (INPUT_ROW);
        row = get_num ();
        printf (INPUT_COL);
@@ -147,13 +237,27 @@ void add_across_word (Puzzle *p)
        print_puzzle (p);
        char ch = getchar ();
 }
+/* confirm exit */
+bool do_confirm_exit ()
+{
+       printf (INPUT_CONFIRM_EXIT);
+       char res[3];
+       fgets (res, 3, stdin);
+       if (toupper(res[0]) == 'Y')
+               return true;
+       else
+               return false;
+} 
 
+/* main loop for the puzzle editor */
 void puzzle_editor_loop (Puzzle *p, const char *filename) 
 {
        bool loop = true;
        while (loop) 
        {
-               print_menu (WHITE, RED, PUZZLE_MENU_TITLE, PUZZLE_EDIT_MENU, 9, 50);
+               char puzzle_title[60];
+               sprintf (puzzle_title, "%s - %s", PUZZLE_MENU_TITLE, filename);
+               print_menu (WHITE, BLUE, puzzle_title, PUZZLE_EDIT_MENU, 14, 50);
                printf (INPUT_CHOICE);
                int ch = get_num ();
                switch (ch)
@@ -161,11 +265,11 @@ void puzzle_editor_loop (Puzzle *p, const char *filename)
                        case 1: print_puzzle (p);
                                        char ch = getchar ();
                                        break;
-                       case 2: add_across_word (p);
+                       case 2: do_add_across_word (p);
                                        break;
-                       case 3: add_down_word (p);
+                       case 3: do_add_down_word (p);
                                        break;
-                       case 4: clear_cell (p);
+                       case 4: do_clear_cell (p);
                                        break;
                        case 5: freeze_puzzle (p);
                                        print_puzzle (p);
@@ -175,43 +279,85 @@ void puzzle_editor_loop (Puzzle *p, const char *filename)
                                        print_puzzle (p);
                                        ch = getchar ();
                                        break;
-                       case 7: set_clue_across_word (p);
-                                       print_puzzle (p);
-                                       ch = getchar ();
+                       case 7: do_set_clue_word (p, ACROSS);
                                        break;
-                       case 8: save_puzzle (p, filename);
+                       case 8: do_set_clue_word (p, DOWN);
+                                       break;
+                       case 9: save_puzzle (p, filename);
                                        printf ("%s\n",FILE_SAVED);
                                        ch = getchar ();
                                        break;
-                       case 9: loop = false;
-                                       break;
+                       case 10: do_set_password (p);
+                                        break;
+                       case 11: do_reset_puzzle (p);
+                                        break;
+                       case 12: do_export_puzzle (p);
+                                        break;
+                       case 13: do_export_clues (p);
+                                        break;
+                       case 14: loop = ! do_confirm_exit ();
+                                        break;
                }
        }
 }
 
 /* open an existing puzzle */
-void open_puzzle (
+void do_open_puzzle (const char *filename
 {
        Puzzle p;
-       printf (INPUT_FILE);
-       char fname[256];
-       fgets(fname, 256, stdin);
-       char* filename = strtok (fname, "\n");
+       /* if no filename is provided, get it from command line */
+       if (filename == NULL) 
+       {
+               printf (INPUT_FILE);
+               char fname[256];
+               fgets(fname, 256, stdin);
+               if (strlen (fname) == 1)
+                       return;
+               filename = strtok (fname, "\n");
+       }
+
        p = load_puzzle (filename);
-       puzzle_editor_loop (&p, filename);
+       
+       if (strcmp (p.hashed_password, "\0") != 0)
+       {
+               char *passwd;
+               passwd = getpass (INPUT_PASSWORD);
+               if (strlen (passwd) == 0)
+                       return;
+                                       
+               if (verify_password (&p, (const char*) passwd))
+                       puzzle_editor_loop (&p, filename);
+               else
+               {
+                       printf (WRONG_PASSWORD);
+                       char ch = getchar ();
+               } 
+       }
+       else
+               puzzle_editor_loop (&p, filename);
 }
 
 /* create a new blank puzzle */
-void new_puzzle (
+void do_new_puzzle (char *filename, int size
 {
        Puzzle p; 
-       printf (INPUT_FILE);
-       char fname[256];
-       fgets (fname, 256, stdin);
-       char* filename = strtok (fname, "\n"); 
-       printf (INPUT_GRID_SIZE);
-       int size;
-       size = get_num ();
+       /* if filename is not provided get it from command line */
+       if (filename == NULL)
+       {
+               printf (INPUT_FILE);
+               char fname[256];
+               fgets (fname, 256, stdin);
+               if (strlen (fname) == 1)
+                       return;
+               filename = strtok (fname, "\n"); 
+       }
+       /* if no size is specified get it from command line */  
+       if (size == -1)
+       {
+               printf (INPUT_GRID_SIZE);
+               size = get_num ();
+       }
+       
        if (size > MAX_PUZZLE_SIZE) 
        { 
                printf (EXCEED_MAX_GRID_SIZE);
@@ -235,9 +381,9 @@ int main_loop ()
                int ch = get_num ();
                switch (ch)
                {
-                       case 1: new_puzzle ();
+                       case 1: do_new_puzzle (NULL, -1);
                                        break;
-                       case 2: open_puzzle ();
+                       case 2: do_open_puzzle (NULL);
                                        break;
                        case 3: exit (0);
                }
@@ -245,6 +391,25 @@ int main_loop ()
 }
 
 int main (int argc, char* argv[]) 
-{      
+{
+       if (argc >= 2)
+       {
+               Puzzle p;
+               switch (argc)
+               {
+                       case 2 : do_open_puzzle (argv[1]);
+                                        break;
+                       case 4 : if (strcmp (argv[2], "new") == 0)
+                                        {
+                                               int grid_size = atoi (argv[3]);
+                                               do_new_puzzle (argv[1], grid_size);
+                                               break;
+                                        }
+                       default: fprintf (stderr, USAGE_LINE_1, argv[0]);
+                                        fprintf (stderr, USAGE_LINE_2);
+                                        fprintf (stderr, USAGE_LINE_3);
+                                        exit (3);
+               }
+       }
        return (main_loop ());
 }