/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
/*
 * Copyright (C) 2005  Novell, Inc.
 *
 * Authors: Michael Zucchi <notzed@novell.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * 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., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <errno.h>
#include <string.h>
#include <bonobo/bonobo-shlib-factory.h>
#include <bonobo/bonobo-control.h>
#include <bonobo/bonobo-i18n.h>
#include <bonobo/bonobo-exception.h>
#include <bonobo/bonobo-arg.h>
#include "evolution-mail-folder.h"
#include "evolution-mail-store.h"
#include "evolution-mail-messageiterator.h"
#include "evolution-mail-session.h"

#include <camel/camel-store.h>
#include <camel/camel-folder.h>
#include <libedataserver/e-msgport.h>

#include "e-corba-utils.h"

#define PARENT_TYPE bonobo_object_get_type ()

static BonoboObjectClass *parent_class = NULL;

#define _PRIVATE(o) (g_type_instance_get_private ((GTypeInstance *)o, evolution_mail_folder_get_type()))

struct _EvolutionMailFolderPrivate {
	CamelFolder *folder;

	guint32 folder_changed;

	EDList listeners;
};

/* GObject methods */

static void
impl_dispose (GObject *object)
{
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(object);

	if (p->folder) {
		camel_object_remove_event(p->folder, p->folder_changed);
		camel_object_unref(p->folder);
		p->folder = NULL;
	}

	e_mail_listener_free(&p->listeners);

	(* G_OBJECT_CLASS (parent_class)->dispose) (object);
}

