Geografic-Information-System/db/db.copy/main.c

/****************************************************************************
 *
 * MODULE:       db.copy
 * AUTHOR(S):    Radim Blazek <radim.blazek gmail.com> (original contributor)
 *               Glynn Clements <glynn gclements.plus.com>, Markus Neteler <neteler itc.it>
 * PURPOSE:      copy a table
 * COPYRIGHT:    (C) 2003-2006 by the GRASS Development Team
 *
 *               This program is free software under the GNU General Public
 *               License (>=v2). Read the file COPYING that comes with GRASS
 *               for details.
 *
 *****************************************************************************/

#include <stdlib.h>
#include <grass/gis.h>
#include <grass/dbmi.h>
#include <grass/glocale.h>


int main(int argc, char **argv)
{
    int ret;
    struct Option *from_driver, *from_database, *from_table;
    struct Option *to_driver, *to_database, *to_table;
    struct Option *where, *select;
    struct GModule *module;
    const char *drv, *db;

    G_gisinit(argv[0]);
    
    /* Set description */
    module = G_define_module();
    G_add_keyword(_("database"));
    G_add_keyword(_("attribute table"));
    G_add_keyword(_("SQL"));
    module->label = _("Copy a table.");
    module->description =
	_("Either 'from_table' (optionally with 'where') can be used "
	  "or 'select' option, but not 'from_table' and 'select' at the same time.");
    
    from_driver = G_define_standard_option(G_OPT_DB_DRIVER);
    from_driver->key = "from_driver";
    from_driver->options = db_list_drivers();
    from_driver->description = _("Input driver name");
    if ((drv = db_get_default_driver_name()))
      from_driver->answer = (char *) drv;

    from_database = G_define_standard_option(G_OPT_DB_DATABASE);
    from_database->key = "from_database";
    from_database->description = _("Input database name");
    if ((db = db_get_default_database_name()))
      from_database->answer = (char *) db;

    from_table = G_define_standard_option(G_OPT_DB_TABLE);
    from_table->key = "from_table";
    from_table->description =
	_("Input table name (only, if 'select' is not used)");

    to_driver = G_define_standard_option(G_OPT_DB_DRIVER);
    to_driver->key = "to_driver";
    to_driver->options = db_list_drivers();
    to_driver->required = NO;
    to_driver->description = _("Output driver name");
    if ((drv = db_get_default_driver_name()))
      to_driver->answer = (char *) drv;

    to_database = G_define_standard_option(G_OPT_DB_DATABASE);
    to_database->key = "to_database";
    to_database->description = _("Output database name");
    if ((db = db_get_default_database_name()))
      to_database->answer = (char *) db;

    to_table = G_define_standard_option(G_OPT_DB_TABLE);
    to_table->key = "to_table";
    to_table->required = YES;
    to_table->description = _("Output table name");
    to_table->gisprompt = "new,dbtable,dbtable";

    where = G_define_standard_option(G_OPT_DB_WHERE);

    select = G_define_option();
    select->key = "select";
    select->type = TYPE_STRING;
    select->required = NO;
    select->label = _("Full select statement (only, if 'from_table' and 'where' is not used)");
    select->description = _("E.g.: SELECT dedek FROM starobince WHERE obec = 'Frimburg'");
    
    if (G_parser(argc, argv))
	exit(EXIT_FAILURE);

    /* Check options and copy tables */
    if (from_table->answer) {
	if (select->answer)
	    G_fatal_error(_("Cannot combine 'from_table' and 'select' options"));

	if (where->answer) {
	    ret =
		db_copy_table_where(from_driver->answer,
				    from_database->answer, from_table->answer,
				    to_driver->answer, to_database->answer,
				    to_table->answer, where->answer);
	}
	else {
	    ret =
		db_copy_table(from_driver->answer, from_database->answer,
			      from_table->answer, to_driver->answer,
			      to_database->answer, to_table->answer);
	}
    }
    else {
	if (!select->answer)
	    G_fatal_error(_("Either 'from_table' or 'select' option must be given."));

	if (where->answer)
	    G_fatal_error(_("Cannot combine 'select' and 'where' options"));

	ret =
	    db_copy_table_select(from_driver->answer, from_database->answer,
				 from_table->answer, to_driver->answer,
				 to_database->answer, to_table->answer,
				 select->answer);
    }

    if (ret == DB_FAILED) {
	G_warning(_("Copy table failed"));
	exit(EXIT_FAILURE);
    }

    exit(EXIT_SUCCESS);
}