/*************************************************************************************************
 * Utility for debugging Villa and its applications
 *                                                      Copyright (C) 2000-2003 Mikio Hirabayashi
 * This file is part of QDBM, Quick Database Manager.
 * QDBM is free software; you can redistribute it and/or modify it under the terms of the GNU
 * Lesser General Public License as published by the Free Software Foundation; either version
 * 2.1 of the License or any later version.  QDBM is distributed in the hope that it will be
 * useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 * details.
 * You should have received a copy of the GNU Lesser General Public License along with QDBM; if
 * not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
 * 02111-1307 USA.
 *************************************************************************************************/


#include <depot.h>
#include <cabin.h>
#include <villa.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#undef TRUE
#define TRUE           1                 /* boolean true */
#undef FALSE
#define FALSE          0                 /* boolean false */


/* for RISC OS */
#if defined(__riscos__) || defined(__riscos)
#include <unixlib/local.h>
int __riscosify_control = __RISCOSIFY_NO_PROCESS;
#endif


/* global variables */
const char *progname;                    /* program name */


/* function prototypes */
int main(int argc, char **argv);
void usage(void);
int isbigendian(void);
char *mystrdup(const char *str);
char *hextoobj(const char *str, int *sp);
char *dectoiobj(const char *str, int *sp);
char *readfile(const char *name, int *sp);
int runcreate(int argc, char **argv);
int runput(int argc, char **argv);
int runout(int argc, char **argv);
int runget(int argc, char **argv);
int runlist(int argc, char **argv);
int runoptimize(int argc, char **argv);
int runinform(int argc, char **argv);
int runremove(int argc, char **argv);
int runeconv(int argc, char **argv);
void pdperror(const char *name);
void printobj(const char *obj, int size);
void printobjhex(const char *obj, int size);
int docreate(const char *name);
int doput(const char *name, const char *kbuf, int ksiz, const char *vbuf, int vsiz,
          int dmode, VLCFUNC cmp);
int doout(const char *name, const char *kbuf, int ksiz, VLCFUNC cmp, int lb);
int doget(const char *name, const char *kbuf, int ksiz, VLCFUNC cmp, int lb, int ox, int nb);
int dolist(const char *name, const char *tbuf, int tsiz, const char *bbuf, int bsiz,
             VLCFUNC cmp, int ki, int ox, int gt, int lt, int max, int desc);
int dooptimize(const char *name);
int doinform(const char *name);
int doremove(const char *name);
int doeconv(const char *name, int mode);


/* main routine */
int main(int argc, char **argv){
  int rv;
  progname = argv[0];
  if(argc < 2) usage();
  rv = 0;
  if(!strcmp(argv[1], "create")){
    rv = runcreate(argc, argv);
  } else if(!strcmp(argv[1], "put")){
    rv = runput(argc, argv);
  } else if(!strcmp(argv[1], "out")){
    rv = runout(argc, argv);
  } else if(!strcmp(argv[1], "get")){
    rv = runget(argc, argv);
  } else if(!strcmp(argv[1], "list")){
    rv = runlist(argc, argv);
  } else if(!strcmp(argv[1], "optimize")){
    rv = runoptimize(argc, argv);
  } else if(!strcmp(argv[1], "inform")){
    rv = runinform(argc, argv);
  } else if(!strcmp(argv[1], "remove")){
    rv = runremove(argc, argv);
  } else if(!strcmp(argv[1], "econv")){
    rv = runeconv(argc, argv);
  } else if(!strcmp(argv[1], "version") || !strcmp(argv[1], "--version")){
    printf("Powered by QDBM version %s, %s endian.\n",
           dpversion, isbigendian() ? "big" : "little");
    printf("Copyright (c) 2000-2003 Mikio Hirabayashi\n");
    rv = 0;
  } else {
    usage();
  }
  return rv;
}


