/*
 *  Copyright (C) 2002 Marco Pesenti Gritti
 *
 *  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, 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 "ephy-types.h"
#include "ephy-history.h"
#include "ephy-file-helpers.h"
#include "ephy-autocompletion-source.h"
#include "ephy-debug.h"

#include <time.h>
#include <string.h>
#include <libgnome/gnome-i18n.h>
#include <libgnomevfs/gnome-vfs-uri.h>

#define EPHY_HISTORY_XML_VERSION "0.1"

/* how often to save the history, in milliseconds */
#define HISTORY_SAVE_INTERVAL (60 * 5 * 1000)

#define HISTORY_PAGE_OBSOLETE_DAYS 30

struct EphyHistoryPrivate
{
	char *xml_file;
	EphyNode *hosts;
	EphyNode *pages;
	EphyNode *last_page;
	GHashTable *hosts_hash;
	GStaticRWLock *hosts_hash_lock;
	GHashTable *pages_hash;
	GStaticRWLock *pages_hash_lock;
	int autosave_timeout;
};

enum
{
        ADD,
	UPDATE,
	REMOVE,
	VISITED,
        LAST_SIGNAL
};

static void
ephy_history_class_init (EphyHistoryClass *klass);
static void
ephy_history_init (EphyHistory *tab);
static void
ephy_history_finalize (GObject *object);
static void
ephy_history_autocompletion_source_init (EphyAutocompletionSourceIface *iface);

static GObjectClass *parent_class = NULL;

static guint ephy_history_signals[LAST_SIGNAL] = { 0 };

GType
ephy_history_get_type (void)
{
        static GType ephy_history_type = 0;

        if (ephy_history_type == 0)
        {
                static const GTypeInfo our_info =
                {
                        sizeof (EphyHistoryClass),
                        NULL, /* base_init */
                        NULL, /* base_finalize */
                        (GClassInitFunc) ephy_history_class_init,
                        NULL,
                        NULL, /* class_data */
                        sizeof (EphyHistory),
                        0, /* n_preallocs */
                        (GInstanceInitFunc) ephy_history_init
                };

		static const GInterfaceInfo autocompletion_source_info =
		{
			(GInterfaceInitFunc) ephy_history_autocompletion_source_init,
			NULL,
			NULL
		};

                ephy_history_type = g_type_register_static (G_TYPE_OBJECT,
							      "EphyHistory",
							      &our_info, 0);

		g_type_add_interface_static (ephy_history_type,
					     EPHY_TYPE_AUTOCOMPLETION_SOURCE,
					     &autocompletion_source_info);
        }

        return ephy_history_type;
}

static void
ephy_history_autocompletion_source_set_basic_key (EphyAutocompletionSource *source,
						  const gchar *basic_key)
{
	/* nothing to do here */
}

static void
ephy_history_autocompletion_source_foreach (EphyAutocompletionSource *source,
					    const gchar *current_text,
					    EphyAutocompletionSourceForeachFunc func,
					    gpointer data)
{
	GPtrArray *children;
	int i;
	EphyHistory *eb = EPHY_HISTORY (source);
	GTime now;

	now = time (NULL);

	children = ephy_node_get_children (eb->priv->pages);
	for (i = 0; i < children->len; i++)
	{
		EphyNode *kid;
		const char *url, *title;
		int last_visit, visits;
		guint32 score;

		kid = g_ptr_array_index (children, i);
		g_assert (EPHY_IS_NODE (kid));

		url = ephy_node_get_property_string
			(kid, EPHY_NODE_PAGE_PROP_LOCATION);
		title = ephy_node_get_property_string
			(kid, EPHY_NODE_PAGE_PROP_TITLE);
		last_visit = ephy_node_get_property_int
			(kid, EPHY_NODE_PAGE_PROP_LAST_VISIT);
		visits = ephy_node_get_property_int
			(kid, EPHY_NODE_PAGE_PROP_VISITS);
		score = MAX (visits - ((now - last_visit) >> 15), 1);

		func (source, url,
		      url, url, FALSE,
		      FALSE, score, data);
	}
	ephy_node_thaw (eb->priv->pages);
}

