libro

ui.c (5404B)

---

 #include <stdio.h>
 
 #include <sqlite3.h>
 
 #include "db.h"
 #include "loc.h"
 
 #define TB_IMPL
 #include "ui.h"
 
 int
 display_books(sqlite3* db, int limit, int offset, int width)
 {
   sqlite3_stmt* stmt;
   if (sqlite3_prepare_v2(db,
                          "SELECT"
                          " lcc, year, title, author"
                          " FROM books"
                          " ORDER BY lcc ASC"
                          " LIMIT ? OFFSET ?;",
                          -1,
                          &stmt,
                          0) != SQLITE_OK)
     return 1;
 
   sqlite3_bind_int(stmt, 1, limit);
   sqlite3_bind_int(stmt, 2, offset);
 
   int c1 = 1;
   int b1 = c1 + 23;
   int c2 = b1 + 2;
   int cw = (width - c2) / 2 - 3;
   int b2 = c2 + cw + 2;
   int c3 = b2 + 2;
 
   int books = 0;
  
   int res;
   for (int y = 2; y < limit + 2; y++) {
     res = sqlite3_step(stmt);
 
     if (res != SQLITE_ROW)
       break;
 
     books++;
 
     tb_print(
       c1, y, TB_DEFAULT, TB_DEFAULT, (const char*)sqlite3_column_text(stmt, 0));
     tb_set_cell(b1, y, 0x2502, TB_DEFAULT, TB_DEFAULT);
 
     const char* title = (const char*)sqlite3_column_text(stmt, 2);
     tb_printf(
       c2, y, TB_DEFAULT, TB_DEFAULT, "%.*s%s",
       cw,
       title,
       strlen(title) > cw ? "\u2026" : "");
     tb_set_cell(b2, y, 0x2502, TB_DEFAULT, TB_DEFAULT);
 
     const char* author = (const char*)sqlite3_column_text(stmt, 3);
     tb_printf(
       c3, y, TB_DEFAULT, TB_DEFAULT, "%.*s%s",
       cw,
       author,
     strlen(author) > cw ? "\u2026" : "");
   }
 
   if (res != SQLITE_DONE) {
     fprintf(stderr, "Failed to fetch books!\n");
     return -1;
   }
 
   if (books == 0) {
     tb_print(c1, 2, TB_DEFAULT, TB_DEFAULT, "No books yet!");
     tb_print(c1, 3, TB_DEFAULT, TB_DEFAULT, "Press A to search for a book by ISBN.");
   }
 
   return 0;
 }
 
 static void
 ui_repaint(struct ui_state* state)
 {
   tb_clear();
 
   size_t out;
   tb_print_ex(0, 0, TB_WHITE | TB_BOLD, TB_YELLOW, &out, " Library ");
   for (int x = out; x < state->w; x++)
     tb_set_cell(x, 0, ' ', TB_DEFAULT, TB_YELLOW);
 
   if (state->state == STATE_VIEW) {
     display_books(state->db, state->h - 2, 0, state->w);
   } else if (state->state == STATE_ADD) {
     tb_print(2, 2, TB_DEFAULT, TB_DEFAULT, "Enter ISBN (ESC to cancel)");
 
     tb_print(2, 3, TB_DEFAULT, TB_DEFAULT, state->input_buffer);
     tb_set_cell(2 + state->input_pos, 3, '_', TB_DEFAULT, TB_DEFAULT);
   }
 
   tb_present();
 }
 
 int
 ui_init(struct ui_state* state, sqlite3* db)
 {
   state->state = STATE_VIEW;
 
   state->db = db;
   state->dirty = 1;
   state->exit = 0;
 
   tb_init();
 
   state->w = tb_width();
   state->h = tb_height();
 
   ui_repaint(state);
 
   return 0;
 }
 
 int
 ui_tick(struct ui_state* state)
 {
   tb_poll_event(&state->ev);
 
   if (state->ev.type == TB_EVENT_RESIZE) {
     state->w = state->ev.w;
     state->h = state->ev.h;
     state->dirty = 1;
   } else if (state->state == STATE_VIEW) {
     if (state->ev.ch == 'q' || state->ev.key == TB_KEY_ESC) {
       state->exit = 1;
     } else if (state->ev.ch == 'a') {
       state->state = STATE_ADD;
       state->dirty = 1;
     }
   } else if (state->state == STATE_ADD) {
     if (state->ev.key == TB_KEY_ESC) {
       state->state = STATE_VIEW;
       state->dirty = 1;
     }
     /* Handle submission */
     else if (state->ev.key == TB_KEY_ENTER) {
       if (state->input_pos == 10 || state->input_pos == 13) {
         /* TODO: Lookup */
         struct lb_book book = { NULL };
         int result = lb_loc_lookup(&book, (char*)&state->input_buffer);
 
         if (result > 0) {
           tb_print(2, 5, TB_GREEN, TB_DEFAULT, "Book found!                                   ");
           tb_print(2, 6, TB_DEFAULT, TB_DEFAULT, book.classification);
           tb_print(2, 7, TB_DEFAULT, TB_DEFAULT, book.title);
           tb_print(2, 8, TB_DEFAULT, TB_DEFAULT, book.author);
 
           int result = db_add(state->db, book.classification, book.title, book.author);
 
           if (result) {
             tb_printf(2, 10, TB_RED, TB_DEFAULT, "Error %d", result);
           } else {
             tb_print(2, 10, TB_GREEN, TB_DEFAULT, "Added to library!");
           }
 
           lb_book_destroy(&book);
 
           /*state->state = STATE_VIEW;
           state->dirty = 1;*/
         }
 
         tb_present();
 
       } else {
         tb_print(2,
                  5,
                  TB_RED,
                  TB_DEFAULT,
                  "The ISBN must be exactly 10 or 13 digits long.");
         tb_present();
       }
     }
     /* Handle typing*/
     else {
       /* Valid input */
       if (state->ev.ch >= '0' && state->ev.ch <= '9' && state->input_pos < 13) {
         state->input_buffer[state->input_pos] = state->ev.ch;
         tb_set_cell(
           2 + state->input_pos, 3, state->ev.ch, TB_DEFAULT, TB_DEFAULT);
         state->input_pos++;
       }
       /* Backspace */
       else if ((state->ev.key == TB_KEY_BACKSPACE ||
                 state->ev.key == TB_KEY_BACKSPACE2) &&
                state->input_pos > 0) {
         state->input_buffer[state->input_pos] = 0;
         tb_set_cell(2 + state->input_pos, 3, ' ', TB_DEFAULT, TB_DEFAULT);
         state->input_pos--;
       }
 
       tb_set_cell(2 + state->input_pos, 3, '_', TB_DEFAULT, TB_DEFAULT);
       tb_present();
     }
   }
 
   /* Repaint */
   if (state->dirty) {
     ui_repaint(state);
     state->dirty = 0;
   }
 
   return 0;
 }
 
 int
 ui_destroy(struct ui_state* state)
 {
   tb_shutdown();
   return 0;
 }