/* print the usage and exit */
void usage(void){
  fprintf(stderr, "%s: administration utility for Villa\n", progname);
  fprintf(stderr, "\n");
  fprintf(stderr, "usage:\n");
  fprintf(stderr, "  %s create name\n", progname);
  fprintf(stderr, "  %s put [-kx|-ki] [-vx|-vi|-vf] [-keep|-dup] name key val\n", progname);
  fprintf(stderr, "  %s out [-l] [-kx|-ki] name key\n", progname);
  fprintf(stderr, "  %s get [-l] [-kx|-ki] [-ox] [-n] name key\n", progname);
  fprintf(stderr, "  %s list [-kx|-ki] [-ox] [-top key] [-bot key] [-gt] [-lt] [-max num] "
          "[-desc] name\n", progname);
  fprintf(stderr, "  %s optimize name\n", progname);
  fprintf(stderr, "  %s inform name\n", progname);
  fprintf(stderr, "  %s remove name\n", progname);
  fprintf(stderr, "  %s econv [-be|-le] name\n", progname);
  fprintf(stderr, "  %s version\n", progname);
  exit(1);
}


/* check whether the byte order of the platform is big endian or not */
int isbigendian(void){
  char buf[sizeof(int)];
  *(int *)buf = 1;
  return !buf[0];
}


/* duplicate a string */
char *mystrdup(const char *str){
  int len;
  char *buf;
  len = strlen(str);
  if(!(buf = malloc(len + 1))) return NULL;
  memcpy(buf, str, len + 1);
  return buf;
}


/* create a binary object from a hexadecimal string */
char *hextoobj(const char *str, int *sp){
  char *buf, mbuf[3];
  int len, i, j;
  len = strlen(str);
  if(!(buf = malloc(len + 1))) return NULL;
  j = 0;
  for(i = 0; i < len; i += 2){
    while(strchr(" \n\r\t\f\v", str[i])){
      i++;
    }
    if((mbuf[0] = str[i]) == '\0') break;
    if((mbuf[1] = str[i+1]) == '\0') break;
    mbuf[2] = '\0';
    buf[j++] = strtol(mbuf, NULL, 16);
  }
  buf[j] = '\0';
  *sp = j;
  return buf;
}


/* create a integer object from a decimal string */
char *dectoiobj(const char *str, int *sp){
  char *buf;
  int num;
  num = atoi(str);
  if(!(buf = malloc(sizeof(int)))) return NULL;
  *(int *)buf = num;
  *sp = sizeof(int);
  return buf;
}


/* read data from file */
char *readfile(const char *name, int *sp){
  FILE *IN;
  int i, c;
  char *buf;
  if(!(IN = fopen(name, "rb"))) return NULL;
  buf = NULL;
  for(i = 0; (c = fgetc(IN)) != EOF; i++){
    if(!(buf = realloc(buf, i + 1))){
      fclose(IN);
      return NULL;
    }
    buf[i] = c;
  }
  fclose(IN);
  *sp = i;
  return buf;
}


/* parse arguments of create command */
int runcreate(int argc, char **argv){
  char *name;
  int i, rv;
  name = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      usage();
    } else if(!name){
      name = argv[i];
    } else {
      usage();
    }
  }
  if(!name) usage();
  rv = docreate(name);
  return rv;
}