static void
ephy_history_emit_data_changed (EphyHistory *eb)
{
	g_signal_emit_by_name (eb, "data-changed");
}

static void
ephy_history_autocompletion_source_init (EphyAutocompletionSourceIface *iface)
{
	iface->foreach = ephy_history_autocompletion_source_foreach;
	iface->set_basic_key = ephy_history_autocompletion_source_set_basic_key;
}

static void
ephy_history_class_init (EphyHistoryClass *klass)
{
        GObjectClass *object_class = G_OBJECT_CLASS (klass);

        parent_class = g_type_class_peek_parent (klass);

        object_class->finalize = ephy_history_finalize;

	ephy_history_signals[VISITED] =
                g_signal_new ("visited",
                              G_OBJECT_CLASS_TYPE (object_class),
                              G_SIGNAL_RUN_FIRST,
                              G_STRUCT_OFFSET (EphyHistoryClass, visited),
                              NULL, NULL,
                              g_cclosure_marshal_VOID__STRING,
                              G_TYPE_NONE,
                              1,
			      G_TYPE_STRING);
}

static void
ephy_history_load (EphyHistory *eb)
{
	xmlDocPtr doc;
	xmlNodePtr root, child;
	char *tmp;

	if (g_file_test (eb->priv->xml_file, G_FILE_TEST_EXISTS) == FALSE)
		return;

	doc = xmlParseFile (eb->priv->xml_file);
	g_assert (doc != NULL);

	root = xmlDocGetRootElement (doc);

	tmp = xmlGetProp (root, "version");
	g_assert (tmp != NULL && strcmp (tmp, EPHY_HISTORY_XML_VERSION) == 0);
	g_free (tmp);

	for (child = root->children; child != NULL; child = child->next)
	{
		EphyNode *node;

		node = ephy_node_new_from_xml (child);
	}

	xmlFreeDoc (doc);
}

static gboolean
page_is_obsolete (EphyNode *node, GDate *now)
{
	int last_visit;
	GDate date;

	last_visit = ephy_node_get_property_int
		(node, EPHY_NODE_PAGE_PROP_LAST_VISIT);

        g_date_clear (&date, 1);
        g_date_set_time (&date, last_visit);

	return (g_date_days_between (&date, now) >=
		HISTORY_PAGE_OBSOLETE_DAYS);
}

static void
remove_obsolete_pages (EphyHistory *eb)
{
	GPtrArray *children;
	int i;
	GTime now;
	GDate current_date;

	now = time (NULL);
        g_date_clear (&current_date, 1);
        g_date_set_time (&current_date, time (NULL));

	children = ephy_node_get_children (eb->priv->pages);
	ephy_node_thaw (eb->priv->pages);
	for (i = 0; i < children->len; i++)
	{
		EphyNode *kid;

		kid = g_ptr_array_index (children, i);

		if (page_is_obsolete (kid, &current_date))
		{
			ephy_history_remove (eb, kid);
		}
	}
}

static void
ephy_history_save (EphyHistory *eb)
{
	xmlDocPtr doc;
	xmlNodePtr root;
	GPtrArray *children;
	int i;

	LOG ("Saving history")

	/* save nodes to xml */
	xmlIndentTreeOutput = TRUE;
	doc = xmlNewDoc ("1.0");

	root = xmlNewDocNode (doc, NULL, "ephy_history", NULL);
	xmlSetProp (root, "version", EPHY_HISTORY_XML_VERSION);
	xmlDocSetRootElement (doc, root);

	children = ephy_node_get_children (eb->priv->hosts);
	for (i = 0; i < children->len; i++)
	{
		EphyNode *kid;

		kid = g_ptr_array_index (children, i);

		ephy_node_save_to_xml (kid, root);
	}
	ephy_node_thaw (eb->priv->hosts);

	children = ephy_node_get_children (eb->priv->pages);
	for (i = 0; i < children->len; i++)
	{
		EphyNode *kid;

		kid = g_ptr_array_index (children, i);

		ephy_node_save_to_xml (kid, root);
	}
	ephy_node_thaw (eb->priv->pages);

	xmlSaveFormatFile (eb->priv->xml_file, doc, 1);
	xmlFreeDoc(doc);
}

