libDatabase
/* $Id$ */
/* Copyright (c) 2013-2020 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_DATABASE
# define PROGNAME_DATABASE "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, int queryc, char * const queryv[]);
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_DATABASE);
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_DATABASE);
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, int queryc, char * const queryv[])
{
int ret = 0;
DatabaseFile database;
Database * db;
int i;
if((db = _database_load(engine, cfile, section)) == NULL)
{
error_print(PROGNAME_DATABASE);
return 2;
}
database.fp = stdout;
for(i = 0; i < queryc; i++)
{
database.first = 1;
database.rows = 0;
if((ret = database_query(db, queryv[i], _database_print,
&database)) != 0)
{
error_print(PROGNAME_DATABASE);
break;
}
printf("(%u rows)\n", database.rows);
}
database_delete(db);
return ret;
}
/* usage */
static int _usage(void)
{
fputs("Usage: " PROGNAME_DATABASE
" -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 < argc)
return _database_query(engine, cfile, section,
argc - optind, &argv[optind]);
return (_database(engine, cfile, section) == 0) ? 0 : 2;
}