Line data    Source code

       1              : /*
       2              :   This file is part of TALER
       3              :   Copyright (C) 2014, 2015, 2020 Taler Systems SA
       4              : 
       5              :   TALER is free software; you can redistribute it and/or modify it under the
       6              :   terms of the GNU General Public License as published by the Free Software
       7              :   Foundation; either version 3, or (at your option) any later version.
       8              : 
       9              :   TALER is distributed in the hope that it will be useful, but WITHOUT ANY
      10              :   WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
      11              :   A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
      12              : 
      13              :   You should have received a copy of the GNU General Public License along with
      14              :   TALER; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
      15              : */
      16              : /**
      17              :  * @file src/merchant-tools/taler-merchant-dbinit.c
      18              :  * @brief Create tables for the merchant database.
      19              :  * @author Florian Dold
      20              :  * @author Marcello Stanisci
      21              :  */
      22              : #include "platform.h"
      23              : #include <taler/taler_util.h>
      24              : #include <gnunet/gnunet_util_lib.h>
      25              : #include "taler/taler_merchant_util.h"
      26              : #include "merchantdb_lib.h"
      27              : #include "merchant-database/create_tables.h"
      28              : #include "merchant-database/drop_tables.h"
      29              : #include "merchant-database/gc.h"
      30              : 
      31              : 
      32              : /**
      33              :  * Return value from main().
      34              :  */
      35              : static int global_ret;
      36              : 
      37              : /**
      38              :  * -r option: do full DB reset
      39              :  */
      40              : static int reset_db;
      41              : 
      42              : /**
      43              :  * -g option: garbage collect DB
      44              :  */
      45              : static int gc_db;
      46              : 
      47              : /**
      48              :  * Main function that will be run.
      49              :  *
      50              :  * @param cls closure
      51              :  * @param args remaining command-line arguments
      52              :  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
      53              :  * @param cfg configuration
      54              :  */
      55              : static void
      56           19 : run (void *cls,
      57              :      char *const *args,
      58              :      const char *cfgfile,
      59              :      const struct GNUNET_CONFIGURATION_Handle *cfg)
      60              : {
      61              :   struct TALER_MERCHANTDB_PostgresContext *pg;
      62              : 
      63           19 :   pg = TALER_MERCHANTDB_connect_admin (cfg);
      64           19 :   if (NULL == pg)
      65              :   {
      66            0 :     fprintf (stderr,
      67              :              "Failed to initialize database connection.\n");
      68            0 :     global_ret = 1;
      69            0 :     return;
      70              :   }
      71           19 :   if (reset_db)
      72              :   {
      73           18 :     if (GNUNET_OK !=
      74           18 :         TALER_MERCHANTDB_drop_tables (pg))
      75              :     {
      76            0 :       global_ret = 1;
      77            0 :       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
      78              :                   "Failed to reset the database\n");
      79            0 :       TALER_MERCHANTDB_disconnect (pg);
      80            0 :       return;
      81              :     }
      82           18 :     gc_db = 0; /* not useful */
      83              :   }
      84           19 :   if (GNUNET_OK !=
      85           19 :       TALER_MERCHANTDB_create_tables (pg))
      86              :   {
      87            0 :     global_ret = 1;
      88            0 :     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
      89              :                 "Failed to initialize tables\n");
      90              :   }
      91           19 :   if (gc_db)
      92              :   {
      93            0 :     if (GNUNET_OK !=
      94            0 :         TALER_MERCHANTDB_gc (pg))
      95              :     {
      96            0 :       global_ret = 1;
      97            0 :       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
      98              :                   "Failed to garbage collect database\n");
      99              :     }
     100            0 :     TALER_MERCHANTDB_disconnect (pg);
     101            0 :     return;
     102              :   }
     103           19 :   TALER_MERCHANTDB_disconnect (pg);
     104              : }
     105              : 
     106              : 
     107              : /**
     108              :  * The main function of the database initialization tool.
     109              :  * Used to initialize the Taler Exchange's database.
     110              :  *
     111              :  * @param argc number of arguments from the command line
     112              :  * @param argv command line arguments
     113              :  * @return 0 ok, 1 on error
     114              :  */
     115              : int
     116           19 : main (int argc,
     117              :       char *const *argv)
     118              : {
     119           19 :   struct GNUNET_GETOPT_CommandLineOption options[] = {
     120           19 :     GNUNET_GETOPT_option_flag ('g',
     121              :                                "gc",
     122              :                                "garbage collect database",
     123              :                                &gc_db),
     124           19 :     GNUNET_GETOPT_option_flag ('r',
     125              :                                "reset",
     126              :                                "reset database (DANGEROUS: all existing data is lost!)",
     127              :                                &reset_db),
     128              : 
     129           19 :     GNUNET_GETOPT_option_version (PACKAGE_VERSION),
     130              :     GNUNET_GETOPT_OPTION_END
     131              :   };
     132              :   enum GNUNET_GenericReturnValue ret;
     133              : 
     134           19 :   ret = GNUNET_PROGRAM_run (
     135              :     TALER_MERCHANT_project_data (),
     136              :     argc, argv,
     137              :     "taler-merchant-dbinit",
     138              :     gettext_noop ("Initialize Taler merchant database"),
     139              :     options,
     140              :     &run, NULL);
     141           19 :   if (GNUNET_SYSERR == ret)
     142            0 :     return 3;
     143           19 :   if (GNUNET_NO == ret)
     144            0 :     return 0;
     145           19 :   return global_ret;
     146              : }
     147              : 
     148              : 
     149              : /* end of taler-merchant-dbinit.c */