Merge branch 'master' of ssh://git.stderr.nl/projects/chimara/chimara
[projects/chimara/chimara.git] / tests / babeltest.c
index 69ed3cca4fa0b256413f7b1111bf0691bc39f2c9..2e8827e5b98f33be1dc8b844d026da06a1c86ae8 100644 (file)
@@ -6,14 +6,18 @@
 #include <glib.h>
 #include <glib/gprintf.h>
 #include <libgda/libgda.h>
-#include <sql-parser/gda-sql-parser.h>
+#include <libgda/sql-parser/gda-sql-parser.h>
+#include <libsoup/soup.h>
 
 typedef struct _metadata {
        const gchar *element_name;
        gchar *ifid;
        gchar *title;
        gchar *author;
-       gchar *year;
+       gchar *firstpublished;
+       gboolean error;
+       gchar *error_message;
+       gchar *error_code;
 } metadata;
 
 void start_element(
@@ -27,11 +31,17 @@ void start_element(
        metadata *md = (metadata*) data;
        md->element_name = element_name;
 
+       if( !strcmp(element_name, "errorCode") ) {
+               md->error = 1;
+               md->error_message = "";
+               md->error_code = "";
+       }
+
        if( !strcmp(element_name, "ifindex") ) {
-               md->ifid = g_strdup("");
-               md->title = g_strdup("");
-               md->author = g_strdup("");
-               md->year = g_strdup("");
+               md->ifid = "";
+               md->title = "";
+               md->author = "";
+               md->firstpublished = "";
        }
 }
 
@@ -43,27 +53,28 @@ void text(
        GError **error)
 {
        metadata *md = (metadata*) data;
-       gchar *stripped_text;
 
-       if( !strcmp(md->element_name, "ifid") ) {
-               stripped_text = g_strstrip( g_strndup(text, text_len) );
-               md->ifid = g_strconcat(md->ifid, stripped_text, NULL);
-               g_free(stripped_text);
+       if( !strcmp(md->element_name, "errorCode") ) {
+               md->error_code = g_strndup(text, text_len);
+       }
+       else if( !strcmp(md->element_name, "errorMessage") ) {
+               md->error_message = g_strndup(text, text_len);
+       }
+       else if( !strcmp(md->element_name, "ifid") ) {
+               if( strlen(md->ifid) < text_len )
+                       md->ifid = g_strndup(text, text_len);
        }
        else if( !strcmp(md->element_name, "title") ) {
-               stripped_text = g_strstrip( g_strndup(text, text_len) );
-               md->title = g_strconcat(md->title, stripped_text, NULL);
-               g_free(stripped_text);
+               if( strlen(md->title) < text_len )
+                       md->title = g_strndup(text, text_len);
        }
        else if( !strcmp(md->element_name, "author") ) {
-               stripped_text = g_strstrip( g_strndup(text, text_len) );
-               md->author = g_strconcat(md->author, stripped_text, NULL);
-               g_free(stripped_text);
+               if( strlen(md->author) < text_len )
+                       md->author = g_strndup(text, text_len);
        }
        else if( !strcmp(md->element_name, "firstpublished") ) {
-               stripped_text = g_strstrip( g_strndup(text, text_len) );
-               md->year = g_strconcat(md->year, stripped_text, NULL);
-               g_free(stripped_text);
+               if( strlen(md->firstpublished) < text_len )
+                       md->firstpublished = g_strndup(text, text_len);
        }
 }
 
@@ -75,7 +86,7 @@ void end_element(
 {
        if( !strcmp(element_name, "ifindex") ) {
                metadata *md = (metadata*) data;
-               printf("IFID: %s\nTitle: %s\nAuthor: %s\nYear: %s\n", md->ifid, md->title, md->author, md->year);
+               printf("IFID: %s\nTitle: %s\nAuthor: %s\nFirst published: %s\n", md->ifid, md->title, md->author, md->firstpublished);
        }
 }
 
@@ -103,37 +114,69 @@ run_sql_non_select(GdaConnection *cnc, const gchar *sql)
 }
 
 int main(int argc, char **argv) {
+       GError *err = NULL;
+       metadata data;
+       data.error = 0;
+
        if(argc < 2) {
                fprintf(stderr, "Usage: %s <story file>\n", argv[0]);
                return 1;
        }
 
+       g_type_init();
+
        babel_init(argv[1]);
        int len = babel_treaty(GET_STORY_FILE_METADATA_EXTENT_SEL, NULL, 0);
-       if(len == 0) {
-               printf("No metadata found.\n");
+       gchar *ifiction;
+       if(len) {
+               printf("Metadata found in file.\n");
+               gchar *buffer = malloc(len * sizeof(gchar));
+               babel_treaty(GET_STORY_FILE_METADATA_SEL, buffer, len);
+               ifiction = g_strndup(buffer, len);
+               g_free(buffer);
+       } else {
+               printf("No metadata found in file, performing IFDB lookup.\n");
+               gchar *ifid = malloc(TREATY_MINIMUM_EXTENT * sizeof(gchar));
+               if( !babel_treaty(GET_STORY_FILE_IFID_SEL, ifid, TREATY_MINIMUM_EXTENT) ) {
+                       fprintf(stderr, "Unable to create an IFID (A serious problem occurred while loading the file).\n");
+                       babel_release();
+                       return 1;
+               }
+               printf("Looking up IFID: %s.\n", ifid);
                babel_release();
-               return 0;
+
+               SoupSession *session = soup_session_async_new();
+               char *uri_string = g_strconcat("http://ifdb.tads.org/viewgame?ifiction&ifid=", ifid, NULL);
+               SoupMessage *message = soup_message_new("GET", uri_string);
+               g_free(uri_string);
+               soup_message_headers_append(message->request_headers, "Connection", "close");
+               if(soup_session_send_message(session, message) != 200)
+                       g_printerr("ERROR: did not get HTTP status 200\n");
+               ifiction = g_strndup(message->response_body->data, message->response_body->length);
+               g_object_unref(message);
+               g_object_unref(session);
        }
 
-       gchar *buffer = malloc(len * sizeof(gchar));
-       babel_treaty(GET_STORY_FILE_METADATA_SEL, buffer, len);
-       g_strchomp(buffer);
-       len = strlen(buffer);
+       ifiction = g_strchomp(ifiction);
 
-       metadata data;
        GMarkupParser xml_parser = {start_element, end_element, text, NULL, NULL};
        GMarkupParseContext *context = g_markup_parse_context_new(&xml_parser, 0, &data, NULL);
 
-       GError *err = NULL;
-       if( g_markup_parse_context_parse(context, buffer, len, &err) == FALSE ) {
+       if( g_markup_parse_context_parse(context, ifiction, strlen(ifiction), &err) == FALSE ) {
                fprintf(stderr, "Metadata parse failed: %s\n", err->message);
        }
 
-       free(buffer);
        g_markup_parse_context_free(context);
+       g_free(ifiction);
+
        babel_release();
 
+       // Check for errors
+       if(data.error) {
+               fprintf(stderr, "ERROR %s: %s\n", data.error_code, data.error_message);
+               return 1;
+       }
+
        // Open DB connection
        GdaConnection *cnc;
        GdaSqlParser *sql_parser;
@@ -152,19 +195,18 @@ int main(int argc, char **argv) {
        g_object_set_data_full(G_OBJECT(cnc), "parser", sql_parser, g_object_unref);
        
        // Create stories table
-       run_sql_non_select(cnc, "DROP TABLE IF EXISTS stories");
-       run_sql_non_select(cnc, "CREATE TABLE stories (ifid text not null primary key, title text, author text, year integer)");
+       //run_sql_non_select(cnc, "DROP TABLE IF EXISTS stories");
+       run_sql_non_select(cnc, "CREATE TABLE IF NOT EXISTS stories (ifid text not null primary key, title text, author text, firstpublished text)");
 
        // Populate the table
        GValue *v1, *v2, *v3, *v4;
        v1 = gda_value_new_from_string(data.ifid, G_TYPE_STRING);
        v2 = gda_value_new_from_string(data.title, G_TYPE_STRING);
        v3 = gda_value_new_from_string(data.author, G_TYPE_STRING);
-       v4 = gda_value_new_from_string(data.year, G_TYPE_UINT);
+       v4 = gda_value_new_from_string(data.firstpublished, G_TYPE_STRING);
 
-       if( !gda_insert_row_into_table(cnc, "stories", &err, "ifid", v1, "title", v2, "author", v3, "year", v4, NULL) ) {
-               g_error("Could not INSERT data into the 'stories' table: %s\n", err && err->message ? err->message : "No details");
-               return 1;
+       if( !gda_insert_row_into_table(cnc, "stories", &err, "ifid", v1, "title", v2, "author", v3, "firstpublished", v4, NULL) ) {
+               g_warning("Could not INSERT data into the 'stories' table: %s\n", err && err->message ? err->message : "No details");
        }
 
        gda_value_free(v1);
@@ -172,6 +214,18 @@ int main(int argc, char **argv) {
        gda_value_free(v3);
        gda_value_free(v4);
 
+       // Dump the table contents
+       GdaDataModel *data_model;
+       GdaStatement *stmt = gda_sql_parser_parse_string(sql_parser, "SELECT * FROM stories", NULL, NULL);
+       data_model = gda_connection_statement_execute_select(cnc, stmt, NULL, &err);
+       if(!data_model)
+               g_error("Could not get the contents of the 'stories' table: %s\n", err && err->message ? err->message : "No details");
+       printf("Dumping library table:\n");
+       gda_data_model_dump(data_model, stdout);
+
+       g_object_unref(stmt);
+       g_object_unref(data_model);
+
        gda_connection_close(cnc);
        return 0;
 }