/* parse arguments of put command */
int runput(int argc, char **argv){
  char *name, *key, *val, *kbuf, *vbuf;
  int i, kx, ki, vx, vi, vf, ksiz, vsiz, rv;
  int dmode;
  name = NULL;
  kx = FALSE;
  ki = FALSE;
  vx = FALSE;
  vi = FALSE;
  vf = FALSE;
  key = NULL;
  val = NULL;
  dmode = VL_DOVER;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      if(!strcmp(argv[i], "-kx")){
        kx = TRUE;
      } else if(!strcmp(argv[i], "-ki")){
        ki = TRUE;
      } else if(!strcmp(argv[i], "-vx")){
        vx = TRUE;
      } else if(!strcmp(argv[i], "-vi")){
        vi = TRUE;
      } else if(!strcmp(argv[i], "-vf")){
        vf = TRUE;
      } else if(!strcmp(argv[i], "-keep")){
        dmode = VL_DKEEP;
      } else if(!strcmp(argv[i], "-dup")){
        dmode = VL_DDUP;
      } else {
        usage();
      }
    } else if(!name){
      name = argv[i];
    } else if(!key){
      key = argv[i];
    } else if(!val){
      val = argv[i];
    } else {
      usage();
    }
  }
  if(!name || !key || !val) usage();
  if(kx){
    kbuf = hextoobj(key, &ksiz);
  } else if(ki){
    kbuf = dectoiobj(key, &ksiz);
  } else {
    kbuf = mystrdup(key);
    ksiz = -1;
  }
  if(vx){
    vbuf = hextoobj(val, &vsiz);
  } else if(vi){
    vbuf = dectoiobj(val, &vsiz);
  } else if(vf){
    vbuf = readfile(val, &vsiz);
  } else {
    vbuf = mystrdup(val);
    vsiz = -1;
  }
  if(kbuf && vbuf){
    rv = doput(name, kbuf, ksiz, vbuf, vsiz, dmode, ki ? VL_CMPINT : VL_CMPLEX);
  } else {
    if(vf){
      fprintf(stderr, "%s: %s: cannot read\n", progname, val);
    } else {
      fprintf(stderr, "%s: out of memory\n", progname);
    }
    rv = 1;
  }
  free(kbuf);
  free(vbuf);
  return rv;
}


/* parse arguments of out command */
int runout(int argc, char **argv){
  char *name, *key, *kbuf;
  int i, kx, ki, lb, ksiz, rv;
  name = NULL;
  kx = FALSE;
  ki = FALSE;
  lb = FALSE;
  key = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      if(!strcmp(argv[i], "-l")){
        lb = TRUE;
      } else if(!strcmp(argv[i], "-kx")){
        kx = TRUE;
      } else if(!strcmp(argv[i], "-ki")){
        ki = TRUE;
      } else {
        usage();
      }
    } else if(!name){
      name = argv[i];
    } else if(!key){
      key = argv[i];
    } else {
      usage();
    }
  }
  if(!name || !key) usage();
  if(kx){
    kbuf = hextoobj(key, &ksiz);
  } else if(ki){
    kbuf = dectoiobj(key, &ksiz);
  } else {
    kbuf = mystrdup(key);
    ksiz = -1;
  }
  if(kbuf){
    rv = doout(name, kbuf, ksiz, ki ? VL_CMPINT : VL_CMPLEX, lb);
  } else {
    fprintf(stderr, "%s: out of memory\n", progname);
    rv = 1;
  }
  free(kbuf);
  return rv;
}


/* parse arguments of get command */
int runget(int argc, char **argv){
  char *name, *key, *kbuf;
  int i, lb, kx, ki, ox, nb, ksiz, rv;
  name = NULL;
  lb = FALSE;
  kx = FALSE;
  ki = FALSE;
  ox = FALSE;
  nb = FALSE;
  key = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      if(!strcmp(argv[i], "-l")){
        lb = TRUE;
      } else if(!strcmp(argv[i], "-kx")){
        kx = TRUE;
      } else if(!strcmp(argv[i], "-ki")){
        ki = TRUE;
      } else if(!strcmp(argv[i], "-ox")){
        ox = TRUE;
      } else if(!strcmp(argv[i], "-n")){
        nb = TRUE;
      } else {
        usage();
      }
    } else if(!name){
      name = argv[i];
    } else if(!key){
      key = argv[i];
    } else {
      usage();
    }
  }
  if(!name || !key) usage();
  if(kx){
    kbuf = hextoobj(key, &ksiz);
  } else if(ki){
    kbuf = dectoiobj(key, &ksiz);
  } else {
    kbuf = mystrdup(key);
    ksiz = -1;
  }
  if(kbuf){
    rv = doget(name, kbuf, ksiz, ki ? VL_CMPINT : VL_CMPLEX, lb, ox, nb);
  } else {
    fprintf(stderr, "%s: out of memory\n", progname);
    rv = 1;
  }
  free(kbuf);
  return rv;
}