static void
hosts_added_cb (EphyNode *node,
	        EphyNode *child,
	        EphyHistory *eb)
{
	g_static_rw_lock_writer_lock (eb->priv->hosts_hash_lock);

	g_hash_table_insert (eb->priv->hosts_hash,
			     (char *) ephy_node_get_property_string (child, EPHY_NODE_PAGE_PROP_LOCATION),
			     child);

	g_static_rw_lock_writer_unlock (eb->priv->hosts_hash_lock);
}

static void
hosts_removed_cb (EphyNode *node,
		  EphyNode *child,
		  EphyHistory *eb)
{
	g_static_rw_lock_writer_lock (eb->priv->hosts_hash_lock);

	g_hash_table_remove (eb->priv->hosts_hash,
			     ephy_node_get_property_string (child, EPHY_NODE_PAGE_PROP_LOCATION));

	g_static_rw_lock_writer_unlock (eb->priv->hosts_hash_lock);
}

static void
pages_added_cb (EphyNode *node,
	        EphyNode *child,
	        EphyHistory *eb)
{
	g_static_rw_lock_writer_lock (eb->priv->pages_hash_lock);

	g_hash_table_insert (eb->priv->pages_hash,
			     (char *) ephy_node_get_property_string (child, EPHY_NODE_PAGE_PROP_LOCATION),
			     child);

	g_static_rw_lock_writer_unlock (eb->priv->pages_hash_lock);
}

static void
pages_removed_cb (EphyNode *node,
		  EphyNode *child,
		  EphyHistory *eb)
{
	g_static_rw_lock_writer_lock (eb->priv->pages_hash_lock);

	g_hash_table_remove (eb->priv->pages_hash,
			     ephy_node_get_property_string (child, EPHY_NODE_PAGE_PROP_LOCATION));

	g_static_rw_lock_writer_unlock (eb->priv->pages_hash_lock);
}

static gboolean
periodic_save_cb (EphyHistory *eh)
{
	remove_obsolete_pages (eh);
	ephy_history_save (eh);

	return TRUE;
}

static void
ephy_history_init (EphyHistory *eb)
{
        eb->priv = g_new0 (EphyHistoryPrivate, 1);

	eb->priv->xml_file = g_build_filename (ephy_dot_dir (),
					       "ephy-history.xml",
					       NULL);

	eb->priv->pages_hash = g_hash_table_new (g_str_hash,
			                          g_str_equal);
	eb->priv->pages_hash_lock = g_new0 (GStaticRWLock, 1);
	g_static_rw_lock_init (eb->priv->pages_hash_lock);

	eb->priv->hosts_hash = g_hash_table_new (g_str_hash,
			                         g_str_equal);
	eb->priv->hosts_hash_lock = g_new0 (GStaticRWLock, 1);
	g_static_rw_lock_init (eb->priv->hosts_hash_lock);

	/* Pages */
	eb->priv->pages = ephy_node_new_with_id (PAGES_NODE_ID);
	ephy_node_ref (eb->priv->pages);
	g_signal_connect_object (G_OBJECT (eb->priv->pages),
				 "child_added",
				 G_CALLBACK (pages_added_cb),
				 G_OBJECT (eb),
				 0);
	g_signal_connect_object (G_OBJECT (eb->priv->pages),
				 "child_removed",
				 G_CALLBACK (pages_removed_cb),
				 G_OBJECT (eb),
				 0);

	/* Hosts */
	eb->priv->hosts = ephy_node_new_with_id (HOSTS_NODE_ID);
	ephy_node_ref (eb->priv->hosts);
	g_signal_connect_object (G_OBJECT (eb->priv->hosts),
				 "child_added",
				 G_CALLBACK (hosts_added_cb),
				 G_OBJECT (eb),
				 0);
	g_signal_connect_object (G_OBJECT (eb->priv->hosts),
				 "child_removed",
				 G_CALLBACK (hosts_removed_cb),
				 G_OBJECT (eb),
				 0);

	ephy_history_load (eb);
	ephy_history_emit_data_changed (eb);

	/* setup the periodic history saving callback */
	eb->priv->autosave_timeout =
		g_timeout_add (HISTORY_SAVE_INTERVAL,
		       (GSourceFunc)periodic_save_cb,
		       eb);
}

