added mysql config file and some initial database installation / upgrade methods

[NeonServV5.git] / src / mysqlConn.c

diff --git a/src/mysqlConn.c b/src/mysqlConn.c
index 66494e7f5055cf333db9d500c9bde2a605c7f2cd..5965e6aafa6d171434bbc11bb3262279f7340a5c 100644 (file)
--- a/src/mysqlConn.c
+++ b/src/mysqlConn.c
@@ -1,5 +1,6 @@
 
 #include "mysqlConn.h"
+#define DATABASE_VERSION "1"
 
 struct used_result {
     MYSQL_RES *result;
@@ -14,12 +15,14 @@ struct escaped_string {
 MYSQL *mysql_conn = NULL;
 static struct used_result *used_results;
 static struct escaped_string *escaped_strings;
+static int mysql_serverport;
+static char *mysql_host, *mysql_user, *mysql_pass, *mysql_base;
 
 void check_mysql() {
     int errid;
     if((errid = mysql_ping(mysql_conn))) {
         if(mysql_errno(mysql_conn) == CR_SERVER_GONE_ERROR) {
-            if(!mysql_real_connect(mysql_conn, MYSQL_HOST, MYSQL_USER, MYSQL_PASS, MYSQL_BASE, MYSQL_PORT, NULL, 0)) {
+            if(!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
                 show_mysql_error();
             }
         } else {
@@ -59,12 +62,86 @@ void mysql_free() {
     escaped_strings = NULL;
 }
 
-void init_mysql() {
+void init_mysql(char *host, int port, char *user, char *pass, char *base) {
+    mysql_host = strdup(host);
+    mysql_serverport = port;
+    mysql_user = strdup(user);
+    mysql_pass = strdup(pass);
+    mysql_base = strdup(base);
     mysql_conn = mysql_init(NULL);
-    if (!mysql_real_connect(mysql_conn, MYSQL_HOST, MYSQL_USER, MYSQL_PASS, MYSQL_BASE, MYSQL_PORT, NULL, 0)) {
+    if (!mysql_real_connect(mysql_conn, mysql_host, mysql_user, mysql_pass, mysql_base, mysql_serverport, NULL, 0)) {
         //error
         show_mysql_error();
     }
+    //check database version...
+    int version = 0;
+    if(!mysql_query(mysql_conn, "SELECT `database_version` FROM `version`")) {
+        MYSQL_RES *res = mysql_use();
+        MYSQL_ROW row;
+        if((row = mysql_fetch_row(res))) {
+            version = atoi(row[0]);
+        }
+    }
+    if(!version) {
+        //CREATE DATABASE
+        FILE *f = fopen("database.sql", "r");
+        mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
+        if (f) {
+            char line[512];
+            char query_buffer[8192];
+            int query_buffer_pos = 0;
+            while (fgets(line, sizeof(line), f)) {
+                query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
+                if(line[(strlen(line) - 1)] == ';') {
+                    mysql_query(mysql_conn, query_buffer);
+                    query_buffer_pos = 0;
+                }
+            }
+            fclose(f);
+        }
+        f = fopen("database.defaults.sql", "r");
+        mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
+        if (f) {
+            char line[2048];
+            char query_buffer[65536];
+            int query_buffer_pos = 0;
+            while (fgets(line, sizeof(line), f)) {
+                query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
+                if(line[(strlen(line) - 1)] == ';') {
+                    mysql_query(mysql_conn, query_buffer);
+                    query_buffer_pos = 0;
+                }
+            }
+            fclose(f);
+        }
+        mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
+        mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
+    }
+    else if(version < atoi(DATABASE_VERSION)) {
+        //UPDATE DATABASE
+        FILE *f = fopen("database.upgrade.sql", "r");
+        mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_ON);
+        if (f) {
+            char line[512];
+            char query_buffer[8192];
+            int query_buffer_pos = 0, use_querys = 0;
+            sprintf(query_buffer, "-- version: %d", version);
+            while (fgets(line, sizeof(line), f)) {
+                if(use_querys) {
+                    query_buffer_pos += sprintf(query_buffer + query_buffer_pos, " %s", line);
+                    if(line[strlen(line) - 1] == ';') {
+                        mysql_query(mysql_conn, query_buffer);
+                        query_buffer_pos = 0;
+                    }
+                } else if(!stricmp(query_buffer, line)) {
+                    use_querys = 1;
+                }
+            }
+            fclose(f);
+        }
+        mysql_set_server_option(mysql_conn, MYSQL_OPTION_MULTI_STATEMENTS_OFF);
+        mysql_query(mysql_conn, "UPDATE `version` SET `database_version` = '" DATABASE_VERSION "'");
+    }
 }
 
 void free_mysql() {