diff pidgin/plugins/cap/cap.c @ 15374:5fe8042783c1

Rename gtk/ and libgaim/ to pidgin/ and libpurple/

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pidgin/plugins/cap/cap.c	Sat Jan 20 02:32:10 2007 +0000
@@ -0,0 +1,965 @@
+/*
+ * Contact Availability Prediction plugin for Gaim
+ *
+ * Copyright (C) 2006 Geoffrey Foster.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
+ * 02111-1307, USA.
+ */
+
+#include "cap.h"
+
+static void generate_prediction(CapStatistics *statistics) {
+	if(statistics->buddy) {
+		if(statistics->prediction == NULL)
+			statistics->prediction = g_malloc(sizeof(CapPrediction));
+		statistics->prediction->probability = generate_prediction_for(statistics->buddy);
+		statistics->prediction->generated_at = time(NULL);
+	}
+}
+
+static double generate_prediction_for(GaimBuddy *buddy) {
+	double prediction = 1.0f;
+	gboolean generated = FALSE;
+	gchar *buddy_name = buddy->name;
+	const gchar *protocol_id = gaim_account_get_protocol_id(buddy->account);
+	const gchar *account_id = gaim_account_get_username(buddy->account);
+	const gchar *status_id = gaim_status_get_id(get_status_for(buddy));
+	time_t t = time(NULL);
+	struct tm *current_time = localtime(&t);
+	int current_minute = current_time->tm_min + current_time->tm_hour * 60;
+	int threshold = gaim_prefs_get_int("/plugins/gtk/cap/threshold");
+	int min_minute = (current_minute - threshold) % 1440;
+	int max_minute = (current_minute + threshold) % 1440;
+	char *sql;
+	sqlite3_stmt *stmt = NULL;
+	const char *tail = NULL;
+	int rc;
+
+
+	sql = sqlite3_mprintf("select sum(success_count) as successes, sum(failed_count) as failures "
+		"from cap_msg_count where "
+		"buddy=%Q and account=%Q and protocol=%Q and minute_val>=%d and minute_val<=%d;",
+		buddy_name,	account_id,	protocol_id, min_minute, max_minute);
+	rc = sqlite3_prepare(_db, sql, -1, &stmt, &tail);
+	if(rc == SQLITE_OK) {
+		int successes = 0;
+		int failures = 0;
+		if(stmt != NULL) {
+			if(sqlite3_step(stmt) == SQLITE_ROW) {
+				successes = sqlite3_column_int(stmt, 0);
+				failures = sqlite3_column_int(stmt, 1);
+				if(failures + successes > 0) {
+					prediction *= ((double)successes/((double)(successes+failures)));
+					generated = TRUE;
+				}
+			}
+			sqlite3_finalize(stmt);
+		}
+	}
+	sqlite3_free(sql);
+
+	sql = sqlite3_mprintf("select sum(success_count) as successes, sum(failed_count) as failures "
+		"from cap_status_count where "
+		"buddy=%Q and account=%Q and protocol=%Q and status=%Q;",
+		buddy_name,	account_id,	protocol_id, status_id);
+	rc = sqlite3_prepare(_db, sql, -1, &stmt, &tail);
+	if(rc == SQLITE_OK) {
+		int successes = 0;
+		int failures = 0;
+		if(stmt != NULL) {
+			if(sqlite3_step(stmt) == SQLITE_ROW) {
+				successes = sqlite3_column_int(stmt, 0);
+				failures = sqlite3_column_int(stmt, 1);
+				if(failures + successes > 0) {
+					prediction *= ((double)successes/((double)(successes+failures)));
+					generated = TRUE;
+				}
+			}
+			sqlite3_finalize(stmt);
+		}
+	}
+	sqlite3_free(sql);
+
+	
+	if(strcmp(gaim_status_get_id(get_status_for(buddy)), "offline") == 0) {
+		/* This is kind of stupid, change it. */
+		if(prediction == 1.0f)
+			prediction = 0.0f;
+	}
+
+	if(generated)
+		return prediction;
+	else
+		return -1;
+}
+
+static CapStatistics * get_stats_for(GaimBuddy *buddy) {
+	gchar *buddy_name;
+	CapStatistics *stats;
+
+	g_return_val_if_fail(buddy != NULL, NULL);
+
+	buddy_name = g_strdup(buddy->name);
+	stats = g_hash_table_lookup(_buddy_stats, buddy_name);
+	if(!stats) {
+		stats = g_malloc(sizeof(CapStatistics));
+		stats->last_message = -1;
+		stats->last_message_status_id = NULL;
+		stats->last_status_id = NULL;
+		stats->prediction = NULL;
+		g_hash_table_insert(_buddy_stats, buddy_name, stats);
+		stats->buddy = buddy;
+		stats->last_seen = -1;
+		stats->last_status_id = "";
+	} else {
+		g_free(buddy_name);
+	}
+	generate_prediction(stats);
+	return stats;
+}
+
+static void destroy_stats(gpointer data) {
+	CapStatistics *stats = data;
+	g_free(stats->prediction);
+	/* g_free(stats->hourly_usage); */
+	/* g_free(stats->daily_usage); */
+	g_free(stats);
+}
+
+static void
+insert_cap_msg_count_success(const char *buddy_name, const char *account, const char *protocol, int minute) {
+	int rc;
+	sqlite3_stmt *stmt;
+	const char *tail;
+	char *sql_select = sqlite3_mprintf("SELECT * FROM cap_msg_count WHERE "
+		"buddy=%Q AND account=%Q AND protocol=%Q AND minute_val=%d;",
+		buddy_name, account, protocol, minute);
+	char *sql_ins_up = NULL;
+
+	gaim_debug_info("cap", "%s\n", sql_select);
+
+	sqlite3_prepare(_db, sql_select, -1, &stmt, &tail);
+	
+	rc = sqlite3_step(stmt);
+
+	if(rc == SQLITE_DONE) {
+		sql_ins_up = sqlite3_mprintf("INSERT INTO cap_msg_count VALUES (%Q, %Q, %Q, %d, %d, %d);",
+			buddy_name, account, protocol, minute, 1, 0);
+	} else if(rc == SQLITE_ROW) {
+		sql_ins_up = sqlite3_mprintf("UPDATE cap_msg_count SET success_count=success_count+1 WHERE "
+			"buddy=%Q AND account=%Q AND protocol=%Q AND minute_val=%d;",
+			buddy_name, account, protocol, minute);
+	} else {
+		gaim_debug_info("cap", "%d\n", rc);
+		sqlite3_finalize(stmt);
+		sqlite3_free(sql_select);
+		return;
+	}
+
+	sqlite3_finalize(stmt);
+	sqlite3_free(sql_select);
+
+	sqlite3_exec(_db, sql_ins_up, NULL, NULL, NULL);
+	sqlite3_free(sql_ins_up);
+}
+
+static void
+insert_cap_status_count_success(const char *buddy_name, const char *account, const char *protocol, const char *status_id) {
+	int rc;
+	sqlite3_stmt *stmt;
+	const char *tail;
+	char *sql_select = sqlite3_mprintf("SELECT * FROM cap_status_count WHERE "
+		"buddy=%Q AND account=%Q AND protocol=%Q AND status=%Q;",
+		buddy_name, account, protocol, status_id);
+	char *sql_ins_up = NULL;
+
+	gaim_debug_info("cap", "%s\n", sql_select);
+
+	sqlite3_prepare(_db, sql_select, -1, &stmt, &tail);
+	
+	rc = sqlite3_step(stmt);
+
+	if(rc == SQLITE_DONE) {
+		sql_ins_up = sqlite3_mprintf("INSERT INTO cap_status_count VALUES (%Q, %Q, %Q, %Q, %d, %d);",
+			buddy_name, account, protocol, status_id, 1, 0);
+	} else if(rc == SQLITE_ROW) {
+		sql_ins_up = sqlite3_mprintf("UPDATE cap_status_count SET success_count=success_count+1 WHERE "
+			"buddy=%Q AND account=%Q AND protocol=%Q AND status=%Q;",
+			buddy_name, account, protocol, status_id);
+	} else {
+		gaim_debug_info("cap", "%d\n", rc);
+		sqlite3_finalize(stmt);
+		sqlite3_free(sql_select);
+		return;
+	}
+
+	sqlite3_finalize(stmt);
+	sqlite3_free(sql_select);
+
+	sqlite3_exec(_db, sql_ins_up, NULL, NULL, NULL);
+	sqlite3_free(sql_ins_up);
+}
+
+static void
+insert_cap_msg_count_failed(const char *buddy_name, const char *account, const char *protocol, int minute) {
+	int rc;
+	sqlite3_stmt *stmt;
+	const char *tail;
+	char *sql_select = sqlite3_mprintf("SELECT * FROM cap_msg_count WHERE "
+		"buddy=%Q AND account=%Q AND protocol=%Q AND minute_val=%d;",
+		buddy_name, account, protocol, minute);
+	char *sql_ins_up = NULL;
+
+	gaim_debug_info("cap", "%s\n", sql_select);
+
+	sqlite3_prepare(_db, sql_select, -1, &stmt, &tail);
+	
+	rc = sqlite3_step(stmt);
+
+	if(rc == SQLITE_DONE) {
+		sql_ins_up = sqlite3_mprintf("INSERT INTO cap_msg_count VALUES (%Q, %Q, %Q, %d, %d, %d);",
+			buddy_name, account, protocol, minute, 0, 1);
+	} else if(rc == SQLITE_ROW) {
+		sql_ins_up = sqlite3_mprintf("UPDATE cap_msg_count SET failed_count=failed_count+1 WHERE "
+			"buddy=%Q AND account=%Q AND protocol=%Q AND minute_val=%d;",
+			buddy_name, account, protocol, minute);
+	} else {
+		gaim_debug_info("cap", "%d\n", rc);
+		sqlite3_finalize(stmt);
+		sqlite3_free(sql_select);
+		return;
+	}
+
+	sqlite3_finalize(stmt);
+	sqlite3_free(sql_select);
+
+	sqlite3_exec(_db, sql_ins_up, NULL, NULL, NULL);
+	sqlite3_free(sql_ins_up);
+}
+
+static void
+insert_cap_status_count_failed(const char *buddy_name, const char *account, const char *protocol, const char *status_id) {
+	int rc;
+	sqlite3_stmt *stmt;
+	const char *tail;
+	char *sql_select = sqlite3_mprintf("SELECT * FROM cap_status_count WHERE "
+		"buddy=%Q AND account=%Q AND protocol=%Q AND status=%Q;",
+		buddy_name, account, protocol, status_id);
+	char *sql_ins_up = NULL;
+
+	gaim_debug_info("cap", "%s\n", sql_select);
+
+	sqlite3_prepare(_db, sql_select, -1, &stmt, &tail);
+	
+	rc = sqlite3_step(stmt);
+
+	if(rc == SQLITE_DONE) {
+		sql_ins_up = sqlite3_mprintf("INSERT INTO cap_status_count VALUES (%Q, %Q, %Q, %Q, %d, %d);",
+			buddy_name, account, protocol, status_id, 0, 1);
+	} else if(rc == SQLITE_ROW) {
+		sql_ins_up = sqlite3_mprintf("UPDATE cap_status_count SET failed_count=failed_count+1 WHERE "
+			"buddy=%Q AND account=%Q AND protocol=%Q AND status=%Q;",
+			buddy_name, account, protocol, status_id);
+	} else {
+		gaim_debug_info("cap", "%d\n", rc);
+		sqlite3_finalize(stmt);
+		sqlite3_free(sql_select);
+		return;
+	}
+
+	sqlite3_finalize(stmt);
+	sqlite3_free(sql_select);
+
+	sqlite3_exec(_db, sql_ins_up, NULL, NULL, NULL);
+	sqlite3_free(sql_ins_up);
+}
+
+static void insert_cap_success(CapStatistics *stats) {
+	gchar *buddy_name = stats->buddy->name;
+	const gchar *protocol_id = gaim_account_get_protocol_id(stats->buddy->account);
+	const gchar *account_id = gaim_account_get_username(stats->buddy->account);
+	const gchar *status_id = (stats->last_message_status_id) ?
+		stats->last_message_status_id :
+		gaim_status_get_id(get_status_for(stats->buddy));
+	struct tm *current_time;
+	int minute;
+	
+	if(stats->last_message == -1) {
+		time_t now = time(NULL);
+		current_time = localtime(&now);
+	} else {
+		current_time = localtime(&stats->last_message);
+	}
+	minute = current_time->tm_min + current_time->tm_hour * 60;
+
+	insert_cap_msg_count_success(buddy_name, account_id, protocol_id, minute);
+	
+	insert_cap_status_count_success(buddy_name, account_id, protocol_id, status_id);
+
+	stats->last_message = -1;
+	stats->last_message_status_id = NULL;
+}
+
+static void insert_cap_failure(CapStatistics *stats) {
+	gchar *buddy_name = stats->buddy->name;
+	const gchar *protocol_id = gaim_account_get_protocol_id(stats->buddy->account);
+	const gchar *account_id = gaim_account_get_username(stats->buddy->account);
+	const gchar *status_id = (stats->last_message_status_id) ?
+		stats->last_message_status_id :
+		gaim_status_get_id(get_status_for(stats->buddy));
+	struct tm *current_time = localtime(&stats->last_message);
+	int minute = current_time->tm_min + current_time->tm_hour * 60;
+
+	insert_cap_msg_count_failed(buddy_name, account_id, protocol_id, minute);
+	
+	insert_cap_status_count_failed(buddy_name, account_id, protocol_id, status_id);
+	
+	stats->last_message = -1;
+	stats->last_message_status_id = NULL;
+}
+
+static gboolean max_message_difference_cb(gpointer data) {
+	CapStatistics *stats = data;
+	gaim_debug_info("cap", "Max Message Difference timeout occured\n");
+	insert_cap_failure(stats);
+	stats->timeout_source_id = 0;
+	return FALSE;
+}
+
+/* Gaim Signal Handlers */
+
+/* sent-im-msg */
+static void sent_im_msg(GaimAccount *account, const char *receiver, const char *message) {
+	GaimBuddy *buddy;
+	guint interval, words;
+
+	buddy = gaim_find_buddy(account, receiver);
+
+	if (buddy == NULL)
+		return;
+
+	interval = gaim_prefs_get_int("/plugins/gtk/cap/max_msg_difference") * 1000 * 60;
+	words = word_count(message);
+
+	CapStatistics *stats = get_stats_for(buddy);
+
+	insert_word_count(gaim_account_get_username(account), receiver, words);
+	stats->last_message = time(NULL);
+	stats->last_message_status_id = gaim_status_get_id(get_status_for(buddy));
+	if(stats->timeout_source_id != 0)
+		g_source_remove(stats->timeout_source_id);
+
+	stats->timeout_source_id = g_timeout_add(interval, max_message_difference_cb, stats);
+}
+
+/* received-im-msg */
+static void
+received_im_msg(GaimAccount *account, char *sender, char *message, GaimConversation *conv, GaimMessageFlags flags) {
+	GaimBuddy *buddy;
+	CapStatistics *stats;
+	/* guint words = word_count(message); */
+
+	buddy = gaim_find_buddy(account, sender);
+
+	if (buddy == NULL)
+		return;
+
+	stats = get_stats_for(buddy);
+
+	/* insert_word_count(sender, buddy_name, words); */
+	
+	/* If we are waiting for a response from a prior message
+	 * then cancel the timeout callback. */
+	if(stats->timeout_source_id != 0) {
+		gaim_debug_info("cap", "Cancelling timeout callback\n");
+		g_source_remove(stats->timeout_source_id);
+		stats->timeout_source_id = 0;
+	}
+
+	insert_cap_success(stats);
+
+	/* Reset the last_message value */
+	stats->last_message = -1;
+	/* Reset the last status id value */
+	stats->last_message_status_id = NULL;
+}
+
+/* buddy-status-changed */
+static void buddy_status_changed(GaimBuddy *buddy, GaimStatus *old_status, GaimStatus *status) {
+	CapStatistics *stats = get_stats_for(buddy);
+	insert_status_change_from_gaim_status(stats, status);
+}
+
+/* buddy-signed-on */
+static void buddy_signed_on(GaimBuddy *buddy) {
+	CapStatistics *stats = get_stats_for(buddy);
+	
+	/* If the statistic object existed but doesn't have a buddy pointer associated
+	 * with it then reassociate one with it. The pointer being null is a result
+	 * of a buddy with existing stats signing off and Gaim sticking around. */
+	if(!stats->buddy) {
+		stats->buddy = buddy;
+	}
+
+	insert_status_change(stats);
+}
+
+/* buddy-signed-off */
+static void buddy_signed_off(GaimBuddy *buddy) {
+	CapStatistics *stats = get_stats_for(buddy);
+
+	/* We don't necessarily want to delete a buddies generated statistics every time they go offline.
+	 * Instead we just set the buddy pointer to null so that when they come back online we can look
+	 * them up again and continue using their statistics.
+	 */
+	insert_status_change(stats);
+	/* stats->buddy = NULL; */
+	stats->last_seen = time(NULL);
+}
+
+static void buddy_idle(GaimBuddy *buddy, gboolean old_idle, gboolean idle) {
+}
+
+static void blist_node_extended_menu(GaimBlistNode *node, GList **menu) {
+	GaimBuddy *buddy;
+	GaimMenuAction *menu_action;
+	gaim_debug_info("cap", "got extended blist menu\n");
+	gaim_debug_info("cap", "is buddy: %d\n", GAIM_BLIST_NODE_IS_BUDDY(node));
+	gaim_debug_info("cap", "is contact: %d\n", GAIM_BLIST_NODE_IS_CONTACT(node));
+	gaim_debug_info("cap", "is group: %d\n", GAIM_BLIST_NODE_IS_GROUP(node));
+	/* Probably only concerned with buddy/contact types. Contacts = meta-buddies (grouped msn/jabber/etc.) */
+	g_return_if_fail(GAIM_BLIST_NODE_IS_BUDDY(node));
+	buddy = (GaimBuddy *)node;
+	menu_action = gaim_menu_action_new(_("Display Statistics"),
+			GAIM_CALLBACK(display_statistics_action_cb), NULL, NULL);
+	*menu = g_list_append(*menu, menu_action);
+}
+
+/* drawing-tooltip */
+static void drawing_tooltip(GaimBlistNode *node, GString *text, gboolean full) {
+	if(node->type == GAIM_BLIST_BUDDY_NODE) {
+		GaimBuddy *buddy = (GaimBuddy *)node;
+		CapStatistics *stats = get_stats_for(buddy);
+		/* get the probability that this buddy will respond and add to the tooltip */
+		if(stats->prediction->probability >= 0.0) {
+			g_string_append_printf(text, "\n<b>%s</b> %3.0f %%", _("Response Probability:"),
+				100 * stats->prediction->probability);
+		} else {
+			g_string_append_printf(text, "\n<b>%s</b> ???", _("Response Probability:"));
+		}
+	}
+}
+
+/* signed-on */
+static void signed_on(GaimConnection *gc) {
+	GaimAccount *account = gaim_connection_get_account(gc);
+	const char *my_gaim_name = gaim_account_get_username(account);
+	gchar *my_name = g_strdup(my_gaim_name);
+	time_t *last_offline = g_hash_table_lookup(_my_offline_times, my_name);
+
+	const gchar *account_id = gaim_account_get_username(account);
+	const gchar *protocol_id = gaim_account_get_protocol_id(account);
+	char *sql;
+
+	sql = sqlite3_mprintf("insert into cap_my_usage values(%Q, %Q, %d, now());", account_id, protocol_id, 1);
+	sqlite3_exec(_db, sql, NULL, NULL, NULL);
+	sqlite3_free(sql);
+
+	if(last_offline) {
+		if(difftime(*last_offline, time(NULL)) > gaim_prefs_get_int("/plugins/gtk/cap/max_seen_difference") * 60) {
+			/* reset all of the last_message times to -1 */
+			g_hash_table_foreach(_my_offline_times, reset_all_last_message_times, NULL);
+		}
+		g_hash_table_remove(_my_offline_times, my_name);
+	}
+	g_free(my_name);
+}
+
+/* signed-off */
+static void signed_off(GaimConnection *gc) {
+	/* Here we record the time you (the user) sign off of an account.
+	 * The account username is the key in the hashtable and the sign off time_t
+	 * (equal to the sign off time) is the value. */
+	GaimAccount *account = gaim_connection_get_account(gc);
+	const char *my_gaim_name = gaim_account_get_username(account);
+	gchar *my_name = g_strdup(my_gaim_name);
+	time_t *offline_time = g_malloc(sizeof(time_t));
+	const gchar *account_id = gaim_account_get_username(account);
+	const gchar *protocol_id = gaim_account_get_protocol_id(account);
+	char *sql;
+
+	sql = sqlite3_mprintf("insert into cap_my_usage values(%Q, %Q, %d, now());", account_id, protocol_id, 0);
+	sqlite3_exec(_db, sql, NULL, NULL, NULL);
+	sqlite3_free(sql);
+
+	time(offline_time);
+	g_hash_table_insert(_my_offline_times, my_name, offline_time);
+}
+
+static void reset_all_last_message_times(gpointer key, gpointer value, gpointer user_data) {
+	CapStatistics *stats = value;
+	stats->last_message = -1;
+}
+
+static GaimStatus * get_status_for(GaimBuddy *buddy) {
+	GaimPresence *presence = gaim_buddy_get_presence(buddy);
+	GaimStatus *status = gaim_presence_get_active_status(presence);
+	return status;
+}
+
+static void create_tables() {
+	int rc;
+	rc = sqlite3_exec(_db,
+		"CREATE TABLE IF NOT EXISTS cap_status ("
+		"	buddy varchar(60) not null,"
+		"	account varchar(60) not null,"
+		"	protocol varchar(60) not null,"
+		"	status varchar(60) not null,"
+		"	event_time datetime not null,"
+		"	primary key (buddy, account, protocol, event_time)"
+		");",
+		NULL, NULL, NULL);
+
+	rc = sqlite3_exec(_db,
+		"create table if not exists cap_message ("
+		"	sender varchar(60) not null,"
+		"	receiver varchar(60) not null,"
+		"	account varchar(60) not null,"
+		"	protocol varchar(60) not null,"
+		"	word_count integer not null,"
+		"	event_time datetime not null,"
+		"	primary key (sender, account, protocol, receiver, event_time)"
+		");",
+		NULL, NULL, NULL);
+
+	rc = sqlite3_exec(_db,
+		"create table if not exists cap_msg_count ("
+		"	buddy varchar(60) not null,"
+		"	account varchar(60) not null,"
+		"	protocol varchar(60) not null,"
+		"	minute_val int not null,"
+		"	success_count int not null,"
+		"	failed_count int not null,"
+		"	primary key (buddy, account, protocol, minute_val)"
+		");",
+	NULL, NULL, NULL);
+
+	rc = sqlite3_exec(_db,
+		"create table if not exists cap_status_count ("
+		"	buddy varchar(60) not null,"
+		"	account varchar(60) not null,"
+		"	protocol varchar(60) not null,"
+		"	status varchar(60) not null,"
+		"	success_count int not null,"
+		"	failed_count int not null,"
+		"	primary key (buddy, account, protocol, status)"
+		");",
+	NULL, NULL, NULL);
+
+	rc = sqlite3_exec(_db,
+		"create table if not exists cap_my_usage ("
+		"	account varchar(60) not null,"
+		"	protocol varchar(60) not null,"
+		"	online tinyint not null,"
+		"	event_time datetime not null,"
+		"	primary key(account, protocol, online, event_time)"
+		");",
+	NULL, NULL, NULL);
+}
+
+static gboolean create_database_connection() {
+	gchar *path;
+	int rc;
+
+	if(_db)
+		return TRUE;
+
+	/* build the path */
+	path = g_build_filename(gaim_user_dir(), "cap.db", (gchar *)NULL);
+
+	/* make database connection here */
+	rc = sqlite3_open(path, &_db);
+	g_free(path);
+	if(rc != SQLITE_OK)
+		return FALSE;
+	
+	/* Add tables here */
+	create_tables();
+	gaim_debug_info("cap", "Database connection successfully made.\n");
+	return TRUE;
+}
+static void destroy_database_connection() {
+	if(_db)
+		sqlite3_close(_db);
+
+	_db = NULL;
+}
+
+static guint word_count(const gchar *string) {
+	/*TODO: doesn't really work, should use regex instead (#include <regex.h>)*/
+	gchar **result = g_strsplit_set(string, " ", -1);
+	guint count = g_strv_length(result);
+
+	g_strfreev(result);
+
+	return count;
+}
+
+static void insert_status_change(CapStatistics *statistics) {
+	insert_status_change_from_gaim_status(statistics, get_status_for(statistics->buddy));
+}
+
+static void insert_status_change_from_gaim_status(CapStatistics *statistics, GaimStatus *status) {
+	char *sql;
+	int rc;
+	const gchar *status_id;
+	const gchar *buddy_name;
+	const gchar *protocol_id;
+	const gchar *account_id;
+
+	/* It would seem that some protocols receive periodic updates of the buddies status.
+	 * Check to make sure the last status is not the same as current status to prevent
+	 * to many duplicated useless database entries. */
+	if(strcmp(statistics->last_status_id, gaim_status_get_id(status)) == 0)
+		return;
+
+	status_id = gaim_status_get_id(status);
+	buddy_name = statistics->buddy->name;
+	protocol_id = gaim_account_get_protocol_id(statistics->buddy->account);
+	account_id = gaim_account_get_username(statistics->buddy->account);
+
+	statistics->last_status_id = gaim_status_get_id(status);
+
+	gaim_debug_info("cap", "Executing: insert into cap_status (buddy, account, protocol, status, event_time) values(%s, %s, %s, %s, now());\n", buddy_name, account_id, protocol_id, status_id);
+
+	sql = sqlite3_mprintf("insert into cap_status values (%Q, %Q, %Q, %Q, now());", buddy_name, account_id, protocol_id, status_id);
+	rc = sqlite3_exec(_db, sql, NULL, NULL, NULL);
+	sqlite3_free(sql);
+}
+
+static void insert_word_count(const char *sender, const char *receiver, guint count) {
+	/* TODO! */
+	/* dbi_result result; */
+	/* result = dbi_conn_queryf(_conn, "insert into cap_message values(\'%s\', \'%s\', %d, now());", sender, receiver, count); */
+}
+
+/* Callbacks */
+void display_statistics_action_cb(GaimBlistNode *node, gpointer data) {
+	GaimBuddy *buddy;
+
+	g_return_if_fail(GAIM_BLIST_NODE_IS_BUDDY(node));
+	buddy = (GaimBuddy *)node;
+	gaim_debug_info("cap", "Statistics for %s requested.\n", buddy->name);
+}
+
+/* Gaim plugin specific code */
+
+static gboolean plugin_load(GaimPlugin *plugin) {
+	_plugin_pointer = plugin;
+	_signals_connected = FALSE;
+
+	/* buddy_stats is a hashtable where strings are keys
+	 * and the keys are a buddies account id (GaimBuddy.name).
+	 * keys/values are automatically deleted */
+	_buddy_stats = g_hash_table_new_full(g_str_hash, g_str_equal, g_free, destroy_stats);
+	
+	/* ? - Can't remember at the moment
+	 */
+	_my_offline_times = g_hash_table_new_full(g_str_hash, g_str_equal, g_free, g_free);
+
+	if(create_database_connection()) {
+		add_plugin_functionality(plugin);
+	}
+	return TRUE;
+}
+
+static void add_plugin_functionality(GaimPlugin *plugin) {
+	if(_signals_connected)
+		return;
+
+	gaim_debug_info("cap", "Adding plugin functionality.\n");
+	
+	/* Connect all the signals */
+	gaim_signal_connect(gaim_conversations_get_handle(), "sent-im-msg", plugin,
+			GAIM_CALLBACK(sent_im_msg), NULL);
+
+	gaim_signal_connect(gaim_conversations_get_handle(), "received-im-msg", plugin,
+			GAIM_CALLBACK(received_im_msg), NULL);
+
+	gaim_signal_connect(gaim_blist_get_handle(), "buddy-status-changed", plugin,
+			GAIM_CALLBACK(buddy_status_changed), NULL);
+
+	gaim_signal_connect(gaim_blist_get_handle(), "buddy-signed-on", plugin,
+			GAIM_CALLBACK(buddy_signed_on), NULL);
+
+	gaim_signal_connect(gaim_blist_get_handle(), "buddy-signed-off", plugin,
+			GAIM_CALLBACK(buddy_signed_off), NULL);
+
+	/*gaim_signal_connect(gaim_blist_get_handle(), "blist-node-extended-menu", plugin,
+			GAIM_CALLBACK(blist_node_extended_menu), NULL);*/
+
+	gaim_signal_connect(gaim_gtk_blist_get_handle(), "drawing-tooltip", plugin,
+			GAIM_CALLBACK(drawing_tooltip), NULL);
+
+	gaim_signal_connect(gaim_connections_get_handle(), "signed-on", plugin,
+			GAIM_CALLBACK(signed_on), NULL);
+	
+	gaim_signal_connect(gaim_connections_get_handle(), "signed-off", plugin,
+			GAIM_CALLBACK(signed_off), NULL);
+
+	gaim_signal_connect(gaim_blist_get_handle(), "buddy-idle-changed", plugin,
+			GAIM_CALLBACK(buddy_idle), NULL);
+
+	_signals_connected = TRUE;
+}
+
+static void cancel_conversation_timeouts(gpointer key, gpointer value, gpointer user_data) {
+	CapStatistics *stats = value;
+	if(stats->timeout_source_id != 0) {
+		g_source_remove(stats->timeout_source_id);
+		stats->timeout_source_id = 0;
+	}
+}
+
+static void remove_plugin_functionality(GaimPlugin *plugin) {
+	if(!_signals_connected)
+		return;
+
+	gaim_debug_info("cap", "Removing plugin functionality.\n");
+
+	/* If there are any timeouts waiting to be processed then cancel them */
+	g_hash_table_foreach(_buddy_stats, cancel_conversation_timeouts, NULL);
+	
+	/* Connect all the signals */
+	gaim_signal_disconnect(gaim_conversations_get_handle(), "sent-im-msg", plugin,
+			GAIM_CALLBACK(sent_im_msg));
+
+	gaim_signal_disconnect(gaim_conversations_get_handle(), "received-im-msg", plugin,
+			GAIM_CALLBACK(received_im_msg));
+
+	gaim_signal_disconnect(gaim_blist_get_handle(), "buddy-status-changed", plugin,
+			GAIM_CALLBACK(buddy_status_changed));
+
+	gaim_signal_disconnect(gaim_blist_get_handle(), "buddy-signed-on", plugin,
+			GAIM_CALLBACK(buddy_signed_on));
+
+	gaim_signal_disconnect(gaim_blist_get_handle(), "buddy-signed-off", plugin,
+			GAIM_CALLBACK(buddy_signed_off));
+
+	/*gaim_signal_disconnect(gaim_blist_get_handle(), "blist-node-extended-menu", plugin,
+			GAIM_CALLBACK(blist_node_extended_menu));*/
+
+	gaim_signal_disconnect(gaim_gtk_blist_get_handle(), "drawing-tooltip", plugin,
+			GAIM_CALLBACK(drawing_tooltip));
+
+	gaim_signal_disconnect(gaim_connections_get_handle(), "signed-on", plugin,
+			GAIM_CALLBACK(signed_on));
+	
+	gaim_signal_disconnect(gaim_connections_get_handle(), "signed-off", plugin,
+			GAIM_CALLBACK(signed_off));
+
+	gaim_signal_disconnect(gaim_blist_get_handle(), "buddy-idle-changed", plugin,
+			GAIM_CALLBACK(buddy_idle));
+
+	_signals_connected = FALSE;
+}
+
+static void write_stats_on_unload(gpointer key, gpointer value, gpointer user_data) {
+	CapStatistics *stats = value;
+	if(stats->last_message != -1 && stats->buddy != NULL) {
+		insert_cap_failure(stats);
+	}
+}
+
+static gboolean plugin_unload(GaimPlugin *plugin) {
+	gaim_debug_info("cap", "CAP plugin unloading\n");
+	
+	/* clean up memory allocations */
+	if(_buddy_stats) {
+		g_hash_table_foreach(_buddy_stats, write_stats_on_unload, NULL);
+		g_hash_table_destroy(_buddy_stats);
+	}
+	 
+	 /* close database connection */
+	 destroy_database_connection();
+
+	return TRUE;
+}
+
+static CapPrefsUI * create_cap_prefs_ui() {
+	CapPrefsUI *ui = g_malloc(sizeof(CapPrefsUI));
+
+	ui->ret = gtk_vbox_new(FALSE, 18);
+	gtk_container_set_border_width(GTK_CONTAINER(ui->ret), 10);
+	ui->cap_vbox = gaim_gtk_make_frame(ui->ret, _("Statistics Configuration"));
+
+	/* msg_difference spinner */
+	ui->msg_difference_label = gtk_label_new(_("Maximum response timeout:"));
+	gtk_misc_set_alignment(GTK_MISC(ui->msg_difference_label), 0, 0.5);
+	ui->msg_difference_input = gtk_spin_button_new_with_range(1, 1440, 1);
+	ui->msg_difference_minutes_label = gtk_label_new(_("minutes"));
+	gtk_misc_set_alignment(GTK_MISC(ui->msg_difference_minutes_label), 0, 0.5);
+
+	/* last_seen spinner */
+	ui->last_seen_label = gtk_label_new(_("Maximum last-seen difference:"));
+	gtk_misc_set_alignment(GTK_MISC(ui->last_seen_label), 0, 0.5);
+	ui->last_seen_input = gtk_spin_button_new_with_range(1, 1440, 1);
+	ui->last_seen_minutes_label = gtk_label_new(_("minutes"));
+	gtk_misc_set_alignment(GTK_MISC(ui->last_seen_minutes_label), 0, 0.5);
+
+	/* threshold spinner */
+	ui->threshold_label = gtk_label_new(_("Threshold:"));
+	gtk_misc_set_alignment(GTK_MISC(ui->threshold_label), 0, 0.5);
+	ui->threshold_input = gtk_spin_button_new_with_range(1, 1440, 1);
+	ui->threshold_minutes_label = gtk_label_new(_("minutes"));
+	gtk_misc_set_alignment(GTK_MISC(ui->threshold_minutes_label), 0, 0.5);
+
+	/* Layout threshold/last-seen/response-timeout input items */
+	ui->table_layout = gtk_table_new(3, 3, FALSE);
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->threshold_label, 0, 1, 0, 1,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->threshold_input, 1, 2, 0, 1,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->threshold_minutes_label, 2, 3, 0, 1,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->msg_difference_label, 0, 1, 1, 2,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->msg_difference_input, 1, 2, 1, 2,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->msg_difference_minutes_label, 2, 3, 1, 2,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->last_seen_label, 0, 1, 2,3,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->last_seen_input, 1, 2, 2, 3,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+	gtk_table_attach(GTK_TABLE(ui->table_layout), ui->last_seen_minutes_label, 2, 3, 2, 3,
+		(GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
+		(GtkAttachOptions) (0), 0, 0);
+
+
+	/* Config window - lay it out */
+	gtk_box_pack_start(GTK_BOX(ui->cap_vbox), ui->table_layout, FALSE, FALSE, 0);
+
+	/* Set the input areas to contain the configuration values from
+	 * gaim prefs.
+	 */
+	if(gaim_prefs_exists("/plugins/gtk/cap/max_msg_difference")) {
+		int max_msg_diff = gaim_prefs_get_int("/plugins/gtk/cap/max_msg_difference");
+		gtk_spin_button_set_value(GTK_SPIN_BUTTON(ui->msg_difference_input),  max_msg_diff);
+	}
+	if(gaim_prefs_exists("/plugins/gtk/cap/max_seen_difference")) {
+		int max_seen_diff = gaim_prefs_get_int("/plugins/gtk/cap/max_seen_difference");
+		gtk_spin_button_set_value(GTK_SPIN_BUTTON(ui->last_seen_input), max_seen_diff);
+	}
+	if(gaim_prefs_exists("/plugins/gtk/cap/threshold")) {
+		int threshold = gaim_prefs_get_int("/plugins/gtk/cap/threshold");
+		gtk_spin_button_set_value(GTK_SPIN_BUTTON(ui->threshold_input), threshold);
+	}
+
+	/* Add the signals */
+	g_signal_connect(G_OBJECT(ui->ret), "destroy",
+		G_CALLBACK(cap_prefs_ui_destroy_cb), ui);
+
+	g_signal_connect(G_OBJECT(ui->msg_difference_input), "value-changed",
+		G_CALLBACK(numeric_spinner_prefs_cb), "/plugins/gtk/cap/max_msg_difference");
+	
+	g_signal_connect(G_OBJECT(ui->last_seen_input), "value-changed",
+		G_CALLBACK(numeric_spinner_prefs_cb), "/plugins/gtk/cap/max_seen_difference");
+	
+	g_signal_connect(G_OBJECT(ui->threshold_input), "value-changed",
+		G_CALLBACK(numeric_spinner_prefs_cb), "/plugins/gtk/cap/threshold");
+	
+	return ui;
+}
+
+static void cap_prefs_ui_destroy_cb(GtkObject *object, gpointer user_data) {
+	CapPrefsUI *ui = user_data;
+	if(_db) {
+		add_plugin_functionality(_plugin_pointer);
+	}
+	g_free(ui);
+}
+
+static void numeric_spinner_prefs_cb(GtkSpinButton *spinbutton, gpointer user_data) {
+	gaim_prefs_set_int(user_data, gtk_spin_button_get_value_as_int(spinbutton));
+}
+
+static GaimGtkPluginUiInfo ui_info = {
+	get_config_frame,
+	0 /* page_num (reserved) */
+};
+
+static GaimPluginInfo info = {
+	GAIM_PLUGIN_MAGIC,
+	GAIM_MAJOR_VERSION,
+	GAIM_MINOR_VERSION,
+	GAIM_PLUGIN_STANDARD,							/**< type		*/
+	GAIM_GTK_PLUGIN_TYPE,							/**< ui_requirement */
+	0,												/**< flags		*/
+	NULL,											/**< dependencies   */
+	GAIM_PRIORITY_DEFAULT,							/**< priority		*/
+	CAP_PLUGIN_ID,									/**< id			*/
+	N_("Contact Availability Prediction"),				/**< name		*/
+	VERSION,										/**< version		*/
+	N_("Contact Availability Prediction plugin."),	/**  summary		*/
+	N_("The contact availability plugin (cap) is used to display statistical information about buddies in a users contact list."),
+	/**  description	*/
+	"Geoffrey Foster <[e-mail]>",	/**< author		*/
+	GAIM_WEBSITE,									/**< homepage		*/
+	plugin_load,									/**< load		*/
+	plugin_unload,									/**< unload		*/
+	NULL,											/**< destroy		*/
+	&ui_info,										/**< ui_info		*/
+	NULL,											/**< extra_info	 */
+	NULL,											/**< prefs_info		*/
+	NULL
+};
+
+static GtkWidget * get_config_frame(GaimPlugin *plugin) {
+	CapPrefsUI *ui = create_cap_prefs_ui();
+
+	/*
+	 * Prevent database stuff from occuring since we are editing values
+	 */
+	remove_plugin_functionality(_plugin_pointer);
+
+	return ui->ret;
+}
+
+static void init_plugin(GaimPlugin *plugin) {
+	gaim_prefs_add_none("/plugins/gtk/cap");
+	gaim_prefs_add_int("/plugins/gtk/cap/max_seen_difference", 1);
+	gaim_prefs_add_int("/plugins/gtk/cap/max_msg_difference", 10);
+	gaim_prefs_add_int("/plugins/gtk/cap/threshold", 5);
+}
+
+GAIM_INIT_PLUGIN(cap, init_plugin, info);