static void
ephy_history_finalize (GObject *object)
{
        EphyHistory *eb;

	g_return_if_fail (IS_EPHY_HISTORY (object));

	eb = EPHY_HISTORY (object);

        g_return_if_fail (eb->priv != NULL);

	ephy_history_save (eb);

	ephy_node_unref (eb->priv->pages);
	ephy_node_unref (eb->priv->hosts);

	g_hash_table_destroy (eb->priv->pages_hash);
	g_static_rw_lock_free (eb->priv->pages_hash_lock);
	g_hash_table_destroy (eb->priv->hosts_hash);
	g_static_rw_lock_free (eb->priv->hosts_hash_lock);

	g_source_remove (eb->priv->autosave_timeout);

        g_free (eb->priv);

	LOG ("Global history finalized");

	G_OBJECT_CLASS (parent_class)->finalize (object);
}

EphyHistory *
ephy_history_new ()
{
	EphyHistory *tab;

	tab = EPHY_HISTORY (g_object_new (EPHY_HISTORY_TYPE, NULL));

	return tab;
}

static void
ephy_history_host_set_title (EphyHistory *eh,
			     EphyNode *host,
			     EphyNode *page,
			     const char *title)
{
	const char *real_url;
	const char *host_url;
	GValue value = { 0, };

	real_url = ephy_node_get_property_string
		(page, EPHY_NODE_PAGE_PROP_LOCATION);
	host_url = ephy_node_get_property_string
		(host, EPHY_NODE_PAGE_PROP_LOCATION);

	if (real_url && host_url &&
	    strcmp (real_url, host_url) == 0)
	{

		g_value_init (&value, G_TYPE_STRING);
		g_value_set_string (&value, title);
		ephy_node_set_property (host, EPHY_NODE_PAGE_PROP_TITLE,
					&value);
		g_value_unset (&value);
	}
}

static void
ephy_history_host_visited (EphyHistory *eh,
			   EphyNode *host,
			   GTime now)
{
	GValue value = { 0, };
	int visits;

	LOG ("Host visited")

	visits = ephy_node_get_property_int
		(host, EPHY_NODE_PAGE_PROP_VISITS);
	if (visits < 0) visits = 0;
	visits++;

	g_value_init (&value, G_TYPE_INT);
	g_value_set_int (&value, visits);
	ephy_node_set_property (host, EPHY_NODE_PAGE_PROP_VISITS,
			        &value);
	g_value_unset (&value);

	g_value_init (&value, G_TYPE_INT);
	g_value_set_int (&value, now);
	ephy_node_set_property (host, EPHY_NODE_PAGE_PROP_LAST_VISIT,
			        &value);
	g_value_unset (&value);
}

