tools/blergtool.c

   1 /* Blerg is (C) 2011 The Dominion of Awesome, and is distributed under a
   2  * BSD-style license.  Please see the COPYING file for details.
   3  */
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include "database.h"
   7 #include "tags.h"
   8
   9 void help() {
  10         printf("Usage: blergtool <store|fetch> <storename> [record]\n");
  11 }
  12
  13 int main(int argc, char *argv[]) {
  14         if (argc < 2) {
  15                 help();
  16                 exit(1);
  17         }
  18
  19         if (strncmp(argv[1], "store", 5) == 0) {
  20                 char *store = argv[2];
  21                 struct blerg *f = blerg_open(store);
  22                 if (!f) {
  23                         printf("Blerg open failed\n");
  24                         exit(1);
  25                 }
  26
  27                 size_t bytes_read = 0;
  28                 char *data = malloc(65536);
  29                 do {
  30                         bytes_read += fread(data + bytes_read, 1, 65536 - bytes_read, stdin);
  31                 } while (bytes_read < 65536 && !feof(stdin));
  32                 int record = blerg_store(f, data, bytes_read);
  33                 blerg_close(f);
  34                 free(data);
  35
  36                 if (record < 0) {
  37                         fprintf(stderr, "Could not store\n");
  38                         exit(1);
  39                 } else {
  40                         fprintf(stderr, "Stored record %d\n", record);
  41                 }
  42         } else if (strncmp(argv[1], "fetch", 5) == 0) {
  43                 char *store = argv[2];
  44                 int record = atoi(argv[3]);
  45                 struct blerg *f = blerg_open(store);
  46                 if (!f) {
  47                         printf("Blerg open failed\n");
  48                         exit(1);
  49                 }
  50
  51                 char *data;
  52                 int size;
  53                 blerg_fetch(f, record, &data, &size);
  54                 blerg_close(f);
  55                 fwrite(data, 1, size, stdout);
  56                 free(data);
  57         } else if (strncmp(argv[1], "list", 4) == 0) {
  58                 char *tag = argv[2];
  59                 int count = 50;
  60                 struct tag *list = tag_list(tag, 0, &count, -1);
  61                 if (list == NULL) {
  62                         printf("No entries");
  63                 } else {
  64                         int i;
  65                         for (i = 0; i < count; i++) {
  66                                 printf("%s %d\n", list[i].author, list[i].record);
  67                         }
  68                         free(list);
  69                 }
  70         } else {
  71                 help();
  72         }
  73
  74         return 0;
  75 }