/*
 * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the author nor the names of any co-contributors
 *      may be used to endorse or promote products derived from this software
 *      without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include "tlog.h"
#include "tmalloc.h"
#include "flatdb.h"

static void
usage() {
	puts(
	"Usage:\n"
	"flatdbtest [-r] -f FILE  [ -p INFO | -g OFFSET | -d OFFSET | -v [ ... ] ]\n"
	"flatdbtest -f FILE  -s"
	);
	exit(1);
}

extern char *optarg;
extern int opterr;

int
main(int argn, char *argv[]) {
	FDB db;
	int i;
	int isopened=0, readonly=0;
	off_t offset;
	int len,rc;
	FDBRecord	*record;	

	opentlog(TL_OPEN_STDERR,TL_DEBUG, NULL);
	opterr=0;

	while((i=getopt(argn,argv,"srf:p:g:d:vh")) != EOF) {
		switch(i) {
			case 'r':
				if (isopened) {FDBClose(&db); usage();}
				readonly=1;
				break;
			case 'f':
				if (isopened) {FDBClose(&db); usage();}
				if ( (rc=FDBOpen(&db, optarg, readonly)) != FDB_OK )
					tlog(TL_CRIT|TL_EXIT, "FDBOpen failed: %d", rc);
				isopened=1;
				break;
			case 'v':
				if (!isopened) {usage();}
				FDBVacuumFreeSpace(&db);
				puts("Vacuum"); 
				break;
			case 'p':
				if (!isopened) {usage();}
				len = RECHDRSZ + strlen(optarg);
				record = (FDBRecord*)tmalloc( len );
				record->length = (size_t)len;
				memcpy( record->data, optarg, len-RECHDRSZ );
				if ( (rc=FDBPut(&db, record, &offset)) != FDB_OK ) {
					FDBClose(&db);
					tlog(TL_CRIT|TL_EXIT, "FDBPut failed: %d", rc);
				}
				tfree(record);
				printf("Put: off:%d len:%d '%s'\n", (int)offset, (int)(len-RECHDRSZ), optarg);
				break;
			case 'g': 
				if (!isopened) {usage();}
				offset = atoi(optarg);
				if ( (rc=FDBGet(&db, offset, 0, &record)) != FDB_OK ) {
					FDBClose(&db);
					tlog(TL_CRIT|TL_EXIT, "FDBGet failed: %d", rc);
				}
				printf("Get: off:%d len:%d '", (int)offset, (int)(record->length - RECHDRSZ));
				fwrite(record->data, record->length - RECHDRSZ, 1, stdout);
				puts("'");
				tfree(record);
				break;
			case 'd':
				if (!isopened) {usage();}
				offset = atoi(optarg);
				if ( (rc=FDBDelete(&db, offset, 0)) != FDB_OK ) {
					FDBClose(&db);
					tlog(TL_CRIT|TL_EXIT, "FDBDelete failed: %d", rc);
				}
				printf("Del: off:%d\n", (int)offset);
				break;
			case 's':
				if (!isopened) {usage();}
				if ( db.listcur ) { 
				puts("List of free space:");
					for(i=0;i<db.listcur;i++)
						printf("\toff: %d\t len: %d\n", (int)(db.space[i].offset), (int)(db.space[i].length));
				} else
					 puts("List of free space is void");
				break;
			case 'h':
			default:
				if (isopened) {FDBClose(&db);}
				usage();
				break;
		}
	}
				
	if (isopened) 
		FDBClose(&db);
	else
		usage();

	return 0;
}