static EphyNode *
ephy_history_add_host (EphyHistory *eh, EphyNode *page)
{
	GnomeVFSURI *vfs_uri = NULL;
	EphyNode *host = NULL;
	const char *host_name = NULL;
	GList *host_locations = NULL, *l;
	GValue value = { 0, };
	const char *url;
	const char *scheme = NULL;
	GTime now;

	now = time (NULL);

	url = ephy_node_get_property_string
		(page, EPHY_NODE_PAGE_PROP_LOCATION);

	vfs_uri = gnome_vfs_uri_new (url);

	if (vfs_uri)
	{
		scheme = gnome_vfs_uri_get_scheme (vfs_uri);
		host_name = gnome_vfs_uri_get_host_name (vfs_uri);
	}

	/* Build an host name */
	if (scheme == NULL || host_name == NULL)
	{
		host_name = _("Others");
		host_locations = g_list_append (host_locations,
						g_strdup ("about:blank"));
	}
	else if (strcmp (url, "file") == 0)
	{
		host_name = _("Local files");
		host_locations = g_list_append (host_locations,
						g_strdup ("file:///"));
	}
	else
	{
		char *location;
		char *tmp;

		location = g_strconcat (gnome_vfs_uri_get_scheme (vfs_uri),
					"://", host_name, "/", NULL);
		host_locations = g_list_append (host_locations, location);


		if (g_str_has_prefix (host_name, "www."))
		{
			tmp = g_strdup (g_utf8_offset_to_pointer (host_name, 4));
		}
		else
		{
			tmp = g_strconcat ("www.", host_name, NULL);
		}
		location = g_strconcat (gnome_vfs_uri_get_scheme (vfs_uri),
					"://", tmp, "/", NULL);
		g_free (tmp);
		host_locations = g_list_append (host_locations, location);
	}

	g_return_val_if_fail (host_locations != NULL, NULL);

	g_static_rw_lock_reader_lock (eh->priv->hosts_hash_lock);

	for (l = host_locations; l != NULL; l = l->next)
	{
		host = g_hash_table_lookup (eh->priv->hosts_hash,
					    (char *)l->data);
		if (host) break;
	}
	g_static_rw_lock_reader_unlock (eh->priv->hosts_hash_lock);

	if (!host)
	{
		host = ephy_node_new ();
		g_value_init (&value, G_TYPE_STRING);
		g_value_set_string (&value, host_name);
		ephy_node_set_property (host, EPHY_NODE_PAGE_PROP_TITLE,
				        &value);
		g_value_unset (&value);

		g_value_init (&value, G_TYPE_STRING);
		g_value_set_string (&value, (char *)host_locations->data);
		ephy_node_set_property (host, EPHY_NODE_PAGE_PROP_LOCATION,
				        &value);
		g_value_unset (&value);

		g_value_init (&value, G_TYPE_INT);
		g_value_set_int (&value, now);
		ephy_node_set_property (host, EPHY_NODE_PAGE_PROP_FIRST_VISIT,
				        &value);
		g_value_unset (&value);

		ephy_node_add_child (eh->priv->hosts, host);
	}

	ephy_history_host_visited (eh, host, now);

	if (vfs_uri)
	{
		gnome_vfs_uri_unref (vfs_uri);
	}

	g_list_foreach (host_locations, (GFunc)g_free, NULL);
	g_list_free (host_locations);

	return host;
}

static void
ephy_history_visited (EphyHistory *eh, EphyNode *node)
{
	GValue value = { 0, };
	GTime now;
	int visits;
	const char *url;
	int host_id;

	now = time (NULL);

	g_assert (EPHY_IS_NODE (node));

	url = ephy_node_get_property_string
		(node, EPHY_NODE_PAGE_PROP_LOCATION);

	visits = ephy_node_get_property_int
		(node, EPHY_NODE_PAGE_PROP_VISITS);
	if (visits < 0) visits = 0;
	visits++;

	g_value_init (&value, G_TYPE_INT);
	g_value_set_int (&value, visits);
	ephy_node_set_property (node, EPHY_NODE_PAGE_PROP_VISITS,
			        &value);
	g_value_unset (&value);

	g_value_init (&value, G_TYPE_INT);
	g_value_set_int (&value, now);
	ephy_node_set_property (node, EPHY_NODE_PAGE_PROP_LAST_VISIT,
			        &value);
	if (visits == 1)
	{
		ephy_node_set_property
			(node, EPHY_NODE_PAGE_PROP_FIRST_VISIT, &value);
	}
	g_value_unset (&value);

	host_id = ephy_node_get_property_int (node, EPHY_NODE_PAGE_PROP_HOST_ID);
	if (host_id >= 0)
	{
		ephy_history_host_visited (eh, ephy_node_get_from_id (host_id), now);
	}

	eh->priv->last_page = node;

	g_signal_emit (G_OBJECT (eh), ephy_history_signals[VISITED], 0, url);
	ephy_history_emit_data_changed (eh);
}

int
ephy_history_get_page_visits (EphyHistory *gh,
			      const char *url)
{
	EphyNode *node;
	int visits = 0;

	node = ephy_history_get_page (gh, url);
	if (node)
	{
		visits = ephy_node_get_property_int
			(node, EPHY_NODE_PAGE_PROP_VISITS);
		if (visits < 0) visits = 0;
	}

	return visits;
}