static void
impl_finalize (GObject *object)
{
	EvolutionMailFolder *emf = (EvolutionMailFolder *)object;
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(object);

	p = p;

	g_warning("EvolutionMailFolder is finalised!\n");

	g_free(emf->full_name);
	g_free(emf->name);

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

/* Evolution.Mail.Folder */

static CORBA_boolean
impl_getProperties(PortableServer_Servant _servant,
		   const Evolution_Mail_PropertyNames* names,
		   Evolution_Mail_Properties **propsp,
		   CORBA_Environment * ev)
{
	EvolutionMailFolder *emf = (EvolutionMailFolder *)bonobo_object_from_servant(_servant);
	int i;
	Evolution_Mail_Properties *props;
	CORBA_boolean ok = CORBA_TRUE;

	*propsp = props = Evolution_Mail_Properties__alloc();
	props->_length = names->_length;
	props->_maximum = props->_length;
	props->_buffer = Evolution_Mail_Properties_allocbuf(props->_maximum);
	CORBA_sequence_set_release(props, CORBA_TRUE);

	for (i=0;i<names->_length;i++) {
		const CORBA_char *name = names->_buffer[i];
		Evolution_Mail_Property *prop = &props->_buffer[i];

		prop->value._release = CORBA_TRUE;

		if (!strcmp(name, "name")) {
			prop->value._type = TC_CORBA_string;
			prop->value._value = CORBA_string_dup(emf->name);
		} else if (!strcmp(name, "full_name")) {
			prop->value._type = TC_CORBA_string;
			prop->value._value = CORBA_string_dup(emf->full_name);
		} else {
			prop->value._type = TC_null;
			ok = CORBA_FALSE;
		}

		prop->name = CORBA_string_dup(name);
	}

	return ok;
}

static Evolution_Mail_MessageIterator
impl_getMessages(PortableServer_Servant _servant, const CORBA_char * pattern, CORBA_Environment * ev)
{
	EvolutionMailFolder *emf = (EvolutionMailFolder *)bonobo_object_from_servant(_servant);
	struct _CamelFolder *folder;
	EvolutionMailMessageIterator *emi;
	Evolution_Mail_MessageIterator iter;

	folder = evolution_mail_folder_get_folder(emf, ev);
	if (folder == NULL)
		return CORBA_OBJECT_NIL;

	emi = evolution_mail_messageiterator_new(folder, pattern);
	camel_object_unref(folder);

	/* NB: How do we destroy the object once we're done? */

	iter = CORBA_Object_duplicate(bonobo_object_corba_objref((BonoboObject *)emi), NULL);

	return iter;
}

static void
impl_changeMessages(PortableServer_Servant _servant, const Evolution_Mail_MessageInfoSets *infos, CORBA_Environment * ev)
{
	EvolutionMailFolder *emf = (EvolutionMailFolder *)bonobo_object_from_servant(_servant);
	struct _CamelFolder *folder;
	int i, j;

	folder = evolution_mail_folder_get_folder(emf, ev);
	if (folder == NULL)
		return;

	camel_folder_freeze(folder);
	for (i=0;i<infos->_length;i++) {
		CamelMessageInfo *mi;
		Evolution_Mail_MessageInfoSet *mis = &infos->_buffer[i];

		mi = camel_folder_get_message_info(folder, mis->uid);
		if (mi == NULL)
			continue;

		if (mis->flagMask)
			camel_message_info_set_flags(mi, mis->flagMask, mis->flagSet);

		for (j=0;j<mis->userFlagSet._length;j++)
			camel_message_info_set_user_flag(mi, mis->userFlagSet._buffer[j], TRUE);
		for (j=0;j<mis->userFlagUnset._length;j++)
			camel_message_info_set_user_flag(mi, mis->userFlagUnset._buffer[j], FALSE);
		for (j=0;j<mis->userTags._length;j++)
			camel_message_info_set_user_tag(mi, mis->userTags._buffer[j].name, mis->userTags._buffer[j].value[0]?mis->userTags._buffer[j].value:NULL);

		camel_message_info_free(mi);
	}
	camel_folder_thaw(folder);

	camel_object_unref(folder);
}

static Evolution_Mail_MessageStream
impl_getMessage(PortableServer_Servant _servant, const CORBA_char * uid, CORBA_Environment *ev)
{
	EvolutionMailFolder *emf = (EvolutionMailFolder *)bonobo_object_from_servant(_servant);
	struct _CamelFolder *folder;
	CamelMimeMessage *msg;
	Evolution_Mail_MessageStream out;
	CamelException ex = { 0 };

	folder = evolution_mail_folder_get_folder(emf, ev);
	if (folder == NULL)
		return CORBA_OBJECT_NIL;

	msg = camel_folder_get_message(folder, uid, &ex);
	if (msg == NULL) {
		e_mail_exception_xfer_camel(ev, &ex);
		out = CORBA_OBJECT_NIL;
	} else {
		out = e_messagestream_from_message(msg, ev);
		camel_object_unref(msg);
	}
	camel_object_unref(folder);

	return out;
}

static void
impl_appendMessage(PortableServer_Servant _servant, const Evolution_Mail_MessageInfoSet*mi, const Evolution_Mail_MessageStream message, CORBA_Environment *ev)
{
	EvolutionMailFolder *emf = (EvolutionMailFolder *)bonobo_object_from_servant(_servant);
	struct _CamelFolder *folder;
	CamelMimeMessage *msg = NULL;
	CamelMessageInfo *info;
	CamelException ex = { 0 };
	CORBA_Environment wev = { 0 };

	folder = evolution_mail_folder_get_folder(emf, ev);
	if (folder == NULL)
		goto fail3;

	msg = e_messagestream_to_message(message, ev);
	if (msg == NULL)
		goto fail2;

	info = e_mail_messageinfoset_to_info(mi);
	camel_folder_append_message(folder, msg, info, NULL, &ex);
	camel_message_info_free(info);

	if (camel_exception_is_set(&ex))
		e_mail_exception_xfer_camel(ev, &ex);

	camel_object_unref(msg);
fail2:
	camel_object_unref(folder);
fail3:
	Evolution_Mail_MessageStream_dispose(message, &wev);
}

/* Initialization */

static void
evolution_mail_folder_class_init (EvolutionMailFolderClass *klass)
{
	POA_Evolution_Mail_Folder__epv *epv = &klass->epv;
	GObjectClass *object_class = G_OBJECT_CLASS (klass);

	parent_class = g_type_class_peek_parent (klass);

	epv->getProperties = impl_getProperties;
	epv->getMessages = impl_getMessages;
	epv->changeMessages = impl_changeMessages;
	epv->getMessage = impl_getMessage;
	epv->appendMessage = impl_appendMessage;

	object_class->dispose = impl_dispose;
	object_class->finalize = impl_finalize;

	g_type_class_add_private(klass, sizeof(struct _EvolutionMailFolderPrivate));
}

static void
evolution_mail_folder_init(EvolutionMailFolder *emf, EvolutionMailFolderClass *klass)
{
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(emf);

	bonobo_object_set_immortal((BonoboObject *)emf, TRUE);
	e_dlist_init(&p->listeners);
}

BONOBO_TYPE_FUNC_FULL (EvolutionMailFolder, Evolution_Mail_Folder, PARENT_TYPE, evolution_mail_folder)

EvolutionMailFolder *
evolution_mail_folder_new(EvolutionMailStore *ems, const char *name, const char *full_name)
{
	EvolutionMailFolder *emf = g_object_new (EVOLUTION_MAIL_TYPE_FOLDER, NULL);

	emf->name = g_strdup(name);
	emf->full_name = g_strdup(full_name);

	emf->store = ems;

	return emf;
}

static void
emf_set_change(Evolution_Mail_FolderChange *change, Evolution_Mail_ChangeType how, CamelFolder *folder, GPtrArray *uids)
{
	int total = 0, i;

	change->type = how;
	change->messages._maximum = uids->len;
	change->messages._buffer = Evolution_Mail_MessageInfos_allocbuf(uids->len);

	for (i=0;i<uids->len;i++) {
		CamelMessageInfo *info = camel_folder_get_message_info(folder, uids->pdata[i]);

		if (info) {
			e_mail_messageinfo_set_message(&change->messages._buffer[total], info);
			camel_message_info_free(info);
			total++;
		} else {
			w(printf("couldn't get info for changed uid '%s'?\n", (char *)uids->pdata[i]));
		}
	}

	change->messages._length = total;
}

static void
emf_folder_changed(CamelObject *o, void *d, void *data)
{
	EvolutionMailFolder *emf = data;
	CamelFolder *folder = (CamelFolder *)o;
	CamelFolderChangeInfo *ci = d;
	int count = 0;
	Evolution_Mail_FolderChanges *changes;

	changes = Evolution_Mail_FolderChanges__alloc();
	changes->_maximum = 3;
	changes->_buffer = Evolution_Mail_FolderChanges_allocbuf(3);
	CORBA_sequence_set_release(changes, TRUE);

	/* could be a race if a new listener is added */
	if (ci->uid_added->len) {
		emf_set_change(&changes->_buffer[count], Evolution_Mail_ADDED, folder, ci->uid_added);
		count++;
	}
	if (ci->uid_removed->len) {
		emf_set_change(&changes->_buffer[count], Evolution_Mail_REMOVED, folder, ci->uid_removed);
		count++;
	}
	if (ci->uid_changed->len) {
		emf_set_change(&changes->_buffer[count], Evolution_Mail_CHANGED, folder, ci->uid_changed);
		count++;
	}

	changes->_length = count;

	evolution_mail_folder_changed(emf, changes);
	CORBA_free(changes);
}

struct _CamelFolder *evolution_mail_folder_get_folder(EvolutionMailFolder *emf, CORBA_Environment *ev)
{
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(emf);
	CamelStore *store;
	CamelException ex;

	if (p->folder == NULL) {
		store = evolution_mail_store_get_store(emf->store, ev);
		if (store == NULL)
			return NULL;

		camel_exception_init(&ex);
		p->folder = camel_store_get_folder(store, emf->full_name, 0, &ex);
		if (p->folder) {
			p->folder_changed = camel_object_hook_event(p->folder, "folder_changed", emf_folder_changed, emf);
		} else {
			e_mail_exception_xfer_camel(ev, &ex);
		}
		camel_object_unref(store);
	}

	if (p->folder)
		camel_object_ref(p->folder);

	return p->folder;
}

int evolution_mail_folder_close_folder(EvolutionMailFolder *emf)
{
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(emf);

	/* FIXME: locking */
	if (p->folder) {
		if (!e_dlist_empty(&p->listeners))
			return -1;

		camel_object_remove_event(p->folder, p->folder_changed);
		camel_object_unref(p->folder);
		p->folder = NULL;
	}

	return 0;
}

void evolution_mail_folder_addlistener(EvolutionMailFolder *emf, Evolution_Mail_FolderListener listener)
{
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(emf);

	/* FIXME: locking */
	e_mail_listener_add(&p->listeners, listener);
}

void
evolution_mail_folder_changed(EvolutionMailFolder *emf, Evolution_Mail_FolderChanges *changes)
{
	struct _EvolutionMailFolderPrivate *p = _PRIVATE(emf);

	if (!e_mail_listener_emit(&p->listeners, (EMailListenerChanged)Evolution_Mail_FolderListener_changed,
				  bonobo_object_corba_objref((BonoboObject *)emf), changes)) {
		d(printf("No more listeners for folder, could dispose store object now\n"));
		evolution_mail_folder_close_folder(emf);
	}
}