/* parse arguments of list command */
int runlist(int argc, char **argv){
  char *name, *top, *bot, *tbuf, *bbuf, *nstr;
  int i, kx, ki, ox, gt, lt, max, desc, tsiz, bsiz, rv;
  name = NULL;
  kx = FALSE;
  ki = FALSE;
  ox = FALSE;
  gt = FALSE;
  lt = FALSE;
  max = -1;
  desc = FALSE;
  top = NULL;
  bot = NULL;
  nstr = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      if(!strcmp(argv[i], "-kx")){
        kx = TRUE;
      } else if(!strcmp(argv[i], "-ki")){
        ki = TRUE;
      } else if(!strcmp(argv[i], "-ox")){
        ox = TRUE;
      } else if(!strcmp(argv[i], "-top")){
        if(++i >= argc) usage();
        top = argv[i];
      } else if(!strcmp(argv[i], "-bot")){
        if(++i >= argc) usage();
        bot = argv[i];
      } else if(!strcmp(argv[i], "-gt")){
        gt = TRUE;
      } else if(!strcmp(argv[i], "-lt")){
        lt = TRUE;
      } else if(!strcmp(argv[i], "-max")){
        if(++i >= argc) usage();
        max = atoi(argv[i]);
      } else if(!strcmp(argv[i], "-desc")){
        desc = TRUE;
      } else {
        usage();
      }
    } else if(!name){
      name = argv[i];
    } else {
      usage();
    }
  }
  if(!name) usage();
  tbuf = NULL;
  bbuf = NULL;
  if(kx){
    if(top) tbuf = hextoobj(top, &tsiz);
    if(bot) bbuf = hextoobj(bot, &bsiz);
  } else if(ki){
    if(top) tbuf = dectoiobj(top, &tsiz);
    if(bot) bbuf = dectoiobj(bot, &bsiz);
  } else {
    if(top){
      tbuf = mystrdup(top);
      tsiz = strlen(tbuf);
    }
    if(bot){
      bbuf = mystrdup(bot);
      bsiz = strlen(bbuf);
    }
  }
  rv = dolist(name, tbuf, tsiz, bbuf, bsiz, ki ? VL_CMPINT : VL_CMPLEX, ki, ox, gt, lt, max, desc);
  free(tbuf);
  free(bbuf);
  return rv;
}


/* parse arguments of optimize command */
int runoptimize(int argc, char **argv){
  char *name;
  int i, rv;
  name = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      usage();
    } else if(!name){
      name = argv[i];
    } else {
      usage();
    }
  }
  if(!name) usage();
  rv = dooptimize(name);
  return rv;
}


/* parse arguments of inform command */
int runinform(int argc, char **argv){
  char *name;
  int i, rv;
  name = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      usage();
    } else if(!name){
      name = argv[i];
    } else {
      usage();
    }
  }
  if(!name) usage();
  rv = doinform(name);
  return rv;
}


/* parse arguments of remove command */
int runremove(int argc, char **argv){
  char *name;
  int i, rv;
  name = NULL;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      usage();
    } else if(!name){
      name = argv[i];
    } else {
      usage();
    }
  }
  if(!name) usage();
  rv = doremove(name);
  return rv;
}