void
ephy_history_add_page (EphyHistory *eb,
		       const char *url)
{
	EphyNode *bm, *node, *host;
	GValue value = { 0, };

	node = ephy_history_get_page (eb, url);
	if (node)
	{
		ephy_history_visited (eb, node);
		return;
	}

	bm = ephy_node_new ();

	g_value_init (&value, G_TYPE_STRING);
	g_value_set_string (&value, url);
	ephy_node_set_property (bm, EPHY_NODE_PAGE_PROP_LOCATION,
			        &value);
	ephy_node_set_property (bm, EPHY_NODE_PAGE_PROP_TITLE,
			        &value);
	g_value_unset (&value);

	host = ephy_history_add_host (eb, bm);

	g_value_init (&value, G_TYPE_INT);
	g_value_set_int (&value, ephy_node_get_id (host));
	ephy_node_set_property (bm, EPHY_NODE_PAGE_PROP_HOST_ID,
			        &value);
	g_value_unset (&value);

	ephy_history_visited (eb, bm);

	ephy_node_add_child (host, bm);
	ephy_node_add_child (eb->priv->pages, bm);
}

EphyNode *
ephy_history_get_page (EphyHistory *eb,
		       const char *url)
{
	EphyNode *node;

	g_static_rw_lock_reader_lock (eb->priv->pages_hash_lock);
	node = g_hash_table_lookup (eb->priv->pages_hash, url);
	g_static_rw_lock_reader_unlock (eb->priv->pages_hash_lock);

	return node;
}

gboolean
ephy_history_is_page_visited (EphyHistory *gh,
			      const char *url)
{
	return (ephy_history_get_page (gh, url) != NULL);
}

void
ephy_history_set_page_title (EphyHistory *gh,
			     const char *url,
			     const char *title)
{
	EphyNode *node;
	guint host_id;
	GValue value = { 0, };

	LOG ("Set page title")

	node = ephy_history_get_page (gh, url);
	if (!node) return;

	g_value_init (&value, G_TYPE_STRING);
	g_value_set_string (&value, title);
	ephy_node_set_property
		(node, EPHY_NODE_PAGE_PROP_TITLE, &value);
	g_value_unset (&value);

	host_id = ephy_node_get_property_int
		(node, EPHY_NODE_PAGE_PROP_HOST_ID);
	if (host_id >= 0)
	{
		ephy_history_host_set_title (gh, ephy_node_get_from_id  (host_id),
					     node, title);
	}
}

void
ephy_history_clear (EphyHistory *gh)
{
	EphyNode *node;

	while ((node = ephy_node_get_nth_child (gh->priv->pages, 0)) != NULL)
	{
		ephy_node_unref (node);
	}

	while ((node = ephy_node_get_nth_child (gh->priv->hosts, 0)) != NULL)
	{
		ephy_node_unref (node);
	}

	ephy_history_save (gh);
}

EphyNode *
ephy_history_get_hosts (EphyHistory *eb)
{
	return eb->priv->hosts;
}

EphyNode *
ephy_history_get_pages (EphyHistory *eb)
{
	return eb->priv->pages;
}

const char *
ephy_history_get_last_page (EphyHistory *gh)
{
	if (gh->priv->last_page == NULL) return NULL;

	return ephy_node_get_property_string
		(gh->priv->last_page, EPHY_NODE_PAGE_PROP_LOCATION);
}

void
ephy_history_remove (EphyHistory *gh, EphyNode *node)
{
	EphyNode *host;
	int host_id;

	LOG ("Remove history item")

	host_id = ephy_node_get_property_int (node, EPHY_NODE_PAGE_PROP_HOST_ID);
	if (host_id < 0)
	{
		EphyNode *tmp;
		while ((tmp = ephy_node_get_nth_child (node, 0)) != NULL)
		{
			ephy_node_unref (tmp);
		}
		ephy_node_unref (node);
		return;
	}

	host = ephy_node_get_from_id (host_id);
	if (ephy_node_get_n_children (host) == 1)
	{
		ephy_node_unref (host);
	}
	else
	{
		ephy_node_unref (node);
	}
}