From 3ff195398245d187f0477ef6c83bd5ed5eb50b03 Mon Sep 17 00:00:00 2001 From: Jon Bergli Heier Date: Mon, 15 Nov 2010 20:19:06 +0100 Subject: Added some ecpg stuff. --- main.c | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'main.c') diff --git a/main.c b/main.c index 5c34642..72a8848 100644 --- a/main.c +++ b/main.c @@ -7,6 +7,7 @@ #include "word.h" #include "nick.h" #include "parsing.h" +#include "pg.h" int main(int argc, char **argv) { /* Set locale. */ @@ -24,8 +25,26 @@ int main(int argc, char **argv) { return 1; } + if(!pg_connect(ircstats_config.db_connection_string)) { + printf("failed to connect\n"); + goto main_free; + } + + if(argc == 2 && strcmp(argv[1], "-i") == 0) { + pg_init(); + goto disconnect; + } + if(argc == 2 && strcmp(argv[1], "-u") == 0) { + pg_upgrade(); + goto disconnect; + } + process(); +disconnect: + pg_disconnect(); + +main_free: nick_free(); cfg_free(); channel_free(); -- cgit v1.2.3