libDatabase/tools/database.c

/* $Id$ */
/* Copyright (c) 2013-2019 Pierre Pronchery <khorben@defora.org> */
/* This file is part of DeforaOS Database libDatabase */
/* This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3 of the License.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>. */



#include <unistd.h>
#include <stdio.h>
#include <System.h>
#include "Database.h"

#ifndef PROGNAME
# define PROGNAME	"database"
#endif


/* database */
/* private */
/* types */
typedef struct _DatabaseFile
{
	FILE * fp;
	int first;
	unsigned int rows;
} DatabaseFile;


/* prototypes */
static int _database(char const * engine, char const * cfile,
		char const * section);
static Database * _database_load(char const * engine, char const * cfile,
		char const * section);
static int _database_print(void * data, int argc, char ** argv,
		char ** columns);
static int _database_query(char const * engine, char const * cfile,
		char const * section, char const * query);

static int _usage(void);


/* functions */
/* database */
static int _database(char const * engine, char const * cfile,
		char const * section)
{
	DatabaseFile database;
	Database * db;
	char buf[BUFSIZ];

	if((db = _database_load(engine, cfile, section)) == NULL)
	{
		error_print(PROGNAME);
		return 2;
	}
	database.fp = stdout;
	while(fgets(buf, sizeof(buf), stdin) != NULL)
	{
		database.first = 1;
		database.rows = 0;
		/* XXX it may not have picked a complete line */
		if(database_query(db, buf, _database_print, &database) != 0)
		{
			error_print(PROGNAME);
			continue;
		}
		fprintf(database.fp, "(%u rows)\n", database.rows);
	}
	database_delete(db);
	return 0;
}


/* database_load */
static Database * _database_load(char const * engine, char const * cfile,
		char const * section)
{
	Database * db;
	Config * config;

	if(cfile == NULL)
		section = NULL;
	if((config = config_new()) == NULL)
		return NULL;
	if(cfile != NULL && config_load(config, cfile) != 0)
		db = NULL;
	else
		db = database_new(engine, config, section);
	config_delete(config);
	return db;
}


/* database_print */
static int _database_print(void * data, int argc, char ** argv, char ** columns)
{
	DatabaseFile * database = data;
	int i;

	database->rows++;
	if(argc == 0)
		return 0;
	if(database->first != 0)
	{
		database->first = 0;
		for(i = 0; i < argc; i++)
			fprintf(database->fp, "|%s", columns[i]);
		fputs("|\n", database->fp);
	}
	for(i = 0; i < argc; i++)
		fprintf(database->fp, "|%s", argv[i]);
	fputs("|\n", database->fp);
	return 0;
}


/* database_query */
static int _database_query(char const * engine, char const * cfile,
		char const * section, char const * query)
{
	int ret;
	DatabaseFile database;
	Database * db;

	if((db = _database_load(engine, cfile, section)) == NULL)
	{
		error_print(PROGNAME);
		return 2;
	}
	database.fp = stdout;
	database.first = 1;
	database.rows = 0;
	if((ret = database_query(db, query, _database_print, &database)) != 0)
	{
		error_print(PROGNAME);
		return ret;
	}
	printf("(%u rows)\n", database.rows);
	database_delete(db);
	return ret;
}


/* usage */
static int _usage(void)
{
	fputs("Usage: " PROGNAME " -d engine [-C configuration [-S section]][query]\n"
"  -d	Database engine to load\n"
"  -C	Connection file to load\n"
"  -S	Section of the connection file to use\n",
stderr);
	return 1;
}


/* public */
/* functions */
/* main */
int main(int argc, char * argv[])
{
	int o;
	char const * engine = NULL;
	char const * cfile = NULL;
	char const * section = NULL;

	while((o = getopt(argc, argv, "C:d:S:")) != -1)
		switch(o)
		{
			case 'C':
				cfile = optarg;
				break;
			case 'd':
				engine = optarg;
				break;
			case 'S':
				section = optarg;
				break;
			default:
				return _usage();
		}
	if(engine == NULL)
		return _usage();
	else if(optind + 1 < argc)
		return _usage();
	else if(optind + 1 == argc)
		return _database_query(engine, cfile, section, argv[optind]);
	return (_database(engine, cfile, section) == 0) ? 0 : 2;
}