/* parse arguments of econv command */
int runeconv(int argc, char **argv){
  char *name;
  int i, be, le, rv;
  name = NULL;
  be = FALSE;
  le = FALSE;
  for(i = 2; i < argc; i++){
    if(!name && argv[i][0] == '-'){
      if(!strcmp(argv[i], "-be")){
        be = TRUE;
      } else if(!strcmp(argv[i], "-le")){
        le = TRUE;
      } else {
        usage();
      }
    } else if(!name){
      name = argv[i];
    } else {
      usage();
    }
  }
  if(!name || (be && le)) usage();
  rv = doeconv(name, be ? 1 : le ? -1 : 0);
  return rv;
}


/* print an error message */
void pdperror(const char *name){
  fprintf(stderr, "%s: %s: %s\n", progname, name, dperrmsg(dpecode));
}


/* print an object */
void printobj(const char *obj, int size){
  int i;
  for(i = 0; i < size; i++){
    putchar(obj[i]);
  }
}


/* print an object as a hexadecimal string */
void printobjhex(const char *obj, int size){
  int i;
  for(i = 0; i < size; i++){
    printf("%s%02X", i > 0 ? " " : "", ((const unsigned char *)obj)[i]);
  }
}


/* perform create command */
int docreate(const char *name){
  VILLA *villa;
  if(!(villa = vlopen(name, VL_OWRITER | VL_OCREAT | VL_OTRUNC, VL_CMPLEX))){
    pdperror(name);
    return 1;
  }
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform put command */
int doput(const char *name, const char *kbuf, int ksiz, const char *vbuf, int vsiz,
          int dmode, VLCFUNC cmp){
  VILLA *villa;
  if(!(villa = vlopen(name, VL_OWRITER, cmp))){
    pdperror(name);
    return 1;
  }
  if(!vlput(villa, kbuf, ksiz, vbuf, vsiz, dmode)){
    pdperror(name);
    vlclose(villa);
    return 1;
  }
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform out command */
int doout(const char *name, const char *kbuf, int ksiz, VLCFUNC cmp, int lb){
  VILLA *villa;
  if(!(villa = vlopen(name, VL_OWRITER, cmp))){
    pdperror(name);
    return 1;
  }
  if(lb){
    if(!vloutlist(villa, kbuf, ksiz)){
      pdperror(name);
      vlclose(villa);
      return 1;
    }
  } else {
    if(!vlout(villa, kbuf, ksiz)){
      pdperror(name);
      vlclose(villa);
      return 1;
    }
  }
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform get command */
int doget(const char *name, const char *kbuf, int ksiz, VLCFUNC cmp, int lb, int ox, int nb){
  VILLA *villa;
  CBLIST *vals;
  char *vbuf;
  int vsiz;
  if(!(villa = vlopen(name, VL_OREADER, cmp))){
    pdperror(name);
    return 1;
  }
  if(lb){
    if(!(vals = vlgetlist(villa, kbuf, ksiz))){
      pdperror(name);
      vlclose(villa);
      return 1;
    }
    while((vbuf = cblistshift(vals, &vsiz)) != NULL){
      if(ox){
        printobjhex(vbuf, vsiz);
      } else {
        printobj(vbuf, vsiz);
      }
      free(vbuf);
      putchar('\n');
    }
    cblistclose(vals);
  } else {
    if(!(vbuf = vlget(villa, kbuf, ksiz, &vsiz))){
      pdperror(name);
      vlclose(villa);
      return 1;
    }
    if(ox){
      printobjhex(vbuf, vsiz);
    } else {
      printobj(vbuf, vsiz);
    }
    free(vbuf);
    if(!nb) putchar('\n');
  }
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform list command */
int dolist(const char *name, const char *tbuf, int tsiz, const char *bbuf, int bsiz,
             VLCFUNC cmp, int ki, int ox, int gt, int lt, int max, int desc){
  VILLA *villa;
  char *kbuf, *vbuf;
  int ksiz, vsiz, show, rv;
  if(!(villa = vlopen(name, VL_OREADER, cmp))){
    pdperror(name);
    return 1;
  }
  if(max < 0) max = vlrnum(villa);
  if(desc){
    if(bbuf){
      vlcurjump(villa, bbuf, bsiz, VL_JBACKWARD);
    } else {
      vlcurlast(villa);
    }
    show = 0;
    while(show < max && (kbuf = vlcurkey(villa, &ksiz)) != NULL){
      if(bbuf && lt){
        if(cmp(kbuf, ksiz, bbuf, bsiz) == 0){
          free(kbuf);
          vlcurnext(villa);
          continue;
        }
        lt = FALSE;
      }
      if(tbuf){
        rv = cmp(kbuf, ksiz, tbuf, tsiz);
        if(rv < 0 || (gt && rv == 0)){
          free(kbuf);
          break;
        }
      }
      if(!(vbuf = vlcurval(villa, &vsiz))){
        free(kbuf);
        break;
      }
      if(ox){
        printobjhex(kbuf, ksiz);
        putchar('\t');
        printobjhex(vbuf, vsiz);
      } else {
        printobj(kbuf, ksiz);
        putchar('\t');
        printobj(vbuf, vsiz);
      }
      putchar('\n');
      free(kbuf);
      free(vbuf);
      show++;
      vlcurprev(villa);
    }
  } else {
    if(tbuf){
      vlcurjump(villa, tbuf, tsiz, VL_JFORWARD);
    } else {
      vlcurfirst(villa);
    }
    show = 0;
    while(show < max && (kbuf = vlcurkey(villa, &ksiz)) != NULL){
      if(tbuf && gt){
        if(cmp(kbuf, ksiz, tbuf, tsiz) == 0){
          free(kbuf);
          vlcurnext(villa);
          continue;
        }
        gt = FALSE;
      }
      if(bbuf){
        rv = cmp(kbuf, ksiz, bbuf, bsiz);
        if(rv > 0 || (lt && rv == 0)){
          free(kbuf);
          break;
        }
      }
      if(!(vbuf = vlcurval(villa, &vsiz))){
        free(kbuf);
        break;
      }
      if(ox){
        printobjhex(kbuf, ksiz);
        putchar('\t');
        printobjhex(vbuf, vsiz);
      } else {
        printobj(kbuf, ksiz);
        putchar('\t');
        printobj(vbuf, vsiz);
      }
      putchar('\n');
      free(kbuf);
      free(vbuf);
      show++;
      vlcurnext(villa);
    }
  }
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform optimize command */
int dooptimize(const char *name){
  VILLA *villa;
  if(!(villa = vlopen(name, VL_OWRITER, VL_CMPLEX))){
    pdperror(name);
    return 1;
  }
  if(!vloptimize(villa)){
    pdperror(name);
    vlclose(villa);
    return 1;
  }
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform inform command */
int doinform(const char *name){
  VILLA *villa;
  char *tmp;
  if(!(villa = vlopen(name, VL_OREADER, VL_CMPLEX))){
    pdperror(name);
    return 1;
  }
  tmp = vlname(villa);
  printf("name: %s\n", tmp ? tmp : "(null)");
  free(tmp);
  printf("file size: %d\n", vlfsiz(villa));
  printf("leaf nodes: %d\n", vllnum(villa));
  printf("non-leaf nodes: %d\n", vlnnum(villa));
  printf("records: %d\n", vlrnum(villa));
  printf("inode number: %d\n", vlinode(villa));
  if(!vlclose(villa)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform remove command */
int doremove(const char *name){
  if(!vlremove(name)){
    pdperror(name);
    return 1;
  }
  return 0;
}


/* perform econv command */
int doeconv(const char *name, int mode){
  int big;
  if(mode > 0){
    big = TRUE;
  } else if(mode < 0){
    big = FALSE;
  } else {
    big = isbigendian();
  }
  if(!vleconv(name, big)){
    pdperror(name);
    return 1;
  }
  return 0;
}



/* END OF FILE */